Initial Commit

pkNX.Structures/Evolution/EvolutionSet6.cs

@@ -0,0 +1,56 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+
+namespace pkNX.Structures
+{
+    /// <summary>
+    /// Generation 6 Evolution Branch Entries
+    /// </summary>
+    public class EvolutionSet6 : EvolutionSet
+    {
+        private const int ENTRY_SIZE = 6;
+        private const int ENTRY_COUNT = 8;
+        public const int SIZE = ENTRY_COUNT * ENTRY_SIZE;
+        private static readonly HashSet<int> argEvos = new HashSet<int> { 6, 8, 16, 17, 18, 19, 20, 21, 22, 29, 30, 32, 33, 34 };
+
+        public EvolutionSet6(byte[] data)
+        {
+            if (data.Length != SIZE)
+                return;
+            PossibleEvolutions = data.GetArray(GetEvo, SIZE);
+        }
+
+        private static EvolutionMethod GetEvo(byte[] data, int offset)
+        {
+            var evo = new EvolutionMethod
+            {
+                Method = (EvolutionType)BitConverter.ToUInt16(data, offset + 0),
+                Argument = BitConverter.ToUInt16(data, offset + 2),
+                Species = BitConverter.ToUInt16(data, offset + 4),
+                // Copy
+                Level = BitConverter.ToUInt16(data, offset + 2),
+            };
+
+            // Argument is used by both Level argument and Item/Move/etc. Clear if appropriate.
+            if (argEvos.Contains((int)evo.Method))
+                evo.Level = 0;
+            return evo;
+        }
+
+        public override byte[] Write()
+        {
+            using (MemoryStream ms = new MemoryStream())
+            using (BinaryWriter bw = new BinaryWriter(ms))
+            {
+                foreach (EvolutionMethod evo in PossibleEvolutions)
+                {
+                    bw.Write((ushort)evo.Method);
+                    bw.Write((ushort)evo.Argument);
+                    bw.Write((ushort)evo.Species);
+                }
+                return ms.ToArray();
+            }
+        }
+    }
+}