common: add new ammo functions to Equipment for grenades

see #145

- AmmoInMagazine2() returns 1 for grenades
- AmmoReserve2() returns 'Owner.LeftAmmo[AmmoType] - 1' for grenades

Author: markus-wa

common/equipment.go

@@ -274,7 +274,7 @@ type Equipment struct {
 	Weapon         EquipmentElement // The type of weapon which the equipment instantiates.
 	Owner          *Player          // The player carrying the equipment, not necessarily the buyer.
 	AmmoType       int              // TODO: Remove this? doesn't seem applicable to CS:GO
-	AmmoInMagazine int              // Amount of bullets in the weapon's magazine
+	AmmoInMagazine int              // Amount of bullets in the weapon's magazine. Deprecated, use AmmoInMagazine2() instead.
 	AmmoReserve    int              // Amount of reserve bullets
 	OriginalString string           // E.g. 'models/weapons/w_rif_m4a1_s.mdl'. Used internally to differentiate alternative weapons (M4A4 / M4A1-S etc.).
 	ZoomLevel      int              // How far the player has zoomed in on the weapon. 0=no zoom, 1=first level, 2=maximum zoom
@@ -295,6 +295,32 @@ func (e Equipment) UniqueID() int64 {
 	return e.uniqueID
 }
 
+// AmmoInMagazine2 returns the ammo left in the magazine.
+// Returns CWeaponCSBase.m_iClip1 for most weapons and 1 for grenades.
+func (e Equipment) AmmoInMagazine2() int {
+	if e.Class() == EqClassGrenade {
+		return 1
+	}
+
+	return e.AmmoInMagazine
+}
+
+// AmmoReserve2 returns the ammo left available for reloading.
+// Returns CWeaponCSBase.m_iPrimaryReserveAmmoCount for most weapons and 'Owner.AmmoLeft[AmmoType] - 1' for grenades.
+// Use AmmoInMagazine2() + AmmoReserve2() to quickly get the amount of grenades a player owns.
+func (e Equipment) AmmoReserve2() int {
+	if e.Class() == EqClassGrenade {
+		if e.Owner != nil {
+			// minus one for 'InMagazine'
+			return e.Owner.AmmoLeft[e.AmmoType] - 1
+		}
+
+		return 0
+	}
+
+	return e.AmmoReserve
+}
+
 // NewEquipment creates a new Equipment and sets the UniqueID.
 //
 // Intended for internal use only.

common/equipment_test.go

@@ -35,3 +35,46 @@ func TestEquipment_Class(t *testing.T) {
 func TestEquipment_UniqueID(t *testing.T) {
 	assert.NotEqual(t, NewEquipment(EqAK47).UniqueID(), NewEquipment(EqAK47).UniqueID(), "UniqueIDs of different equipment instances should be different")
 }
+
+func TestEquipment_AmmoInMagazine2_Default(t *testing.T) {
+	wep := &Equipment{AmmoInMagazine: 1}
+
+	assert.Equal(t, 1, wep.AmmoInMagazine2())
+}
+
+func TestEquipment_AmmoInMagazine2_Grenade(t *testing.T) {
+	wep := &Equipment{
+		Weapon: EqFlash,
+	}
+
+	assert.Equal(t, 1, wep.AmmoInMagazine2())
+}
+
+func TestEquipment_AmmoReserve2_Default(t *testing.T) {
+	wep := &Equipment{
+		AmmoReserve: 1,
+	}
+
+	assert.Equal(t, 1, wep.AmmoReserve2())
+}
+
+func TestEquipment_AmmoReserve2_Grenade(t *testing.T) {
+	owner := new(Player)
+	owner.AmmoLeft[1] = 2
+	wep := &Equipment{
+		Weapon:         EqFlash,
+		AmmoInMagazine: -1,
+		Owner:          owner,
+		AmmoType:       1,
+	}
+
+	assert.Equal(t, 1, wep.AmmoReserve2())
+}
+
+func TestEquipment_AmmoReserve2_Grenade_OwnerNil(t *testing.T) {
+	wep := &Equipment{
+		Weapon: EqFlash,
+	}
+
+	assert.Equal(t, 0, wep.AmmoReserve2())
+}

common/player.go

@@ -4,6 +4,7 @@ import (
 	"time"
 
 	"github.com/golang/geo/r3"
+
 	st "github.com/markus-wa/demoinfocs-golang/sendtables"
 )
 
@@ -26,7 +27,7 @@ type Player struct {
 	RoundStartEquipmentValue    int
 	ActiveWeaponID              int                // Used internally to set the active weapon, see ActiveWeapon()
 	RawWeapons                  map[int]*Equipment // All weapons the player is currently carrying
-	AmmoLeft                    [32]int            // Ammo left in the various weapons, index corresponds to key of RawWeapons
+	AmmoLeft                    [32]int            // Ammo left for special weapons (e.g. grenades), index corresponds Equipment.AmmoType
 	Entity                      st.IEntity
 	AdditionalPlayerInformation *AdditionalPlayerInformation // Mostly scoreboard information such as kills, deaths, etc.
 	ViewDirectionX              float32