leon/knifer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch to using lazy parser

Browse Source
This commit is contained in:
Lol3rrr
2024-10-18 17:37:02 +02:00
parent 6c098b412a
commit 201cead3f1
4 changed files with 240 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.lock generated
View File

@@ -695,7 +695,7 @@ dependencies = [
[[package]]
name = "csdemo"
version = "0.1.0"
source = "git+https://github.com/Lol3rrr/csdemo.git#a0eab7c68f14e4e846823d0468ef196cf4c85fa5"
source = "git+https://github.com/Lol3rrr/csdemo.git#4b1cb505667fd024174ba91fa19f9dd7a410fd30"
dependencies = [
"bitter",
"phf",

28
analysis/src/endofgame.rs
View File

@@ -33,13 +33,10 @@ pub struct PlayerStats {
pub fn parse(buf: &[u8]) -> Result<EndOfGame, ()> {
let tmp = csdemo::Container::parse(buf).map_err(|e| ())?;
let output = csdemo::parser::parse(
csdemo::FrameIterator::parse(tmp.inner),
csdemo::parser::EntityFilter::all(),
)
.map_err(|e| ())?;
let output = csdemo::lazyparser::LazyParser::new(tmp);
let header = &output.header;
let header = output.file_header().ok_or(())?;
let player_info = output.player_info();
let mut player_stats = std::collections::HashMap::<_, PlayerStats>::new();
let mut pawn_to_player =
@@ -47,7 +44,7 @@ pub fn parse(buf: &[u8]) -> Result<EndOfGame, ()> {
let mut track = false;
let mut player_life = std::collections::HashMap::<_, u8>::new();
for event in output.events.iter() {
for event in output.events().filter_map(|e| e.ok()) {
match event {
csdemo::DemoEvent::GameEvent(gevent) => {
match gevent.as_ref() {
@@ -84,15 +81,10 @@ pub fn parse(buf: &[u8]) -> Result<EndOfGame, ()> {
track = false;
}
csdemo::game_event::GameEvent::PlayerDeath(pdeath) if track => {
player_death(pdeath, &output.player_info, &mut player_stats);
player_death(pdeath, &player_info, &mut player_stats);
}
csdemo::game_event::GameEvent::PlayerHurt(phurt) if track => {
player_hurt(
phurt,
&output.player_info,
&mut player_stats,
&mut player_life,
);
player_hurt(phurt, &player_info, &mut player_stats, &mut player_life);
}
_ => {}
};
@@ -104,8 +96,7 @@ pub fn parse(buf: &[u8]) -> Result<EndOfGame, ()> {
let mut teams = std::collections::HashMap::<i32, TeamInfo>::new();
let mut entity_to_team = std::collections::HashMap::new();
for tick_state in output.entity_states.ticks {
for state in tick_state.states {
for (tick, state) in output.entities().filter_map(|e| e.ok()) {
let team = match csdemo::structured::ccsteam::CCSTeam::try_from(&state) {
Ok(t) => t,
Err(_) => continue,
@@ -130,7 +121,7 @@ pub fn parse(buf: &[u8]) -> Result<EndOfGame, ()> {
let team_number = player_ids
.iter()
.filter_map(|p| output.player_info.get(*p).map(|p| p.team))
.filter_map(|p| player_info.get(*p).map(|p| p.team))
.next()
.unwrap();
@@ -147,12 +138,11 @@ pub fn parse(buf: &[u8]) -> Result<EndOfGame, ()> {
team_entry.end_score = score as usize;
}
}
}
let mut players: Vec<_> = player_stats
.into_iter()
.filter_map(|(id, stats)| {
let player = output.player_info.get(&id)?;
let player = player_info.get(&id)?;
Some((
PlayerInfo {

39
analysis/src/heatmap.rs
View File

@@ -79,16 +79,13 @@ impl From<u32> for PawnID {
pub fn parse(config: &Config, buf: &[u8]) -> Result<HeatMapOutput, ()> {
let tmp = csdemo::Container::parse(buf).map_err(|e| ())?;
let output = csdemo::parser::parse(
csdemo::FrameIterator::parse(tmp.inner),
csdemo::parser::EntityFilter::all(),
)
.map_err(|e| ())?;
let output = csdemo::lazyparser::LazyParser::new(tmp);
let pawn_ids = {
let mut tmp = std::collections::HashMap::<PawnID, _>::new();
for event in output.events.iter() {
for event in output.events().filter_map(|e| e.ok()) {
let entry = match event {
csdemo::DemoEvent::GameEvent(ge) => match ge.as_ref() {
csdemo::game_event::GameEvent::PlayerSpawn(pspawn) => {
@@ -120,12 +117,12 @@ pub fn parse(config: &Config, buf: &[u8]) -> Result<HeatMapOutput, ()> {
let mut player_cells = std::collections::HashMap::new();
let mut heatmaps = std::collections::HashMap::new();
for tick_state in output.entity_states.ticks.iter() {
let _tracing_guard = tracing::debug_span!("Tick", tick=?tick_state.tick).entered();
for (tick, state) in output.entities().filter_map(|s| s.ok()) {
let _tracing_guard = tracing::debug_span!("Tick", ?tick).entered();
process_tick(
config,
tick_state,
&state,
&pawn_ids,
&mut teams,
&mut player_lifestate,
@@ -139,7 +136,7 @@ pub fn parse(config: &Config, buf: &[u8]) -> Result<HeatMapOutput, ()> {
Ok(HeatMapOutput {
player_heatmaps: heatmaps,
player_info: output.player_info,
player_info: output.player_info(),
})
}
@@ -147,7 +144,7 @@ pub const MAX_COORD: f32 = (1 << 14) as f32;
fn process_tick(
config: &Config,
tick_state: &csdemo::parser::EntityTickStates,
entity_state: &csdemo::parser::entities::EntityState,
pawn_ids: &std::collections::HashMap<PawnID, csdemo::UserId>,
teams: &mut std::collections::HashMap<PawnID, String>,
player_lifestate: &mut std::collections::HashMap<csdemo::UserId, u32>,
@@ -155,11 +152,10 @@ fn process_tick(
player_cells: &mut std::collections::HashMap<csdemo::UserId, (u32, u32, u32)>,
heatmaps: &mut std::collections::HashMap<(csdemo::UserId, String), HeatMap>,
) {
for entity_state in tick_state
.states
.iter()
.filter(|s| matches!(s.class.as_ref(), "CCSPlayerPawn" | "CCSTeam"))
{
if !matches!(entity_state.class.as_ref(), "CCSPlayerPawn" | "CCSTeam") {
return;
}
if entity_state.class.as_ref() == "CCSTeam" {
let raw_team_name = match entity_state
.get_prop("CCSTeam.m_szTeamname")
@@ -170,7 +166,7 @@ fn process_tick(
.flatten()
{
Some(n) => n,
None => continue,
None => return,
};
for prop in entity_state
@@ -182,17 +178,17 @@ fn process_tick(
teams.insert(prop, raw_team_name.clone());
}
continue;
return;
}
let pawn_id = PawnID::from(entity_state.id);
let user_id = match pawn_ids.get(&pawn_id).cloned() {
Some(id) => id,
None => continue,
None => return,
};
let team = match teams.get(&pawn_id).cloned() {
Some(t) => t,
None => continue,
None => return,
};
let _inner_guard = tracing::trace_span!("Entity", entity_id=?entity_state.id).entered();
@@ -300,7 +296,7 @@ fn process_tick(
// 0 means alive
if lifestate != 0 {
continue;
return;
}
// tracing::trace!("Coord (X, Y, Z): {:?} -> {:?}", (x_coord, y_coord, z_coord), (x_cell, y_cell));
@@ -310,7 +306,6 @@ fn process_tick(
.or_insert(HeatMap::new(config.cell_size));
heatmap.increment(x_cell, y_cell);
}
}
impl core::fmt::Display for HeatMap {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {

23
analysis/src/perround.rs
View File

@@ -70,15 +70,13 @@ pub struct PerRound {
pub fn parse(buf: &[u8]) -> Result<PerRound, ()> {
let tmp = csdemo::Container::parse(buf).map_err(|e| ())?;
let output = csdemo::parser::parse(
csdemo::FrameIterator::parse(tmp.inner),
csdemo::parser::EntityFilter::all(),
)
.map_err(|e| ())?;
let output = csdemo::lazyparser::LazyParser::new(tmp);
let player_info = output.player_info();
let mut rounds: Vec<Round> = Vec::new();
for tick in output.entity_states.ticks.iter() {
for state in tick.states.iter() {
for (tick, state) in output.entities().filter_map(|e| e.ok()) {
let round_start_count = state
.get_prop("CCSGameRulesProxy.CCSGameRules.m_nRoundStartCount")
.map(|v| v.value.as_u32())
@@ -87,7 +85,7 @@ pub fn parse(buf: &[u8]) -> Result<PerRound, ()> {
if rounds.len() < (round_start_count - 1) as usize {
rounds.push(Round {
winreason: WinReason::StillInProgress,
start: tick.tick,
start: tick,
end: u32::MAX,
events: Vec::new(),
});
@@ -100,7 +98,7 @@ pub fn parse(buf: &[u8]) -> Result<PerRound, ()> {
.flatten();
if let Some(round_end_count) = round_end_count {
if rounds.len() == (round_end_count - 1) as usize {
rounds.last_mut().unwrap().end = tick.tick;
rounds.last_mut().unwrap().end = tick;
}
}
@@ -117,13 +115,12 @@ pub fn parse(buf: &[u8]) -> Result<PerRound, ()> {
}
}
}
}
let mut rounds_iter = rounds.iter_mut();
let mut current_tick = 0;
let mut current_round = rounds_iter.next().unwrap();
'events: for event in output.events.iter() {
'events: for event in output.events().filter_map(|e| e.ok()) {
match event {
csdemo::DemoEvent::Tick(tick) => {
current_tick = tick.tick();
@@ -154,8 +151,8 @@ pub fn parse(buf: &[u8]) -> Result<PerRound, ()> {
None => died.clone(),
};
let died_player = output.player_info.get(&died).unwrap();
let attacker_player = output.player_info.get(&attacker).unwrap();
let died_player = player_info.get(&died).unwrap();
let attacker_player = player_info.get(&attacker).unwrap();
RoundEvent::Kill {
attacker: attacker_player.xuid,