Merge pull request #17 from nasshu2916/feature/convert-anim-to-dmx-binary

AnimationClip から DMXのバイナリデータに変換できるようにする

Author: nasshu2916

Assets/ArtNet/Editor/DmxRecorder/ConvertAnimInspector.cs

@@ -1,3 +1,4 @@
+using System.IO;
 using UnityEditor;
 using UnityEngine;
 
@@ -16,15 +17,22 @@ public override void OnInspectorGUI()
 
             EditorGUILayout.BeginVertical(GUI.skin.box);
             {
-                if (GUILayout.Button("Convert"))
+                if (GUILayout.Button("ConvertAnim"))
                 {
-                    Convert(convertAnim);
+                    ConvertAnim(convertAnim);
+                }
+
+                GUILayout.Space(5);
+
+                if (GUILayout.Button("ConvertPacket"))
+                {
+                    ConvertPacket(convertAnim);
                 }
             }
             EditorGUILayout.EndVertical();
         }
 
-        private static void Convert(ConvertAnim convertAnim)
+        private static void ConvertAnim(ConvertAnim convertAnim)
         {
             var binary = convertAnim.binary;
             if (!binary)
@@ -44,5 +52,41 @@ private static void Convert(ConvertAnim convertAnim)
 
             Debug.Log("Conversion complete");
         }
+
+        private static void ConvertPacket(ConvertAnim convertAnim)
+        {
+            var binary = convertAnim.binary;
+            if (!binary)
+            {
+                Debug.LogError("Binary is null");
+                return;
+            }
+            if (string.IsNullOrEmpty(convertAnim.OutputDirectory))
+            {
+                Debug.LogError("Output directory is null or empty");
+                return;
+            }
+
+            var timelineSettingPath = convertAnim.OutputDirectory + "/DmxTimeline.asset";
+            var dmxTimelineSetting = AssetDatabase.LoadAssetAtPath(timelineSettingPath, typeof(DmxTimelineSetting)) as DmxTimelineSetting;
+            if (dmxTimelineSetting is null)
+            {
+                Debug.LogError("DmxTimelineSetting is null");
+                return;
+            }
+
+            Debug.Log($"ArtNet Recorder: {dmxTimelineSetting.DmxTimelines.Count} timelines found");
+            var timelineConverter = new TimelineConverter(dmxTimelineSetting);
+            var dmxPackets = timelineConverter.ToDmxPackets();
+            var storeData = RecordData.Serialize(dmxPackets);
+
+
+            var path = convertAnim.OutputDirectory + "/DmxPackets.bytes";
+            var exists = File.Exists(path);
+            File.WriteAllBytes(path, storeData);
+            var message = exists ? "Data updated" : "Data stored";
+            Debug.Log($"ArtNet Recorder: {message} at {path}");
+            Debug.Log("Conversion complete");
+        }
     }
 }

Assets/ArtNet/Editor/DmxRecorder/RecordData.cs

@@ -15,7 +15,7 @@ public static class RecordData
         private static readonly byte[] ReservedBuffer = new byte[11];
         private static byte Version = 0x01;
 
-        public static byte[] Serialize(List<(int time, DmxPacket packet)> dmxPackets)
+        public static byte[] Serialize(IReadOnlyList<(int time, DmxPacket packet)> dmxPackets)
         {
             var startTime = dmxPackets.Select(x => x.time).OrderBy(x => x).First();
             using var memoryStream = new MemoryStream();

Assets/ArtNet/Editor/DmxRecorder/TimelineConverter.cs

@@ -21,6 +21,14 @@ public TimelineConverter(IReadOnlyCollection<(int time, DmxPacket packet)> packe
             }
         }
 
+        public TimelineConverter(DmxTimelineSetting dmxTimelineSetting)
+        {
+            foreach (var timelineElement in dmxTimelineSetting.DmxTimelines)
+            {
+                Timelines.Add(new TimelineUniverse(timelineElement.Universe, timelineElement.DmxTimelineClip));
+            }
+        }
+
         public void SaveDmxTimelineClips(string directory)
         {
             if (System.IO.Directory.Exists(directory) == false)
@@ -51,6 +59,11 @@ public void SaveDmxTimelineClips(string directory)
             AssetDatabase.Refresh();
         }
 
+        public List<(int time, DmxPacket packet)> ToDmxPackets()
+        {
+            return Timelines.SelectMany(x => x.ToDmxPackets()).OrderBy(x => x.time).ToList();
+        }
+
         private static void SaveAsset<T>(T asset, string directory, string fileName) where T : UnityEngine.Object
         {
             var path = $"{directory}/{fileName}";
@@ -77,6 +90,71 @@ public TimelineUniverse(int universe, IReadOnlyCollection<(int time, DmxPacket p
             }
         }
 
+        public TimelineUniverse(int universe, AnimationClip clip)
+        {
+            Universe = universe;
+            var curveBindings = AnimationUtility.GetCurveBindings(clip);
+            ChannelDmxFrameData = new List<DmxFrameData>[512];
+            for (var i = 0; i < ChannelDmxFrameData.Length; i++)
+            {
+                var propertyName = $"Ch{i + 1:D3}";
+
+                var curve = curveBindings
+                    .Where(binding => binding.propertyName == propertyName)
+                    .Select(binding => AnimationUtility.GetEditorCurve(clip, binding))
+                    .FirstOrDefault();
+
+                if (curve is null) continue;
+
+                ChannelDmxFrameData[i] = curve.keys.Select(x => new DmxFrameData((int) (x.time * 1000), (byte) x.value)).ToList();
+            }
+        }
+
+        public IEnumerable<int> AllFrameTimes()
+        {
+            return ChannelDmxFrameData.SelectMany(x => x.Select(frameData => frameData.Millisecond)).Distinct();
+        }
+
+        public byte FrameValue(int channel, int time)
+        {
+            var dmxFrameData = ChannelDmxFrameData[channel];
+
+            // If there is a frame data at the exact time, return it
+            foreach (var frameData in dmxFrameData.Where(frameData => frameData.Millisecond == time))
+            {
+                return frameData.Value;
+            }
+
+            // if there is no frame data, return 0
+            if (dmxFrameData.Count == 0) return 0;
+
+            // if time is out of range, return the first or last value
+            if (time < dmxFrameData[0].Millisecond) return dmxFrameData[0].Value;
+            if (time > dmxFrameData[^1].Millisecond) return dmxFrameData[^1].Value;
+
+            // return the estimated value from frames around the specified time.
+
+            // Find the frame data before and after the specified time
+            var prev = dmxFrameData[0];
+            var next = dmxFrameData[0];
+            foreach (var frameData in dmxFrameData)
+            {
+                if (frameData.Millisecond > time)
+                {
+                    next = frameData;
+                    break;
+                }
+
+                prev = frameData;
+            }
+
+            // Calculate the estimated value
+            var prevDiff = next.Value - prev.Value;
+            var prevDiffTime = next.Millisecond - prev.Millisecond;
+            var timeDiff = time - prev.Millisecond;
+            return (byte) (prev.Value + (prevDiff * timeDiff / prevDiffTime));
+        }
+
         public AnimationClip ToAnimationClip()
         {
             var curves = ConvertAnimationCurves();
@@ -144,6 +222,35 @@ private static bool IsOmittedFrame(
 
             return Math.Abs((float) prevDiff / prevDiffTime - (float) nextDiff / nextDiffTime) <= tolerance;
         }
+
+        public IEnumerable<(int time, DmxPacket packet)> ToDmxPackets()
+        {
+            byte sequence = 0;
+            var packets = new List<(int time, DmxPacket packet)>();
+            var allFrameTimes = AllFrameTimes().OrderBy(x => x).ToList();
+
+            foreach (var time in allFrameTimes)
+            {
+                var dmx = new byte[512];
+                for (var i = 0; i < ChannelDmxFrameData.Length; i++)
+                {
+                    dmx[i] = FrameValue(i, time);
+                }
+
+                var packet = new DmxPacket { Sequence = sequence, Universe = (ushort) Universe, Dmx = dmx };
+                packets.Add((time, packet));
+                if (sequence >= 255)
+                {
+                    sequence = 0;
+                }
+                else
+                {
+                    sequence++;
+                }
+            }
+
+            return packets;
+        }
     }
 
     public struct DmxFrameData