Add Ammo counts to Weapon

Source/Nakatomi/Weapon.cpp

@@ -14,6 +14,8 @@ void AWeapon::BeginPlay()
 {
 	Super::BeginPlay();
 
+	SetAmmoCountToDefault();
+	
 }
 
 USkeletalMeshComponent* AWeapon::GetSkeletalMesh()
@@ -65,3 +67,43 @@ void AWeapon::SetFireSound(USoundBase* USoundBase)
 {
 	FireSound = USoundBase;
 }
+
+int AWeapon::GetAmmoCount()
+{
+	return AmmoCount;
+}
+
+void AWeapon::SetAmmoCount(int ammoCount)
+{
+	AmmoCount = ammoCount;
+
+	if (AmmoCount > WeaponProperties.MaxAmmo)
+	{
+		AmmoCount = WeaponProperties.MaxAmmo;
+	}
+}
+
+void AWeapon::SetAmmoCountToDefault()
+{
+	AmmoCount = WeaponProperties.DefaultAmmo;
+}
+
+void AWeapon::IncrementAmmoCount(int ammoCount)
+{
+	AmmoCount += ammoCount;
+
+	if (AmmoCount > WeaponProperties.MaxAmmo)
+	{
+		AmmoCount = WeaponProperties.MaxAmmo;
+	}
+}
+
+void AWeapon::DecrementAmmoCount(int ammoCount)
+{
+	AmmoCount -= ammoCount;
+
+	if (AmmoCount < 0)
+	{
+		AmmoCount = 0;
+	}
+}

Source/Nakatomi/Weapon.h

@@ -31,6 +31,12 @@ struct FWeaponProperties
 	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon Properties")
 	float WeaponChangeTime = 2.0f;
 
+	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon Properties")
+	int MaxAmmo = 1;
+
+	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon Properties")
+	int DefaultAmmo = 1;
+
 	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon Properties")
 	int ProjectilesPerShot = 1;
 
@@ -66,6 +72,9 @@ protected:
 	UPROPERTY(EditDefaultsOnly)
 	USoundBase* FireSound;
 
+	UPROPERTY(VisibleAnywhere)
+	int AmmoCount;
+
 public:	
 	// Sets default values for this actor's properties
 	AWeapon();
@@ -94,4 +103,14 @@ public:
 	USoundBase* GetFireSound();
 
 	void SetFireSound(USoundBase* USoundBase);
+
+	int GetAmmoCount();
+
+	void SetAmmoCount(int ammoCount);
+
+	void SetAmmoCountToDefault();
+
+	void IncrementAmmoCount(int ammoCount);
+
+	void DecrementAmmoCount(int ammoCount);
 };