Weapon Selection Modifiers

src/g_game.cpp

@@ -315,7 +315,7 @@ CCMD (slot)
 		if (slot < NUM_WEAPON_SLOTS && mo)
 		{
 			// Needs to be redone
-			IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, PickWeapon)
+			IFVM(PlayerPawn, FindWeapon)
 			{
 				VMValue param[] = { mo, slot, !(dmflags2 & DF2_DONTCHECKAMMO) };
 				VMReturn ret((void**)&SendItemUse);
@@ -367,7 +367,7 @@ CCMD (weapnext)
 	if (mo)
 	{
 		// Needs to be redone
-		IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, PickNextWeapon)
+		IFVM(PlayerPawn, FindNextWeapon)
 		{
 			VMValue param[] = { mo };
 			VMReturn ret((void**)&SendItemUse);
@@ -394,7 +394,7 @@ CCMD (weapprev)
 	if (mo)
 	{
 		// Needs to be redone
-		IFVIRTUALPTRNAME(mo, NAME_PlayerPawn, PickPrevWeapon)
+		IFVM(PlayerPawn, FindPrevWeapon)
 		{
 			VMValue param[] = { mo };
 			VMReturn ret((void**)&SendItemUse);

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;
+	}
 
 }
 

wadsrc/static/zscript/actors/player/player.zs

@@ -2334,6 +2334,31 @@ 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;
+	}
+
+	virtual Weapon PostFindWeapon(int slot, bool checkammo, Weapon orig, Weapon mod)
+	{
+		return mod;
+	}
+
+	Weapon FindWeapon(int slot, bool checkammo)
+	{
+		let orig = PickWeapon(slot, checkammo);
+		let mod = CallModifyPickWeapon(slot, checkammo, orig);
+		return PostFindWeapon(slot, checkammo, orig, mod);
+	}
+
 	//===========================================================================
 	//
 	// FindMostRecentWeapon
@@ -2438,6 +2463,31 @@ 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;
+	}
+
+	virtual Weapon PostFindNextWeapon(Weapon orig, Weapon mod)
+	{
+		return mod;
+	}
+
+	Weapon FindNextWeapon()
+	{
+		let orig = PickNextWeapon();
+		let mod = CallModifyPickNextWeapon(orig);
+		return PostFindNextWeapon(orig, mod);
+	}
+
 	//===========================================================================
 	//
 	// FWeaponSlots :: PickPrevWeapon
@@ -2491,6 +2541,31 @@ 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;
+	}
+
+	virtual Weapon PostFindPrevWeapon(Weapon orig, Weapon mod)
+	{
+		return mod;
+	}
+
+	Weapon FindPrevWeapon()
+	{
+		let orig = PickPrevWeapon();
+		let mod = CallModifyPickPrevWeapon(orig);
+		return PostFindPrevWeapon(orig, mod);
+	}
+
 	//============================================================================
 	//
 	// P_BobWeapon