StageManagementSubsystem.h

이 파일의 문서화 페이지로 가기

#pragma once
#include "CK_UE.h"
#include "Component/Spawn/EnemySpawnDirectorSystemComponent.h"
#include "Data/Stage/StageManifest.h"
#include "Interface/Level/StreamingEventListenerInterface.h"
#include "Level/Spawn/SpawnHub.h"
#include "Subsystems/WorldSubsystem.h"
#include "StageManagementSubsystem.generated.h"
 
 
struct TargetableQuery;
struct FStageEventContext;
struct FStoryboardEntry;
class UStageManagementSubsystem;
class UStageDictionary;
class UStageEventListenerComponent;
class UStageEventProcessor;
class UStoryboardDefinition;
class ULevelStreamingData;
 
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE(FOnAllWavesCleared);
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE(FOnNextWaveCountdownStopped);
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_OneParam(FOnEnemyRemainingCountChanged, int32, RemainingCount);
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_OneParam(FOnNextWaveCountdownStarted, float, SecondsRemaining);
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_OneParam(FOnWaveManagerReadyToSpawning, const FWaveDataCollection&, WaveDataCollection);
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_TwoParams(FOnWaveStarted, const FWaveDataCollection&, WaveDataCollection, const int32, CurrentWaveIndex);
 
DECLARE_DYNAMIC_MULTICAST_DELEGATE_ThreeParams(FOnWaveCleared, const FWaveDataCollection&, WaveDataCollection, const int32, ClearedWaveIndex, const int32, NextWaveIndex);
 
 
USTRUCT(BlueprintType)
struct FStageProgressState
{
    GENERATED_BODY()
 
public:
    void IncreaseCurrentWaveRemainingEnemyCount(const int32 Count)
    {
        CurrentWaveRemainingEnemyCount += FMath::Max(Count, 0);
    }
 
    void DecreaseCurrentWaveRemainingEnemyCount(const int32 Count)
    {
        CurrentWaveRemainingEnemyCount = FMath::Max(CurrentWaveRemainingEnemyCount - FMath::Max(Count, 0), 0);
    }
 
    void SetCurrentStageID(const FName InStageID)
    {
        CurrentStageID = InStageID;
    }
 
    void SetNextWaveCountdownEndTime(const float InEndTime)
    {
        NextWaveCountdownEndTime = InEndTime;
    }
 
    void SetCurrentWaveRemainingEnemyCount(const int32 Count)
    {
        CurrentWaveRemainingEnemyCount = FMath::Max(Count, 0);
    }
 
    void Reset()
    {
        CurrentStageID = NAME_None;
        NextWaveCountdownEndTime = 0.0f;
        CurrentWaveRemainingEnemyCount = 0;
    }
 
public:
    FORCEINLINE FName GetCurrentStageID() const
    {
        return CurrentStageID;
    }
 
    FORCEINLINE float GetNextWaveCountdownEndTime() const
    {
        return NextWaveCountdownEndTime;
    }
 
    FORCEINLINE int32 GetCurrentWaveRemainingEnemyCount() const
    {
        return CurrentWaveRemainingEnemyCount;
    }
 
private:
    UPROPERTY(Transient)
    FName CurrentStageID = NAME_None;
 
    UPROPERTY(Transient)
    float NextWaveCountdownEndTime = 0.f;
 
    UPROPERTY(Transient)
    int32 CurrentWaveRemainingEnemyCount = 0;
};
 
 
USTRUCT(Blueprintable, BlueprintType)
struct FStageManagementHandle
{
    GENERATED_BODY()
 
public:
    FStageManagementHandle(UStageManagementSubsystem* StageManagementSubsystem);
    FStageManagementHandle() { }
 
public:
    void ResumeSpawning() const
    {
        SetSpawningEnabled(true);
    }
 
    void PauseSpawning() const
    {
        SetSpawningEnabled(false);
    }
 
private:
    void SetSpawningEnabled(bool bEnabled) const;
 
public:
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnWaveStarted OnWaveStarted;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnWaveCleared OnWaveCleared;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnAllWavesCleared OnAllWavesCleared;
 
    UPROPERTY(BlueprintAssignable, Category = "Enemy")
    FOnSpawnedEnemyKilled OnSpawnedEnemyKilled;
 
private:
    UPROPERTY(Transient)
    TObjectPtr<UStageManagementSubsystem> StageManagementSubsystem;
};
 
 
UCLASS()
class CK_UE_API UStageManagementSubsystem : public UWorldSubsystem, public IStreamingEventListenerInterface
{
    GENERATED_BODY()
 
public:
    virtual void Deinitialize() override;
    virtual void OnLevelLoaded(ULevel* NewLevel, const ULevelStreamingData* LevelStreamingData) override;
    virtual void OnLevelUnloaded() override;
    virtual void OnLevelStreamingStarted() override { }
    virtual void OnLevelStreamingEnded() override { }
 
    void SetCurrentWaveRemainingEnemyCount(int32 Count);
    void IncreaseCurrentWaveRemainingEnemyCount(int32 Count);
    void DecreaseCurrentWaveRemainingEnemyCount(int32 Count = 1);
    void NotifyNextWaveCountdownStarted(float Duration);
    void NotifyNextWaveCountdownStopped();
 
    ASpawnHub* GetOrCreateSpawnHub();
    TargetableQuery GetStageTargetableQueryHandle() const;
 
public:
    UFUNCTION(BlueprintPure, Category = "Stage|Enemy")
    int32 GetTotalSpawnRemainingCount() const;
 
    UFUNCTION(BlueprintPure, Category = "Stage|Enemy")
    int32 GetCurrentWaveRemainingEnemyCount() const;
 
    UFUNCTION(BlueprintPure, Category = "Stage|Enemy")
    int32 GetSpawnedEnemyCount() const;
 
    UFUNCTION(BlueprintCallable, Category = "Stage|Enemy")
    void NotifyEnemyRemainingCountChanged();
 
    UFUNCTION(BlueprintPure, Category = "Stage|Wave")
    float GetNextWaveCountdownRemainingTime() const;
 
    UFUNCTION(BlueprintPure, Category = "Stage|Wave")
    bool IsNextWaveCountdownActive() const;
 
    UFUNCTION(BlueprintPure, Category = "Stage")
    FStageManagementHandle GetStageManagementHandle();
 
public:
    void RegisterListener(const FName EventID, UStageEventListenerComponent* Listener);
    void UnregisterListener(const FName EventID, UStageEventListenerComponent* Listener);
    void DispatchStageEvent(const FName EventID, const AActor* Caller, const AActor* Sender);
 
    int32 GetProcessorTriggerCount(const UStageEventProcessor* Processor) const;
    int32 IncrementProcessorTriggerCount(const UStageEventProcessor* Processor);
    int32 DecrementProcessorTriggerCount(const UStageEventProcessor* Processor);
    void NotifyProcessorTriggered(const FName EventID, const AActor* Caller, const UStageEventProcessor* Processor);
 
    FName GetLastPlayerTriggeredEventID() const;
    bool RollbackLastPlayerTriggeredProcessorCounts(const FName EventID);
 
public:
    UFUNCTION(BlueprintCallable, Category = "StageEvent")
    void ResetProcessorTriggerCount(UStageEventProcessor* Processor);
 
    UFUNCTION(BlueprintCallable, Category = "StageEvent")
    void ResetAllProcessorTriggerCounts();
 
public:
    UFUNCTION(BlueprintPure, Category = "Stage")
    FORCEINLINE ASpawnHub* GetSpawnHub() const
    {
        return SpawnHub;
    }
 
    FORCEINLINE FName GetCurrentStageID() const
    {
        return StageProgress.GetCurrentStageID();
    }
 
public:
    template <CSpawnSystem TSpawnSystem>
    TSpawnSystem* GetOrCreateSpawnSystem();
    
private:
    bool EnsureStageDictionaryLoaded();
    const FStoryboardEntry* ResolveEntry(const FName EventID) const;
    const UStoryboardDefinition* ResolveStoryboard() const;
    bool IsPlayerEventCaller(const AActor* Caller) const;
    FStageEventContext BuildStageEventContext(FName EventID);
 
public:
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnWaveManagerReadyToSpawning ReadyToSpawn;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnWaveStarted OnWaveStarted;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnWaveCleared OnWaveCleared;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnAllWavesCleared OnAllWavesCleared;
 
    UPROPERTY(BlueprintAssignable, Category = "Enemy")
    FOnSpawnedEnemyKilled OnSpawnedEnemyKilled;
 
    UPROPERTY(BlueprintAssignable, Category = "Enemy")
    FOnEnemyRemainingCountChanged OnEnemyRemainingCountChanged;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnNextWaveCountdownStarted OnNextWaveCountdownStarted;
 
    UPROPERTY(BlueprintAssignable, Category = "Wave")
    FOnNextWaveCountdownStopped OnNextWaveCountdownStopped;
 
protected:
    UPROPERTY(Transient)
    TObjectPtr<ASpawnHub> SpawnHub;
 
private:
    UPROPERTY(Transient)
    FStageProgressState StageProgress;
 
    UPROPERTY(Transient)
    TObjectPtr<UStageDictionary> CachedStageDictionary;
 
    TArray<TWeakObjectPtr<const UStageEventProcessor>> LastPlayerTriggeredProcessors;
    TMap<TWeakObjectPtr<const UStageEventProcessor>, int32> ProcessorTriggerCounts;
    TMap<FName, TArray<TWeakObjectPtr<UStageEventListenerComponent>>> ListenersByEventID;
 
    FName LastPlayerTriggeredEventID = NAME_None;
    int32 PlayerTriggeredDispatchSerial = 0;
    int32 ActivePlayerTriggeredDispatchSerial = INDEX_NONE;
    int32 LastPlayerTriggeredDispatchSerial = INDEX_NONE;
 
public:
    friend struct FStageManagementHandle;
};
 
 
template <CSpawnSystem TSpawnSystem>
TSpawnSystem* UStageManagementSubsystem::GetOrCreateSpawnSystem()
{
    ASpawnHub* Hub = GetOrCreateSpawnHub();
    return Hub ? Hub->GetOrCreateSpawnSystem<TSpawnSystem>() : nullptr;
}