Add project files.

2022-09-27 17:40:31 +04:00
parent 871d908948
commit a476ace7d7
305 changed files with 71340 additions and 84 deletions
--- a/AssetStudio/ResourceReader.cs
+++ b/AssetStudio/ResourceReader.cs
@@ -0,0 +1,91 @@
+using System.IO;
+
+namespace AssetStudio
+{
+    public class ResourceReader
+    {
+        private bool needSearch;
+        private string path;
+        private SerializedFile assetsFile;
+        private long offset;
+        private long size;
+        private BinaryReader reader;
+
+        public int Size { get => (int)size; }
+
+        public ResourceReader(string path, SerializedFile assetsFile, long offset, long size)
+        {
+            needSearch = true;
+            this.path = path;
+            this.assetsFile = assetsFile;
+            this.offset = offset;
+            this.size = size;
+        }
+
+        public ResourceReader(BinaryReader reader, long offset, long size)
+        {
+            this.reader = reader;
+            this.offset = offset;
+            this.size = size;
+        }
+
+        private BinaryReader GetReader()
+        {
+            if (needSearch)
+            {
+                var resourceFileName = Path.GetFileName(path);
+                if (assetsFile.assetsManager.resourceFileReaders.TryGetValue(resourceFileName, out reader))
+                {
+                    needSearch = false;
+                    return reader;
+                }
+                var assetsFileDirectory = Path.GetDirectoryName(assetsFile.fullName);
+                var resourceFilePath = Path.Combine(assetsFileDirectory, resourceFileName);
+                if (!File.Exists(resourceFilePath))
+                {
+                    var findFiles = Directory.GetFiles(assetsFileDirectory, resourceFileName, SearchOption.AllDirectories);
+                    if (findFiles.Length > 0)
+                    {
+                        resourceFilePath = findFiles[0];
+                    }
+                }
+                if (File.Exists(resourceFilePath))
+                {
+                    needSearch = false;
+                    reader = new BinaryReader(File.OpenRead(resourceFilePath));
+                    assetsFile.assetsManager.resourceFileReaders.Add(resourceFileName, reader);
+                    return reader;
+                }
+                throw new FileNotFoundException($"Can't find the resource file {resourceFileName}");
+            }
+            else
+            {
+                return reader;
+            }
+        }
+
+        public byte[] GetData()
+        {
+            var binaryReader = GetReader();
+            binaryReader.BaseStream.Position = offset;
+            return binaryReader.ReadBytes((int)size);
+        }
+
+        public void GetData(byte[] buff)
+        {
+            var binaryReader = GetReader();
+            binaryReader.BaseStream.Position = offset;
+            binaryReader.Read(buff, 0, (int)size);
+        }
+
+        public void WriteData(string path)
+        {
+            var binaryReader = GetReader();
+            binaryReader.BaseStream.Position = offset;
+            using (var writer = File.OpenWrite(path))
+            {
+                binaryReader.BaseStream.CopyTo(writer, size);
+            }
+        }
+    }
+}