RoveSoSimulatorInputManager.h

#pragma once
 
#include "UObject/Class.h"
#include "UObject/ObjectMacros.h"
#include "Containers/Map.h"
#include "RoveSoSimulatorInputManager.generated.h"
 
class UInputAction;
class UInputMappingContext;
struct FInputActionInstance;
class UEnhancedInputComponent;
class ARoveSoSimulatorPlayerController;
 
//#region InputActions
USTRUCT(BlueprintType)
struct FInputActionsUniversal
{
  GENERATED_BODY()
 
  // --- INPUT AXES - UNIVERSAL ---
  UPROPERTY(EditDefaultsOnly, Category="Input - Universal", meta=(Dispatch="Float"))
  class UInputAction* Translation;
  UPROPERTY(EditDefaultsOnly, Category="Input - Universal", meta=(Dispatch="Float"))
  class UInputAction* Rotation;
  UPROPERTY(EditDefaultsOnly, Category="Input - Universal", meta=(Dispatch="Float"))
  class UInputAction* Zoom;
  UPROPERTY(EditDefaultsOnly, Category="Input - Universal", meta=(Dispatch="Cycle"))
  class UInputAction* ControlModeCycle;
  UPROPERTY(EditDefaultsOnly, Category="Input - Universal", meta=(Dispatch="Cycle"))
  class UInputAction* CameraCycle;
  // --- INPUT ACTIONS - UNIVERSAL ---
  UPROPERTY(EditDefaultsOnly, Category="Input - Universal", meta=(Dispatch="Void"))
  class UInputAction* Exit;
};
 
USTRUCT(BlueprintType)
struct FInputActionsRoverDrive
{
  GENERATED_BODY()
 
  // --- INPUT AXES - ROVER DRIVE ---
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Float"))
  class UInputAction* LeftThrottle;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Float"))
  class UInputAction* RightThrottle;
  // --- INPUT ACTIONS - ROVER DRIVE---
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* SetSpawnPoint;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* GotoSpawnPoint;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* ResetSpline;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* ToggleSpline;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* TogglePointcloud;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* ResetRover;
  UPROPERTY(EditDefaultsOnly, Category="Input - Rover Drive", meta=(Dispatch="Void"))
  class UInputAction* FlipRover;
};
 
USTRUCT(BlueprintType)
struct FInputActionsArm
{
  GENERATED_BODY()
 
  // --- INPUT AXES - ARM ---
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // X-Axis
  class UInputAction* J1;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Shoulder-bicep
  class UInputAction* J2;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Bicep-forearm
  class UInputAction* J3;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Forearm Roll
  class UInputAction* J4;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Wrist up/down
  class UInputAction* J5;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Gripper Roll
  class UInputAction* J6;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Gripper Open / Close
  class UInputAction* GripperOpenClose;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Float"))
  // Solenoid In / Out
  class UInputAction* SolenoidExtension;
  // --- INPUT ACTIONS - ARM---
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Void"))
  class UInputAction* LaserOn;
  UPROPERTY(EditDefaultsOnly, Category="Input - Arm", meta=(Dispatch="Void"))
  class UInputAction* LaserOff;
};
 
USTRUCT(BlueprintType)
struct FActions
{
  GENERATED_BODY()
 
  UPROPERTY(EditDefaultsOnly)
  FInputActionsUniversal Universal;
  UPROPERTY(EditDefaultsOnly)
  FInputActionsRoverDrive RoverDrive;
  UPROPERTY(EditDefaultsOnly)
  FInputActionsArm Arm;
};
 
USTRUCT(BlueprintType)
struct FInputMappingContexts
{
  GENERATED_BODY()
 
  UPROPERTY(EditDefaultsOnly)
  UInputMappingContext* Universal;
  UPROPERTY(EditDefaultsOnly)
  UInputMappingContext* RoverDrive;
  UPROPERTY(EditDefaultsOnly)
  UInputMappingContext* Arm;
};
//#endregion
 
/** 
 * @brief Manager class which collects all input callbacks into one location and allows any piece of code to attach a handler.
 */
UCLASS(Abstract, Blueprintable, BlueprintType)
class URoveSoSimulatorInputManager : public UObject
{
  GENERATED_BODY()
 
public:
  UFUNCTION()
  void SetPlayerController(ARoveSoSimulatorPlayerController* newPlayerController) { PlayerController = newPlayerController; }
  UFUNCTION()
  void SetupInputBindings(UEnhancedInputComponent* EI);
  void SetupBindings(UEnhancedInputComponent* EI, const UStruct* structType, void* structData);
 
  using FVoidCallback = TFunction<void()>;
  using FBoolCallback = TFunction<void(const bool)>;
  using FIntCallback = TFunction<void(const int)>;
  using FFloatCallback = TFunction<void(const float)>;
  TMap<const UInputAction*, FVoidCallback> VoidDispatch;
  TMap<const UInputAction*, FBoolCallback> BoolDispatch;
  TMap<const UInputAction*, FBoolCallback> CycleDispatch; // The passed bool will be true if cycling forward or false if cycling backwards
  TMap<const UInputAction*, FIntCallback> IntDispatch;
  TMap<const UInputAction*, FFloatCallback> FloatDispatch;
 
  UPROPERTY(EditDefaultsOnly, BlueprintReadOnly)
  FActions Actions;
  UPROPERTY(EditDefaultsOnly, BlueprintReadOnly)
  FInputMappingContexts Mappings;
 
private:
  URoveSoSimulatorInputManager();
  UPROPERTY()
  ARoveSoSimulatorPlayerController* PlayerController;
 
  UFUNCTION()
  void OnVoidInput(const FInputActionInstance& Instance);
  UFUNCTION()
  void OnBoolInput(const FInputActionInstance& Instance);
  UFUNCTION()
  void OnCycleInput(const FInputActionInstance& Instance);
  UFUNCTION()
  void OnIntInput(const FInputActionInstance& Instance);
  UFUNCTION()
  void OnFloatInput(const FInputActionInstance& Instance);
};