From 09e4cd0f4104df1787ca9da65655dbe5241a3e45 Mon Sep 17 00:00:00 2001
From: Louis Hobbs <bazzadobs@live.co.uk>
Date: Mon, 12 Jun 2023 22:09:27 +0100
Subject: [PATCH] Add extra weapon pickup creation check for enemies

---
 Source/Nakatomi/EnemyAIController.cpp | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/Source/Nakatomi/EnemyAIController.cpp b/Source/Nakatomi/EnemyAIController.cpp
index d4623c5..4b5d065 100644
--- a/Source/Nakatomi/EnemyAIController.cpp
+++ b/Source/Nakatomi/EnemyAIController.cpp
@@ -91,10 +91,13 @@ void AEnemyAIController::OnDeath(FDamageInfo info)
 
 	enemy->SetLifeSpan(10.0f);
 
-	auto weaponPickup = GetWorld()->SpawnActor<AWeaponPickup>();
-	weaponPickup->SetActorLocation(enemy->GetActorLocation());
-	weaponPickup->SetWeapon(enemy->DefaultWeaponInventory[enemy->GetCurrentInventorySlot()]);
-	weaponPickup->SetWeaponProperties(*enemy->CurrentWeapon->GetWeaponProperties());
+	if (enemy->DefaultWeaponInventory.Num() > 0)
+	{
+		auto weaponPickup = GetWorld()->SpawnActor<AWeaponPickup>();
+		weaponPickup->SetActorLocation(enemy->GetActorLocation());
+		weaponPickup->SetWeapon(enemy->DefaultWeaponInventory[enemy->GetCurrentInventorySlot()]);
+		weaponPickup->SetWeaponProperties(*enemy->CurrentWeapon->GetWeaponProperties());
+	}
 }
 
 void AEnemyAIController::OnPerceptionUpdated(const TArray<AActor*>& actors)