add transfer animation option to recorder settings

Author: vkovec

com.unity.formats.fbx/Editor/Sources/Recorders/FbxRecorder/FbxRecorder.cs

@@ -62,6 +62,8 @@ protected override void EndRecording(RecordingSession session)
 
                 AnimationUtility.SetAnimationClips(animator, new AnimationClip[] { clip });
                 var exportSettings = new ExportModelSettingsSerialize();
+                exportSettings.SetAnimationSource(settings.TransferAnimationSource);
+                exportSettings.SetAnimationDest(settings.TransferAnimationDest);
                 var toInclude = ExportSettings.Include.ModelAndAnim;
                 if (!settings.ExportGeometry)
                 {

com.unity.formats.fbx/Editor/Sources/Recorders/FbxRecorder/FbxRecorderSettings.cs

@@ -1,4 +1,4 @@
-using System.Collections;
+using System;
 using System.Collections.Generic;
 using UnityEngine;
 using UnityEditor.Recorder;
@@ -7,7 +7,7 @@
 namespace UnityEditor.Formats.Fbx.Exporter
 {
     [RecorderSettings(typeof(FbxRecorder), "FBX", "fbx_recorder")]
-    public class FbxRecorderSettings : RecorderSettings
+    internal class FbxRecorderSettings : RecorderSettings
     {
         [SerializeField] bool m_exportGeometry = true;
         public bool ExportGeometry
@@ -22,6 +22,167 @@ public bool ExportGeometry
             }
         }
 
+        [SerializeField]
+        private string m_animSourceBindingId;
+        [SerializeField]
+        private string m_animDestBindingId;
+
+        public Transform TransferAnimationSource
+        {
+            get
+            {
+                if (string.IsNullOrEmpty(m_animSourceBindingId))
+                    return null;
+
+                return GetBinding(m_animSourceBindingId);
+            }
+
+            set
+            {
+                if (!TransferAnimationSourceIsValid(value))
+                {
+                    return;
+                }
+                if (string.IsNullOrEmpty(m_animSourceBindingId))
+                    m_animSourceBindingId = GenerateBindingId();
+
+                SetBinding(m_animSourceBindingId, value);
+            }
+        }
+
+        public Transform TransferAnimationDest
+        {
+            get
+            {
+                if (string.IsNullOrEmpty(m_animDestBindingId))
+                    return null;
+
+                return GetBinding(m_animDestBindingId);
+            }
+
+            set
+            {
+                if (!TransferAnimationDestIsValid(value))
+                {
+                    return;
+                }
+                if (string.IsNullOrEmpty(m_animDestBindingId))
+                    m_animDestBindingId = GenerateBindingId();
+
+                SetBinding(m_animDestBindingId, value);
+            }
+        }
+
+        static Transform GetBinding(string id)
+        {
+            // return BindingManager.Get(m_BindingId) as GameObject;
+            var method = Type.GetType("UnityEditor.Recorder.BindingManager,Unity.Recorder.Editor").GetMethod("Get", System.Reflection.BindingFlags.Static | System.Reflection.BindingFlags.Public);
+            return method.Invoke(null, new object[] { id }) as Transform;
+        }
+
+        static void SetBinding(string id, UnityEngine.Object obj)
+        {
+            // BindingManager.Set(m_BindingId, value);
+            var method = Type.GetType("UnityEditor.Recorder.BindingManager,Unity.Recorder.Editor").GetMethod("Set", System.Reflection.BindingFlags.Static | System.Reflection.BindingFlags.Public);
+            method.Invoke(null, new object[] { id, obj });
+        }
+
+        static string GenerateBindingId()
+        {
+            return GUID.Generate().ToString();
+        }
+
+        /// <summary>
+        /// Determines whether p is an ancestor to t.
+        /// </summary>
+        /// <returns><c>true</c> if p is ancestor to t; otherwise, <c>false</c>.</returns>
+        /// <param name="p">P.</param>
+        /// <param name="t">T.</param>
+        protected bool IsAncestor(Transform p, Transform t)
+        {
+            var curr = t;
+            while (curr != null)
+            {
+                if (curr == p)
+                {
+                    return true;
+                }
+                curr = curr.parent;
+            }
+            return false;
+        }
+
+        /// <summary>
+        /// Determines whether t1 and t2 are in the same hierarchy.
+        /// </summary>
+        /// <returns><c>true</c> if t1 is in same hierarchy as t2; otherwise, <c>false</c>.</returns>
+        /// <param name="t1">T1.</param>
+        /// <param name="t2">T2.</param>
+        protected bool IsInSameHierarchy(Transform t1, Transform t2)
+        {
+            return (IsAncestor(t1, t2) || IsAncestor(t2, t1));
+        }
+
+        protected bool TransferAnimationSourceIsValid(Transform newValue)
+        {
+            if (!newValue)
+            {
+                return true;
+            }
+
+            var selectedGO = m_AnimationInputSettings.gameObject;
+
+            if (!selectedGO)
+            {
+                Debug.LogWarning("FbxRecorderSettings: no Objects selected for export, can't transfer animation");
+                return false;
+            }
+
+            // source must be ancestor to dest
+            if (TransferAnimationDest && !IsAncestor(newValue, TransferAnimationDest))
+            {
+                Debug.LogWarningFormat("FbxRecorderSettings: Source {0} must be an ancestor of {1}", newValue.name, TransferAnimationDest.name);
+                return false;
+            }
+            // must be in same hierarchy as selected GO
+            if (!selectedGO || !IsInSameHierarchy(newValue, selectedGO.transform))
+            {
+                Debug.LogWarningFormat("FbxRecorderSettings: Source {0} must be in the same hierarchy as {1}", newValue.name, selectedGO ? selectedGO.name : "the selected object");
+                return false;
+            }
+            return true;
+        }
+
+        protected bool TransferAnimationDestIsValid(Transform newValue)
+        {
+            if (!newValue)
+            {
+                return true;
+            }
+
+            var selectedGO = m_AnimationInputSettings.gameObject;
+
+            if (!selectedGO)
+            {
+                Debug.LogWarning("FbxRecorderSettings: no Objects selected for export, can't transfer animation");
+                return false;
+            }
+
+            // source must be ancestor to dest
+            if (TransferAnimationSource && !IsAncestor(TransferAnimationSource, newValue))
+            {
+                Debug.LogWarningFormat("FbxRecorderSettings: Destination {0} must be a descendant of {1}", newValue.name, TransferAnimationSource.name);
+                return false;
+            }
+            // must be in same hierarchy as selected GO
+            if (!selectedGO || !IsInSameHierarchy(newValue, selectedGO.transform))
+            {
+                Debug.LogWarningFormat("FbxRecorderSettings: Destination {0} must be in the same hierarchy as {1}", newValue.name, selectedGO ? selectedGO.name : "the selected object");
+                return false;
+            }
+            return true;
+        }
+
         [SerializeField] AnimationInputSettings m_AnimationInputSettings = new AnimationInputSettings();
 
         public AnimationInputSettings animationInputSettings

com.unity.formats.fbx/Editor/Sources/Recorders/FbxRecorder/FbxRecorderSettingsEditor.cs

@@ -17,5 +17,22 @@ protected override void FileTypeAndFormatGUI()
 
             settings.ExportGeometry = EditorGUILayout.Toggle("Export Geometry", settings.ExportGeometry);
         }
+
+        protected override void OnEncodingGui()
+        {
+            base.OnEncodingGui();
+
+            DrawSeparator();
+
+            EditorGUILayout.LabelField(new GUIContent(
+                "Transfer Animation",
+                "Transfer transform animation from source to destination. Animation on objects between source and destination will also be transferred to destination."
+            ));
+
+            FbxRecorderSettings settings = target as FbxRecorderSettings;
+
+            settings.TransferAnimationSource = EditorGUILayout.ObjectField("Source", settings.TransferAnimationSource, typeof(Transform), allowSceneObjects: true) as Transform;
+            settings.TransferAnimationDest = EditorGUILayout.ObjectField("Destination", settings.TransferAnimationDest, typeof(Transform), allowSceneObjects: true) as Transform;
+        }
     }
 }