Skip to main content

subtr_actor/stats/analysis_graph/nodes/
movement.rs

1use super::*;
2use crate::stats::calculators::*;
3use crate::*;
4
5/// Tracks per-player movement classification/stats from player and vertical state during live play.
6pub struct MovementNode {
7    calculator: MovementCalculator,
8}
9
10impl MovementNode {
11    pub fn new() -> Self {
12        Self {
13            calculator: MovementCalculator::new(),
14        }
15    }
16}
17
18impl Default for MovementNode {
19    fn default() -> Self {
20        Self::new()
21    }
22}
23
24impl AnalysisNode for MovementNode {
25    type State = MovementCalculator;
26
27    fn name(&self) -> &'static str {
28        "movement"
29    }
30
31    fn emitted_events(&self) -> &'static [crate::stats::calculators::EmittedEvent] {
32        crate::stats::calculators::MOVEMENT_EMITTED_EVENTS
33    }
34
35    fn dependencies(&self) -> NodeDependencies {
36        vec![
37            frame_info_dependency(),
38            player_frame_state_dependency(),
39            player_vertical_state_dependency(),
40            live_play_dependency(),
41        ]
42    }
43
44    fn evaluate(&mut self, ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
45        let live_play_state = ctx.get::<LivePlayState>()?;
46        self.calculator.update(
47            ctx.get::<FrameInfo>()?,
48            ctx.get::<PlayerFrameState>()?,
49            ctx.get::<PlayerVerticalState>()?,
50            live_play_state,
51        )
52    }
53
54    fn finish(&mut self, _ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
55        self.calculator.flush_pending_events();
56        Ok(())
57    }
58
59    fn state(&self) -> &Self::State {
60        &self.calculator
61    }
62}
63
64pub(crate) fn boxed_default() -> Box<dyn AnalysisNodeDyn> {
65    Box::new(MovementNode::new())
66}