brk_computer 0.3.0-alpha.2

A Bitcoin dataset computer built on top of brk_indexer
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

use brk_error::Result;
use brk_types::Version;
use vecdb::Database;

use super::{MacdChain, RsiChain, Vecs};
use crate::{
    indexes,
    internal::{PerBlock, PercentPerBlock, RatioPerBlock, WindowsTo1m},
};

const VERSION: Version = Version::new(4);

impl RsiChain {
    fn forced_import(
        db: &Database,
        tf: &str,
        version: Version,
        indexes: &indexes::Vecs,
    ) -> Result<Self> {
        macro_rules! import {
            ($name:expr) => {
                PerBlock::forced_import(db, &format!("rsi_{}_{}", $name, tf), version, indexes)?
            };
        }

        macro_rules! percent_import {
            ($name:expr) => {
                PercentPerBlock::forced_import(
                    db,
                    &format!("rsi_{}_{}", $name, tf),
                    version,
                    indexes,
                )?
            };
        }

        let average_gain = import!("average_gain");
        let average_loss = import!("average_loss");

        let rsi = PercentPerBlock::forced_import(db, &format!("rsi_{tf}"), version, indexes)?;

        Ok(Self {
            gains: import!("gains"),
            losses: import!("losses"),
            average_gain,
            average_loss,
            rsi,
            rsi_min: percent_import!("min"),
            rsi_max: percent_import!("max"),
            stoch_rsi: percent_import!("stoch"),
            stoch_rsi_k: percent_import!("stoch_k"),
            stoch_rsi_d: percent_import!("stoch_d"),
        })
    }
}

impl MacdChain {
    fn forced_import(
        db: &Database,
        tf: &str,
        version: Version,
        indexes: &indexes::Vecs,
    ) -> Result<Self> {
        let line = PerBlock::forced_import(db, &format!("macd_line_{tf}"), version, indexes)?;
        let signal = PerBlock::forced_import(db, &format!("macd_signal_{tf}"), version, indexes)?;

        let histogram =
            PerBlock::forced_import(db, &format!("macd_histogram_{tf}"), version, indexes)?;

        Ok(Self {
            ema_fast: PerBlock::forced_import(
                db,
                &format!("macd_ema_fast_{tf}"),
                version,
                indexes,
            )?,
            ema_slow: PerBlock::forced_import(
                db,
                &format!("macd_ema_slow_{tf}"),
                version,
                indexes,
            )?,
            line,
            signal,
            histogram,
        })
    }
}

impl Vecs {
    pub(crate) fn forced_import(
        db: &Database,
        version: Version,
        indexes: &indexes::Vecs,
    ) -> Result<Self> {
        let v = version + VERSION;

        let rsi = WindowsTo1m::try_from_fn(|tf| {
            RsiChain::forced_import(db, tf, v + Version::TWO, indexes)
        })?;
        let macd = WindowsTo1m::try_from_fn(|tf| MacdChain::forced_import(db, tf, v, indexes))?;

        let pi_cycle = RatioPerBlock::forced_import_raw(db, "pi_cycle", v, indexes)?;

        Ok(Self {
            rsi,
            pi_cycle,
            macd,
        })
    }
}