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