Stamina-Module-cpp-UE/StaminaComponent.cpp at main · 2kiss/Stamina-Module-cpp-UE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Fill out your copyright notice in the Description page of Project Settings.


#include "StaminaComponent.h"


// Sets default values for this component's properties
UStaminaComponent::UStaminaComponent()
{
	PrimaryComponentTick.bCanEverTick = true;
	bIsSprinting = false;
	bCanRegenerate = true;
	CurrentStamina = 100.f;
}

void UStaminaComponent::BeginPlay()
{
	Super::BeginPlay();
}


void UStaminaComponent::Initialize(UCharacterStatsAsset* StatsAsset)
{
	if (StatsAsset)
	{
		Stats = StatsAsset;
		CurrentStamina = Stats->MaxStamina;

		OnStaminaChanged.Broadcast(CurrentStamina, Stats->MaxStamina);
	}
}

void UStaminaComponent::TickComponent(float DeltaTime, ELevelTick TickType, FActorComponentTickFunction* ThisTickFunction)
{
	Super::TickComponent(DeltaTime, TickType, ThisTickFunction);

	if (!Stats) return;

	if (bIsSprinting)
	{
		float Drain = Stats->StaminaDrainPerSprint * DeltaTime;
		CurrentStamina = FMath::Clamp(CurrentStamina - Drain, 0.0f, Stats->MaxStamina);

		if (CurrentStamina <= 0.0f)
		{

			bIsSprinting = false;
			OnStaminaExhausted.Broadcast();
		}
	}
	else if (bCanRegenerate && CurrentStamina < Stats->MaxStamina)
	{
		float Regen = Stats->StaminaRegenPerSecond * DeltaTime;
		CurrentStamina = FMath::Clamp(CurrentStamina + Regen, 0.0f, Stats->MaxStamina);
	}

	if (CurrentStamina < Stats->MaxStamina || bIsSprinting)
	{
		OnStaminaChanged.Broadcast(CurrentStamina, Stats->MaxStamina);
	}
}

void UStaminaComponent::SetSprinting(bool bNewSprinting)
{
	if (bIsSprinting == bNewSprinting) return;

	bIsSprinting = bNewSprinting;

	if (bIsSprinting)
	{
		bCanRegenerate = false;
		GetWorld()->GetTimerManager().ClearTimer(RegenDelayTimer);
	}
	else
	{
		if (Stats)
		{
			GetWorld()->GetTimerManager().SetTimer(RegenDelayTimer, this, &UStaminaComponent::EnableRegeneration, Stats->StaminaRegenDelay, false);
		}
	}
}

bool UStaminaComponent::TryConsumeStamina(float Amount)
{
	if (CurrentStamina >= Amount)
	{
		CurrentStamina -= Amount;

		bCanRegenerate = false;
		if (Stats)
		{
			GetWorld()->GetTimerManager().SetTimer(RegenDelayTimer, this, &UStaminaComponent::EnableRegeneration, Stats->StaminaRegenDelay, false);
		}

		OnStaminaChanged.Broadcast(CurrentStamina, Stats ? Stats->MaxStamina : 100.f);
		return true;
	}

	return false;
}

void UStaminaComponent::EnableRegeneration()
{
	bCanRegenerate = true;
}

float UStaminaComponent::GetStaminaPercent() const
{
	if (Stats && Stats->MaxStamina > 0.f)
	{
		return CurrentStamina / Stats->MaxStamina;
	}
	return 0.f;
}