common: try to improve accuracy of Player.IsDucking()

Author: markus-wa

pkg/demoinfocs/common/player.go

@@ -187,8 +187,9 @@ func (p *Player) IsScoped() bool {
 }
 
 // IsDucking returns true if the player is currently crouching.
+// See also: Flags().Ducking() & Flags().DuckingKeyPressed()
 func (p *Player) IsDucking() bool {
-	return getBool(p.Entity, "localdata.m_Local.m_bDucking")
+	return p.Flags().Ducking() && p.Flags().DuckingKeyPressed()
 }
 
 // HasDefuseKit returns true if the player currently has a defuse kit in his inventory.
@@ -284,6 +285,51 @@ func (p *Player) Velocity() r3.Vector {
 	}
 }
 
+// see https://github.com/ValveSoftware/source-sdk-2013/blob/master/mp/src/public/const.h#L146-L188
+const (
+	flOnGround = 1 << iota
+	flDucking
+	flAnimDucking
+)
+
+// PlayerFlags wraps m_fFlags and provides accessors for the various known flags a player may have set.
+type PlayerFlags uint32
+
+func (pf PlayerFlags) Get(f PlayerFlags) bool {
+	return pf&f != 0
+}
+
+// OnGround returns true if the player is touching the ground.
+// See m_fFlags FL_ONGROUND https://github.com/ValveSoftware/source-sdk-2013/blob/master/mp/src/public/const.h#L146-L188
+func (pf PlayerFlags) OnGround() bool {
+	return pf.Get(flOnGround)
+}
+
+// Ducking returns true if the player is/was fully crouched.
+//   Fully ducked: Ducking() && DuckingKeyPressed()
+//   Previously fully ducked, unducking in progress: Ducking() && !DuckingKeyPressed()
+//   Fully unducked: !Ducking() && !DuckingKeyPressed()
+//   Previously fully unducked, ducking in progress: !Ducking() && DuckingKeyPressed()
+// See m_fFlags FL_DUCKING https://github.com/ValveSoftware/source-sdk-2013/blob/master/mp/src/public/const.h#L146-L188
+func (pf PlayerFlags) Ducking() bool {
+	return pf.Get(flDucking)
+}
+
+// DuckingKeyPressed returns true if the player is holding the crouch key pressed.
+//   Fully ducked: Ducking() && DuckingKeyPressed()
+//   Previously fully ducked, unducking in progress: Ducking() && !DuckingKeyPressed()
+//   Fully unducked: !Ducking() && !DuckingKeyPressed()
+//   Previously fully unducked, ducking in progress: !Ducking() && DuckingKeyPressed()
+// See m_fFlags FL_ANIMDUCKING https://github.com/ValveSoftware/source-sdk-2013/blob/master/mp/src/public/const.h#L146-L188
+func (pf PlayerFlags) DuckingKeyPressed() bool {
+	return pf.Get(flAnimDucking)
+}
+
+// Flags returns flags currently set on m_fFlags.
+func (p *Player) Flags() PlayerFlags {
+	return PlayerFlags(getInt(p.Entity, "m_fFlags"))
+}
+
 /////////////////////////////
 // CCSPlayerResource stuff //
 /////////////////////////////

pkg/demoinfocs/common/player_test.go

@@ -209,15 +209,33 @@ func TestPlayer_IsAirborne(t *testing.T) {
 }
 
 func TestPlayer_IsDucking(t *testing.T) {
-	pl := playerWithProperty("localdata.m_Local.m_bDucking", st.PropertyValue{IntVal: 0})
+	pl := playerWithProperty("m_fFlags", st.PropertyValue{IntVal: 0})
 
 	assert.False(t, pl.IsDucking())
 
-	pl = playerWithProperty("localdata.m_Local.m_bDucking", st.PropertyValue{IntVal: 1})
+	pl = playerWithProperty("m_fFlags", st.PropertyValue{IntVal: 1 << 1})
+
+	assert.False(t, pl.IsDucking())
+
+	pl = playerWithProperty("m_fFlags", st.PropertyValue{IntVal: 1 << 2})
+
+	assert.False(t, pl.IsDucking())
+
+	pl = playerWithProperty("m_fFlags", st.PropertyValue{IntVal: 1<<1 | 1<<2})
 
 	assert.True(t, pl.IsDucking())
 }
 
+func TestPlayerFlags_OnGround(t *testing.T) {
+	pl := playerWithProperty("m_fFlags", st.PropertyValue{IntVal: 0})
+
+	assert.False(t, pl.Flags().OnGround())
+
+	pl = playerWithProperty("m_fFlags", st.PropertyValue{IntVal: 1})
+
+	assert.True(t, pl.Flags().OnGround())
+}
+
 func TestPlayer_HasDefuseKit(t *testing.T) {
 	pl := playerWithProperty("m_bHasDefuser", st.PropertyValue{IntVal: 0})