Skip to main content

subtr_actor/stats/analysis_graph/nodes/
touch.rs

1use super::*;
2use crate::stats::calculators::*;
3use crate::*;
4
5pub struct TouchNode {
6    calculator: TouchCalculator,
7}
8
9impl TouchNode {
10    pub fn new() -> Self {
11        Self {
12            calculator: TouchCalculator::new(),
13        }
14    }
15}
16
17impl Default for TouchNode {
18    fn default() -> Self {
19        Self::new()
20    }
21}
22
23impl AnalysisNode for TouchNode {
24    type State = TouchCalculator;
25
26    fn name(&self) -> &'static str {
27        "touch"
28    }
29
30    fn dependencies(&self) -> NodeDependencies {
31        vec![
32            frame_info_dependency(),
33            ball_frame_state_dependency(),
34            player_frame_state_dependency(),
35            player_vertical_state_dependency(),
36            rotation_dependency(),
37            touch_state_dependency(),
38            possession_state_dependency(),
39            fifty_fifty_state_dependency(),
40            frame_events_state_dependency(),
41            live_play_dependency(),
42        ]
43    }
44
45    fn evaluate(&mut self, ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
46        let touch_state = ctx.get::<TouchState>()?;
47        self.calculator.update(
48            ctx.get::<FrameInfo>()?,
49            ctx.get::<BallFrameState>()?,
50            ctx.get::<PlayerFrameState>()?,
51            ctx.get::<PlayerVerticalState>()?,
52            ctx.get::<RotationCalculator>()?,
53            touch_state,
54            ctx.get::<PossessionState>()?,
55            ctx.get::<FiftyFiftyState>()?,
56            ctx.get::<FrameEventsState>()?,
57            ctx.get::<LivePlayState>()?,
58        )
59    }
60
61    fn finish(&mut self, _ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
62        self.calculator.finish();
63        Ok(())
64    }
65
66    fn state(&self) -> &Self::State {
67        &self.calculator
68    }
69}
70
71pub(crate) fn boxed_default() -> Box<dyn AnalysisNodeDyn> {
72    Box::new(TouchNode::new())
73}