stock-trek 0.8.9

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

use crate::{
    cex::capability::{CexCapability, HasRequiredCapabilities, combine_capabilities},
    commands::command::{Command, CommandTrait},
    conditions::condition::Condition,
    error::result::StockTrekResult,
    resolved_context::ResolvedContext,
};
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize)]
pub struct IfCommand {
    condition: Condition,
    if_true: Command,
    if_false: Command,
}

impl IfCommand {
    pub fn new(condition: Condition, if_true: Command, if_false: Command) -> Command {
        Box::new(Self {
            condition,
            if_true,
            if_false,
        })
    }
}

#[typetag::serde]
impl CommandTrait for IfCommand {
    fn execute(&self, c: &ResolvedContext) -> StockTrekResult<()> {
        let condition = self.condition.test(c)?;
        if condition {
            self.if_true.execute(c)?;
        } else {
            self.if_false.execute(c)?;
        }
        Ok(())
    }
}

impl HasRequiredCapabilities for IfCommand {
    fn required_capabilities(&self) -> Vec<CexCapability> {
        combine_capabilities(&[self.if_false.as_ref(), self.if_true.as_ref()])
    }
}