max/ScenePartition
1
0
Fork 0
generated from max/template-unity-project
Code Issues 13 Pull Requests Packages Projects 1 Releases Wiki Activity

Compare commits

base: max:ReadableFileNames
Branches Tags
max:main
max:RootOrder
max:ReadableFileNames
..
compare: max:main
Branches Tags
max:RootOrder
max:main
max:ReadableFileNames

3 Commits
ReadableFi ... main

Author SHA1 Message Date
max
8453143918 Comment 2023-08-10 22:27:10 +02:00
max
1b9c367221 Prevent 'scene changed on disk' popup 
Prevent 'scene changed on disk' popup when loading/additive loading.
 2023-08-10 22:24:10 +02:00
max
4fe602e932 Save scene on save 
- Check if the user wants to save the dirty scene on save
 2023-08-10 21:41:57 +02:00
1 changed files with 9 additions and 189 deletions
Show Stats Expand all files Collapse all files

198
Editor/ScenePartitionSO.cs
View File

@ -92,7 +92,7 @@ private void LoadScenePartitions(SortedSet<ulong> partitionIds)
// Add always load ids. // Add always load ids.
SortedSet<ulong> baseIds = GetAlwaysLoadIds(); SortedSet<ulong> baseIds = GetAlwaysLoadIds();
foreach (var id in baseIds) foreach (ulong id in baseIds)
{ {
partitionIds.Add(id); partitionIds.Add(id);
} }
@ -129,6 +129,8 @@ private void LoadScenePartitions(SortedSet<ulong> partitionIds)
} }
AssetDatabase.Refresh(); AssetDatabase.Refresh();
// 'Reload' the scene to prevent the user getting the popup 'Scene has been changed on disk'.
EditorSceneManager.OpenScene(scenePath, OpenSceneMode.Single);
} }
} }
@ -139,35 +141,33 @@ public void Save()
{ {
using (new ProfilerUtility.ProfilerScope($"{nameof(Save)}")) using (new ProfilerUtility.ProfilerScope($"{nameof(Save)}"))
{ {
// Check if the user wants to save the scene if dirty.
if (!EditorSceneManager.SaveCurrentModifiedScenesIfUserWantsTo()) return;
DeleteLoadedPartitions(); // Delete the loaded partitions from disk so we can write the new ones. DeleteLoadedPartitions(); // Delete the loaded partitions from disk so we can write the new ones.
const string objectIdPattern = @"&(\d+)"; string pattern = @"&(\d+)";
string dataPath = ScenePartitionUtils.GetDataPath(this); string dataPath = ScenePartitionUtils.GetDataPath(this);
string scenePath = ScenePartitionUtils.GetScenePath(this); string scenePath = ScenePartitionUtils.GetScenePath(this);
// Read the data from the scene file. // Read the data from the scene file.
string[] sceneData = File.ReadAllLines(scenePath); string[] sceneData = File.ReadAllLines(scenePath);
Dictionary<ulong, string> sceneObjectNameById = GetSceneObjectNames(in sceneData);
// Split it into blocks. // Split it into blocks.
int lastIndex = sceneData.Length; int lastIndex = sceneData.Length;
for (int i = sceneData.Length - 1; i >= 0; i--) for (int i = sceneData.Length - 1; i >= 0; i--)
{ {
if (sceneData[i].StartsWith("---")) // --- is the start of a new yaml document. if (sceneData[i].StartsWith("---")) // --- is the start of a new yaml document.
{ {
Match match = Regex.Match(sceneData[i], objectIdPattern); Match match = Regex.Match(sceneData[i], pattern);
if (match.Success) if (match.Success)
{ {
// Extract the file number // Extract the file number
string id = match.Groups[1].Value; string id = match.Groups[1].Value;
ulong objectId = ulong.Parse(id);
GetObjectInfo(in sceneData, in sceneObjectNameById, i, lastIndex, objectId, out string objectInfo);
// Write data to disk. // Write data to disk.
File.WriteAllLines($"{dataPath}/{SceneName}-{id}{objectInfo}.yaml", sceneData[i..lastIndex]); File.WriteAllLines($"{dataPath}/{SceneName}-{id}.yaml", sceneData[i..lastIndex]);
} }
lastIndex = i; lastIndex = i;
@ -382,185 +382,5 @@ public void ClearCache()
{ {
data = null; data = null;
} }
private Dictionary<ulong, string> GetSceneObjectNames(in string[] sceneData)
{
using ProfilerUtility.ProfilerScope profilerScope = new(nameof(GetSceneObjectNames));
Dictionary<ulong, string> sceneObjectNameById = new Dictionary<ulong, string>();
const string objectIdPattern = @"&(\d+)";
int lastIndex = sceneData.Length;
for (int i = sceneData.Length - 1; i >= 0; i--)
{
bool foundName = false;
{ // GameObjects.
const string gameObjectHeaderPattern = "--- !u!1 &";
if (!foundName && sceneData[i].StartsWith(gameObjectHeaderPattern))
{
Match match = Regex.Match(sceneData[i], objectIdPattern);
if (!match.Success) continue;
if (!ulong.TryParse(match.Groups[1].Value, out ulong id)) continue;
for (int j = i; j < lastIndex; j++)
{
const string namePattern = "m_Name: ";
int nameStartIndex = sceneData[j].LastIndexOf(namePattern);
if (nameStartIndex < 0) continue;
nameStartIndex += namePattern.Length;
string name = sceneData[j][nameStartIndex..];
sceneObjectNameById.Add(id, name);
foundName = true;
break;
}
}
}
{ // Prefabs.
const string prefabHeaderPattern = "--- !u!1001 &";
if (!foundName && sceneData[i].StartsWith(prefabHeaderPattern))
{
Match match = Regex.Match(sceneData[i], objectIdPattern);
if (!match.Success) continue;
if (!ulong.TryParse(match.Groups[1].Value, out ulong id)) continue;
// Get name form property override in Scene.
for (int j = i; j < lastIndex; j++)
{
const string propertyPattern = "propertyPath: m_Name";
int propertyStartIndex = sceneData[j].LastIndexOf(propertyPattern);
if (propertyStartIndex < 0) continue;
const string valuePattern = "value: ";
int valueStartIndex = sceneData[j + 1].LastIndexOf(valuePattern);
if (valueStartIndex < 0) continue;
valueStartIndex += valuePattern.Length;
string name = sceneData[j + 1][valueStartIndex..];
sceneObjectNameById.Add(id, name);
foundName = true;
break;
}
// Get name from prefab in AssetDatabase.
if (!foundName)
{
// Find the match using regex
const string guidPattern = @"guid:\s(\w+)";
Match guidMatch = Regex.Match(sceneData[lastIndex - 1], guidPattern);
if (guidMatch.Success)
{
// Extract the GUID
string guid = guidMatch.Groups[1].Value;
string path = AssetDatabase.GUIDToAssetPath(guid);
string name = Path.GetFileNameWithoutExtension(path);
sceneObjectNameById.Add(id, name);
foundName = true;
}
}
}
}
if (sceneData[i].StartsWith("---"))
{
lastIndex = i;
continue;
}
}
return sceneObjectNameById;
}
private bool TryGetObjectTypeName(string data, out string objectTypeName)
{
objectTypeName = data.Remove(data.Length - 1);
return true;
}
private void GetObjectInfo(in string[] sceneData, in Dictionary<ulong, string> sceneObjectNameById, int index, int lastIndex, ulong objectId, out string objectInfo)
{
using ProfilerUtility.ProfilerScope profilerScope = new(nameof(GetObjectInfo));
objectInfo = "";
// Try get scene object name.
bool foundSceneObjectName = false;
// If it is a gameObject or prefab try to get the name directly.
const string gameObjectHeaderPattern = "--- !u!1 &";
const string prefabHeaderPattern = "--- !u!1001 &";
if (!foundSceneObjectName && sceneData[index].StartsWith(gameObjectHeaderPattern) || sceneData[index].StartsWith(prefabHeaderPattern))
{
if (sceneObjectNameById.TryGetValue(objectId, out string sceneObjectName))
{
objectInfo += $"-{sceneObjectName}";
foundSceneObjectName = true;
}
}
if (!foundSceneObjectName)
{
// Most components have a property that links to the GameObject.
for (int j = index; j < lastIndex; j++)
{
const string gameObjectPattern = @"m_GameObject:\s{fileID:\s(\d+)}";
// Find the match using regex
Match gameObjectMatch = Regex.Match(sceneData[j], gameObjectPattern);
if (!gameObjectMatch.Success) continue;
if (ulong.TryParse(gameObjectMatch.Groups[1].Value, out ulong fileNumber))
{
if (fileNumber == 0) break; // 0 == nothing.
if (!sceneObjectNameById.TryGetValue(fileNumber, out string sceneObjectName)) break;
objectInfo += $"-{sceneObjectName}";
foundSceneObjectName = true;
}
break;
}
}
if (!foundSceneObjectName)
{
// Some have a property that links to the PrefabInstance.
for (int j = index; j < lastIndex; j++)
{
const string prefabInstancePattern = @"m_PrefabInstance:\s{fileID:\s(\d+)}";
// Find the match using regex
Match prefabInstanceMatch = Regex.Match(sceneData[j], prefabInstancePattern);
if (!prefabInstanceMatch.Success) continue;
if (ulong.TryParse(prefabInstanceMatch.Groups[1].Value, out ulong fileNumber))
{
if (fileNumber == 0) break; // 0 == nothing.
if (!sceneObjectNameById.TryGetValue(fileNumber, out string sceneObjectName)) break;
objectInfo += $"-{sceneObjectName}";
foundSceneObjectName = true;
}
break;
}
}
// Try get object type name.
if (TryGetObjectTypeName(sceneData[index + 1], out string objectTypeName))
{
objectInfo += $"-{objectTypeName}";
}
}
} }
} }