Add bit size of column attribute to type information

wowdev · Jan 5, 2025 · 1af829e · 1af829e
1 parent 723f901
commit 1af829e
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/DBCD.IO/Attributes/SizeInBitsAttribute.cs b/DBCD.IO/Attributes/SizeInBitsAttribute.cs
@@ -0,0 +1,13 @@
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace DBCD.IO.Attributes
+{
+    public class SizeInBitsAttribute: Attribute
+    {
+        public readonly ushort Size;
+
+        public SizeInBitsAttribute(ushort size) => Size = size;
+    }
+}
diff --git a/DBCD.IO/DBParser.cs b/DBCD.IO/DBParser.cs
@@ -5,6 +5,9 @@
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+using System.Runtime.CompilerServices;
+
+[assembly:InternalsVisibleTo("DBCD")]
 
 namespace DBCD.IO
 {
@@ -23,6 +26,7 @@ public class DBParser
         public uint LayoutHash => _reader.LayoutHash;
         public int IdFieldIndex => _reader.IdFieldIndex;
         public DB2Flags Flags => _reader.Flags;
+        internal ColumnMetaData[] ColumnMeta => _reader.ColumnMeta;
         #endregion
 
         public DBParser(string fileName) : this(File.Open(fileName, FileMode.Open, FileAccess.Read, FileShare.Read)) { }

diff --git a/DBCD/DBCDBuilder.cs b/DBCD/DBCDBuilder.cs
@@ -77,6 +77,7 @@ internal Tuple<Type, DBCDInfo> Build(DBParser dbcReader, Stream dbd, string name
             var columns = new List<string>(fields.Length);
             bool localiseStrings = locale != Locale.None;
 
+            var metadataIndex = 0;
             foreach (var fieldDefinition in fields)
             {
                 var columnDefinition = databaseDefinition.columnDefinitions[fieldDefinition.name];
@@ -100,6 +101,15 @@ internal Tuple<Type, DBCDInfo> Build(DBParser dbcReader, Stream dbd, string name
                 if (fieldDefinition.isID)
                 {
                     AddAttribute<IndexAttribute>(field, fieldDefinition.isNonInline);
+                } 
+
+                if (!fieldDefinition.isNonInline)
+                {
+                    if (metadataIndex < dbcReader.ColumnMeta.Length)
+                    {
+                        AddAttribute<SizeInBitsAttribute>(field, dbcReader.ColumnMeta[metadataIndex].Size);
+                    }
+                    metadataIndex++;
                 }
 
                 if (fieldDefinition.arrLength > 1)