Skip to content

Commit 2889cdb

Browse files
committed
Fix wrong EquipmentElement.EquipmentClass() values
Values were off by one as Go would round down instead of up Fixes #19
1 parent b9bb35e commit 2889cdb

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

common/constants.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,12 @@ const (
143143
EqHE EquipmentElement = 506
144144
)
145145

146+
// Class returns the class of the equipment.
147+
// E.g. pistol, smg, heavy etc.
148+
func (e EquipmentElement) Class() EquipmentClass {
149+
return EquipmentClass((int(e) + 99) / 100)
150+
}
151+
146152
// EquipmentClass constants give information about the type of an equipment (SMG, Rifle, Grenade etc.).
147153
// Note: EquipmentElement / 100 = EquipmentClass
148154
const (

common/structs.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ type Equipment struct {
103103
// Class returns the class of the equipment.
104104
// E.g. pistol, smg, heavy etc.
105105
func (e Equipment) Class() EquipmentClass {
106-
return EquipmentClass(int(e.Weapon) / 100)
106+
return e.Weapon.Class()
107107
}
108108

109109
// NewEquipment is a wrapper for NewSkinEquipment to create weapons without skins.

0 commit comments

Comments
 (0)