AssetStudio/Unity Studio/Unity Studio Classes/Helpers.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using static Unity_Studio.UnityStudio;


namespace Unity_Studio
{
    public class PPtr
    {
        //m_FileID 0 means current file
        public int m_FileID;
        //m_PathID acts more like a hash in some games
        public long m_PathID;
    }

    public static class PPtrHelpers
    {
        public static PPtr ReadPPtr(this AssetsFile sourceFile)
        {
            var result = new PPtr();
            var reader = sourceFile.assetsFileReader;

            int FileID = reader.ReadInt32();
            if (FileID >= 0 && FileID < sourceFile.sharedAssetsList.Count)
            {
                var sharedFile = sourceFile.sharedAssetsList[FileID];
                var index = sharedFile.Index;
                if (index == -2)
                {
                    var name = sharedFile.fileName.ToUpper();
                    if (!sharedFileIndex.TryGetValue(name, out index))
                    {
                        index = assetsfileList.FindIndex(aFile => aFile.upperFileName == name);
                        sharedFileIndex.Add(name, index);
                    }
                    sharedFile.Index = index;
                }
                result.m_FileID = index;
            }

            result.m_PathID = sourceFile.fileGen < 14 ? reader.ReadInt32() : reader.ReadInt64();

            return result;
        }

        public static bool TryGetPD(this List<AssetsFile> assetsfileList, PPtr m_elm, out AssetPreloadData result)
        {
            result = null;

            if (m_elm != null && m_elm.m_FileID >= 0 && m_elm.m_FileID < assetsfileList.Count)
            {
                AssetsFile sourceFile = assetsfileList[m_elm.m_FileID];

                //TryGetValue should be safe because m_PathID is 0 when initialized and PathID values range from 1
                if (sourceFile.preloadTable.TryGetValue(m_elm.m_PathID, out result)) { return true; }
            }

            return false;
        }

        public static bool TryGetTransform(this List<AssetsFile> assetsfileList, PPtr m_elm, out Transform m_Transform)
        {
            m_Transform = null;

            if (m_elm != null && m_elm.m_FileID >= 0 && m_elm.m_FileID < assetsfileList.Count)
            {
                AssetsFile sourceFile = assetsfileList[m_elm.m_FileID];

                if (sourceFile.TransformList.TryGetValue(m_elm.m_PathID, out m_Transform)) { return true; }
            }

            return false;
        }

        public static bool TryGetGameObject(this List<AssetsFile> assetsfileList, PPtr m_elm, out GameObject m_GameObject)
        {
            m_GameObject = null;

            if (m_elm != null && m_elm.m_FileID >= 0 && m_elm.m_FileID < assetsfileList.Count)
            {
                AssetsFile sourceFile = assetsfileList[m_elm.m_FileID];

                if (sourceFile.GameObjectList.TryGetValue(m_elm.m_PathID, out m_GameObject)) { return true; }
            }

            return false;
        }

        public static void ParseGameObject(this List<AssetsFile> assetsfileList, GameObject m_GameObject)
        {
            foreach (var m_Component in m_GameObject.m_Components)
            {
                if (m_Component.m_FileID >= 0 && m_Component.m_FileID < assetsfileList.Count)
                {
                    AssetsFile sourceFile = assetsfileList[m_Component.m_FileID];
                    if (sourceFile.preloadTable.TryGetValue(m_Component.m_PathID, out var asset))
                    {
                        switch (asset.Type2)
                        {
                            case 4: //Transform
                                {
                                    m_GameObject.m_Transform = m_Component;
                                    break;
                                }
                            case 23: //MeshRenderer
                                {
                                    m_GameObject.m_MeshRenderer = m_Component;
                                    break;
                                }
                            case 33: //MeshFilter
                                {
                                    m_GameObject.m_MeshFilter = m_Component;
                                    break;
                                }
                            case 137: //SkinnedMeshRenderer
                                {
                                    m_GameObject.m_SkinnedMeshRenderer = m_Component;
                                    break;
                                }
                        }
                    }
                }
            }
        }
    }

    class TexEnv
    {
        public string name;
        public PPtr m_Texture;
        public float[] m_Scale;
        public float[] m_Offset;
    }

    class strFloatPair
    {
        public string first;
        public float second;
    }

    class strColorPair
    {
        public string first;
        public float[] second;
    }

    public static class StringExtensions
    {
        /// <summary>
        /// Compares the string against a given pattern.
        /// </summary>
        /// <param name="str">The string.</param>
        /// <param name="pattern">The pattern to match, where "*" means any sequence of characters, and "?" means any single character.</param>
        /// <returns><c>true</c> if the string matches the given pattern; otherwise <c>false</c>.</returns>
        public static bool Like(this string str, string pattern)
        {
            return new Regex(
                "^" + Regex.Escape(pattern).Replace(@"\*", ".*").Replace(@"\?", ".") + "$",
                RegexOptions.IgnoreCase | RegexOptions.Singleline
            ).IsMatch(str);
        }
    }
}