Add project files.

AssetStudio/Classes/PPtr.cs

@@ -0,0 +1,152 @@
+using System;
+
+namespace AssetStudio
+{
+    public sealed class PPtr<T> : IYAMLExportable
+        where T : Object 
+    {
+        public int m_FileID;
+        public long m_PathID;
+
+        private SerializedFile assetsFile;
+        private int index = -2; //-2 - Prepare, -1 - Missing
+
+        public PPtr(ObjectReader reader)
+        {
+            m_FileID = reader.ReadInt32();
+            m_PathID = reader.m_Version < SerializedFileFormatVersion.Unknown_14 ? reader.ReadInt32() : reader.ReadInt64();
+            assetsFile = reader.assetsFile;
+        }
+
+        public PPtr(int fileID, long pathID, SerializedFile assetsFile)
+        {
+            m_FileID = fileID;
+            m_PathID = pathID;
+            this.assetsFile = assetsFile;
+        }
+
+        public YAMLNode ExportYAML()
+        {
+            var node = new YAMLMappingNode();
+            node.Style = MappingStyle.Flow;
+            node.Add("fileID", m_FileID);
+            return node;
+        }
+
+        private bool TryGetAssetsFile(out SerializedFile result)
+        {
+            result = null;
+            if (m_FileID == 0)
+            {
+                result = assetsFile;
+                return true;
+            }
+
+            if (m_FileID > 0 && m_FileID - 1 < assetsFile.m_Externals.Count)
+            {
+                var assetsManager = assetsFile.assetsManager;
+                var assetsFileList = assetsManager.assetsFileList;
+                var assetsFileIndexCache = assetsManager.assetsFileIndexCache;
+
+                if (index == -2)
+                {
+                    var m_External = assetsFile.m_Externals[m_FileID - 1];
+                    var name = m_External.fileName;
+                    if (!assetsFileIndexCache.TryGetValue(name, out index))
+                    {
+                        index = assetsFileList.FindIndex(x => x.fileName.Equals(name, StringComparison.OrdinalIgnoreCase));
+                        assetsFileIndexCache.Add(name, index);
+                    }
+                }
+
+                if (index >= 0)
+                {
+                    result = assetsFileList[index];
+                    return true;
+                }
+            }
+
+            return false;
+        }
+
+        public bool TryGet(out T result)
+        {
+            if (TryGetAssetsFile(out var sourceFile))
+            {
+                if (sourceFile.ObjectsDic.TryGetValue(m_PathID, out var obj))
+                {
+                    if (obj is T variable)
+                    {
+                        result = variable;
+                        return true;
+                    }
+                }
+            }
+
+            result = null;
+            return false;
+        }
+
+        public bool TryGet<T2>(out T2 result) where T2 : Object
+        {
+            if (TryGetAssetsFile(out var sourceFile))
+            {
+                if (sourceFile.ObjectsDic.TryGetValue(m_PathID, out var obj))
+                {
+                    if (obj is T2 variable)
+                    {
+                        result = variable;
+                        return true;
+                    }
+                }
+            }
+
+            result = null;
+            return false;
+        }
+
+        public void Set(T m_Object)
+        {
+            var name = m_Object.assetsFile.fileName;
+            if (string.Equals(assetsFile.fileName, name, StringComparison.OrdinalIgnoreCase))
+            {
+                m_FileID = 0;
+            }
+            else
+            {
+                m_FileID = assetsFile.m_Externals.FindIndex(x => string.Equals(x.fileName, name, StringComparison.OrdinalIgnoreCase));
+                if (m_FileID == -1)
+                {
+                    assetsFile.m_Externals.Add(new FileIdentifier
+                    {
+                        fileName = m_Object.assetsFile.fileName
+                    });
+                    m_FileID = assetsFile.m_Externals.Count;
+                }
+                else
+                {
+                    m_FileID += 1;
+                }
+            }
+
+            var assetsManager = assetsFile.assetsManager;
+            var assetsFileList = assetsManager.assetsFileList;
+            var assetsFileIndexCache = assetsManager.assetsFileIndexCache;
+
+            if (!assetsFileIndexCache.TryGetValue(name, out index))
+            {
+                index = assetsFileList.FindIndex(x => x.fileName.Equals(name, StringComparison.OrdinalIgnoreCase));
+                assetsFileIndexCache.Add(name, index);
+            }
+
+            m_PathID = m_Object.m_PathID;
+        }
+
+        public PPtr<T2> CastTo<T2>() where T2 : Object
+        {
+            return new PPtr<T2>(m_FileID, m_PathID, assetsFile);
+        }
+
+        public bool IsNull => m_PathID == 0 || m_FileID < 0;
+    }
+}