Input buffer

Source/GameBaseFramework/Private/Animation/GBFAnimNotifyState_InputBuffer.cpp

@@ -0,0 +1,77 @@
+#include "Animation/GBFAnimNotifyState_InputBuffer.h"
+
+#include "GBFLog.h"
+
+#include <Components/SkeletalMeshComponent.h>
+
+void UGBFAnimNotifyState_InputBuffer::NotifyBegin( USkeletalMeshComponent * mesh_component, UAnimSequenceBase * animation, float total_duration, const FAnimNotifyEventReference & event_reference )
+{
+    Super::NotifyBegin( mesh_component, animation, total_duration, event_reference );
+
+    if ( auto * ability_input_buffer_component = GetAbilityInputBufferComponent( mesh_component ) )
+    {
+        ability_input_buffer_component->StartMonitoring( InputTagsToCheck, TriggerPriority );
+    }
+}
+
+void UGBFAnimNotifyState_InputBuffer::NotifyEnd( USkeletalMeshComponent * mesh_component, UAnimSequenceBase * animation, const FAnimNotifyEventReference & event_reference )
+{
+    Super::NotifyEnd( mesh_component, animation, event_reference );
+
+    if ( auto * ability_input_buffer_component = GetAbilityInputBufferComponent( mesh_component ) )
+    {
+        ability_input_buffer_component->StopMonitoring();
+    }
+}
+
+UGBFAbilityInputBufferComponent * UGBFAnimNotifyState_InputBuffer::GetAbilityInputBufferComponent_Implementation( const USkeletalMeshComponent * mesh_component ) const
+{
+    if ( mesh_component == nullptr )
+    {
+        return nullptr;
+    }
+
+    const auto * owner = mesh_component->GetOwner();
+    if ( owner == nullptr )
+    {
+        UE_LOG( LogGBF, Error, TEXT( "Ability Input Buffer Notify : No Owner found" ) );
+        return nullptr;
+    }
+
+    auto * ability_input_buffer_component = owner->FindComponentByClass< UGBFAbilityInputBufferComponent >();
+
+    if ( ability_input_buffer_component != nullptr )
+    {
+        return ability_input_buffer_component;
+    }
+
+    // Check all parent
+    auto * parent = owner->GetParentActor();
+    while ( parent )
+    {
+        ability_input_buffer_component = parent->FindComponentByClass< UGBFAbilityInputBufferComponent >();
+        if ( ability_input_buffer_component != nullptr )
+        {
+            return ability_input_buffer_component;
+        }
+        parent = parent->GetParentActor();
+    }
+
+    // Check all attached actors
+    TArray< AActor * > actors;
+    owner->GetAttachedActors( actors, true, true );
+    actors.RemoveAll( []( AActor * actor ) {
+        return !Cast< APawn >( actor );
+    } );
+    for ( auto & child : actors )
+    {
+        ability_input_buffer_component = child->FindComponentByClass< UGBFAbilityInputBufferComponent >();
+        if ( ability_input_buffer_component != nullptr )
+        {
+            return ability_input_buffer_component;
+        }
+    }
+
+    UE_LOG( LogGBF, Error, TEXT( "No Ability Input Buffer Component found" ) );
+    return nullptr;
+}

Source/GameBaseFramework/Private/Characters/Components/GBFAbilityInputBufferComponent.cpp

@@ -0,0 +1,230 @@
+#include "Characters/Components/GBFAbilityInputBufferComponent.h"
+
+#include "Characters/Components/GBFHeroComponent.h"
+#include "Characters/Components/GBFPawnExtensionComponent.h"
+#include "GAS/Components/GBFAbilitySystemComponent.h"
+#include "Input/GBFInputComponent.h"
+
+UGBFAbilityInputBufferComponent::UGBFAbilityInputBufferComponent( const FObjectInitializer & ObjectInitializer ) :
+    Super( ObjectInitializer )
+{
+    TriggerPriority = ETriggerPriority::LastTriggeredInput;
+    this->PrimaryComponentTick.bStartWithTickEnabled = false;
+    this->PrimaryComponentTick.bCanEverTick = true;
+}
+
+void UGBFAbilityInputBufferComponent::TickComponent( float DeltaTime, ELevelTick TickType, FActorComponentTickFunction * ThisTickFunction )
+{
+    Super::TickComponent( DeltaTime, TickType, ThisTickFunction );
+    MonitoringTime += DeltaTime;
+    if ( !ensureAlwaysMsgf( MonitoringTime <= MaxMonitoringTime, TEXT( "Ability Input Buffer didn't call Stop Monitor 5 secs after activation, please call it manually !" ) ) )
+    {
+        StopMonitoring();
+    }
+}
+
+void UGBFAbilityInputBufferComponent::StartMonitoring( FGameplayTagContainer input_tags_to_check, ETriggerPriority trigger_priority )
+{
+    if ( input_tags_to_check.IsEmpty() )
+    {
+        return;
+    }
+
+    Reset();
+    TriggerPriority = trigger_priority;
+    InputTagsToCheck = input_tags_to_check;
+    BindActions();
+
+#if !UE_BUILD_SHIPPING
+    SetComponentTickEnabled( true );
+#endif
+}
+
+void UGBFAbilityInputBufferComponent::StopMonitoring()
+{
+    RemoveBinds();
+    TryToTriggerAbility();
+    Reset();
+}
+
+void UGBFAbilityInputBufferComponent::Reset()
+{
+    TriggeredTags.Reset();
+    InputTagsToCheck.Reset();
+    BindHandles.Reset();
+    MonitoringTime = 0.0f;
+
+#if !UE_BUILD_SHIPPING
+    SetComponentTickEnabled( false );
+#endif
+}
+
+void UGBFAbilityInputBufferComponent::BindActions()
+{
+    if ( InputTagsToCheck.IsEmpty() )
+    {
+        return;
+    }
+
+    auto * pawn = GetPawn< APawn >();
+    if ( pawn == nullptr )
+    {
+        return;
+    }
+
+    auto * input_component = Cast< UEnhancedInputComponent >( pawn->InputComponent );
+    if ( input_component == nullptr )
+    {
+        return;
+    }
+
+    const auto * hero_component = UGBFHeroComponent::FindHeroComponent( pawn );
+    if ( hero_component == nullptr )
+    {
+        return;
+    }
+
+    for ( auto & input_config : hero_component->GetBoundActionsByInputconfig() )
+    {
+        for ( auto & tag : InputTagsToCheck )
+        {
+            if ( const auto * input_action = input_config.Key->FindAbilityInputActionForTag( tag ) )
+            {
+                BindHandles.Add( input_component->BindAction( input_action, ETriggerEvent::Triggered, this, &ThisClass::AbilityInputTagPressed, tag ).GetHandle() );
+            }
+        }
+    }
+}
+
+void UGBFAbilityInputBufferComponent::RemoveBinds()
+{
+    auto * pawn = GetPawn< APawn >();
+    if ( pawn == nullptr )
+    {
+        return;
+    }
+
+    if ( auto * input_component = Cast< UEnhancedInputComponent >( pawn->InputComponent ) )
+    {
+        for ( auto & handle : BindHandles )
+        {
+            input_component->RemoveBindingByHandle( handle );
+        }
+    }
+}
+
+void UGBFAbilityInputBufferComponent::AbilityInputTagPressed( FGameplayTag input_tag )
+{
+    TriggeredTags.Add( input_tag );
+}
+
+bool UGBFAbilityInputBufferComponent::TryToTriggerAbility()
+{
+    if ( TriggeredTags.IsEmpty() )
+    {
+        return false;
+    }
+
+    auto * pawn = GetPawn< APawn >();
+    if ( pawn == nullptr )
+    {
+        return false;
+    }
+
+    const auto * pawn_ext_comp = UGBFPawnExtensionComponent::FindPawnExtensionComponent( pawn );
+    if ( pawn_ext_comp == nullptr )
+    {
+        return false;
+    }
+
+    if ( auto * asc = pawn_ext_comp->GetGBFAbilitySystemComponent() )
+    {
+        for ( auto & tagged_ability_tag : InputTagsToCheck )
+        {
+            auto * tagged_ability = asc->FindAbilityClassWithInputTag( tagged_ability_tag );
+            asc->CancelAbility( tagged_ability );
+        }
+
+        // Try to activate ability in priority order
+        FGameplayTag tag = TryToGetInputTagWithPriority();
+
+        while ( tag.IsValid() )
+        {
+            if ( auto * ability = asc->FindAbilityClassWithInputTag( tag ) )
+            {
+                asc->CancelAbility( ability );
+                if ( asc->TryActivateAbilityByClass( ability->GetClass() ) )
+                {
+                    return true;
+                }
+            }
+
+            tag = TryToGetInputTagWithPriority();
+        }
+    }
+    return false;
+}
+
+FGameplayTag UGBFAbilityInputBufferComponent::TryToGetInputTagWithPriority()
+{
+    if ( TriggeredTags.IsEmpty() )
+    {
+        return FGameplayTag::EmptyTag;
+    }
+
+    switch ( TriggerPriority )
+    {
+        case ETriggerPriority::LastTriggeredInput:
+            return GetLastTriggeredInput();
+
+        case ETriggerPriority::MostTriggeredInput:
+            return GetMostTriggeredInput();
+
+        default:
+            return FGameplayTag::EmptyTag;
+    }
+}
+
+FGameplayTag UGBFAbilityInputBufferComponent::GetLastTriggeredInput()
+{
+    FGameplayTag first_tag = TriggeredTags[ 0 ];
+    TriggeredTags.Remove( first_tag );
+    return first_tag;
+}
+
+FGameplayTag UGBFAbilityInputBufferComponent::GetMostTriggeredInput()
+{
+    TMap< int, FGameplayTag > triggered_tag_map;
+
+    // Remove all to get count easily
+    for ( auto & tag_to_remove : InputTagsToCheck )
+    {
+        int count = TriggeredTags.Remove( tag_to_remove );
+        triggered_tag_map.Add( count, tag_to_remove );
+    }
+
+    // Get most triggered input
+    int max = -1;
+    for ( auto & input : triggered_tag_map )
+    {
+        if ( input.Key > max )
+        {
+            max = input.Key;
+        }
+    }
+
+    FGameplayTag most_triggered_tag = triggered_tag_map.FindAndRemoveChecked( max );
+
+    // Sort to keep order if first ability fails
+    triggered_tag_map.Remove( 0 );
+    triggered_tag_map.KeySort(
+        []( const int & a, const int & b ) {
+            return a > b;
+        } );
+    for ( auto & i : triggered_tag_map )
+    {
+        TriggeredTags.Add( i.Value );
+    }
+
+    return most_triggered_tag;
+}

Source/GameBaseFramework/Private/Characters/Components/GBFHeroComponent.cpp

@@ -224,6 +224,11 @@ void UGBFHeroComponent::ClearAbilityCameraMode( const FGameplayAbilitySpecHandle
     }
 }
 
+TMap< const UGBFInputConfig *, TArray< uint32 > > UGBFHeroComponent::GetBoundActionsByInputconfig() const
+{
+    return BoundActionsByInputConfig;
+}
+
 void UGBFHeroComponent::OnRegister()
 {
     Super::OnRegister();

Source/GameBaseFramework/Private/GAS/Components/GBFAbilitySystemComponent.cpp

@@ -379,6 +379,24 @@ FGameplayAbilitySpecHandle UGBFAbilitySystemComponent::FindAbilitySpecHandleForC
     return FGameplayAbilitySpecHandle();
 }
 
+UGameplayAbility * UGBFAbilitySystemComponent::FindAbilityClassWithInputTag( FGameplayTag input_tag )
+{
+    if ( !input_tag.IsValid() )
+    {
+        return nullptr;
+    }
+
+    for ( const auto & ability_spec : ActivatableAbilities.Items )
+    {
+        if ( ability_spec.Ability && ability_spec.DynamicAbilityTags.HasTagExact( input_tag ) )
+        {
+            return ability_spec.Ability;
+        }
+    }
+
+    return nullptr;
+}
+
 void UGBFAbilitySystemComponent::OurCancelAllAbilities()
 {
     static const auto GameplayTagContainer = FGameplayTagContainer::CreateFromArray(

Source/GameBaseFramework/Public/Animation/GBFAnimNotifyState_InputBuffer.h

@@ -0,0 +1,30 @@
+#pragma once
+
+#include "Characters/Components/GBFAbilityInputBufferComponent.h"
+
+#include <Animation/AnimNotifies/AnimNotifyState.h>
+#include <CoreMinimal.h>
+
+#include "GBFAnimNotifyState_InputBuffer.generated.h"
+
+class UGBFAbilityInputBufferComponent;
+
+UCLASS( DisplayName = "Ability Input Buffer Window" )
+class GAMEBASEFRAMEWORK_API UGBFAnimNotifyState_InputBuffer : public UAnimNotifyState
+{
+    GENERATED_BODY()
+
+public:
+    void NotifyBegin( USkeletalMeshComponent * mesh_component, UAnimSequenceBase * animation, float total_duration, const FAnimNotifyEventReference & event_reference ) override;
+    void NotifyEnd( USkeletalMeshComponent * mesh_component, UAnimSequenceBase * animation, const FAnimNotifyEventReference & event_reference ) override;
+
+protected:
+    UFUNCTION( BlueprintNativeEvent )
+    UGBFAbilityInputBufferComponent * GetAbilityInputBufferComponent( const USkeletalMeshComponent * mesh_component ) const;
+
+private:
+    UPROPERTY( EditAnywhere )
+    ETriggerPriority TriggerPriority;
+    UPROPERTY( EditAnywhere, Meta = ( Categories = "Input" ) )
+    FGameplayTagContainer InputTagsToCheck;
+};