RRSceneDirector.h

Go to the documentation of this file.

 
#pragma once
 
 
 
#include "CoreMinimal.h"
 
 
 
// UE
 
#include "Engine/DirectionalLight.h"
 
#include "Engine/PostProcessVolume.h"
 
#include "GameFramework/GameStateBase.h"
 
 
 
// RapyutaSimulationPlugins
 
#include "Core/RRActorCommon.h"
 
#include "Core/RRBaseActor.h"
 
#include "Core/RRCamera.h"
 
#include "Core/RRGameSingleton.h"
 
#include "Core/RRPlayerController.h"
 
#include "Core/RRTypeUtils.h"
 
#include "RapyutaSimulationPlugins.h"
 
 
 
#include "RRSceneDirector.generated.h"
 
 
 
DECLARE_DELEGATE_OneParam(FOnSpawnedActorsSettled, bool /*bIsForNewOperationBatch*/);
 
 
 
#define UE_LOG_WITH_SCENE_ID(CategoryName, Verbosity, ...)                                                                       \
 
    {                                                                                                                            \
 
        UE_LOG_WITH_INFO(CategoryName, Verbosity, TEXT("SceneInstance[%d] %s"), SceneInstanceId, *FString::Printf(__VA_ARGS__)); \
 
    }
 
 
 
class RAPYUTASIMULATIONPLUGINS_API ARRSceneDirector : public ARRBaseActor
 
{
 
 
 
public:
 
    ARRSceneDirector();
 
 
 
 
    int32 OperationBatchId = 0;
 
 
 
 
    FString SceneName;
 
 
 
 
    APostProcessVolume* MainPostProcessVolume = nullptr;
 
 
 
 
    ARRCamera* SceneCamera = nullptr;
 
 
 
 
    bool HasSceneInitialized()
 
    {
 
        return bSceneInitialized;
 
    }
 
 
 
    FOnSpawnedActorsSettled OnSpawnedActorsSettled;
 
 
 
    virtual bool HasOperationCompleted(bool bIsLogged = false);
 
 
 
 
    double DataCollectionTimeStamp = 0.f;
 
 
 
 
    TArray<int32> SceneEntityMaskValueList;
 
 
 
protected:
 
    virtual bool Initialize() override;
 
 
 
 
 
 
 
 
    virtual bool InitializeOperation();
 
 
 
    virtual void RunOperation();
 
 
 
    virtual void ContinueOperation(bool bIsLastOperationSuccessful, bool bContinueRGBRandomizing)
 
    {
 
    }
 
 
 
    virtual void OnDataCollectionPhaseDone(bool bIsFinalDataCollectingPhase);
 
    virtual void EndSceneInstance();
 
 
 
 
    FTimerHandle InitializationTimerHandle;
 
 
 
 
    FTimerHandle DataCollectionTimerHandle;
 
    // Also use this if needed: FTimerDelegate DataCollectionTimerDelegate;
 
 
 
 
    uint8 bIsDataCollecting : 1;
 
 
 
    virtual void DoDataCollecting()
 
    {
 
        bIsDataCollecting = true;
 
    }
 
 
 
 
    uint8 bIsOperating : 1;
 
 
 
 
    int32 OperationBatchLoopLeft = 0;
 
    virtual void SpawnActors()
 
    {
 
    }
 
 
 
    virtual void ResetScene();
 
 
 
private:
 
    void TryInitializeOperation();
 
 
 
 
    FDateTime LastTimeStamp;
 
 
 
 
    uint8 bSceneInitialized : 1;
 
};