subtr_actor/stats/analysis_graph/nodes/
movement.rs1use super::*;
2use crate::stats::calculators::*;
3use crate::*;
4
5pub 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}