136 lines
3.0 KiB
C++
136 lines
3.0 KiB
C++
// Unreal Fighting Engine by Kevin Poretti
|
|
|
|
#include "FEStateMachineComponent.h"
|
|
|
|
// FE includes
|
|
#include "FEState.h"
|
|
|
|
UFEStateMachineComponent::UFEStateMachineComponent()
|
|
{
|
|
// Don't use Unreal's tick instead use a fixed tick
|
|
PrimaryComponentTick.bCanEverTick = false;
|
|
}
|
|
|
|
|
|
void UFEStateMachineComponent::Initialize()
|
|
{
|
|
for(const TSubclassOf<UFEState>& CurrState : DefaultStates)
|
|
{
|
|
UFEState* TempState = AddState(CurrState);
|
|
if(!CurrentState) // first state to be created is the entry into this state machine
|
|
{
|
|
CurrentState = TempState;
|
|
CurrentState->Enter();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void UFEStateMachineComponent::InitActorInfo(AActor* InOwner, AActor* InAvatar)
|
|
{
|
|
Owner = InOwner;
|
|
Avatar = InAvatar;
|
|
}
|
|
|
|
|
|
UFEState* UFEStateMachineComponent::AddState(TSubclassOf<UFEState> StateClassToAdd)
|
|
{
|
|
UFEState* TempState = NewObject<UFEState>(this, StateClassToAdd);
|
|
if(TempState)
|
|
{
|
|
States.Add(TempState);
|
|
TempState->InitActorInfo(Owner, Avatar);
|
|
return TempState;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
void UFEStateMachineComponent::AddStates(const TArray<TSubclassOf<UFEState>>& StateClassesToAdd)
|
|
{
|
|
for(const TSubclassOf<UFEState>& CurrState : StateClassesToAdd)
|
|
{
|
|
AddState(CurrState);
|
|
}
|
|
}
|
|
|
|
|
|
void UFEStateMachineComponent::RemoveState(FName StateToRemove)
|
|
{
|
|
UE_LOG(LogTemp, Error, TEXT("UFEStateMachineComponent::RemoveState is not yet implemented"));
|
|
}
|
|
|
|
void UFEStateMachineComponent::SwitchStates(UFEState* NewState)
|
|
{
|
|
check(NewState);
|
|
|
|
CurrentState->Exit();
|
|
CurrentState = NewState;
|
|
CurrentState->Enter();
|
|
}
|
|
|
|
|
|
FName UFEStateMachineComponent::GetCurrentStateName() const
|
|
{
|
|
return CurrentState ? CurrentState->Name : NAME_None;
|
|
}
|
|
|
|
|
|
UFEState* UFEStateMachineComponent::FindStateWithName(FName StateName)
|
|
{
|
|
for (UFEState* CurrState : States)
|
|
{
|
|
if(CurrState ->Name == StateName)
|
|
{
|
|
return CurrState;
|
|
}
|
|
}
|
|
|
|
UE_LOG(LogTemp, Warning,
|
|
TEXT("Could not find state in state machine with name %s on %s"), *StateName.ToString(), *Owner->GetName());
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
void UFEStateMachineComponent::BeginPlay()
|
|
{
|
|
Super::BeginPlay();
|
|
|
|
Initialize();
|
|
}
|
|
|
|
|
|
void UFEStateMachineComponent::FixedTick(float OneFrame)
|
|
{
|
|
// Should we switch states?
|
|
|
|
for(UFEState* CurrState : States)
|
|
{
|
|
// Check if the state is enabled
|
|
|
|
// Check state entry conditions if there are any
|
|
|
|
// Check input conditions if there are any
|
|
|
|
// If all state entry conditions are good and at least one input condition is good then we can transition
|
|
|
|
// Lastly just check if the state we're about to transition into isn't the current state.
|
|
// It is OK to transition if state's "CanTransitionToSelf" is true
|
|
|
|
// SwitchStates(NewState);
|
|
// return;
|
|
}
|
|
|
|
// CurrentState should never be null
|
|
// TODO: Should probably assert or whatever UE's equivalent is
|
|
check(CurrentState);
|
|
|
|
// Tick current state
|
|
CurrentState->FixedTick(OneFrame);
|
|
|
|
// Debug
|
|
}
|
|
|