ST_MIDIManager.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Main/Base/ST_AudioControlManagerBase.h"
#include "MIDI/ST_MIDIControlInterface.h"
#include "Main/ST_AudioControlSettings.h"
#include "MIDIDeviceController.h"
#include "MIDIDeviceInputController.h"
#include "MIDIDeviceOutputController.h"
#include "MIDIDeviceControllerBase.h"
#include "ST_MIDIManager.generated.h"

UCLASS()
class ST_AUDIOCONTROL_API UST_MIDIManager : public UST_AudioControlManagerBase, public IST_MIDIControlInterface
{
    GENERATED_BODY()
 
public:
    UST_MIDIManager();


    virtual void OnMIDINoteOn_Implementation(int32 Timestamp, int32 Channel, int32 Note, int32 Velocity) override;
    virtual void OnMIDINoteOff_Implementation(int32 Timestamp, int32 Channel, int32 Note, int32 Velocity) override;
    virtual void OnMIDIPitchBend_Implementation(FIntPoint PitchBendRange, int32 Timestamp, int32 Channel, int32 Pitch) override;
    virtual void OnMIDIAftertouch_Implementation(int32 Timestamp, int32 Channel, int32 Note, int32 Amount) override;
    virtual void OnMIDIControlChange_Implementation(FIntPoint ControlChangeRange, int32 Timestamp, int32 Channel, int32 Type, int32 Value) override;
    virtual void OnMIDIProgramChange_Implementation(int32 Timestamp, int32 Channel, int32 ControlID, int32 Velocity) override;
    virtual void OnMIDIChannelAftertouch_Implementation(int32 Timestamp, int32 Channel, int32 Amount) override;

    UFUNCTION()
    void OnMIDIEventHandler(UMIDIDeviceController* HandledMIDIDeviceController, int32 Timestamp, EMIDIEventType EventType, int32 Channel, int32 ControlID, int32 Velocity, int32 RawEventType);

    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDINoteOnHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 Note, int32 Velocity);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDINoteOffHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 Note, int32 Velocity);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDIPitchBendHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 Pitch);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDIAftertouchHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 Note, int32 Amount);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDIControlChangeHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 Type, int32 Value);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDIProgramChangeHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 ControlID, int32 Velocity);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDIChannelAftertouchHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Channel, int32 Amount);
 
    UFUNCTION(Category = "ShaderTech|AudioControl|MIDI|Receive")
    void OnMIDIRawEventHandler(UMIDIDeviceInputController* MIDIDeviceController, int32 Timestamp, int32 Type, int32 Channel, int32 MessageData1, int32 MessageData2);

    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDIEvent(EMIDIEventType EventType, int32 Channel, int32 data1, int32 data2);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDINoteOn(int32 Channel, int32 Note, int32 Velocity);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDINoteOff(int32 Channel, int32 Note, int32 Velocity);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDIPitchBend(int32 Channel, int32 Pitch);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDINoteAftertouch(int32 Channel, int32 Note, float Amount);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDIControlChange(int32 Channel, int32 Type, int32 Value);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDIProgramChange(int32 Channel, int32 ProgramNumber);
 
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI|Send")
    void SendMIDIChannelAftertouch(int32 Channel, float Amount);

    bool ManageMIDIDeviceController(FName NewDeviceName);

    bool UnmanageMIDIDeviceController(FName NewDeviceName);

    FMIDIRanges GetMIDIRanges() const;
 
protected:
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|AudioControl|MIDI")
    FMIDIRanges MIDIRanges;
 
public:
    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI")
    static UST_MIDIManager* GetInstance(const UObject* context);

    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI")
    void SetControlChangeValue(FIntPoint NewRange);

    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioControl|MIDI")
    void SetPitchBendValue(FIntPoint NewRange);
 
protected:
    virtual void OnManagedItemRegistered(UObject* ManagedItem, bool bFirstManagedItemRegistered) override;

    virtual void OnManagedItemDeregistered(UObject* ManagedItem, bool bAllManagedItemDeregistered) override;

    virtual void OnControllerRegistered(FST_AudioController Controller, bool bFirstControllerRegistered) override;

    virtual void OnControllerDeregistered(FST_AudioController Controller, bool bAllControllersDeregistered) override;
};
 
inline FMIDIRanges UST_MIDIManager::GetMIDIRanges() const
{
    return MIDIRanges;
}
 
inline void UST_MIDIManager::SetControlChangeValue(FIntPoint NewRange)
{
    MIDIRanges.ControlChangeRange = NewRange;
}
 
inline void UST_MIDIManager::SetPitchBendValue(FIntPoint NewRange)
{
    MIDIRanges.PitchBendRange = NewRange;
}