stock-trek 0.4.12

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

use crate::{
    error::{
        general::GeneralError,
        result::{StockTrekError, StockTrekResult},
    },
    resolved_context::ResolvedContext,
    values::value::{NumberValue, NumberValueTrait},
};
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum BinaryOperator {
    Add,
    Sub,
    Mul,
    Div,
    Mod,
    Pow,
    Log,
    Atan2,
}

#[derive(Serialize, Deserialize)]
pub struct BinaryCalculationValue {
    left: NumberValue,
    operator: BinaryOperator,
    right: NumberValue,
}

impl BinaryCalculationValue {
    pub fn new(left: NumberValue, operator: BinaryOperator, right: NumberValue) -> NumberValue {
        Box::new(Self {
            left,
            operator,
            right,
        })
    }
}

#[typetag::serde]
impl NumberValueTrait for BinaryCalculationValue {
    fn number(&self, c: &ResolvedContext) -> StockTrekResult<f64> {
        let left_value = self.left.number(c)?;
        let right_value = self.right.number(c)?;
        let calculation_result = match self.operator {
            BinaryOperator::Add => left_value + right_value,
            BinaryOperator::Atan2 => left_value.atan2(right_value),
            BinaryOperator::Div => {
                if right_value == 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(
                        "Divide: A divisor =0 is not supported, it would give +/- infinity"
                            .to_string(),
                    )));
                }
                left_value / right_value
            }
            BinaryOperator::Pow => {
                if left_value < 0.0 && right_value.fract() != 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(
                        format!(
                        "Power: Combining a base <0 {} with a non-integer exponent {} is not supported, it would give a complex number",
                        left_value,
                        right_value,
                        )
                    )));
                }
                left_value.powf(right_value)
            }
            BinaryOperator::Log => {
                if left_value == 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(
                        "Log: An argument =0 is not supported, it would give undefined".to_string(),
                    )));
                }
                if right_value == 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(
                        "Log: A base =0 is not supported, it would give undefined".to_string(),
                    )));
                }
                if right_value == 1.0 {
                    return Err(StockTrekError::General(GeneralError::Message(
                        "Log: A base =1 is not supported, it would give undefined".to_string(),
                    )));
                }
                if left_value < 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(format!(
                        "Log: An argument <0 {} is not supported, it would give a complex number",
                        left_value
                    ))));
                }
                if right_value < 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(format!(
                        "Log: A base <0 {} is not supported, it would give a complex number",
                        right_value
                    ))));
                }
                left_value.log(right_value)
            }
            BinaryOperator::Mod => {
                if right_value == 0.0 {
                    return Err(StockTrekError::General(GeneralError::Message(
                        "Modulo: A modulus =0 is not supported, it would give undefined"
                            .to_string(),
                    )));
                }
                left_value % right_value
            }
            BinaryOperator::Mul => left_value * right_value,
            BinaryOperator::Sub => left_value - right_value,
        };
        Ok(calculation_result)
    }
}