Merge pull request #104 from Kamzik123/an2-serialization

Added an2 serialization code
Greavesy1899 · Aug 9, 2024 · 88b6e08 · 88b6e08
2 parents 129ae6b + 0d13a9a
commit 88b6e08
Show file tree

Hide file tree

Showing 5 changed files with 103 additions and 8 deletions.
diff --git a/Mafia2Libs/Core/IO/FileAnimation2.cs b/Mafia2Libs/Core/IO/FileAnimation2.cs
@@ -14,6 +14,7 @@ public FileAnimation2(FileInfo info) : base(info)
         public override bool Open()
         {
             Animation2 anim = new(file.FullName);
+            //anim.WriteToFile(file.FullName + "_test");
             return true;
         }
 

diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Animations/AnimTrack.cs b/Mafia2Libs/ResourceTypes/FileTypes/Animations/AnimTrack.cs
@@ -157,7 +157,7 @@ private void Quantize(float duration)
             Duration = duration;
             NumKeyFrames = (short)KeyFrames.Length;
 
-            var refQuat = KeyFrames.Length > 0 ? KeyFrames[0].value : Quaternion.Identity;
+            var refQuat = KeyFrames.Length > 0 ? Quaternion.Inverse(KeyFrames[0].value) : Quaternion.Identity;
 
             PackedReferenceQuat = PackReferenceQuaternion(refQuat);
             refQuat = UnpackReferenceQuaternion(PackedReferenceQuat);
@@ -458,7 +458,7 @@ private float GetOptimalScale(Quaternion invRefQuat)
             for (int i = 0; i < KeyFrames.Length; i++)
             {
                 var frame = KeyFrames[i];
-                var q = invRefQuat * frame.value;
+                var q = invRefQuat * Quaternion.Inverse(frame.value);
                 List<float> values = new() { Math.Abs(q.X), Math.Abs(q.Y), Math.Abs(q.Z), Math.Abs(q.W) };
                 values.Remove(values.Max());
                 var temp = values.Max();
@@ -686,8 +686,8 @@ public void Dequantize()
 
         public void Quantize()
         {
-            Scale = ComputeScale();
-            Scale = Scale > 1.0f ? Scale - 1.0f : 1.0f;
+            Scale = GetOptimalScale();
+            Scale = Scale > 0.0f ? Scale : 1.0f;
 
             var data = new BigInteger();
             var chunkSize = 96;
@@ -717,16 +717,20 @@ public void Quantize()
             Data = data.ToByteArray();
         }
 
-        public float ComputeScale()
+        public float GetOptimalScale()
         {
             float scale = 0.0f;
 
-            foreach (var frame in KeyFrames)
+            for (int i = 0; i < KeyFrames.Length; i++)
             {
-                scale += Math.Abs(frame.value.X) + Math.Abs(frame.value.Y) + Math.Abs(frame.value.Z);
+                var frame = KeyFrames[i];
+                var v = frame.value;
+                List<float> values = new() { Math.Abs(v.X), Math.Abs(v.Y), Math.Abs(v.Z) };
+                var temp = values.Max();
+                scale = temp > scale ? temp : scale;
             }
 
-            return (float)Math.Round(scale * 10 / KeyFrames.Length);
+            return scale;
         }
     }
 }
diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Animations/Animation2.cs b/Mafia2Libs/ResourceTypes/FileTypes/Animations/Animation2.cs
@@ -1,4 +1,5 @@
 using ResourceTypes.ModelHelpers.ModelExporter;
+using System;
 using System.IO;
 using Utils.Logging;
 
@@ -97,6 +98,67 @@ public void Read(BinaryReader br)
             ToolkitAssert.Ensure(br.BaseStream.Position == br.BaseStream.Length, "Animation2: Failed to reach EOF.");
         }
 
+        public void WriteToFile(string fileName)
+        {
+            using (MemoryStream ms = new())
+            {
+                using (BinaryWriter bw = new(ms))
+                {
+                    Write(bw);
+                }
+
+                File.WriteAllBytes(fileName, ms.ToArray());
+            }
+        }
+
+        public void Write(BinaryWriter bw)
+        {
+            int Count = Header.RootBoneID != 0 ? (Tracks.Length - 1) : Tracks.Length;
+            Header.Count = (short)Count;
+            Header.NumEvents = (short)Events.Length;
+
+            Header.Write(bw);
+            bw.Write(IsDataPresent);
+
+            foreach (var val in Events)
+            {
+                val.Write(bw);
+            }
+
+            bw.Write(Unk00);
+            bw.Write(Unk01);
+            bw.Write((short)Count);
+
+            foreach (var track in Tracks)
+            {
+                track.Write(bw);
+            }
+
+            if (UnkShorts00.Length < Unk01)
+            {
+                short[] newUnk00Shorts = new short[Unk01];
+                Array.Copy(UnkShorts00, 0, newUnk00Shorts, 0, UnkShorts00.Length);
+                UnkShorts00 = newUnk00Shorts;
+            }
+
+            if (UnkShorts01.Length < Count)
+            {
+                short[] newUnk01Shorts = new short[Count];
+                Array.Copy(UnkShorts01, 0, newUnk01Shorts, 0, UnkShorts01.Length);
+                UnkShorts01 = newUnk01Shorts;
+            }
+
+            foreach (var val in UnkShorts00)
+            {
+                bw.Write(val);
+            }
+
+            foreach (var val in UnkShorts01)
+            {
+                bw.Write(val);
+            }
+        }
+
         private void ConvertToMTB()
         {
             MT_ObjectBundle NewBundle = new MT_ObjectBundle();

diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Animations/Event.cs b/Mafia2Libs/ResourceTypes/FileTypes/Animations/Event.cs
@@ -34,5 +34,11 @@ public void Read(BinaryReader br)
             EventID = br.ReadUInt32();
             Time = br.ReadSingle();
         }
+
+        public void Write(BinaryWriter bw)
+        {
+            bw.Write(EventID);
+            bw.Write(Time);
+        }
     }
 }
diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Animations/Header.cs b/Mafia2Libs/ResourceTypes/FileTypes/Animations/Header.cs
@@ -75,5 +75,27 @@ public void Read(BinaryReader br)
             Unk09 = br.ReadByte();
             RootBoneID = br.ReadUInt64();
         }
+
+        public void Write(BinaryWriter bw)
+        {
+            bw.Write(SkeletonID);
+            bw.Write(Version);
+            bw.Write(Magic);
+            bw.Write(NumEvents);
+            bw.Write(Unk00);
+            Unk01.WriteToFile(bw);
+            bw.Write(Unk02);
+            bw.Write(Unk03);
+            bw.Write(Unk04);
+            bw.Write(Unk05);
+            bw.Write(Unk06);
+            bw.Write(Unk07);
+            bw.Write(Hash);
+            bw.Write(Unk08);
+            bw.Write(Duration);
+            bw.Write(Count);
+            bw.Write(Unk09);
+            bw.Write(RootBoneID);
+        }
     }
 }