MassSubsystemBase.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Subsystems/WorldSubsystem.h"
#include "MassTypeManager.h"
#include "Subsystems/SubsystemCollection.h"
#include "MassEntityConcepts.h"
#include "MassSubsystemBase.generated.h"
 
#define UE_API MASSENTITY_API
 
struct FMassEntityManager;
 
namespace UE::Mass
{
    namespace Subsystems
    {
        struct FInitializationState
        {
            uint8 bInitializeCalled : 1 = false;
            uint8 bPostInitializeCalled : 1 = false;
            uint8 bOnWorldBeginPlayCalled : 1 = false;
        };
 
        MASSENTITY_API void RegisterSubsystemType(FSubsystemCollectionBase& Collection, TSubclassOf<USubsystem> SubsystemClass, FSubsystemTypeTraits&& Traits);
        MASSENTITY_API void RegisterSubsystemType(TSharedRef<FMassEntityManager> EntityManager, TSubclassOf<USubsystem> SubsystemClass, FSubsystemTypeTraits&& Traits);
    }
}
 
UCLASS(Abstract, config = Mass, defaultconfig, MinimalAPI)
class UMassSubsystemBase : public UWorldSubsystem
{
    GENERATED_BODY()
 
public:
    static UE_API bool AreRuntimeMassSubsystemsAllowed(UObject* Outer);
    UE::Mass::Subsystems::FInitializationState GetInitializationState() const { return InitializationState; }
 
protected:
    //~USubsystem interface
    UE_API virtual bool ShouldCreateSubsystem(UObject* Outer) const override;
    UE_API virtual void Initialize(FSubsystemCollectionBase& Collection) override;
    UE_API virtual void PostInitialize() override;
    UE_API virtual void Deinitialize() override;
    UE_API virtual void OnWorldBeginPlay(UWorld& InWorld) override;
    //~End of USubsystem interface
 
    UE_API void HandleLateCreation();
 
    template <UE::Mass::CSubsystem T>
    void OverrideSubsystemTraits(FSubsystemCollectionBase& Collection)
    {
        UE::Mass::Subsystems::RegisterSubsystemType(Collection, T::StaticClass(), UE::Mass::FSubsystemTypeTraits::Make<T>());
    }
 
    UE::Mass::Subsystems::FInitializationState InitializationState;
};
 
UCLASS(Abstract, config = Mass, defaultconfig, MinimalAPI)
class UMassTickableSubsystemBase : public UTickableWorldSubsystem
{
    GENERATED_BODY()
 
public:
    UE::Mass::Subsystems::FInitializationState GetInitializationState() const { return InitializationState; }
 
protected:
    //~USubsystem interface
    UE_API virtual bool ShouldCreateSubsystem(UObject* Outer) const override;
    UE_API virtual void Initialize(FSubsystemCollectionBase& Collection) override;
    UE_API virtual void PostInitialize() override;
    UE_API virtual void Deinitialize() override;
    UE_API virtual void OnWorldBeginPlay(UWorld& InWorld) override;
    //~End of USubsystem interface
 
    template <UE::Mass::CSubsystem T>
    void OverrideSubsystemTraits(FSubsystemCollectionBase& Collection)
    {
        UE::Mass::Subsystems::RegisterSubsystemType(Collection, T::StaticClass(), UE::Mass::FSubsystemTypeTraits::Make<T>());
    }
 
    UE_API void HandleLateCreation();
 
private:
    UE::Mass::Subsystems::FInitializationState InitializationState;
};
 
#undef UE_API