daily update on 20240506 by nitload youmu

源代码/StgSharp/Data/SSD/SSDraw.cs

@@ -52,17 +52,17 @@ namespace StgSharp.Data.FileIO
         public const int SegmentCapacity = 16 * 16;
 
         private Area home;
-
-        private SSDrawHead _head;
-        private string _name; 
-        private SSDSegment currentSegment;
         private int previousHash;
         private List<SSDSegment> segments;
 
+        private SSDrawHead _head;
+        private SSDSegment currentSegment;
+        private string _name;
+
         internal SSDraw()
         {
             segments = new List<SSDSegment>();
-            
+
         }
 
         public SSDraw(string name)
@@ -73,14 +73,11 @@ namespace StgSharp.Data.FileIO
 
         public int ID
         {
-            get { return _head._head.globalID; }
-            set { _head._head.globalID = value; }
+            get => _head._head.globalID;
+            set => _head._head.globalID = value;
         }
 
-        internal SSDrawHead Head
-        {
-            get{return _head; }
-        }
+        internal SSDrawHead Head => _head;
 
         //TODO 分析头，加载文件头，进一步加载全部的数据块
 
@@ -96,7 +93,7 @@ namespace StgSharp.Data.FileIO
                 return null;
             }
             SSDraw ret = new SSDraw();
-            foreach (var item in head)
+            foreach ((int, int) item in head)
             {
 
             }
@@ -105,8 +102,8 @@ namespace StgSharp.Data.FileIO
 
         public void CreateRawFile(string outputFolder, bool rewrite)
         {
-            string route =
-                $"{((outputFolder == string.Empty) ? " " : ($"{outputFolder}\\"))}{$"{_name}.{FileType}"}";
+            string route = (outputFolder == string.Empty) ? ($"{_name}.{FileType}") :
+                ($"{outputFolder} \\ {_name}.{FileType}");
             if (File.Exists(route) && !rewrite)
             {
                 throw new Exception("This SSDRAW has already existed.");
@@ -114,9 +111,7 @@ namespace StgSharp.Data.FileIO
             using (File.Create(route)) { }
             using (File.Open(route, FileMode.Truncate)) { }
             FileStream f = File.Open(route, FileMode.Open);
-            int size = 0;
-            f.Write(_head.GetBytes(out size), 0, size);
-            foreach (var _head in segments)
+            foreach (SSDSegment _head in segments)
             {
                 f.Write(_head.GetBytes(), 0, 18 * 16);
             }
@@ -133,7 +128,7 @@ namespace StgSharp.Data.FileIO
             }
             FileStream stream = File.Open(route, FileMode.Open);
             return new SSDraw(string.Empty);
-            
+
         }
 
         public void WriteRaw(byte[] dataStream)
@@ -172,15 +167,12 @@ namespace StgSharp.Data.FileIO
     }
 
 
-    public unsafe class SSDrawHead : IEnumerable<int>
+    public unsafe class SSDrawHead : IEnumerable<(int, int)>
     {
 
-        private List<int> freeSegments;
+        private SSDSegment segmentRange;
         private string name;
 
-        private List<(int, int)> segmentRange;
-        internal SSDSegmentHead _head;
-
         internal SSDrawHead(string name)
         {
             int hash;
@@ -192,14 +184,16 @@ namespace StgSharp.Data.FileIO
                 hash = BitConverter.ToInt32(bytes, 0);
             }
             _head.previousHash = hash;
-            segmentRange = new List<(int, int)>();
-            freeSegments = new List<int>();
         }
 
         public SSDrawHead()
         {
-            segmentRange = new List<(int, int)>();
-            freeSegments = new List<int>();
+        }
+
+        public (int begin, int end) this[byte index]
+        {
+            get => segmentRange.ReadData<(int, int)>(index);
+            set => segmentRange.WriteData<(int, int)>(index, value);
         }
 
         public byte DiscIndex
@@ -214,85 +208,49 @@ namespace StgSharp.Data.FileIO
             internal set => _head.selfDefine_1 = value;
         }
 
-        public void AddSingleSegment(int id)
+        internal ref SSDSegmentHead _head
         {
-            freeSegments.Add(id);
+            get => ref segmentRange.Head;
         }
 
-        public unsafe byte[] GetBytes(out int size)
+        internal byte pairCount
         {
-            byte[] namecode = Encoding.UTF8.GetBytes(name);
-            size = 16 + 4 +
-                (segmentRange.Count * 8) +
-                (freeSegments.Count * 4) +
-                namecode.Length;
-            byte[] ret = new byte[size];
-            fixed (byte* bptr = ret)
+            get => (byte)(_head.size / 8);
+            set
             {
-                *(SSDSegmentHead*)bptr = _head;
-                *(bptr + 16) = (byte)segmentRange.Count;
-                *(bptr + 17) = (byte)namecode.Length;
-                *(bptr + 18) = (byte)freeSegments.Count;
-                int t = 20;
-                for (int i = 0; i < segmentRange.Count; i++, t += 8)
+                if (value > 32)
                 {
-                    *(int*)(bptr + t) = segmentRange[i].Item1;
-                    *(int*)(bptr + t + 4) = segmentRange[i].Item2;
-                }
-                for (int i = 0; i < freeSegments.Count; i++, t += 4)
-                {
-                    *(int*)(bptr + t) = freeSegments[i];
-                }
-                for (int i = 0; i < namecode.Length; i++, t++)
-                {
-                    *(bptr + t) = namecode[i];
+                    throw new ArgumentOutOfRangeException();
                 }
+                _head.size = (byte)(value * 8);
             }
-            return ret;
         }
 
-        public void WriteRange(int begin, int end)
+
+        public unsafe byte[] GetBytes()
         {
-            segmentRange.Add((begin, end));
+            return segmentRange.GetBytes();
         }
 
-        internal void MarkSegment(int id)
-        {
-            freeSegments.Add(id);
-        }
 
         IEnumerator IEnumerable.GetEnumerator()
         {
-            foreach (var tuple in segmentRange)
+            for (int i = 0; i < pairCount; i++)
             {
-                for (int i = tuple.Item1; i <= tuple.Item2; i++)
-                {
-                    yield return i;
-                }
-            }
-
-            foreach (var item in freeSegments)
-            {
-                yield return item;
+                yield return segmentRange.ReadData<(int, int)>(i);
             }
         }
 
-        IEnumerator<int> IEnumerable<int>.GetEnumerator()
+        IEnumerator<(int, int)> IEnumerable<(int, int)>.GetEnumerator()
         {
-            foreach (var tuple in segmentRange)
+            for (int i = 0; i < pairCount; i++)
             {
-                for (int i = tuple.Item1; i <= tuple.Item2; i++)
-                {
-                    yield return i;
-                }
-            }
-
-            foreach (var item in freeSegments)
-            {
-                yield return item;
+                yield return segmentRange.ReadData<(int, int)>(i);
             }
         }
 
+
+
     }