CSGOWTF/csgowtfd
3
0
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity

added detailed parsing of player hurt events

Browse Source
This commit is contained in:
Giovanni Harting
2021-10-11 21:37:10 +02:00
parent 3ff65bc5d7
commit 4b7851da83
28 changed files with 14660 additions and 978 deletions
Download Patch File Download Diff File Expand all files Collapse all files

120
csgo/demo_parser.go
View File

@@ -6,6 +6,7 @@ import (
"csgowtfd/ent"
"csgowtfd/ent/match"
"csgowtfd/ent/player"
"csgowtfd/ent/stats"
"fmt"
"github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs"
"github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs/common"
@@ -143,53 +144,45 @@ func (p *DemoParser) parseWorker() {
}
killMap := make(map[uint64]int, 10)
eqMap := make(map[uint64][]*struct {
Eq int
HitGroup int
Dmg uint
To uint64
})
gameStarted := false
demoParser := demoinfocs.NewParser(fDemo)
// onPlayerHurt
demoParser.RegisterEventHandler(func(e events.PlayerHurt) {
if e.Attacker == nil || e.Player == nil || !gameStarted {
if e.Attacker == nil || e.Player == nil || e.Weapon == nil || !gameStarted {
return
}
tAttacker := p.getMatchPlayerBySteamID(tStats, e.Attacker.SteamID64)
if e.Attacker.Team == e.Player.Team {
tAttacker.Extended.Dmg.Team += e.HealthDamageTaken
return
} else {
tAttacker.Extended.Dmg.Enemy += e.HealthDamageTaken
if _, ok := eqMap[e.Attacker.SteamID64]; !ok {
eqMap[e.Attacker.SteamID64] = make([]*struct {
Eq int
HitGroup int
Dmg uint
To uint64
}, 0)
}
switch e.Weapon.Type {
case common.EqDecoy:
tAttacker.Extended.Dmg.UD.Decoy += e.HealthDamageTaken
case common.EqSmoke:
tAttacker.Extended.Dmg.UD.Smoke += e.HealthDamageTaken
case common.EqHE:
tAttacker.Extended.Dmg.UD.HE += e.HealthDamageTaken
case common.EqMolotov, common.EqIncendiary:
tAttacker.Extended.Dmg.UD.Flames += e.HealthDamageTaken
case common.EqFlash:
tAttacker.Extended.Dmg.UD.Flash += e.HealthDamageTaken
found := false
for _, di := range eqMap[e.Attacker.SteamID64] {
if di.Eq == int(e.Weapon.Type) && di.HitGroup == int(e.HitGroup) {
di.Dmg += uint(e.HealthDamageTaken)
found = true
}
}
switch e.HitGroup {
case events.HitGroupHead:
tAttacker.Extended.Dmg.HitGroup.Head += e.HealthDamageTaken
case events.HitGroupChest:
tAttacker.Extended.Dmg.HitGroup.Chest += e.HealthDamageTaken
case events.HitGroupStomach:
tAttacker.Extended.Dmg.HitGroup.Stomach += e.HealthDamageTaken
case events.HitGroupLeftArm:
tAttacker.Extended.Dmg.HitGroup.LeftArm += e.HealthDamageTaken
case events.HitGroupRightArm:
tAttacker.Extended.Dmg.HitGroup.RightArm += e.HealthDamageTaken
case events.HitGroupLeftLeg:
tAttacker.Extended.Dmg.HitGroup.LeftLeg += e.HealthDamageTaken
case events.HitGroupRightLeg:
tAttacker.Extended.Dmg.HitGroup.RightLeg += e.HealthDamageTaken
case events.HitGroupGear:
tAttacker.Extended.Dmg.HitGroup.Gear += e.HealthDamageTaken
if !found {
eqMap[e.Attacker.SteamID64] = append(eqMap[e.Attacker.SteamID64], &struct {
Eq int
HitGroup int
Dmg uint
To uint64
}{Eq: int(e.Weapon.Type), HitGroup: int(e.HitGroup), Dmg: uint(e.HealthDamageTaken), To: e.Player.SteamID64})
}
})
@@ -213,13 +206,13 @@ func (p *DemoParser) parseWorker() {
switch killDiff {
case 2:
tPlayer.Extended.MultiKills.Duo++
tPlayer.Mk2++
case 3:
tPlayer.Extended.MultiKills.Triple++
tPlayer.Mk3++
case 4:
tPlayer.Extended.MultiKills.Quad++
tPlayer.Mk4++
case 5:
tPlayer.Extended.MultiKills.Pent++
tPlayer.Mk5++
}
killMap[IGP.SteamID64] = IGP.Kills()
}
@@ -237,20 +230,20 @@ func (p *DemoParser) parseWorker() {
// team flash
if e.Attacker.Team == e.Player.Team && e.Attacker.SteamID64 != e.Player.SteamID64 {
tAttacker.Extended.Flash.Total.Team++
tAttacker.Extended.Flash.Duration.Team += e.Player.FlashDuration
tAttacker.FlashTotalTeam++
tAttacker.FlashDurationTeam += e.Player.FlashDuration
}
// own flash
if e.Attacker.SteamID64 == e.Player.SteamID64 {
tAttacker.Extended.Flash.Total.Self++
tAttacker.Extended.Flash.Duration.Self += e.Player.FlashDuration
tAttacker.FlashTotalSelf++
tAttacker.FlashDurationSelf += e.Player.FlashDuration
}
// enemy flash
if e.Attacker.Team != e.Player.Team {
tAttacker.Extended.Flash.Total.Enemy++
tAttacker.Extended.Flash.Duration.Enemy += e.Player.FlashDuration
tAttacker.FlashTotalEnemy++
tAttacker.FlashDurationEnemy += e.Player.FlashDuration
}
})
@@ -263,8 +256,19 @@ func (p *DemoParser) parseWorker() {
if demoPlayer != nil && demoPlayer.SteamID64 != 0 {
tMatchPlayer := p.getMatchPlayerBySteamID(tStats, demoPlayer.SteamID64)
tMatchPlayer.Extended.Crosshair = demoPlayer.CrosshairCode()
tMatchPlayer.Extended.Color = demoPlayer.Color().String()
tMatchPlayer.Crosshair = demoPlayer.CrosshairCode()
switch demoPlayer.Color() {
case common.Yellow:
tMatchPlayer.Color = stats.ColorYellow
case common.Green:
tMatchPlayer.Color = stats.ColorGreen
case common.Purple:
tMatchPlayer.Color = stats.ColorPurple
case common.Blue:
tMatchPlayer.Color = stats.ColorBlue
case common.Orange:
tMatchPlayer.Color = stats.ColorOrange
}
}
}
})
@@ -279,8 +283,8 @@ func (p *DemoParser) parseWorker() {
if e.Player != nil && e.SteamID64() != 0 {
tMatchPlayer := p.getMatchPlayerBySteamID(tStats, e.SteamID64())
tMatchPlayer.Extended.Rank.Old = e.RankOld
tMatchPlayer.Extended.Rank.New = e.RankNew
tMatchPlayer.RankOld = e.RankOld
tMatchPlayer.RankNew = e.RankNew
}
})
@@ -300,12 +304,28 @@ func (p *DemoParser) parseWorker() {
for _, tMatchPlayer := range tStats {
p.lock.Lock()
err := tMatchPlayer.Update().SetExtended(tMatchPlayer.Extended).Exec(context.Background())
nMatchPLayer, err := tMatchPlayer.Update().
SetDmgTeam(tMatchPlayer.DmgTeam).
SetDmgEnemy(tMatchPlayer.DmgEnemy).
SetUdHe(tMatchPlayer.UdHe).
SetUdFlash(tMatchPlayer.UdFlash).
SetUdSmoke(tMatchPlayer.UdSmoke).
SetUdFlames(tMatchPlayer.UdFlames).
Save(context.Background())
p.lock.Unlock()
if err != nil {
log.Errorf("[DP] Unable to update player %d in database: %v", tMatchPlayer.Edges.Players.ID, err)
continue
}
for _, eqDmg := range eqMap[tMatchPlayer.PlayerStats] {
p.lock.Lock()
err := p.db.WeaponStats.Create().SetStat(nMatchPLayer).SetDmg(eqDmg.Dmg).SetVictim(eqDmg.To).SetHitGroup(eqDmg.HitGroup).SetEqType(eqDmg.Eq).Exec(context.Background())
p.lock.Unlock()
if err != nil {
log.Errorf("[DP] Unable to create WeaponStat: %v", err)
}
}
}
log.Infof("[DP] Parsed %d (took %s/%s)", demo.MatchId, downloadTime, time.Now().Sub(startTime))