Added battery play state

Author: weeeBox

Source/BatteryCollector/BatteryCollectorGameMode.cpp

@@ -22,6 +22,8 @@ ABatteryCollectorGameMode::ABatteryCollectorGameMode()
 void ABatteryCollectorGameMode::BeginPlay()
 {
 	Super::BeginPlay();
+	SetCurrentState(EBatteryPlayState::EPlaying);
+
 	ABatteryCollectorCharacter* MyCharacter = Cast<ABatteryCollectorCharacter>(UGameplayStatics::GetPlayerPawn(this, 0));
 	if (MyCharacter)
 	{
@@ -48,17 +50,38 @@ void ABatteryCollectorGameMode::Tick(float DeltaTime)
 	ABatteryCollectorCharacter* MyCharacter = Cast<ABatteryCollectorCharacter>(UGameplayStatics::GetPlayerPawn(this, 0));
 	if (MyCharacter)
 	{
+		// If our power is greater than needed to win, set the game's state to Won
+		if (MyCharacter->GetCurrentPower() > PowerToWin)
+		{
+			SetCurrentState(EBatteryPlayState::EWon);
+		}
 		// if the character's power is positive
-		if (MyCharacter->GetCurrentPower() > 0)
+		else if (MyCharacter->GetCurrentPower() > 0)
 		{
 			// decrease the character's power using the decay rate
 			MyCharacter->UpdatePower(-DeltaTime*DecayRate*(MyCharacter->GetInitialPower()));
 		}
+		else
+		{
+			SetCurrentState(EBatteryPlayState::EGameOver);
+		}
+
 	}
 
 }
 
 float ABatteryCollectorGameMode::GetPowerToWin() const
 {
 	return PowerToWin;
-}
+}
+
+EBatteryPlayState ABatteryCollectorGameMode::GetCurrentState() const
+{
+	return CurrentState;
+}
+
+void ABatteryCollectorGameMode::SetCurrentState(EBatteryPlayState NewState)
+{
+	//set current state
+	CurrentState = NewState;
+}

Source/BatteryCollector/BatteryCollectorGameMode.h

@@ -3,6 +3,17 @@
 #include "GameFramework/GameMode.h"
 #include "BatteryCollectorGameMode.generated.h"
 
+
+//enum to store the current state of gameplay
+UENUM(BlueprintType)
+enum class EBatteryPlayState
+{
+	EPlaying,
+	EGameOver,
+	EWon,
+	EUnknown
+};
+
 UCLASS(minimalapi)
 class ABatteryCollectorGameMode : public AGameMode
 {
@@ -19,6 +30,13 @@ class ABatteryCollectorGameMode : public AGameMode
 
 	virtual void BeginPlay() override;
 
+	/** Returns the current playing state */
+	UFUNCTION(BlueprintPure, Category = "Power")
+	EBatteryPlayState GetCurrentState() const;
+
+	/** Sets a new playing state */
+	void SetCurrentState(EBatteryPlayState NewState);
+
 protected:
 	/**The rate at which the character loses power */
 	UPROPERTY(EditDefaultsOnly, BlueprintReadWrite, Category = "Power", Meta = (BlueprintProtected = "true"))
@@ -35,6 +53,10 @@ class ABatteryCollectorGameMode : public AGameMode
 	/** The instance of the HUD */
 	UPROPERTY()
 	class UUserWidget* CurrentWidget;
+
+private:
+	/**Keeps track of the current playing state */
+	EBatteryPlayState CurrentState;
 };