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