VRTPMobile.h

#pragma once

#include "CoreMinimal.h"
#include "Components/ActorComponent.h"
#include "Components/PrimitiveComponent.h"
#include "Components/SceneCaptureComponentCube.h"
#include "Engine/TextureRenderTargetCube.h"
#include "Engine/DataAsset.h"
#include "Engine/TextureCube.h"
#include "VRTPMobile.generated.h"

UENUM(BlueprintType)
enum class EVRTPMBackgroundMode : uint8
{
    MM_COLOR        UMETA(DisplayName = "Color"),
    MM_SKYBOX       UMETA(DisplayName = "Skybox")
};

UENUM(BlueprintType)
enum class EVRTPMMaskMode : uint8
{
    MM_OFF          UMETA(DisplayName = "Off"),
    MM_MASK         UMETA(DisplayName = "Mask"),
    MM_WINDOW       UMETA(DisplayName = "Window"),
    MM_PORTAL       UMETA(DisplayName = "Portal"),
};

USTRUCT(BlueprintType)
struct FVRTPMPreset
{
    GENERATED_USTRUCT_BODY()

    
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Capture")
    TSubclassOf<class AActor> SkyboxBlueprint;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Capture")
    UTextureCube* CubeMapOverride;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Post Process")
    UMaterial* PostProcessMaterial;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Iris")
    UStaticMesh* IrisMesh;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings")
    FLinearColor EffectColor;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float EffectCoverage;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float EffectFeather;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings")
    EVRTPMBackgroundMode BackgroundMode;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings")
    bool ApplyEffectColor;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings")
    bool ForceEffect;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings|Mask Settings")
    EVRTPMMaskMode MaskMode;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Effect Settings|Mask Settings", meta = (ClampMin = "0", ClampMax = "255"))
    int32 StencilIndex;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Angular Velocity")
    bool bUseAngularVelocity;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float AngularStrength;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "180.0"))
    float AngularMin;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "180.0"))
    float AngularMax;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float AngularSmoothing;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Velocity")
    bool bUseVelocity;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float VelocityStrength;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float VelocityMin;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float VelocityMax;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float VelocitySmoothing;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Acceleration")
    bool bUseAcceleration;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float AccelerationStrength;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float AccelerationMin;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float AccelerationMax;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float AccelerationSmoothing;

    FVRTPMPreset()
    {
        SkyboxBlueprint = NULL;
        CubeMapOverride = NULL;
        PostProcessMaterial = NULL;
        IrisMesh = NULL;
        EffectColor = FLinearColor::Black;
        EffectCoverage = 0;
        EffectFeather = 0;
        BackgroundMode = EVRTPMBackgroundMode::MM_COLOR;
        ApplyEffectColor = false;
        ForceEffect = false;
        MaskMode = EVRTPMMaskMode::MM_OFF;
        StencilIndex = 200;
        bUseAngularVelocity = false;
        AngularStrength = 0;
        AngularMin = 0;
        AngularMax = 0;
        AngularSmoothing = 0;
        bUseVelocity = false;
        VelocityStrength = 0;
        VelocityMin = 0;
        VelocityMax = 0;
        VelocitySmoothing = 0;
        bUseAcceleration = false;
        AccelerationStrength = 0;
        AccelerationMin = 0;
        AccelerationMax = 0;
        AccelerationSmoothing = 0;
    }
};

UCLASS()
class UVRTPMPresetData : public UDataAsset
{
    GENERATED_BODY()

public:
    UPROPERTY(EditAnywhere)
    FVRTPMPreset Data;
};

UCLASS(Blueprintable, meta = (BlueprintSpawnableComponent), ClassGroup = MotionController, HideCategories = ("VRTunnellingProMobile"))
class UVRTunnellingProMobile : public UActorComponent
{
    GENERATED_BODY()

public: 

    UVRTunnellingProMobile();

#if WITH_EDITOR
    virtual void PreEditChange(UProperty* PropertyAboutToChange) override;
    virtual void PostEditChangeProperty(FPropertyChangedEvent& PropertyChangedEvent) override;
#endif 
    virtual void OnComponentDestroyed(bool bDestroyingHierarchy) override;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "VR Tunnelling|Effect Preset")
    UVRTPMPresetData* Preset;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "VR Tunnelling|Effect Preset")
    bool bEnablePreset;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, AdvancedDisplay, Category = "VR Tunnelling")
    TSubclassOf<class AActor> SkyboxBlueprint;
    UPROPERTY(EditAnywhere)
    TSubclassOf<class AActor> SkyboxBlueprintSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, AdvancedDisplay, Category = "VR Tunnelling")
    UTextureCube* CubeMapOverride;
    UPROPERTY(EditAnywhere)
    UTextureCube* CubeMapOverrideSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, AdvancedDisplay, Category = "VR Tunnelling")
    UMaterial* PostProcessMaterial;
    UPROPERTY(EditAnywhere)
    UMaterial* PostProcessMaterialSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, AdvancedDisplay, Category = "VR Tunnelling")
    UStaticMesh* IrisMesh;
    UPROPERTY(EditAnywhere)
    UStaticMesh* IrisMeshSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings")
    FLinearColor EffectColor;
    UPROPERTY(EditAnywhere)
    FLinearColor EffectColorSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float EffectCoverage;
    UPROPERTY(EditAnywhere)
    float EffectCoverageSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float EffectFeather;
    UPROPERTY(EditAnywhere)
    float EffectFeatherSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings")
    EVRTPMBackgroundMode BackgroundMode;
    UPROPERTY(EditAnywhere)
    EVRTPMBackgroundMode BackgroundModeSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings")
    bool ApplyEffectColor;
    UPROPERTY(EditAnywhere)
    bool ApplyEffectColorSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings")
    bool ForceEffect;
    UPROPERTY(EditAnywhere)
    bool ForceEffectSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings|Mask Settings")
    EVRTPMMaskMode MaskMode;
    UPROPERTY(EditAnywhere)
    EVRTPMMaskMode MaskModeSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Effect Settings|Mask Settings", meta = (ClampMin = "0", ClampMax = "255"))
    int32 StencilIndex;
    UPROPERTY(EditAnywhere)
    int32 StencilIndexSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Angular Velocity")
    bool bUseAngularVelocity;
    UPROPERTY(EditAnywhere)
    bool bUseAngularVelocitySwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float AngularStrength;
    UPROPERTY(EditAnywhere)
    float AngularStrengthSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "180.0"))
    float AngularMin;
    UPROPERTY(EditAnywhere)
    float AngularMinSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "180.0"))
    float AngularMax;
    UPROPERTY(EditAnywhere)
    float AngularMaxSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Angular Velocity", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float AngularSmoothing;
    UPROPERTY(EditAnywhere)
    float AngularSmoothingSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Velocity")
    bool bUseVelocity;
    UPROPERTY(EditAnywhere)
    bool bUseVelocitySwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float VelocityStrength;
    UPROPERTY(EditAnywhere)
    float VelocityStrengthSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float VelocityMin;
    UPROPERTY(EditAnywhere)
    float VelocityMinSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float VelocityMax;
    UPROPERTY(EditAnywhere)
    float VelocityMaxSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Velocity", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float VelocitySmoothing;
    UPROPERTY(EditAnywhere)
    float VelocitySmoothingSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Acceleration")
    bool bUseAcceleration;
    UPROPERTY(EditAnywhere)
    bool bUseAccelerationSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "10.0"))
    float AccelerationStrength;
    UPROPERTY(EditAnywhere)
    float AccelerationStrengthSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float AccelerationMin;
    UPROPERTY(EditAnywhere)
    float AccelerationMinSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "1000.0"))
    float AccelerationMax;
    UPROPERTY(EditAnywhere)
    float AccelerationMaxSwap;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, SimpleDisplay, Category = "VR Tunnelling|Motion Settings|Acceleration", meta = (ClampMin = "0.0", ClampMax = "1.0"))
    float AccelerationSmoothing;
    UPROPERTY(EditAnywhere)
    float AccelerationSmoothingSwap;

    USceneCaptureComponentCube* SceneCaptureCube;
    UTextureRenderTargetCube* TC;
    float HFov;
    float VFov;
    UMaterialInstanceDynamic* PostProcessMID;
    AActor* Skybox;
    UStaticMeshComponent* Iris;
    UMaterialInstanceDynamic* IrisOuterMID;
    UMaterialInstanceDynamic* IrisInnerMID;
    bool CaptureInit;

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void ApplyPreset(UVRTPMPresetData* NewPreset);

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void SetBackgroundMode(EVRTPMBackgroundMode NewBackgroundMode);

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void SetMaskMode(EVRTPMMaskMode NewMaskMode);

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void SetEffectColor(FLinearColor NewColor);

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void ApplyColor(bool Enabled);

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void SetStencilMask(int32 NewStencilIndex, bool UpdateMaskedObjects = true);

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void UpdateMaskedObjects();

    UFUNCTION(BlueprintCallable, Category = "VR Tunnelling")
    void SetFeather(float NewFeather);

protected:
    virtual void BeginPlay() override;

public: 
    virtual void TickComponent(float DeltaTime, ELevelTick TickType, FActorComponentTickFunction* ThisTickFunction) override;

private:
    FVector LastForward;
    FVector LastPosition;
    float LastSpeed;

    float AngleSmoothed;
    float VelocitySmoothed;
    float AccelerationSmoothed;

    void CacheSettings();
    void InitCapture();
    void InitSkybox();
    void InitIris();
    void InitFromPreset();
    void SetPresetData(UVRTPMPresetData* NewPreset);
    void UpdateEffectSettings();

    void CalculateMotion(float DeltaTime);
    void ApplyBackgroundMode();
    void ApplyMaskMode();
    void ApplyStencilMasks();
};