57 lines
1.6 KiB
C#
57 lines
1.6 KiB
C#
|
using System;
|
||
|
|
using UnityEditor.Build.Pipeline.Utilities;
|
||
|
|
using UnityEditor.SceneManagement;
|
||
|
|
using System.Linq;
|
||
|
|
|
||
|
|
namespace UnityEditor.Build.Utilities
|
||
|
|
{
|
||
|
|
/// <summary>
|
||
|
|
/// Cleans up scenes in the scene manager.
|
||
|
|
/// </summary>
|
||
|
|
public class SceneStateCleanup : IDisposable
|
||
|
|
{
|
||
|
|
SceneSetup[] m_Scenes;
|
||
|
|
|
||
|
|
bool m_Disposed;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Creates a new scene state cleanup object.
|
||
|
|
/// </summary>
|
||
|
|
public SceneStateCleanup()
|
||
|
|
{
|
||
|
|
m_Scenes = EditorSceneManager.GetSceneManagerSetup();
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Disposes of the scene state cleanup instance.
|
||
|
|
/// </summary>
|
||
|
|
public void Dispose()
|
||
|
|
{
|
||
|
|
Dispose(true);
|
||
|
|
GC.SuppressFinalize(this);
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Disposes of the scene state cleanup instance.
|
||
|
|
/// </summary>
|
||
|
|
/// <param name="disposing">Set to true to reset the scenes list. Set to false to leave the scenes list as is.</param>
|
||
|
|
protected virtual void Dispose(bool disposing)
|
||
|
|
{
|
||
|
|
if (m_Disposed)
|
||
|
|
return;
|
||
|
|
|
||
|
|
m_Disposed = true;
|
||
|
|
|
||
|
|
if (disposing)
|
||
|
|
{
|
||
|
|
// Test runner injects scenes, so we strip those here
|
||
|
|
var scenes = m_Scenes.Where(s => !string.IsNullOrEmpty(s.path)).ToArray();
|
||
|
|
if (!scenes.IsNullOrEmpty())
|
||
|
|
EditorSceneManager.RestoreSceneManagerSetup(scenes);
|
||
|
|
else
|
||
|
|
EditorSceneManager.NewScene(NewSceneSetup.DefaultGameObjects);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|