subtr_actor/stats/analysis_graph/nodes/
possession.rs

1use super::*;
2use crate::stats::calculators::*;
3use crate::*;
4
5pub struct PossessionNode {
6    calculator: PossessionCalculator,
7}
8
9impl PossessionNode {
10    pub fn new() -> Self {
11        Self {
12            calculator: PossessionCalculator::new(),
13        }
14    }
15}
16
17impl Default for PossessionNode {
18    fn default() -> Self {
19        Self::new()
20    }
21}
22
23impl AnalysisNode for PossessionNode {
24    type State = PossessionCalculator;
25
26    fn name(&self) -> &'static str {
27        "possession"
28    }
29
30    fn dependencies(&self) -> NodeDependencies {
31        vec![
32            frame_info_dependency(),
33            ball_frame_state_dependency(),
34            possession_state_dependency(),
35            live_play_dependency(),
36        ]
37    }
38
39    fn evaluate(&mut self, ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
40        self.calculator.update(
41            ctx.get::<FrameInfo>()?,
42            ctx.get::<BallFrameState>()?,
43            ctx.get::<PossessionState>()?,
44            ctx.get::<LivePlayState>()?,
45        )
46    }
47
48    fn finish(&mut self, _ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
49        self.calculator.flush_pending_event();
50        Ok(())
51    }
52
53    fn state(&self) -> &Self::State {
54        &self.calculator
55    }
56}
57
58pub(crate) fn boxed_default() -> Box<dyn AnalysisNodeDyn> {
59    Box::new(PossessionNode::new())
60}
subtr_actor/stats/analysis_graph/nodes/possession.rs

subtr_actor/stats/analysis_graph/nodes/
possession.rs
