Skip to main content

subtr_actor/stats/analysis_graph/nodes/
fifty_fifty.rs

1use super::*;
2use crate::stats::calculators::*;
3use crate::*;
4
5pub struct FiftyFiftyNode {
6    calculator: FiftyFiftyCalculator,
7}
8
9impl FiftyFiftyNode {
10    pub fn new() -> Self {
11        Self {
12            calculator: FiftyFiftyCalculator::new(),
13        }
14    }
15}
16
17impl Default for FiftyFiftyNode {
18    fn default() -> Self {
19        Self::new()
20    }
21}
22
23impl AnalysisNode for FiftyFiftyNode {
24    type State = FiftyFiftyCalculator;
25
26    fn name(&self) -> &'static str {
27        "fifty_fifty"
28    }
29
30    fn dependencies(&self) -> NodeDependencies {
31        vec![fifty_fifty_state_dependency()]
32    }
33
34    fn evaluate(&mut self, ctx: &AnalysisStateContext<'_>) -> SubtrActorResult<()> {
35        let fifty_fifty_state = ctx.get::<FiftyFiftyState>()?;
36        self.calculator.update(fifty_fifty_state)
37    }
38
39    fn state(&self) -> &Self::State {
40        &self.calculator
41    }
42}
43
44pub(crate) fn boxed_default() -> Box<dyn AnalysisNodeDyn> {
45    Box::new(FiftyFiftyNode::new())
46}