perf: additional performance improvements

bdunderscore · Nov 29, 2024 · d765815 · d765815
1 parent 3918c20
commit d765815
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 14 deletions.
diff --git a/Editor/API/BuildContext.cs b/Editor/API/BuildContext.cs
@@ -129,9 +129,6 @@ public BuildContext(GameObject obj, string assetRootPath, bool isClone = true)
 
             if (assetRootPath != null)
             {
-                // Ensure the target directory exists
-                Directory.CreateDirectory(assetRootPath);
-
                 AssetSaver = new AssetSaver(assetRootPath, avatarName);
             }
             else

diff --git a/Editor/API/Serialization/AssetSaver.cs b/Editor/API/Serialization/AssetSaver.cs
@@ -38,22 +38,55 @@ internal AssetSaver(string generatedAssetsRoot, string avatarName, int assetsPer
 
             if (AssetDatabase.IsValidFolder(subAssetPath))
             {
+                using var _2 = new ProfilerScope("DeleteFolder");
                 AssetDatabase.DeleteAsset(rootPath);
             }
+
+            using var _ = new ProfilerScope("Asset editing");
+            try
+            {
+                AssetDatabase.StartAssetEditing();
+
+                // Ensure directory exists recursively
+                var pathParts = subAssetPath.Split('/');
+                var currentDir = pathParts[0];
+
+                for (int i = 1; i < pathParts.Length; i++)
+                {
+                    var nextDir = currentDir + "/" + pathParts[i];
+                    if (!AssetDatabase.IsValidFolder(nextDir))
+                    {
+                        if (Directory.Exists(nextDir))
+                        {
+                            // Sometimes the asset database can be unaware of a folder on disk, so refresh it.
+                            // This is quite expensive, so only do it if necessary.
+                            Debug.Log("Force refresh due to " + nextDir);
+                            using var _2 = new ProfilerScope("AssetDatabase.Refresh");
+                            AssetDatabase.Refresh();
+                        }
+                        else
+                        {
+                            using var _2 = new ProfilerScope("CreateFolder");
+                            AssetDatabase.CreateFolder(currentDir, pathParts[i]);
+                        }
+                    }
+
+                    currentDir = nextDir;
+                }
+
+                var rootAssetPath = AssetDatabase.GenerateUniqueAssetPath(rootPath + "/" + avatarName + ".asset");
+                _rootAsset = ScriptableObject.CreateInstance<GeneratedAssets>();
+                AssetDatabase.CreateAsset(_rootAsset, rootAssetPath);
+                _currentSubContainer = CreateAssetContainer();
 
-            // Ensure directory exists recursively
-            if (!AssetDatabase.IsValidFolder(subAssetPath))
+                _assetCount = 0;
+            }
+            finally
             {
-                Directory.CreateDirectory(subAssetPath);
-                AssetDatabase.Refresh();
+                using var _2 = new ProfilerScope("StopAssetEditing");
+
+                AssetDatabase.StopAssetEditing();
             }
-
-            var rootAssetPath = AssetDatabase.GenerateUniqueAssetPath(rootPath + "/" + avatarName + ".asset");
-            _rootAsset = ScriptableObject.CreateInstance<GeneratedAssets>();
-            AssetDatabase.CreateAsset(_rootAsset, rootAssetPath);
-            _currentSubContainer = CreateAssetContainer();
-
-            _assetCount = 0;
         }
 
         public void SaveAsset(UnityEngine.Object? obj)