Basic split saving and loading of scene files

2023-06-24 14:27:50 +02:00
parent a335984b1c
commit bdbfaf34c0
16 changed files with 300 additions and 0 deletions
--- a/Editor/ScenePartitionUtils.cs
+++ b/Editor/ScenePartitionUtils.cs
@ -0,0 +1,94 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using UnityEditor;
+using UnityEditor.SceneManagement;
+using UnityEngine;
+using UnityEngine.SceneManagement;
+
+namespace VertexColor.ScenePartition.Editor
+{
+    public static class ScenePartitionUtils
+    {
+        public static string GetDataPath(ScenePartitionSO scenePartitionSO)
+        {
+            string dataPath = Path.Combine(Application.dataPath, $"../Data/Scenes/{scenePartitionSO.name}");
+
+            if (!Directory.Exists(dataPath))
+            {
+                Directory.CreateDirectory(dataPath);
+            }
+
+            return dataPath;
+        }
+
+        public static string GetScenePath(ScenePartitionSO scenePartitionSO)
+        {
+            string scenePath = AssetDatabase.GetAssetOrScenePath(scenePartitionSO.sceneAsset);
+
+            return scenePath;
+        }
+
+        public static void Load(ScenePartitionSO scenePartitionSO)
+        {
+            string dataPath = GetDataPath(scenePartitionSO);
+            string scenePath = GetScenePath(scenePartitionSO);
+
+            List<string> data = new List<string>();
+            List<string> files = Directory.GetFiles(dataPath).ToList();
+            files.Sort();
+
+            for (int i = 0; i < files.Count; i++)
+            {
+                data.AddRange(File.ReadAllLines(files[i]));
+            }
+
+            File.WriteAllLines(scenePath, data);
+
+            AssetDatabase.Refresh();
+        }
+
+        public static void Save(ScenePartitionSO scenePartitionSO)
+        {
+            string dataPath = GetDataPath(scenePartitionSO);
+            string scenePath = GetScenePath(scenePartitionSO);
+
+            string[] data = File.ReadAllLines(scenePath);
+
+            int lastIndex = data.Length;
+            for (int i = data.Length - 1; i >= 0; i--)
+            {
+                if (data[i].StartsWith("---")) // --- is the start of a new yaml document.
+                {
+                    int idStartIndex = data[i].IndexOf(" &") + 2; // & is the start of the object id.
+                    int idLength = data[i].Length;
+
+                    File.WriteAllLines($"{dataPath}/{scenePartitionSO.sceneName}-{data[i][idStartIndex..idLength]}.yaml", data[i..lastIndex]);
+
+                    lastIndex = i;
+                }
+            }
+
+            File.WriteAllLines($"{dataPath}/{scenePartitionSO.sceneName}.yaml", data[0..lastIndex]);
+        }
+
+        public static void Unload(ScenePartitionSO scenePartitionSO)
+        {
+            string dataPath = GetDataPath(scenePartitionSO);
+            string scenePath = GetScenePath(scenePartitionSO);
+
+            Scene scene = EditorSceneManager.OpenScene(scenePath, OpenSceneMode.Single);
+
+            GameObject[] gameObjects = scene.GetRootGameObjects();
+
+            for (int i = gameObjects.Length - 1; i >= 0; i--)
+            {
+                Object.DestroyImmediate(gameObjects[i]);
+            }
+
+            EditorSceneManager.SaveScene(scene);
+
+            AssetDatabase.Refresh();
+        }
+    }
+}