Add inventory virtuals for weapon selection

RicardoLuis0 · Oct 21, 2024 · 1bb560a · 1bb560a
1 parent d0e6bce
commit 1bb560a
Show file tree

Hide file tree

Showing 3 changed files with 71 additions and 4 deletions.
diff --git a/src/g_game.cpp b/src/g_game.cpp
@@ -313,7 +313,7 @@ CCMD (slot)
 		if (slot < NUM_WEAPON_SLOTS && mo)
 		{
 			// Needs to be redone
-			IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, PickWeapon)
+			IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, FindWeapon)
 			{
 				VMValue param[] = { mo, slot, !(dmflags2 & DF2_DONTCHECKAMMO) };
 				VMReturn ret((void**)&SendItemUse);
@@ -365,7 +365,7 @@ CCMD (weapnext)
 	if (mo)
 	{
 		// Needs to be redone
-		IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, PickNextWeapon)
+		IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, FindNextWeapon)
 		{
 			VMValue param[] = { mo };
 			VMReturn ret((void**)&SendItemUse);
@@ -392,7 +392,7 @@ CCMD (weapprev)
 	if (mo)
 	{
 		// Needs to be redone
-		IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, PickPrevWeapon)
+		IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, FindPrevWeapon)
 		{
 			VMValue param[] = { mo };
 			VMReturn ret((void**)&SendItemUse);

diff --git a/wadsrc/static/zscript/actors/inventory/inventory.zs b/wadsrc/static/zscript/actors/inventory/inventory.zs
@@ -1366,7 +1366,20 @@ class Inventory : Actor
 		}
 	}
 
-
+	virtual Weapon ModifyPickWeapon(int slot, bool checkammo, Weapon originalPick)
+	{
+		return originalPick;
+	}
+
+	virtual Weapon ModifyPickNextWeapon(Weapon originalPick)
+	{
+		return originalPick;
+	}
+
+	virtual Weapon ModifyPickPrevWeapon(Weapon originalPick)
+	{
+		return originalPick;
+	}
 
 }
 

diff --git a/wadsrc/static/zscript/actors/player/player.zs b/wadsrc/static/zscript/actors/player/player.zs
@@ -2334,6 +2334,24 @@ class PlayerPawn : Actor
 		return ReadyWeapon;
 	}
 
+	Weapon CallModifyPickWeapon(int slot, bool checkammo, Weapon pick)
+	{
+		let cur = inv;
+
+		if(cur) do
+		{
+			pick = cur.ModifyPickWeapon(slot, checkammo, pick);
+		}
+		while(cur = cur.inv)
+
+		return pick;
+	}
+
+	Weapon FindWeapon(int slot, bool checkammo)
+	{
+		return CallModifyPickWeapon(slot, checkammo, PickWeapon(slot, checkammo));
+	}
+
 	//===========================================================================
 	//
 	// FindMostRecentWeapon
@@ -2438,6 +2456,24 @@ class PlayerPawn : Actor
 		return ReadyWeapon;
 	}
 
+	Weapon CallModifyPickNextWeapon(Weapon pick)
+	{
+		let cur = inv;
+
+		if(cur) do
+		{
+			pick = cur.ModifyPickNextWeapon(pick);
+		}
+		while(cur = cur.inv)
+
+		return pick;
+	}
+
+	Weapon FindNextWeapon()
+	{
+		return CallModifyPickNextWeapon(PickNextWeapon());
+	}
+
 	//===========================================================================
 	//
 	// FWeaponSlots :: PickPrevWeapon
@@ -2491,6 +2527,24 @@ class PlayerPawn : Actor
 		return player.ReadyWeapon;
 	}
 
+	Weapon CallModifyPickPrevWeapon(Weapon pick)
+	{
+		let cur = inv;
+
+		if(cur) do
+		{
+			pick = cur.ModifyPickPrevWeapon(pick);
+		}
+		while(cur = cur.inv)
+
+		return pick;
+	}
+
+	Weapon FindPrevWeapon()
+	{
+		return CallModifyPickPrevWeapon(PickPrevWeapon());
+	}
+
 	//============================================================================
 	//
 	// P_BobWeapon