1 files changed, 161 insertions, 0 deletions

diff --git a/src/animation/AnimBlendAssocGroup.cpp b/src/animation/AnimBlendAssocGroup.cpp
new file mode 100644
index 00000000..16749504
--- /dev/null
+++ b/src/animation/AnimBlendAssocGroup.cpp
@@ -0,0 +1,161 @@
+#include "common.h"
+#include "patcher.h"
+#include "ModelInfo.h"
+#include "AnimManager.h"
+#include "RpAnimBlend.h"
+#include "AnimBlendAssociation.h"
+#include "AnimBlendAssocGroup.h"
+
+CAnimBlendAssocGroup::CAnimBlendAssocGroup(void)
+{
+	assocList = nil;
+	numAssociations = 0;
+}
+
+CAnimBlendAssocGroup::~CAnimBlendAssocGroup(void)
+{
+	DestroyAssociations();
+}
+
+void
+CAnimBlendAssocGroup::DestroyAssociations(void)
+{
+	if(assocList){
+		delete[] assocList;
+		assocList = nil;
+		numAssociations = 0;
+	}
+}
+
+CAnimBlendAssociation*
+CAnimBlendAssocGroup::GetAnimation(uint32 id)
+{
+	return &assocList[id];
+}
+
+CAnimBlendAssociation*
+CAnimBlendAssocGroup::GetAnimation(const char *name)
+{
+	int i;
+	for(i = 0; i < numAssociations; i++)
+		if(strcmpi(assocList[i].hierarchy->name, name) == 0)
+			return &assocList[i];
+	return nil;
+}
+
+
+CAnimBlendAssociation*
+CAnimBlendAssocGroup::CopyAnimation(uint32 id)
+{
+	CAnimBlendAssociation *anim = GetAnimation(id);
+	if(anim == nil)
+		return nil;
+	CAnimManager::UncompressAnimation(anim->hierarchy);
+	return new CAnimBlendAssociation(*anim);
+}
+
+CAnimBlendAssociation*
+CAnimBlendAssocGroup::CopyAnimation(const char *name)
+{
+	CAnimBlendAssociation *anim = GetAnimation(name);
+	if(anim == nil)
+		return nil;
+	CAnimManager::UncompressAnimation(anim->hierarchy);
+	return new CAnimBlendAssociation(*anim);
+}
+
+int
+strcmpIgnoringDigits(const char *s1, const char *s2)
+{
+	char c1, c2;
+
+	for(;;){
+		c1 = *s1;
+		c2 = *s2;
+		if(c1) s1++;
+		if(c2) s2++;
+		if(c1 == '\0' && c2 == '\0')
+			return 1;
+		if(islower(c1)) c1 = toupper(c1);
+		if(islower(c2)) c2 = toupper(c2);
+		if(isdigit(c1) && isdigit(c2))
+			continue;
+		if(c1 != c2)
+			return 0;
+	}
+}
+
+CBaseModelInfo*
+GetModelFromName(const char *name)
+{
+	int i;
+	CBaseModelInfo *mi;
+
+	for(i = 0; i < MODELINFOSIZE; i++){
+		mi = CModelInfo::GetModelInfo(i);
+		if(mi->GetRwObject() && RwObjectGetType(mi->GetRwObject()) == rpCLUMP &&
+		   strcmpIgnoringDigits(mi->GetName(), name))
+			return mi;
+	}
+	return nil;
+}
+
+void
+CAnimBlendAssocGroup::CreateAssociations(const char *name)
+{
+	int i;
+	CAnimBlock *animBlock;
+
+	if(assocList)
+		DestroyAssociations();
+
+	animBlock = CAnimManager::GetAnimationBlock(name);
+	assocList = new CAnimBlendAssociation[animBlock->numAnims];
+	numAssociations = 0;
+
+	for(i = 0; i < animBlock->numAnims; i++){
+		CAnimBlendHierarchy *anim = CAnimManager::GetAnimation(animBlock->firstIndex + i);
+		CBaseModelInfo *model = GetModelFromName(anim->name);
+		printf("Associated anim %s with model %s\n", anim->name, model->GetName());
+		if(model){
+			RpClump *clump = (RpClump*)model->CreateInstance();
+			RpAnimBlendClumpInit(clump);
+			assocList[i].Init(clump, anim);
+			RpClumpDestroy(clump);
+			assocList[i].animId = i;
+		}
+	}
+	numAssociations = animBlock->numAnims;
+}
+
+// Create associations from hierarchies for a given clump
+void
+CAnimBlendAssocGroup::CreateAssociations(const char *blockName, RpClump *clump, char **animNames, int numAssocs)
+{
+	int i;
+	CAnimBlock *animBlock;
+
+	if(assocList)
+		DestroyAssociations();
+
+	animBlock = CAnimManager::GetAnimationBlock(blockName);
+	assocList = new CAnimBlendAssociation[animBlock->numAnims];
+
+	numAssociations = 0;
+	for(i = 0; i < numAssocs; i++){
+		assocList[i].Init(clump, CAnimManager::GetAnimation(animNames[i], animBlock));
+		assocList[i].animId = i;
+	}
+	numAssociations = numAssocs;
+}
+
+
+
+STARTPATCHES
+	InjectHook(0x4013D0, (CAnimBlendAssociation *(CAnimBlendAssocGroup::*)(uint32))&CAnimBlendAssocGroup::GetAnimation, PATCH_JUMP);
+	InjectHook(0x401300, (CAnimBlendAssociation *(CAnimBlendAssocGroup::*)(const char*))&CAnimBlendAssocGroup::GetAnimation, PATCH_JUMP);
+	InjectHook(0x401420, (CAnimBlendAssociation *(CAnimBlendAssocGroup::*)(uint32))&CAnimBlendAssocGroup::CopyAnimation, PATCH_JUMP);
+	InjectHook(0x4013E0, (CAnimBlendAssociation *(CAnimBlendAssocGroup::*)(const char*))&CAnimBlendAssocGroup::CopyAnimation, PATCH_JUMP);
+	InjectHook(0x401130, (void (CAnimBlendAssocGroup::*)(const char*))&CAnimBlendAssocGroup::CreateAssociations, PATCH_JUMP);
+	InjectHook(0x401220, (void (CAnimBlendAssocGroup::*)(const char*, RpClump*, char**, int))&CAnimBlendAssocGroup::CreateAssociations, PATCH_JUMP);
+ENDPATCHES