From 1fd93cf228ab2574b01a78eef968fe5f26a3fc68 Mon Sep 17 00:00:00 2001
From: "davedx@gmail.com"
 <davedx@gmail.com@6c4fd544-2939-11df-bb46-9574ba5d0bfa>
Date: Tue, 4 Jan 2011 17:41:56 +0000
Subject: [PATCH] [Changed] Added tagged joints. Will post blog explaining it

---
 .../badlogic/gdx/graphics/animation/Animator.java  |  9 ++--
 .../badlogic/gdx/graphics/keyframed/Keyframe.java  |  6 +++
 .../gdx/graphics/keyframed/KeyframeAnimator.java   | 54 ++++++++++++++++++++
 .../gdx/graphics/keyframed/KeyframedModel.java     | 57 ++++++++++++++++++++++
 4 files changed, 122 insertions(+), 4 deletions(-)

diff --git a/gdx/src/com/badlogic/gdx/graphics/animation/Animator.java b/gdx/src/com/badlogic/gdx/graphics/animation/Animator.java
index 5c556719c..ac29151ed 100644
--- a/gdx/src/com/badlogic/gdx/graphics/animation/Animator.java
+++ b/gdx/src/com/badlogic/gdx/graphics/animation/Animator.java
@@ -40,12 +40,13 @@ public abstract class Animator {
 		mCurrentAnim = anim;
 		mAnimLoop = loop;
 
+		mAnimPos = mFrameDelta = 0.f;
+		mCurrentFrameIdx = -1;
+		mNextFrameIdx = -1;
+
 		if(mCurrentAnim != null)
 		{
 			mAnimLen = mCurrentAnim.getLength(); 
-			mAnimPos = mFrameDelta = 0.f;
-			mCurrentFrameIdx = -1;
-			mNextFrameIdx = -1;
 		}	
 	}
 	
@@ -106,7 +107,7 @@ public abstract class Animator {
 				mFrameDelta = 0.f;
 				mCurrentFrameIdx = currentFrameIdx;
 			}
-	
+
 			mFrameDelta += dt;
 			
 			setInterpolationFrames();
diff --git a/gdx/src/com/badlogic/gdx/graphics/keyframed/Keyframe.java b/gdx/src/com/badlogic/gdx/graphics/keyframed/Keyframe.java
index d68dbb1d5..9b46b315a 100644
--- a/gdx/src/com/badlogic/gdx/graphics/keyframed/Keyframe.java
+++ b/gdx/src/com/badlogic/gdx/graphics/keyframed/Keyframe.java
@@ -12,6 +12,9 @@
  */
 package com.badlogic.gdx.graphics.keyframed;
 
+import com.badlogic.gdx.math.Quaternion;
+import com.badlogic.gdx.math.Vector3;
+
 /**
  * Container for the geometry of a single animation keyframe.
  * 
@@ -23,4 +26,7 @@ public class Keyframe {
 	public short[][] Indices = null;
 	public boolean IndicesSet = false;
 	public boolean IndicesSent = false;
+	
+	public Vector3[] TaggedJointPos = null;
+	public Quaternion[] TaggedJoint = null;
 }
diff --git a/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframeAnimator.java b/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframeAnimator.java
index df5a623dd..df8cd2909 100644
--- a/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframeAnimator.java
+++ b/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframeAnimator.java
@@ -12,8 +12,12 @@
  */
 package com.badlogic.gdx.graphics.keyframed;
 
+import com.badlogic.gdx.Gdx;
 import com.badlogic.gdx.graphics.animation.Animator;
 import com.badlogic.gdx.graphics.loaders.md5.MD5Animation;
+import com.badlogic.gdx.graphics.loaders.md5.MD5Quaternion;
+import com.badlogic.gdx.math.Quaternion;
+import com.badlogic.gdx.math.Vector3;
 
 /**
  * An animation controller for keyframed animations.
@@ -70,12 +74,31 @@ public class KeyframeAnimator extends Animator {
 		R.Indices[idx] = new short[numIndices];
 	}
 	
+	/**
+	 * Set the number of tagged joints for allocation
+	 * @param num
+	 */
+	public void setNumTaggedJoints(int num)
+	{
+		// allocate space for joint data in the result keyframe
+		R.TaggedJointPos = new Vector3[num];
+		for(int i=0; i<num; i++)
+			R.TaggedJointPos[i] = new Vector3();
+		R.TaggedJoint = new Quaternion[num];
+		for(int i=0; i<num; i++)
+			R.TaggedJoint[i] = new Quaternion(0,0,0,0);
+	}
+	
 	@Override
 	protected void setInterpolationFrames() {
 		A = ((KeyframeAnimation)mCurrentAnim).mKeyframes[mCurrentFrameIdx];
 		B = ((KeyframeAnimation)mCurrentAnim).mKeyframes[mNextFrameIdx];
 	}
 
+
+	static MD5Quaternion jointAOrient = new MD5Quaternion();
+	static MD5Quaternion jointBOrient = new MD5Quaternion();
+
 	//TODO: Optimise further if possible - this is the CPU bottleneck for animation
 	@Override
 	protected void interpolate()
@@ -128,6 +151,37 @@ public class KeyframeAnimator extends Animator {
 			}
 		}
 		R.IndicesSet = true;
+		
+		//interpolate any tagged joints
+		for(int tj = 0; tj<A.TaggedJoint.length; tj++)
+		{
+			//position
+			float PAX = A.TaggedJointPos[tj].x;
+			float PAY = A.TaggedJointPos[tj].y;
+			float PAZ = A.TaggedJointPos[tj].z;
+			float PBX = B.TaggedJointPos[tj].x;
+			float PBY = B.TaggedJointPos[tj].y;
+			float PBZ = B.TaggedJointPos[tj].z;
+
+			R.TaggedJointPos[tj].x = PAX + (PBX - PAX)*t;
+			R.TaggedJointPos[tj].y = PAY + (PBY - PAY)*t;
+			R.TaggedJointPos[tj].z = PAZ + (PBZ - PAZ)*t;
+			
+			//orientation
+			jointAOrient.x = A.TaggedJoint[tj].x;
+			jointAOrient.y = A.TaggedJoint[tj].y;
+			jointAOrient.z = A.TaggedJoint[tj].z;
+			jointAOrient.w = A.TaggedJoint[tj].w;
+			jointBOrient.x = B.TaggedJoint[tj].x;
+			jointBOrient.y = B.TaggedJoint[tj].y;
+			jointBOrient.z = B.TaggedJoint[tj].z;
+			jointBOrient.w = B.TaggedJoint[tj].w;
+			jointAOrient.slerp(jointBOrient, t);
+			R.TaggedJoint[tj].x = jointAOrient.x;
+			R.TaggedJoint[tj].y = jointAOrient.y;
+			R.TaggedJoint[tj].z = jointAOrient.z;
+			R.TaggedJoint[tj].w = jointAOrient.w;
+		}
 	}
 
 	/**
diff --git a/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframedModel.java b/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframedModel.java
index ed4414a3c..dc84057a1 100644
--- a/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframedModel.java
+++ b/gdx/src/com/badlogic/gdx/graphics/keyframed/KeyframedModel.java
@@ -14,6 +14,8 @@ package com.badlogic.gdx.graphics.keyframed;
 
 import java.io.DataInputStream;
 import java.io.DataOutputStream;
+import java.util.ArrayList;
+import java.util.Iterator;
 
 import com.badlogic.gdx.graphics.GL10;
 import com.badlogic.gdx.graphics.Material;
@@ -22,8 +24,11 @@ import com.badlogic.gdx.graphics.VertexAttributes;
 import com.badlogic.gdx.graphics.animation.Animator;
 import com.badlogic.gdx.graphics.loaders.md5.MD5Animation;
 import com.badlogic.gdx.graphics.loaders.md5.MD5Animator;
+import com.badlogic.gdx.graphics.loaders.md5.MD5Joints;
 import com.badlogic.gdx.graphics.loaders.md5.MD5Model;
 import com.badlogic.gdx.graphics.loaders.md5.MD5Renderer;
+import com.badlogic.gdx.math.Quaternion;
+import com.badlogic.gdx.math.Vector3;
 import com.badlogic.gdx.utils.GdxRuntimeException;
 import com.badlogic.gdx.utils.ObjectMap;
 
@@ -41,6 +46,7 @@ public class KeyframedModel {
 	private KeyframeAnimator mAnimator = null;
 	private Mesh[] mTarget = null;
 	private int mNumMeshes = 0;
+	private ArrayList<String> mTaggedJointNames = new ArrayList<String>();
 	
 	/**
 	 * Gets the {@link KeyframeAnimator} for this model.
@@ -71,6 +77,16 @@ public class KeyframedModel {
 			mMaterials[i] = mats[i];
 		}
 	}
+	
+	/**
+	 * Sets the tagged joints for this model's animations. Tagged joints have their data preserved post-sampling.
+	 * @param joints
+	 *           An array of joint names.
+	 */
+	public void setTaggedJoints(ArrayList<String> joints)
+	{
+		mTaggedJointNames = joints;
+	}
 
 	//TODO: Split this out to an MD5toKeyframe loader in com.badlogic.gdx.graphics.loaders
 	/**
@@ -115,6 +131,11 @@ public class KeyframedModel {
 			Keyframe k = new Keyframe();
 			k.Vertices = new float[mNumMeshes][];
 			k.Indices = new short[mNumMeshes][];
+			if(mTaggedJointNames.size() > 0)
+			{
+				k.TaggedJointPos = new Vector3[mTaggedJointNames.size()];
+				k.TaggedJoint = new Quaternion[mTaggedJointNames.size()];
+			}
 			
 			for(int m=0; m<mNumMeshes; m++)
 			{
@@ -130,6 +151,7 @@ public class KeyframedModel {
 				if(mTarget[m] == null)
 				{
 					mAnimator.setKeyframeDimensions(m, numVertices, numIndices);
+					mAnimator.setNumTaggedJoints(mTaggedJointNames.size());
 
 					VertexAttributes attribs = renderer.getMesh().getVertexAttributes();
 					mTarget[m] = new Mesh(false, numVertices, numIndices, attribs);
@@ -138,6 +160,31 @@ public class KeyframedModel {
 				}
 			}
 			
+			//store tagged joints.
+			MD5Joints skel = animator.getSkeleton();
+			for(int tj=0; tj<mTaggedJointNames.size(); tj++)
+			{
+				String name = mTaggedJointNames.get(tj);
+				for(int j=0; j<skel.numJoints; j++)
+				{
+					if(name.equals(skel.names[j]))
+					{
+						int idx = j*8;
+						float p = skel.joints[idx];
+						float x = skel.joints[idx+1];
+						float y = skel.joints[idx+2];
+						float z = skel.joints[idx+3];
+						k.TaggedJointPos[tj] = new Vector3(x, y, z);
+						float qx = skel.joints[idx+4];
+						float qy = skel.joints[idx+5];
+						float qz = skel.joints[idx+6];
+						float qw = skel.joints[idx+7];
+						k.TaggedJoint[tj] = new Quaternion(qx, qy, qz, qw);
+						break;
+					}
+				}
+			}
+			
 			a.mKeyframes[i] = k;
 			
 			animator.update(sampleRate);
@@ -149,6 +196,16 @@ public class KeyframedModel {
 		mAnimations.put(animKey, a);
 	}
 	
+	public void getJointData(int tagIndex, Vector3 pos, Quaternion orient)
+	{
+		Keyframe kf = mAnimator.getInterpolatedKeyframe();
+		pos.set(kf.TaggedJointPos[tagIndex]);
+		orient.x = kf.TaggedJoint[tagIndex].x;
+		orient.y = kf.TaggedJoint[tagIndex].y;
+		orient.z = kf.TaggedJoint[tagIndex].z;
+		orient.w = kf.TaggedJoint[tagIndex].w;
+	}
+	
 	/**
 	 * Set the current playing animation.
 	 * @param animKey
-- 
2.11.0