subtr_actor/stats/analysis_graph/
collector.rs

1use super::graph::AnalysisGraph;
2use crate::stats::calculators::FrameInput;
3use crate::*;
4
5#[allow(dead_code)]
6pub struct AnalysisNodeCollector {
7    graph: AnalysisGraph,
8    last_sample_time: Option<f32>,
9    last_replay_meta_player_count: Option<usize>,
10    last_demolish_count: usize,
11    last_boost_pad_event_count: usize,
12    last_touch_event_count: usize,
13    last_player_stat_event_count: usize,
14    last_goal_event_count: usize,
15}
16
17#[allow(dead_code)]
18impl AnalysisNodeCollector {
19    pub fn new(mut graph: AnalysisGraph) -> Self {
20        graph.register_input_state::<FrameInput>();
21        Self {
22            graph,
23            last_sample_time: None,
24            last_replay_meta_player_count: None,
25            last_demolish_count: 0,
26            last_boost_pad_event_count: 0,
27            last_touch_event_count: 0,
28            last_player_stat_event_count: 0,
29            last_goal_event_count: 0,
30        }
31    }
32
33    pub fn graph(&self) -> &AnalysisGraph {
34        &self.graph
35    }
36
37    pub fn graph_mut(&mut self) -> &mut AnalysisGraph {
38        &mut self.graph
39    }
40
41    pub fn into_graph(self) -> AnalysisGraph {
42        self.graph
43    }
44}
45
46impl Collector for AnalysisNodeCollector {
47    fn process_frame(
48        &mut self,
49        processor: &ReplayProcessor,
50        _frame: &boxcars::Frame,
51        frame_number: usize,
52        current_time: f32,
53    ) -> SubtrActorResult<TimeAdvance> {
54        let player_count = processor.player_count();
55        if self.last_replay_meta_player_count != Some(player_count) {
56            self.graph.on_replay_meta(&processor.get_replay_meta()?)?;
57            self.last_replay_meta_player_count = Some(player_count);
58        }
59
60        let dt = self
61            .last_sample_time
62            .map(|last_time| (current_time - last_time).max(0.0))
63            .unwrap_or(0.0);
64        let frame_input = FrameInput::aggregate(
65            processor,
66            frame_number,
67            current_time,
68            dt,
69            self.last_demolish_count,
70            self.last_boost_pad_event_count,
71            self.last_touch_event_count,
72            self.last_player_stat_event_count,
73            self.last_goal_event_count,
74        );
75        self.graph.evaluate_with_state(&frame_input)?;
76        self.last_sample_time = Some(current_time);
77        self.last_demolish_count = processor.demolishes.len();
78        self.last_boost_pad_event_count = processor.boost_pad_events.len();
79        self.last_touch_event_count = processor.touch_events.len();
80        self.last_player_stat_event_count = processor.player_stat_events.len();
81        self.last_goal_event_count = processor.goal_events.len();
82
83        Ok(TimeAdvance::NextFrame)
84    }
85
86    fn finish_replay(&mut self, _processor: &ReplayProcessor) -> SubtrActorResult<()> {
87        self.graph.finish()
88    }
89}
subtr_actor/stats/analysis_graph/collector.rs

subtr_actor/stats/analysis_graph/
collector.rs
