/*
*
* def flip_euler(euler, rotation_mode):
* ret = euler.copy()
* inner_axis = rotation_mode[0]
* outer_axis = rotation_mode[2]
* middle_axis = rotation_mode[1]
*
* ret[euler_axis_index(z)] += pi
* ret[euler_axis_index(x)] += pi
* ret[euler_axis_index(y)] *= -1
* ret[euler_axis_index(y)] += pi
* return ret
* */
/// <summary>
///
/// </summary>
private static void RemoveUnusedTracks(HSD_JOBJ model, HSD_AnimJoint animJoint)
{
JointAnimManager manager = new JointAnimManager();
manager.FromAnimJoint(animJoint);
var joints = model.BreathFirstList;
if (joints.Count != manager.Nodes.Count)
{
return;
}
for (int i = 0; i < joints.Count; i++)
{
var fobjs = manager.Nodes[i].Tracks;
var toRem = new List <FOBJ_Player>();
foreach (var f in fobjs)
{
bool remove = false;
switch (f.JointTrackType)
{
case JointTrackType.HSD_A_J_TRAX: remove = RemoveTrack(f, joints[i].TX); break;
case JointTrackType.HSD_A_J_TRAY: remove = RemoveTrack(f, joints[i].TY); break;
case JointTrackType.HSD_A_J_TRAZ: remove = RemoveTrack(f, joints[i].TZ); break;
case JointTrackType.HSD_A_J_ROTX: remove = RemoveTrack(f, joints[i].RX); break;
case JointTrackType.HSD_A_J_ROTY: remove = RemoveTrack(f, joints[i].RY); break;
case JointTrackType.HSD_A_J_ROTZ: remove = RemoveTrack(f, joints[i].RZ); break;
case JointTrackType.HSD_A_J_SCAX: remove = RemoveTrack(f, joints[i].SX); break;
case JointTrackType.HSD_A_J_SCAY: remove = RemoveTrack(f, joints[i].SY); break;
case JointTrackType.HSD_A_J_SCAZ: remove = RemoveTrack(f, joints[i].SZ); break;
}
if (remove)
{
toRem.Add(f);
}
}
foreach (var v in toRem)
{
fobjs.Remove(v);
}
}
var newAnimJoint = manager.ToAnimJoint(model, AOBJ_Flags.ANIM_LOOP);
animJoint._s = newAnimJoint._s;
}
/// <summary>
///
/// </summary>
/// <param name="source"></param>
/// <param name="target"></param>
/// <param name="animation"></param>
/// <returns></returns>
public static JointAnimManager Retarget(HSD_JOBJ source, HSD_JOBJ target, JointAnimManager animation, JointMap sourceMap = null, JointMap targetMap = null)
{
var sourceManager = new JOBJManager();
sourceManager.SetJOBJ(source);
var sourceInverseWorldTransforms = new List <Matrix4>();
for (int i = 0; i < sourceManager.JointCount; i++)
{
sourceInverseWorldTransforms.Add(sourceManager.GetWorldTransform(i).Inverted());
}
sourceManager.Animation = animation;
var targetManager = new JOBJManager();
targetManager.SetJOBJ(target);
JointAnimManager newAnim = new JointAnimManager();
newAnim.FrameCount = sourceManager.Animation.FrameCount;
var targetWorldTransforms = new List <Matrix4>();
for (int i = 0; i < targetManager.JointCount; i++)
{
targetWorldTransforms.Add(targetManager.GetWorldTransform(i));
newAnim.Nodes.Add(new AnimNode());
}
targetManager.Animation = newAnim;
for (int f = 0; f <= sourceManager.Animation.FrameCount; f++)
{
sourceManager.Frame = f - 1;
sourceManager.UpdateNoRender();
targetManager.Frame = f;
targetManager.UpdateNoRender();
// bake animation on target skeleton
for (int i = 0; i < sourceManager.JointCount; i++)
{
int targetIndex = i;
int sourceIndex = i;
// remap bone if joint maps are present
if (sourceMap != null && targetMap != null && !string.IsNullOrEmpty(sourceMap[i]))
{
targetIndex = targetMap.IndexOf(sourceMap[i]);
}
if (targetIndex == -1)
{
continue;
}
int targetParentIndex = targetManager.ParentIndex(targetManager.GetJOBJ(targetIndex));
var inverseSourceWorldRotation = sourceInverseWorldTransforms[sourceIndex];
var sourceAnimatedWorldRotation = sourceManager.GetWorldTransform(sourceIndex);
var targetWorldRotation = targetWorldTransforms[targetIndex];
var rel = targetWorldRotation *
inverseSourceWorldRotation *
sourceAnimatedWorldRotation;
if (targetParentIndex != -1)
{
rel *= targetManager.GetWorldTransform(targetParentIndex).Inverted();
}
var rot = Math3D.ToEulerAngles(rel.ExtractRotation().Inverted());
var targetJOBJ = targetManager.GetJOBJ(targetIndex);
var sourceJOBJ = sourceManager.GetJOBJ(sourceIndex);
sourceManager.Animation.GetAnimatedState(f - 1, sourceIndex, sourceJOBJ, out float TX, out float TY, out float TZ, out float RX, out float RY, out float RZ, out float SX, out float SY, out float SZ);
var relTranslate = new Vector3(sourceJOBJ.TX - TX, sourceJOBJ.TY - TY, sourceJOBJ.TZ - TZ);
//relTranslate = Vector3.TransformNormal(relTranslate, Matrix4.CreateFromQuaternion(Math3D.FromEulerAngles(sourceJOBJ.RZ, sourceJOBJ.RY, sourceJOBJ.RX).Inverted()));
//relTranslate = Vector3.TransformNormal(relTranslate, Matrix4.CreateFromQuaternion(Math3D.FromEulerAngles(targetJOBJ.RZ, targetJOBJ.RY, targetJOBJ.RX)));
var relScale = new Vector3(sourceJOBJ.SX - SX, sourceJOBJ.SY - SY, sourceJOBJ.SZ - SZ);
//relScale = Vector3.TransformNormal(relScale, inverseSourceWorldRotation);
//relScale = Vector3.TransformNormal(relScale, targetWorldRotation);
AddKey(targetIndex, f, targetJOBJ.TX - relTranslate.X, JointTrackType.HSD_A_J_TRAX, newAnim);
AddKey(targetIndex, f, targetJOBJ.TY - relTranslate.Y, JointTrackType.HSD_A_J_TRAY, newAnim);
AddKey(targetIndex, f, targetJOBJ.TZ - relTranslate.Z, JointTrackType.HSD_A_J_TRAZ, newAnim);
AddKey(targetIndex, f, rot.X, JointTrackType.HSD_A_J_ROTX, newAnim);
AddKey(targetIndex, f, rot.Y, JointTrackType.HSD_A_J_ROTY, newAnim);
AddKey(targetIndex, f, rot.Z, JointTrackType.HSD_A_J_ROTZ, newAnim);
AddKey(targetIndex, f, targetJOBJ.SX - relScale.X, JointTrackType.HSD_A_J_SCAX, newAnim);
AddKey(targetIndex, f, targetJOBJ.SY - relScale.Y, JointTrackType.HSD_A_J_SCAY, newAnim);
AddKey(targetIndex, f, targetJOBJ.SZ - relScale.Z, JointTrackType.HSD_A_J_SCAZ, newAnim);
targetManager.UpdateNoRender();
}
}
EulerFilter(newAnim);
var targetAnim = newAnim.ToAnimJoint(target, AOBJ_Flags.ANIM_LOOP);
AnimationCompressor.AdaptiveCompressAnimation(targetAnim, targetMap);
RemoveUnusedTracks(target, targetAnim);
newAnim.FromAnimJoint(targetAnim);
// apply additional single frame filter check
for (int i = 0; i < sourceManager.JointCount; i++)
{
int targetIndex = i;
int sourceIndex = i;
// remap bone if joint maps are present
if (sourceMap != null && targetMap != null && !string.IsNullOrEmpty(sourceMap[i]))
{
targetIndex = targetMap.IndexOf(sourceMap[i]);
}
if (targetIndex == -1)
{
continue;
}
var sourceNode = animation.Nodes[sourceIndex];
var targetNode = newAnim.Nodes[targetIndex];
FOBJ_Player sourceXRot = null;
FOBJ_Player sourceYRot = null;
FOBJ_Player sourceZRot = null;
foreach (var t in sourceNode.Tracks)
{
if (t.JointTrackType == JointTrackType.HSD_A_J_ROTX)
{
sourceXRot = t;
}
if (t.JointTrackType == JointTrackType.HSD_A_J_ROTY)
{
sourceYRot = t;
}
if (t.JointTrackType == JointTrackType.HSD_A_J_ROTZ)
{
sourceZRot = t;
}
}
if (sourceXRot != null && sourceYRot != null && sourceZRot != null &&
sourceXRot.Keys.Count == 1 && sourceYRot.Keys.Count == 1 && sourceZRot.Keys.Count == 1)
{
foreach (var t in targetNode.Tracks)
{
if (t.JointTrackType == JointTrackType.HSD_A_J_ROTX ||
t.JointTrackType == JointTrackType.HSD_A_J_ROTY ||
t.JointTrackType == JointTrackType.HSD_A_J_ROTZ)
{
t.Keys.RemoveAll(e => e.Frame > 0);
t.Keys[0].InterpolationType = GXInterpolationType.HSD_A_OP_KEY;
}
}
}
}
return(newAnim);
}
