GlobalDeferredLoadingSystem.cs

using Arch.Core;
using Arch.SystemGroups;
using Arch.SystemGroups.DefaultSystemGroups;
using DCL.AvatarRendering.Emotes;
using DCL.AvatarRendering.Wearables.Components;
using DCL.AvatarRendering.Wearables.Components.Intentions;
using DCL.AvatarRendering.Wearables.Helpers;
using DCL.Ipfs;
using DCL.Optimization.PerformanceBudgeting;
using DCL.Profiles;
using ECS.Groups;
using ECS.SceneLifeCycle;
using ECS.SceneLifeCycle.Components;
using ECS.SceneLifeCycle.SceneDefinition;
using ECS.SceneLifeCycle.Systems;
using ECS.StreamableLoading.AssetBundles;
using ECS.StreamableLoading.AssetBundles.InitialSceneState;
using ECS.StreamableLoading.AudioClips;
using ECS.StreamableLoading.DeferredLoading;
using ECS.StreamableLoading.GLTF;
using ECS.StreamableLoading.Textures;
using SceneRunner.Scene;

namespace DCL.GlobalPartitioning
{
    /// <summary>
    ///     Weighs asset, definitions and scenes loading against each other according to their partition in the global world
    /// </summary>
    [UpdateInGroup(typeof(PresentationSystemGroup))]
    [UpdateAfter(typeof(PrepareGlobalAssetBundleLoadingParametersSystem))]
    [UpdateBefore(typeof(LoadGlobalSystemGroup))]
    public partial class GlobalDeferredLoadingSystem : DeferredLoadingSystem
    {
        private static readonly QueryDescription[] COMPONENT_HANDLERS_SCENES_ASSETS;
        private static readonly QueryDescription[] COMPONENT_HANDLERS_SCENES;

        private readonly IScenesCache scenesCache;
        private readonly Entity playerEntity;

        static GlobalDeferredLoadingSystem()
        {
            COMPONENT_HANDLERS_SCENES_ASSETS = new[]
            {
                CreateQuery<GetSceneDefinitionList, SceneDefinitions>(),
                CreateQuery<GetSceneDefinition, SceneEntityDefinition>(),
                CreateQuery<GetISSDescriptor, ISSDescriptor>(),
                CreateQuery<GetSceneFacadeIntention, ISceneFacade>(),
                CreateQuery<GetWearableDTOByPointersIntention, WearablesDTOList>(),
                CreateQuery<GetTrimmedWearableByParamIntention, IWearable[]>(),
                CreateQuery<GetSmartWearableSceneIntention, GetSmartWearableSceneIntention.Result>(),
                CreateQuery<GetAssetBundleManifestIntention, SceneAssetBundleManifest>(),
                CreateQuery<GetAssetBundleIntention, AssetBundleData>(),
                CreateQuery<GetGLTFIntention, GLTFData>(),
                CreateQuery<GetTextureIntention, TextureData>(),
                CreateQuery<GetEmotesDTOByPointersFromRealmIntention, EmotesDTOList>(),
                CreateQuery<GetTrimmedEmotesByParamIntention, TrimmedEmotesResponse>(),
                CreateQuery<GetAudioClipIntention, AudioClipData>(),
                CreateQuery<GetProfilesBatchIntent, ProfilesBatchResult>(),
            };

            COMPONENT_HANDLERS_SCENES = new[]
            {
                CreateQuery<GetSceneDefinitionList, SceneDefinitions>(),
                CreateQuery<GetSceneDefinition, SceneEntityDefinition>(),
                CreateQuery<GetISSDescriptor, ISSDescriptor>(),
            };
        }

        public GlobalDeferredLoadingSystem(World world, IReleasablePerformanceBudget releasablePerformanceLoadingBudget, IPerformanceBudget memoryBudget, IScenesCache scenesCache, Entity playerEntity)
            : base(world, COMPONENT_HANDLERS_SCENES, releasablePerformanceLoadingBudget, memoryBudget)
        {
            this.scenesCache = scenesCache;
            this.playerEntity = playerEntity;
        }

        protected override void Update(float t)
        {
            FilterHandlersIfInTeleport();
            base.Update(t);
        }

        private void FilterHandlersIfInTeleport()
        {
            bool downloadOnlySceneMetadata = false;
            //We check if the player is teleporting, and if the scene we want to teleport to has started.
            //If so, only scene metadata will be allowed to de downloaded
            TeleportUtils.PlayerTeleportingState teleportParcel = TeleportUtils.GetTeleportParcel(World, playerEntity);

            if (teleportParcel.IsTeleporting
                && scenesCache.TryGetByParcel(teleportParcel.Parcel, out ISceneFacade scene)
                && scene.SceneStateProvider.State == SceneState.Running)
                downloadOnlySceneMetadata = true;

            sameBoatQueries = downloadOnlySceneMetadata ? COMPONENT_HANDLERS_SCENES : COMPONENT_HANDLERS_SCENES_ASSETS;
        }
    }
}