stock-trek 0.4.0

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

use crate::{
    predicates::predicate::{Predicate, PredicateTrait},
    resolved_context::ResolvedContext,
    util::serde_ordering,
    values::value::NumberValue,
};
use anyhow::{bail, Result};
use serde::{Deserialize, Serialize};
use std::cmp::Ordering;

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

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

#[typetag::serde]
impl PredicateTrait for ComparePredicate {
    fn test(&self, context: &ResolvedContext) -> Result<bool> {
        let left_value = self.left.number(context)?;
        let right_value = self.right.number(context)?;
        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 => bail!("Failed to compare {} and {}", left_value, right_value),
        }
    }
}