[AnimationEditor] Restructured sdk/type initialization

CadeEvs · Sep 22, 2022 · 8582620 · 8582620
1 parent b2d0792
commit 8582620
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 25 deletions.
diff --git a/Plugins/AnimationEditorPlugin/Editors/ViewModels/AssetBankViewModel.cs b/Plugins/AnimationEditorPlugin/Editors/ViewModels/AssetBankViewModel.cs
@@ -37,9 +37,7 @@ public object Data
         public AssetBankViewModel(FrostyBaseEditor inOwner)
         {
             m_owner = inOwner;
-
-            AssetBankTypeLibrary.Initialize();
-
+
             m_entries = new List<AssetBankFileEntry>();
             foreach (AssetBankFileEntry entry in App.AssetManager.EnumerateCustomAssets("assetbank"))
             {

diff --git a/Plugins/AnimationEditorPlugin/Managers/AssetBankFileManager.cs b/Plugins/AnimationEditorPlugin/Managers/AssetBankFileManager.cs
@@ -19,12 +19,15 @@ public class AssetBankFileManager : ICustomAssetManager
 
         #region -- ICustomAssetManager --
 
-        public bool ShouldInitializeOnStartup => false;
+        public bool ShouldInitializeOnStartup => true;
 
         public void Initialize(ILogger logger)
         {
             logger.Log("Loading asset banks");
 
+            // initial type library
+            AssetBankTypeLibrary.Initialize();
+
             uint totalCount = App.AssetManager.GetResCount((uint)ResourceType.AssetBank);
             uint index = 0;
             foreach (ResAssetEntry resEntry in App.AssetManager.EnumerateRes((uint)ResourceType.AssetBank))
@@ -111,28 +114,10 @@ public void Initialize(ILogger logger)
                                 banks = new List<Bank>();
                             }
 
-                            using (AssetBankModuleWriter writer = new AssetBankModuleWriter("AssetBankClasses.dll", banks))
-                            {
-                                writer.Write(App.FileSystemManager.Head);
-                            }
-
-                            if (File.Exists("AssetBankClasses.dll"))
-                            {
-                                FileInfo fi = new FileInfo(".\\AssetBankTmpProfiles\\" + ProfilesLibrary.SDKFilename + ".dll");
-                                if (!fi.Directory.Exists)
-                                {
-                                    Directory.CreateDirectory(fi.Directory.FullName);
-                                }
-                                if (fi.Exists)
-                                {
-                                    File.Delete(fi.FullName);
-                                }
-
-                                File.Move("AssetBankClasses.dll", fi.FullName);
-                            }
-                            else
+                            // write sdk if one doesn't exist
+                            if (!File.Exists("AssetBankProfiles/" + ProfilesLibrary.SDKFilename + ".dll"))
                             {
-                                Console.WriteLine("Failed to produce SDK");
+                                WriteSdk(banks);
                             }
 
                             /*
@@ -213,5 +198,32 @@ public void OnCommand(string command, params object[] value)
         }
 
         #endregion
+
+        private void WriteSdk(List<Bank> banks)
+        {
+            using (AssetBankModuleWriter writer = new AssetBankModuleWriter("AssetBankClasses.dll", banks))
+            {
+                writer.Write(App.FileSystemManager.Head);
+            }
+
+            if (File.Exists("AssetBankClasses.dll"))
+            {
+                FileInfo fi = new FileInfo(".\\AssetBankTmpProfiles\\" + ProfilesLibrary.SDKFilename + ".dll");
+                if (!fi.Directory.Exists)
+                {
+                    Directory.CreateDirectory(fi.Directory.FullName);
+                }
+                if (fi.Exists)
+                {
+                    File.Delete(fi.FullName);
+                }
+
+                File.Move("AssetBankClasses.dll", fi.FullName);
+            }
+            else
+            {
+                Console.WriteLine("Failed to produce SDK");
+            }
+        }
     }
 }