stock-trek 0.6.0

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

use crate::{
    asset_id::AssetId,
    exchange_id::ExchangeId,
    market_data::market::Market,
    statistics::{
        advanced::Advanced, decompose::Decompose, evaluation::Evaluation,
        exponential_smoothing::ExponentialSmoothing, filter::Filter, frequency::Frequency,
        hypothesis::Hypothesis, moving_average::MovingAverage, stats::Stats,
        time_series::TimeSeries, transformation::Transformation, wavelet::Wavelet,
    },
};
use std::collections::HashMap;

pub struct StrategyContext {
    market_data: HashMap<ExchangeId, MarketDataByBaseContext>,
    pub stats: Stats,
}

impl StrategyContext {
    pub fn new(market_data: HashMap<ExchangeId, MarketDataByBaseContext>) -> Self {
        Self {
            market_data,
            stats: Stats {
                advanced: Advanced,
                decompose: Decompose,
                evaluation: Evaluation,
                exponential_smoothing: ExponentialSmoothing,
                filter: Filter,
                frequency: Frequency,
                hypothesis: Hypothesis,
                moving_average: MovingAverage,
                time_series: TimeSeries,
                transformation: Transformation,
                wavelet: Wavelet,
            },
        }
    }
    pub fn exchange_markets_for<'context>(
        &'context self,
        base: &'context AssetId,
        quote: &'context AssetId,
    ) -> impl Iterator<Item = (ExchangeId, &'context Market)> + 'context {
        self.market_data
            .iter()
            .filter_map(move |(exchange, by_base)| {
                by_base
                    .markets_by_base
                    .get(base)
                    .and_then(|by_quote| by_quote.markets_by_quote.get(quote))
                    .map(|market| (exchange.clone(), market))
            })
    }
    pub fn market_for(
        &self,
        exchange_id: &ExchangeId,
        base: &AssetId,
        quote: &AssetId,
    ) -> Option<&Market> {
        self.market_data
            .get(exchange_id)
            .and_then(|m| m.markets_by_base.get(base))
            .and_then(|m| m.markets_by_quote.get(quote))
    }
}

pub struct MarketDataByBaseContext {
    markets_by_base: HashMap<AssetId, MarketDataByQuoteContext>,
}

impl MarketDataByBaseContext {
    pub fn new(markets_by_base: HashMap<AssetId, MarketDataByQuoteContext>) -> Self {
        Self { markets_by_base }
    }
}

pub struct MarketDataByQuoteContext {
    markets_by_quote: HashMap<AssetId, Market>,
}

impl MarketDataByQuoteContext {
    pub fn new(markets_by_quote: HashMap<AssetId, Market>) -> Self {
        Self { markets_by_quote }
    }
}