WXGame/Blacksmith/Packages/com.tuyoogame.yooasset@2.2.4-preview/Runtime/ResourceManager/Operation/UnloadSceneOperation.cs

using UnityEngine;
using UnityEngine.SceneManagement;

namespace YooAsset
{
    /// <summary>
    /// 场景卸载异步操作类
    /// </summary>
    public sealed class UnloadSceneOperation : AsyncOperationBase
    {
        private enum ESteps
        {
            None,
            CheckError,
            PrepareDone,
            UnLoadScene,
            Done,
        }

        private ESteps _steps = ESteps.None;
        private readonly string _error;
        private readonly ProviderOperation _provider;
        private AsyncOperation _asyncOp = null;

        internal UnloadSceneOperation(string error)
        {
            _error = error;
        }
        internal UnloadSceneOperation(ProviderOperation provider)
        {
            _error = null;
            _provider = provider;

            // 注意：卸载场景前必须先解除挂起操作
            if (provider is DatabaseSceneProvider)
            {
                var temp = provider as DatabaseSceneProvider;
                temp.UnSuspendLoad();
            }
            else if (provider is BundledSceneProvider)
            {
                var temp = provider as BundledSceneProvider;
                temp.UnSuspendLoad();
            }
            else
            {
                throw new System.NotImplementedException();
            }
        }
        internal override void InternalOnStart()
        {
            _steps = ESteps.CheckError;
        }
        internal override void InternalOnUpdate()
        {
            if (_steps == ESteps.None || _steps == ESteps.Done)
                return;

            if (_steps == ESteps.CheckError)
            {
                if (string.IsNullOrEmpty(_error) == false)
                {
                    _steps = ESteps.Done;
                    Status = EOperationStatus.Failed;
                    Error = _error;
                    return;
                }

                _steps = ESteps.PrepareDone;
            }

            if (_steps == ESteps.PrepareDone)
            {
                if (_provider.IsDone == false)
                    return;

                if (_provider.SceneObject.IsValid() == false)
                {
                    _steps = ESteps.Done;
                    Status = EOperationStatus.Failed;
                    Error = "Scene is invalid !";
                    return;
                }

                if (_provider.SceneObject.isLoaded == false)
                {
                    _steps = ESteps.Done;
                    Status = EOperationStatus.Failed;
                    Error = "Scene is not loaded !";
                    return;
                }

                _steps = ESteps.UnLoadScene;
            }

            if (_steps == ESteps.UnLoadScene)
            {
                if (_asyncOp == null)
                {
                    _asyncOp = SceneManager.UnloadSceneAsync(_provider.SceneObject);
                    _provider.ResourceMgr.UnloadSubScene(_provider.SceneName);
                }

                Progress = _asyncOp.progress;
                if (_asyncOp.isDone == false)
                    return;

                _provider.ResourceMgr.TryUnloadUnusedAsset(_provider.MainAssetInfo);
                _steps = ESteps.Done;
                Status = EOperationStatus.Succeed;
            }
        }
    }
}