+Unity Runtime codeHEADmaster

1 files changed, 202 insertions, 0 deletions

diff --git a/Runtime/Animation/CharacterTestFixture.h b/Runtime/Animation/CharacterTestFixture.h
new file mode 100644
index 0000000..cd72c5e
--- /dev/null
+++ b/Runtime/Animation/CharacterTestFixture.h
@@ -0,0 +1,202 @@
+#pragma once
+
+#if ENABLE_UNIT_TESTS
+
+#include "Runtime/Testing/Testing.h"
+#include "Runtime/Testing/TestFixtures.h"
+
+#include "Runtime/Misc/GameObjectUtility.h"
+#include "Runtime/Animation/Avatar.h"
+#include "Runtime/Animation/AvatarBuilder.h"
+#include "Runtime/Animation/Animator.h"
+#include "Runtime/Utilities/PathNameUtility.h"
+#include "Runtime/Filters/Deformation/SkinnedMeshFilter.h"
+
+#include "Runtime/Animation/OptimizeTransformHierarchy.h"
+
+
+// y
+// |
+// |
+// 3   b1_2_2
+// |    |       mr2
+// |    |        |
+// 2   b1_2----b1_2_1                 b2_1_2-----b2_1_2_1
+// |  	|               mr1              |
+// |    |               |                |
+// 1    b1-----b1_1---b1_1_1     b2-----b2_1------b2_1_1
+// | 
+// 0____1_______2_______3_________4_______5_________6_________ x
+//
+// smr1 {b1(root), b1_1, b1_1_1, b1_2, b1_2_1, b1_2_2}
+// smr2 {b2_1_1, b2_1_2, b2_1_2_1}
+
+struct TransformDescriptor {
+	const char* path;
+	float t[3];
+	float r[3];
+	float s[3];
+};
+const TransformDescriptor BONE_ARRAY[] = {
+	{"b1",										{1,1,0}, {0,0,0}, {1,1,1} },
+	{"b1/b1_1",									{2,1,0}, {0,0,0}, {1,1,1} },
+	{"b1/b1_1/b1_1_1",							{3,1,0}, {0,0,0}, {1,1,1} },
+	{"b1/b1_2",									{1,2,0}, {0,0,0}, {1,1,1} },
+	{"b1/b1_2/b1_2_1",							{2,2,0}, {0,0,0}, {1,1,1} },
+	{"b1/b1_2/b1_2_1/tobeStrip1",				{5,5,5}, {0,0,0}, {1,1,1} },
+	{"b1/b1_2/b1_2_2",							{1,3,0}, {0,0,0}, {1,1,1} },
+	{"b2",										{4,1,0}, {0,0,0}, {1,1,1} },
+	{"b2/b2_1",									{5,1,0}, {0,0,0}, {1,1,1} },
+	{"b2/b2_1/b2_1_1",							{6,1,0}, {0,0,0}, {1,1,1} },
+	{"b2/b2_1/b2_1_2",							{5,2,0}, {0,0,0}, {1,1,1} },
+	{"b2/b2_1/b2_1_2/b2_1_2_1",					{6,2,0}, {0,0,0}, {1,1,1} },
+	{"b2/b2_1/b2_1_2/b2_1_2_1/tobeStrip3",		{6,6,6}, {0,0,0}, {1,1,1} },
+	{"tobeStrip2",								{7,7,7}, {0,0,0}, {1,1,1} },
+};
+const int BONE_COUNT = sizeof(BONE_ARRAY)/sizeof(BONE_ARRAY[0]);
+const TransformDescriptor MESH_RENDERER_ARRAY[] = {
+	{"b1/b1_1/b1_1_1/mr1",						{3,1.5,0}, {0,0,0}, {1,1,1} },
+	{"b1/b1_2/b1_2_1/mr2",						{2,2.5,0}, {0,0,0}, {1,1,1} },
+};
+const int MESH_RENDERER_COUNT = sizeof(MESH_RENDERER_ARRAY)/sizeof(MESH_RENDERER_ARRAY[0]);
+const TransformDescriptor SKINNED_MESH_RENDERER_ARRAY[] = {
+	{"smr1",									{8,8,8}, {0,0,0}, {1,1,1} },
+	{"b2/b2_1/smr2",							{9,9,9}, {0,0,0}, {1,1,1} },
+};
+const int SKINNED_MESH_RENDERER_COUNT = sizeof(SKINNED_MESH_RENDERER_ARRAY)/sizeof(SKINNED_MESH_RENDERER_ARRAY[0]);
+
+class CharacterTestFixture : public TestFixtureBase
+{
+protected:
+	Unity::GameObject*	root;
+	Avatar*				avatar;
+	Avatar*				unstrippedAvatar;
+
+	CharacterTestFixture():root(NULL), avatar(NULL), unstrippedAvatar(NULL) {}
+
+	~CharacterTestFixture()
+	{
+		DestroyObjects();
+	}
+
+	void MakeCharacter(const UnityStr* extraExposedPaths = NULL, int extraExposedPathCount = 0)
+	{
+		root = &CreateGameObjectWithHideFlags("root", true, 0, "Transform", "Animator", NULL);
+		Transform& rootTr = root->GetComponent(Transform);
+		AttachGameObjects(rootTr);
+
+		CreateAvatars(extraExposedPaths, extraExposedPathCount);
+
+		Animator& animator = root->GetComponent(Animator);
+		animator.SetAvatar(avatar);
+		animator.AwakeFromLoad(kDefaultAwakeFromLoad);
+	}
+
+	void DestroyObjects()
+	{
+		if (root)
+		{
+			DestroyObjectHighLevel(root);
+			root = NULL;
+		}
+		if (avatar)
+		{
+			DestroyObjectHighLevel(avatar);
+			avatar = NULL;
+		}
+		if (unstrippedAvatar)
+		{
+			DestroyObjectHighLevel(unstrippedAvatar);
+			unstrippedAvatar = NULL;
+		}
+	}
+
+	void CreateAvatars(const UnityStr* extraExposedPaths = NULL, int extraExposedPathCount = 0)
+	{
+		HumanDescription dummyHd;
+		std::string errStr;
+
+		AvatarBuilder::Options options;
+		options.avatarType = kGeneric;
+		options.readTransform = true;
+
+		//// 1. create unstripped avatar
+		//unstrippedAvatar = NEW_OBJECT (Avatar);
+		//unstrippedAvatar->Reset();
+		//errStr = AvatarBuilder::BuildAvatar(*unstrippedAvatar, *root, true, dummyHd, options);
+		//CHECK_EQUAL (string(""), errStr);
+		//unstrippedAvatar->AwakeFromLoad(kDefaultAwakeFromLoad);
+
+		//// 2. strip
+		//RemoveUnnecessaryTransforms(*root, 
+		//	&dummyHd,
+		//	extraExposedPaths,
+		//	extraExposedPathCount,
+		//	true);
+
+		// 3. create stripped avatar
+		avatar = NEW_OBJECT (Avatar);
+		avatar->Reset();
+
+		errStr = AvatarBuilder::BuildAvatar(*avatar,
+			*root, true, dummyHd, options);
+		CHECK_EQUAL (string(""), errStr);
+		avatar->AwakeFromLoad(kDefaultAwakeFromLoad);
+	}
+
+	void AttachGameObjects(Transform& rootTr)
+	{
+		dynamic_array<PPtr<Transform> > smrBones[SKINNED_MESH_RENDERER_COUNT];
+
+		dynamic_array<TransformDescriptor> transforms;
+		for (int i = 0; i < BONE_COUNT; i++)	transforms.push_back(BONE_ARRAY[i]);
+		for (int i = 0; i < MESH_RENDERER_COUNT; i++)		transforms.push_back(MESH_RENDERER_ARRAY[i]);
+		for (int i = 0; i < SKINNED_MESH_RENDERER_COUNT; i++)		transforms.push_back(SKINNED_MESH_RENDERER_ARRAY[i]);
+		for (int i = 0; i < transforms.size(); i++)
+		{
+			string path(transforms[i].path);
+
+			string parentPath = DeleteLastPathNameComponent(path);
+			Transform* parent = FindRelativeTransformWithPath(rootTr, parentPath.c_str());
+			CHECK(parent != NULL);
+
+			string goName = GetLastPathNameComponent(path);
+			GameObject& go = CreateGameObjectWithHideFlags(goName, true, 0, "Transform", NULL);
+			Transform& tr = go.GetComponent(Transform);
+			tr.SetPosition(Vector3f(transforms[i].t[0], transforms[i].t[1], transforms[i].t[2]));
+			tr.SetParent(parent);
+
+			if (goName.compare(0, strlen("b1"), "b1") == 0)
+				smrBones[0].push_back(&tr);
+			else if (goName.compare(0, strlen("b2_1_"), "b2_1_") == 0)
+				smrBones[1].push_back(&tr);
+		}
+
+		for (int i = 0; i < MESH_RENDERER_COUNT; i++)
+		{
+			Transform* tr = FindRelativeTransformWithPath(rootTr, MESH_RENDERER_ARRAY[i].path);
+			AddComponent(tr->GetGameObject(), "MeshRenderer");
+		}
+
+		for (int i = 0; i < SKINNED_MESH_RENDERER_COUNT; i++)
+		{
+			Transform* tr = FindRelativeTransformWithPath(rootTr, SKINNED_MESH_RENDERER_ARRAY[i].path);
+			AddComponent(tr->GetGameObject(), "SkinnedMeshRenderer");
+			SkinnedMeshRenderer& smr = tr->GetComponent(SkinnedMeshRenderer);
+			if (i == 0)
+				smr.SetRootBone(smrBones[i][0]);
+			smr.SetBones(smrBones[i]);
+		}
+	}
+
+	int GetAllChildrenCount(Transform& tr)
+	{
+		int count = tr.GetChildrenCount();
+		for (int i = 0; i < tr.GetChildrenCount(); i++)
+			count += GetAllChildrenCount(tr.GetChild(i));
+		return count;
+	}
+
+};
+
+#endif