chapaty 0.1.4

A software to backtest trading strategies.
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

use crate::{
    bot::indicator_data_pair::IndicatorDataPair,
    enums::{data::HdbSourceDirKind, indicator::TradingIndicatorKind},
};
#[derive(Clone)]
pub struct FileNameResolver {
    indicator_data_pair: Option<IndicatorDataPair>,
    simulation_data: HdbSourceDirKind,
}

impl FileNameResolver {
    pub fn new(simulation_data: HdbSourceDirKind) -> Self {
        Self {
            indicator_data_pair: None,
            simulation_data,
        }
    }

    pub fn with_indicator_data_pair(self, indicator_data_pair: IndicatorDataPair) -> Self {
        Self {
            indicator_data_pair: Some(indicator_data_pair),
            ..self
        }
    }

    pub fn get_filename(&self) -> String {
        self.indicator_data_pair.clone().map_or_else(
            || self.simulation_data.to_string(),
            |_| self.generate_file_name(),
        )
    }

    fn generate_file_name(&self) -> String {
        match self.indicator_data_pair.as_ref().unwrap().data {
            HdbSourceDirKind::Tick => self.trading_indicator_from_tick_data(),
            HdbSourceDirKind::AggTrades => self.trading_indicator_from_agg_trades_data(),
            ohlc_variant => self.trading_indicator_from_ohlc_variant(&ohlc_variant),
        }
    }

    fn trading_indicator_from_tick_data(&self) -> String {
        match self.indicator_data_pair.clone().unwrap().indicator {
            TradingIndicatorKind::Poc(_)
            | TradingIndicatorKind::ValueAreaHigh(_)
            | TradingIndicatorKind::ValueAreaLow(_) => format!("vol-tick"),
        }
    }

    fn trading_indicator_from_agg_trades_data(&self) -> String {
        match self.indicator_data_pair.clone().unwrap().indicator {
            TradingIndicatorKind::Poc(_)
            | TradingIndicatorKind::ValueAreaHigh(_)
            | TradingIndicatorKind::ValueAreaLow(_) => format!("vol-aggTrades"),
        }
    }

    fn trading_indicator_from_ohlc_variant(&self, ohlc_variant: &HdbSourceDirKind) -> String {
        match self.indicator_data_pair.clone().unwrap().indicator {
            TradingIndicatorKind::Poc(_)
            | TradingIndicatorKind::ValueAreaHigh(_)
            | TradingIndicatorKind::ValueAreaLow(_) => {
                format!("tpo-{}", ohlc_variant.split_ohlc_dir_in_parts().1)
            }
        }
    }
}