stock-trek 0.8.4

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
114
115
116
117
118
119
120
121

use crate::{
    error::{
        result::{StockTrekError, StockTrekResult},
        stats::StatsError,
    },
    resolved_context::ResolvedContext,
    values::value::{NumberValue, NumberValueTrait},
};
use serde::{Deserialize, Serialize};
use strum::Display;

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

#[derive(Clone, 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 clone_box(&self) -> NumberValue {
        Box::new(self.clone())
    }
    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::Stats(StatsError::DivisionByZero {
                        function: "Div",
                        detail: "divisor = 0 would produce +/- infinity".to_string(),
                    }));
                }
                left_value / right_value
            }
            BinaryOperator::Pow => {
                if left_value < 0.0 && right_value.fract() != 0.0 {
                    return Err(StockTrekError::Stats(StatsError::ComplexPowerResult {
                        operator: "Pow",
                        base: left_value,
                        exponent: right_value,
                    }));
                }
                left_value.powf(right_value)
            }
            BinaryOperator::Log => {
                if left_value == 0.0 {
                    return Err(StockTrekError::Stats(StatsError::UndefinedLogarithm {
                        operator: "Log",
                        detail: "argument = 0 would produce undefined".to_string(),
                    }));
                }
                if right_value == 0.0 {
                    return Err(StockTrekError::Stats(StatsError::UndefinedLogarithm {
                        operator: "Log",
                        detail: "base = 0 would produce undefined".to_string(),
                    }));
                }
                if right_value == 1.0 {
                    return Err(StockTrekError::Stats(StatsError::UndefinedLogarithm {
                        operator: "Log",
                        detail: "base = 1 would produce undefined".to_string(),
                    }));
                }
                if left_value < 0.0 {
                    return Err(StockTrekError::Stats(StatsError::ComplexLogarithm {
                        operator: "Log",
                        detail: format!(
                            "argument {} < 0 would produce a complex number",
                            left_value
                        ),
                    }));
                }
                if right_value < 0.0 {
                    return Err(StockTrekError::Stats(StatsError::ComplexLogarithm {
                        operator: "Log",
                        detail: format!("base {} < 0 would produce a complex number", right_value),
                    }));
                }
                left_value.log(right_value)
            }
            BinaryOperator::Mod => {
                if right_value == 0.0 {
                    return Err(StockTrekError::Stats(StatsError::DivisionByZero {
                        function: "Mod",
                        detail: "divisor = 0 would produce +/- infinity".to_string(),
                    }));
                }
                left_value % right_value
            }
            BinaryOperator::Mul => left_value * right_value,
            BinaryOperator::Sub => left_value - right_value,
        };
        Ok(calculation_result)
    }
}