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

use crate::enums::indicator::TradingIndicatorKind;
use polars::prelude::DataFrame;
use std::collections::HashMap;

#[derive(Clone)]
pub struct PreTradeData {
    pub market_sim_data: DataFrame,
    pub indicators: HashMap<TradingIndicatorKind, DataFrame>,
}

pub struct PreTradeDataBuilder {
    market_sim_data: Option<DataFrame>,
    indicators: Option<HashMap<TradingIndicatorKind, DataFrame>>,
}

impl PreTradeDataBuilder {
    pub fn new() -> Self {
        Self {
            market_sim_data: None,
            indicators: None,
        }
    }

    pub fn with_market_sim_data(self, market_sim_data: DataFrame) -> Self {
        Self {
            market_sim_data: Some(market_sim_data),
            ..self
        }
    }

    pub fn with_indicators(self, indicators: HashMap<TradingIndicatorKind, DataFrame>) -> Self {
        Self {
            indicators: Some(indicators),
            ..self
        }
    }

    pub fn build(self) -> PreTradeData {
        PreTradeData {
            market_sim_data: self.market_sim_data.unwrap(),
            indicators: self.indicators.unwrap(),
        }
    }
}