stock-trek 0.8.2

Stock Trek time-series analysis
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use crate::{
    conditions::condition::{Condition, ConditionTrait},
    error::{
        result::{StockTrekError, StockTrekResult},
        stats::StatsError,
    },
    resolved_context::ResolvedContext,
    util::serde_ordering,
    values::value::NumberValue,
};
use serde::{Deserialize, Serialize};
use std::cmp::Ordering;

#[derive(Serialize, Deserialize)]
pub struct CompareCondition {
    left: NumberValue,
    #[serde(with = "serde_ordering")]
    comparison: Ordering,
    right: NumberValue,
}

impl CompareCondition {
    pub fn new(left: NumberValue, comparison: Ordering, right: NumberValue) -> Condition {
        Box::new(Self {
            left,
            comparison,
            right,
        })
    }
}

#[typetag::serde]
impl ConditionTrait for CompareCondition {
    fn test(&self, c: &ResolvedContext) -> StockTrekResult<bool> {
        let left_value = self.left.number(c)?;
        let right_value = self.right.number(c)?;
        match left_value.partial_cmp(&right_value) {
            Some(Ordering::Less) => Ok(self.comparison.is_le()),
            Some(Ordering::Equal) => Ok(self.comparison.is_eq()),
            Some(Ordering::Greater) => Ok(self.comparison.is_ge()),
            None => Err(StockTrekError::Stats(StatsError::IncomparableValues {
                left: left_value,
                right: right_value,
            })),
        }
    }
}