| | 1 | | using System; |
| | 2 | | using System.Collections; |
| | 3 | | using System.Linq; |
| | 4 | | using DCL.Helpers; |
| | 5 | | using UnityEngine; |
| | 6 | | using System.Collections.Generic; |
| | 7 | | using DCL.Models; |
| | 8 | | using MainScripts.DCL.Analytics.PerformanceAnalytics; |
| | 9 | |
|
| | 10 | | namespace DCL |
| | 11 | | { |
| | 12 | | public class AssetPromise_AB_GameObject : AssetPromise_WithUrl<Asset_AB_GameObject> |
| | 13 | | { |
| 44 | 14 | | public AssetPromiseSettings_Rendering settings = new AssetPromiseSettings_Rendering(); |
| | 15 | | AssetPromise_AB subPromise; |
| | 16 | | Coroutine loadingCoroutine; |
| | 17 | |
|
| 120 | 18 | | private BaseVariable<FeatureFlag> featureFlags => DataStore.i.featureFlags.flags; |
| | 19 | | private const string AB_LOAD_ANIMATION = "ab_load_animation"; |
| | 20 | | private bool doTransitionAnimation; |
| | 21 | |
|
| 44 | 22 | | public AssetPromise_AB_GameObject(string contentUrl, string hash) : base(contentUrl, hash) |
| | 23 | | { |
| 44 | 24 | | featureFlags.OnChange += OnFeatureFlagChange; |
| 44 | 25 | | OnFeatureFlagChange(featureFlags.Get(), null); |
| 44 | 26 | | } |
| | 27 | |
|
| 368 | 28 | | private void OnFeatureFlagChange(FeatureFlag current, FeatureFlag previous) { doTransitionAnimation = current.Is |
| | 29 | |
|
| 38 | 30 | | protected override void OnLoad(Action OnSuccess, Action<Exception> OnFail) { loadingCoroutine = CoroutineStarter |
| | 31 | |
|
| | 32 | | protected override bool AddToLibrary() |
| | 33 | | { |
| 12 | 34 | | if (!library.Add(asset)) |
| | 35 | | { |
| 0 | 36 | | return false; |
| | 37 | | } |
| | 38 | |
|
| 12 | 39 | | if (settings.forceNewInstance) |
| | 40 | | { |
| 2 | 41 | | asset = (library as AssetLibrary_AB_GameObject).GetCopyFromOriginal(asset.id); |
| | 42 | | } |
| | 43 | | else |
| | 44 | | { |
| 10 | 45 | | asset = library.Get(asset.id); |
| | 46 | | } |
| | 47 | |
|
| | 48 | | //NOTE(Brian): Call again this method because we are replacing the asset. |
| 12 | 49 | | settings.ApplyBeforeLoad(asset.container.transform); |
| | 50 | |
|
| 12 | 51 | | return true; |
| | 52 | | } |
| | 53 | |
|
| | 54 | | protected override void OnReuse(Action OnSuccess) |
| | 55 | | { |
| 25 | 56 | | asset.renderers = MeshesInfoUtils.ExtractUniqueRenderers(asset.container); |
| 25 | 57 | | asset.Show(OnSuccess); |
| 25 | 58 | | } |
| | 59 | |
|
| | 60 | | protected override void OnAfterLoadOrReuse() |
| | 61 | | { |
| 37 | 62 | | asset.renderers = MeshesInfoUtils.ExtractUniqueRenderers(asset.container); |
| 37 | 63 | | settings.ApplyAfterLoad(asset.container.transform); |
| 37 | 64 | | } |
| | 65 | |
|
| 88 | 66 | | protected override void OnBeforeLoadOrReuse() { settings.ApplyBeforeLoad(asset.container.transform); } |
| | 67 | |
|
| | 68 | | protected override void OnCancelLoading() |
| | 69 | | { |
| 7 | 70 | | if (loadingCoroutine != null) |
| | 71 | | { |
| 3 | 72 | | PerformanceAnalytics.ABTracker.TrackCancelled(); |
| 3 | 73 | | CoroutineStarter.Stop(loadingCoroutine); |
| 3 | 74 | | loadingCoroutine = null; |
| | 75 | | } |
| | 76 | |
|
| 7 | 77 | | if (asset != null) |
| 7 | 78 | | UnityEngine.Object.Destroy(asset.container); |
| | 79 | |
|
| 7 | 80 | | AssetPromiseKeeper_AB.i.Forget(subPromise); |
| 7 | 81 | | } |
| | 82 | |
|
| | 83 | | public override string ToString() |
| | 84 | | { |
| 0 | 85 | | if (subPromise != null) |
| 0 | 86 | | return $"{subPromise.ToString()} ... AB_GameObject state = {state}"; |
| | 87 | | else |
| 0 | 88 | | return $"subPromise == null? state = {state}"; |
| | 89 | | } |
| | 90 | |
|
| | 91 | | public IEnumerator LoadingCoroutine(Action OnSuccess, Action<Exception> OnFail) |
| | 92 | | { |
| 19 | 93 | | PerformanceAnalytics.ABTracker.TrackLoading(); |
| 19 | 94 | | subPromise = new AssetPromise_AB(contentUrl, hash, asset.container.transform); |
| 19 | 95 | | bool success = false; |
| 19 | 96 | | Exception loadingException = null; |
| 34 | 97 | | subPromise.OnSuccessEvent += (x) => success = true; |
| | 98 | |
|
| 19 | 99 | | subPromise.OnFailEvent += ( ab, exception) => |
| | 100 | | { |
| 1 | 101 | | loadingException = exception; |
| 1 | 102 | | success = false; |
| 1 | 103 | | }; |
| | 104 | |
|
| 19 | 105 | | asset.ownerPromise = subPromise; |
| 19 | 106 | | AssetPromiseKeeper_AB.i.Keep(subPromise); |
| | 107 | |
|
| 19 | 108 | | yield return subPromise; |
| | 109 | |
|
| 16 | 110 | | if (success) |
| | 111 | | { |
| 15 | 112 | | yield return InstantiateABGameObjects(); |
| | 113 | |
|
| 15 | 114 | | if (subPromise.asset == null || asset.container == null) |
| 3 | 115 | | success = false; |
| | 116 | | } |
| | 117 | |
|
| 16 | 118 | | loadingCoroutine = null; |
| | 119 | |
|
| 16 | 120 | | if (success) |
| | 121 | | { |
| 12 | 122 | | PerformanceAnalytics.ABTracker.TrackLoaded(); |
| 12 | 123 | | OnSuccess?.Invoke(); |
| | 124 | | } |
| | 125 | | else |
| | 126 | | { |
| 4 | 127 | | PerformanceAnalytics.ABTracker.TrackFailed(); |
| 4 | 128 | | loadingException ??= new Exception($"AB sub-promise asset or container is null. Asset: {subPromise.asset |
| 4 | 129 | | OnFail?.Invoke(loadingException); |
| | 130 | | } |
| 16 | 131 | | } |
| | 132 | |
|
| | 133 | | private IEnumerator InstantiateABGameObjects() |
| | 134 | | { |
| 15 | 135 | | var goList = subPromise.asset.GetAssetsByExtensions<GameObject>("glb", "ltf"); |
| | 136 | |
|
| 15 | 137 | | if (goList.Count == 0) |
| | 138 | | { |
| 2 | 139 | | if (asset.container != null) |
| 2 | 140 | | UnityEngine.Object.Destroy(asset.container); |
| | 141 | |
|
| 2 | 142 | | asset.container = null; |
| | 143 | |
|
| 2 | 144 | | AssetPromiseKeeper_AB.i.Forget(subPromise); |
| | 145 | |
|
| 2 | 146 | | yield break; |
| | 147 | | } |
| | 148 | |
|
| 50 | 149 | | for (int i = 0; i < goList.Count; i++) |
| | 150 | | { |
| 13 | 151 | | if (loadingCoroutine == null) |
| | 152 | | break; |
| | 153 | |
|
| 13 | 154 | | if (asset.container == null) |
| | 155 | | break; |
| | 156 | |
|
| 12 | 157 | | GameObject assetBundleModelGO = UnityEngine.Object.Instantiate(goList[i], asset.container.transform); |
| | 158 | |
|
| 12 | 159 | | asset.renderers = MeshesInfoUtils.ExtractUniqueRenderers(assetBundleModelGO); |
| 12 | 160 | | asset.materials = MeshesInfoUtils.ExtractUniqueMaterials(asset.renderers); |
| 12 | 161 | | asset.SetTextures(MeshesInfoUtils.ExtractUniqueTextures(asset.materials)); |
| | 162 | |
|
| 12 | 163 | | UploadMeshesToGPU(MeshesInfoUtils.ExtractUniqueMeshes(asset.renderers)); |
| 12 | 164 | | asset.totalTriangleCount = MeshesInfoUtils.ComputeTotalTriangles(asset.renderers, asset.meshToTriangleCo |
| | 165 | |
|
| | 166 | | //NOTE(Brian): Renderers are enabled in settings.ApplyAfterLoad |
| 12 | 167 | | yield return MaterialCachingHelper.Process(asset.renderers.ToList(), enableRenderers: false, settings.ca |
| | 168 | |
|
| 12 | 169 | | var animators = MeshesInfoUtils.ExtractUniqueAnimations(assetBundleModelGO); |
| 12 | 170 | | asset.animationClipSize = subPromise.asset.metrics.animationsEstimatedSize; |
| 12 | 171 | | asset.meshDataSize = subPromise.asset.metrics.meshesEstimatedSize; |
| | 172 | |
|
| 24 | 173 | | foreach (var animator in animators) |
| | 174 | | { |
| 0 | 175 | | animator.cullingType = AnimationCullingType.AlwaysAnimate; |
| | 176 | | } |
| | 177 | |
|
| | 178 | | #if UNITY_EDITOR |
| 12 | 179 | | assetBundleModelGO.name = subPromise.asset.GetName(); |
| | 180 | | #endif |
| | 181 | |
|
| 12 | 182 | | yield return null; |
| | 183 | |
|
| 12 | 184 | | yield return SetMaterialTransition(); |
| 12 | 185 | | } |
| 13 | 186 | | } |
| | 187 | |
|
| | 188 | | private void UploadMeshesToGPU(HashSet<Mesh> meshesList) |
| | 189 | | { |
| 48 | 190 | | foreach ( Mesh mesh in meshesList ) |
| | 191 | | { |
| 12 | 192 | | if ( !mesh.isReadable ) |
| | 193 | | continue; |
| | 194 | |
|
| 12 | 195 | | asset.meshToTriangleCount[mesh] = mesh.triangles.Length; |
| 12 | 196 | | asset.meshes.Add(mesh); |
| | 197 | |
|
| 12 | 198 | | Physics.BakeMesh(mesh.GetInstanceID(), false); |
| 12 | 199 | | mesh.UploadMeshData(true); |
| | 200 | | } |
| 12 | 201 | | } |
| | 202 | |
|
| | 203 | | protected override Asset_AB_GameObject GetAsset(object id) |
| | 204 | | { |
| 25 | 205 | | if (settings.forceNewInstance) |
| | 206 | | { |
| 9 | 207 | | return ((AssetLibrary_AB_GameObject) library).GetCopyFromOriginal(id); |
| | 208 | | } |
| | 209 | | else |
| | 210 | | { |
| 16 | 211 | | return base.GetAsset(id); |
| | 212 | | } |
| | 213 | | } |
| | 214 | |
|
| | 215 | | internal override void OnForget() |
| | 216 | | { |
| 32 | 217 | | base.OnForget(); |
| 32 | 218 | | featureFlags.OnChange -= OnFeatureFlagChange; |
| 32 | 219 | | } |
| | 220 | |
|
| | 221 | | IEnumerator SetMaterialTransition(Action OnSuccess = null) |
| | 222 | | { |
| 12 | 223 | | if (settings.visibleFlags != AssetPromiseSettings_Rendering.VisibleFlags.INVISIBLE && doTransitionAnimation) |
| | 224 | | { |
| 0 | 225 | | MaterialTransitionController[] materialTransitionControllers = new MaterialTransitionController[asset.re |
| 0 | 226 | | int index = 0; |
| 0 | 227 | | foreach (Renderer assetRenderer in asset.renderers) |
| | 228 | | { |
| 0 | 229 | | MaterialTransitionController transition = assetRenderer.gameObject.AddComponent<MaterialTransitionCo |
| 0 | 230 | | materialTransitionControllers[index] = transition; |
| 0 | 231 | | transition.delay = 0; |
| 0 | 232 | | transition.OnDidFinishLoading(assetRenderer.sharedMaterial); |
| | 233 | |
|
| 0 | 234 | | index++; |
| | 235 | | } |
| | 236 | | // Wait until MaterialTransitionController finishes its effect |
| 0 | 237 | | yield return new WaitUntil(() => IsTransitionFinished(materialTransitionControllers)); |
| | 238 | | } |
| 12 | 239 | | OnSuccess?.Invoke(); |
| 12 | 240 | | } |
| | 241 | |
|
| | 242 | | private bool IsTransitionFinished(MaterialTransitionController[] matTransitions) |
| | 243 | | { |
| 0 | 244 | | bool finishedTransition = true; |
| | 245 | |
|
| 0 | 246 | | for (int i = 0; i < matTransitions.Length; i++) |
| | 247 | | { |
| 0 | 248 | | if (matTransitions[i] != null) |
| | 249 | | { |
| 0 | 250 | | finishedTransition = false; |
| | 251 | |
|
| 0 | 252 | | break; |
| | 253 | | } |
| | 254 | | } |
| | 255 | |
|
| 0 | 256 | | return finishedTransition; |
| | 257 | | } |
| | 258 | |
|
| | 259 | | } |
| | 260 | |
|
| | 261 | | } |