| | 1 | | using System.Collections.Generic; |
| | 2 | | using Cysharp.Threading.Tasks; |
| | 3 | | using DCL.Controllers; |
| | 4 | | using DCL.ECS7.InternalComponents; |
| | 5 | | using DCL.ECSRuntime; |
| | 6 | | using DCL.Models; |
| | 7 | | using UnityEngine; |
| | 8 | |
|
| | 9 | | namespace DCL.ECSComponents |
| | 10 | | { |
| | 11 | | public class MaterialHandler : IECSComponentHandler<PBMaterial> |
| | 12 | | { |
| | 13 | | private PBMaterial lastModel = null; |
| | 14 | | internal AssetPromise_Material promiseMaterial; |
| | 15 | |
|
| 12 | 16 | | private readonly Queue<AssetPromise_Material> activePromises = new Queue<AssetPromise_Material>(); |
| | 17 | | private readonly IInternalECSComponent<InternalMaterial> materialInternalComponent; |
| | 18 | |
|
| 12 | 19 | | public MaterialHandler(IInternalECSComponent<InternalMaterial> materialInternalComponent) |
| | 20 | | { |
| 12 | 21 | | this.materialInternalComponent = materialInternalComponent; |
| 12 | 22 | | } |
| | 23 | |
|
| 1 | 24 | | public void OnComponentCreated(IParcelScene scene, IDCLEntity entity) { } |
| | 25 | |
|
| | 26 | | public void OnComponentRemoved(IParcelScene scene, IDCLEntity entity) |
| | 27 | | { |
| 3 | 28 | | materialInternalComponent.RemoveFor(scene, entity, new InternalMaterial() { material = null }); |
| | 29 | |
|
| 6 | 30 | | while (activePromises.Count > 0) |
| | 31 | | { |
| 3 | 32 | | AssetPromiseKeeper_Material.i.Forget(activePromises.Dequeue()); |
| | 33 | | } |
| 3 | 34 | | } |
| | 35 | |
|
| | 36 | | public void OnComponentModelUpdated(IParcelScene scene, IDCLEntity entity, PBMaterial model) |
| | 37 | | { |
| 22 | 38 | | if (lastModel != null && lastModel.Equals(model)) |
| 2 | 39 | | return; |
| | 40 | |
|
| 20 | 41 | | lastModel = model; |
| | 42 | |
|
| 20 | 43 | | TextureUnion texture = model.Pbr != null ? model.Pbr.Texture : model.Unlit != null ? model.Unlit.Texture : n |
| | 44 | |
|
| | 45 | | AssetPromise_Material_Model promiseModel; |
| 20 | 46 | | AssetPromise_Material_Model.Texture? albedoTexture = texture != null ? CreateMaterialPromiseTextureModel ( |
| | 47 | | texture.GetTextureUrl(scene), |
| | 48 | | texture.GetWrapMode(), |
| | 49 | | texture.GetFilterMode() |
| | 50 | | ) : null; |
| | 51 | |
|
| 20 | 52 | | if (model.Pbr != null) |
| | 53 | | { |
| 19 | 54 | | AssetPromise_Material_Model.Texture? alphaTexture = model.Pbr.AlphaTexture != null ? CreateMaterialPromi |
| | 55 | | model.Pbr.AlphaTexture.GetTextureUrl(scene), |
| | 56 | | model.Pbr.AlphaTexture.GetWrapMode(), |
| | 57 | | model.Pbr.AlphaTexture.GetFilterMode() |
| | 58 | | ) : null; |
| | 59 | |
|
| 19 | 60 | | AssetPromise_Material_Model.Texture? emissiveTexture = model.Pbr.EmissiveTexture != null ? CreateMateria |
| | 61 | | model.Pbr.EmissiveTexture.GetTextureUrl(scene), |
| | 62 | | model.Pbr.EmissiveTexture.GetWrapMode(), |
| | 63 | | model.Pbr.EmissiveTexture.GetFilterMode() |
| | 64 | | ) : null; |
| | 65 | |
|
| 19 | 66 | | AssetPromise_Material_Model.Texture? bumpTexture = model.Pbr.BumpTexture != null ? CreateMaterialPromise |
| | 67 | | model.Pbr.BumpTexture.GetTextureUrl(scene), |
| | 68 | | model.Pbr.BumpTexture.GetWrapMode(), |
| | 69 | | model.Pbr.BumpTexture.GetFilterMode() |
| | 70 | | ) : null; |
| | 71 | |
|
| 19 | 72 | | promiseModel = CreatePBRMaterialPromiseModel(model, albedoTexture, alphaTexture, emissiveTexture, bumpTe |
| | 73 | | } |
| | 74 | | else |
| | 75 | | { |
| 1 | 76 | | promiseModel = CreateBasicMaterialPromiseModel(model, albedoTexture); |
| | 77 | | } |
| | 78 | |
|
| 20 | 79 | | promiseMaterial = new AssetPromise_Material(promiseModel); |
| 20 | 80 | | promiseMaterial.OnSuccessEvent += materialAsset => |
| | 81 | | { |
| 19 | 82 | | materialInternalComponent.PutFor(scene, entity, new InternalMaterial() |
| | 83 | | { |
| | 84 | | material = materialAsset.material |
| | 85 | | }); |
| | 86 | |
|
| | 87 | | // Run task to forget previous material after update to avoid forgetting a |
| | 88 | | // material that has not be changed from the renderers yet, since material change |
| | 89 | | // is done by a system during update |
| 19 | 90 | | UniTask.RunOnThreadPool(async () => |
| | 91 | | { |
| 57 | 92 | | await UniTask.Yield(PlayerLoopTiming.LastPostLateUpdate); |
| 19 | 93 | | ForgetPreviousPromises(activePromises, materialAsset); |
| 19 | 94 | | }); |
| 19 | 95 | | }; |
| 20 | 96 | | promiseMaterial.OnFailEvent += (material, exception) => |
| | 97 | | { |
| 0 | 98 | | ForgetPreviousPromises(activePromises, material); |
| 0 | 99 | | }; |
| 20 | 100 | | activePromises.Enqueue(promiseMaterial); |
| 20 | 101 | | AssetPromiseKeeper_Material.i.Keep(promiseMaterial); |
| 20 | 102 | | } |
| | 103 | |
|
| | 104 | | private static void ForgetPreviousPromises(Queue<AssetPromise_Material> promises, Asset_Material currentAppliedM |
| | 105 | | { |
| 19 | 106 | | if (promises.Count <= 1) |
| 8 | 107 | | return; |
| | 108 | |
|
| 19 | 109 | | while (promises.Count > 1 && promises.Peek().asset != currentAppliedMaterial) |
| | 110 | | { |
| 8 | 111 | | AssetPromiseKeeper_Material.i.Forget(promises.Dequeue()); |
| | 112 | | } |
| 11 | 113 | | } |
| | 114 | |
|
| | 115 | | private static AssetPromise_Material_Model CreatePBRMaterialPromiseModel(PBMaterial model, |
| | 116 | | AssetPromise_Material_Model.Texture? albedoTexture, AssetPromise_Material_Model.Texture? alphaTexture, |
| | 117 | | AssetPromise_Material_Model.Texture? emissiveTexture, AssetPromise_Material_Model.Texture? bumpTexture) |
| | 118 | | { |
| 19 | 119 | | return AssetPromise_Material_Model.CreatePBRMaterial(albedoTexture, alphaTexture, emissiveTexture, bumpTextu |
| | 120 | | model.GetAlphaTest(), model.GetAlbedoColor().ToUnityColor(), model.GetEmissiveColor().ToUnityColor(), |
| | 121 | | model.GetReflectiveColor().ToUnityColor(), (AssetPromise_Material_Model.TransparencyMode)model.GetTransp |
| | 122 | | model.GetRoughness(), model.GetGlossiness(), model.GetSpecularIntensity(), |
| | 123 | | model.GetEmissiveIntensity(), model.GetDirectIntensity()); |
| | 124 | | } |
| | 125 | |
|
| | 126 | | private static AssetPromise_Material_Model CreateBasicMaterialPromiseModel(PBMaterial model, AssetPromise_Materi |
| | 127 | | { |
| 1 | 128 | | return AssetPromise_Material_Model.CreateBasicMaterial(albedoTexture, model.GetAlphaTest()); |
| | 129 | | } |
| | 130 | |
|
| | 131 | | private static AssetPromise_Material_Model.Texture? CreateMaterialPromiseTextureModel(string textureUrl, UnityEn |
| | 132 | | { |
| 11 | 133 | | if (string.IsNullOrEmpty(textureUrl)) |
| 2 | 134 | | return null; |
| | 135 | |
|
| 9 | 136 | | return new AssetPromise_Material_Model.Texture(textureUrl, wrapMode, filterMode); |
| | 137 | | } |
| | 138 | | } |
| | 139 | | } |