Mirror/brk
1
0
Fork 0
mirror of https://github.com/bitcoinresearchkit/brk.git synced 2026-07-13 12:08:13 -07:00
Code Issues Packages Projects Releases Wiki Activity

global: snapshot

Browse Source
This commit is contained in:
nym21
2026-03-05 18:08:10 +01:00
parent 2ae542ecdb
commit 266342cd98
10 changed files with 684 additions and 556 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/brk_computer/src/internal/from_height/ratio/extended.rs
+8 -8
View File
@@ -6,7 +6,7 @@ use vecdb::{Database, Exit, ReadableVec, Rw, StorageMode};
use crate::{blocks, indexes, prices};
use super::{ComputedFromHeightRatio, ComputedFromHeightRatioExtension};
use super::{ComputedFromHeightRatio, ComputedFromHeightRatioPercentiles};
#[derive(Deref, DerefMut, Traversable)]
pub struct ComputedFromHeightRatioExtended<M: StorageMode = Rw> {
@@ -15,7 +15,7 @@ pub struct ComputedFromHeightRatioExtended<M: StorageMode = Rw> {
#[traversable(flatten)]
pub base: ComputedFromHeightRatio<M>,
#[traversable(flatten)]
pub extended: ComputedFromHeightRatioExtension<M>,
pub percentiles: ComputedFromHeightRatioPercentiles<M>,
}
impl ComputedFromHeightRatioExtended {
@@ -27,11 +27,13 @@ impl ComputedFromHeightRatioExtended {
) -> Result<Self> {
Ok(Self {
base: ComputedFromHeightRatio::forced_import(db, name, version, indexes)?,
extended: ComputedFromHeightRatioExtension::forced_import(db, name, version, indexes)?,
percentiles: ComputedFromHeightRatioPercentiles::forced_import(
db, name, version, indexes,
)?,
})
}
/// Compute ratio and all extended metrics from an externally-provided metric price (in cents).
/// Compute ratio and all percentile metrics from an externally-provided metric price (in cents).
pub(crate) fn compute_rest(
&mut self,
blocks: &blocks::Vecs,
@@ -43,10 +45,8 @@ impl ComputedFromHeightRatioExtended {
let close_price = &prices.price.cents.height;
self.base
.compute_ratio(starting_indexes, close_price, metric_price, exit)?;
self.extended
.compute_rest(blocks, starting_indexes, exit, &self.base.ratio.height)?;
self.extended
.compute_cents_bands(starting_indexes, metric_price, exit)?;
self.percentiles
.compute(blocks, starting_indexes, exit, &self.base.ratio.height, metric_price)?;
Ok(())
}
}
crates/brk_computer/src/internal/from_height/ratio/full.rs
+51
View File
@@ -0,0 +1,51 @@
use brk_error::Result;
use brk_traversable::Traversable;
use brk_types::{Cents, Height, Indexes, Version};
use derive_more::{Deref, DerefMut};
use vecdb::{Database, Exit, ReadableVec, Rw, StorageMode};
use crate::{blocks, indexes, prices};
use super::{ComputedFromHeightRatioExtended, ComputedFromHeightRatioStdDevBands};
#[derive(Deref, DerefMut, Traversable)]
pub struct ComputedFromHeightRatioFull<M: StorageMode = Rw> {
#[deref]
#[deref_mut]
#[traversable(flatten)]
pub base: ComputedFromHeightRatioExtended<M>,
#[traversable(flatten)]
pub std_dev: ComputedFromHeightRatioStdDevBands<M>,
}
impl ComputedFromHeightRatioFull {
pub(crate) fn forced_import(
db: &Database,
name: &str,
version: Version,
indexes: &indexes::Vecs,
) -> Result<Self> {
Ok(Self {
base: ComputedFromHeightRatioExtended::forced_import(db, name, version, indexes)?,
std_dev: ComputedFromHeightRatioStdDevBands::forced_import(
db, name, version, indexes,
)?,
})
}
/// Compute ratio, percentiles, and all stddev bands from an externally-provided metric price (in cents).
pub(crate) fn compute_rest(
&mut self,
blocks: &blocks::Vecs,
prices: &prices::Vecs,
starting_indexes: &Indexes,
exit: &Exit,
metric_price: &impl ReadableVec<Height, Cents>,
) -> Result<()> {
self.base
.compute_rest(blocks, prices, starting_indexes, exit, metric_price)?;
self.std_dev
.compute(blocks, starting_indexes, exit, &self.base.base.ratio.height, metric_price)?;
Ok(())
}
}
crates/brk_computer/src/internal/from_height/ratio/mod.rs
+6 -2
View File
@@ -1,10 +1,14 @@
mod extended;
mod extension;
mod full;
mod percentiles;
mod price_extended;
mod std_dev_bands;
pub use extended::*;
pub use extension::*;
pub use full::*;
pub use percentiles::*;
pub use price_extended::*;
pub use std_dev_bands::*;
use brk_error::Result;
use brk_traversable::Traversable;
crates/brk_computer/src/internal/from_height/ratio/extension.rs → crates/brk_computer/src/internal/from_height/ratio/percentiles.rs
+6 -57
View File
@@ -8,13 +8,13 @@ use vecdb::{
use crate::{
blocks, indexes,
internal::{ComputedFromHeightStdDevExtended, ExpandingPercentiles, Price, PriceTimesRatioBp32Cents},
internal::{ExpandingPercentiles, Price, PriceTimesRatioBp32Cents},
};
use super::{super::ComputedFromHeight, ComputedFromHeightRatio};
#[derive(Traversable)]
pub struct ComputedFromHeightRatioExtension<M: StorageMode = Rw> {
pub struct ComputedFromHeightRatioPercentiles<M: StorageMode = Rw> {
pub ratio_sma_1w: ComputedFromHeightRatio<M>,
pub ratio_sma_1m: ComputedFromHeightRatio<M>,
pub ratio_pct99: ComputedFromHeightRatio<M>,
@@ -30,18 +30,13 @@ pub struct ComputedFromHeightRatioExtension<M: StorageMode = Rw> {
pub ratio_pct2_price: Price<ComputedFromHeight<Cents, M>>,
pub ratio_pct1_price: Price<ComputedFromHeight<Cents, M>>,
pub ratio_sd: ComputedFromHeightStdDevExtended<M>,
pub ratio_sd_4y: ComputedFromHeightStdDevExtended<M>,
pub ratio_sd_2y: ComputedFromHeightStdDevExtended<M>,
pub ratio_sd_1y: ComputedFromHeightStdDevExtended<M>,
#[traversable(skip)]
expanding_pct: ExpandingPercentiles,
}
const VERSION: Version = Version::new(4);
impl ComputedFromHeightRatioExtension {
impl ComputedFromHeightRatioPercentiles {
pub(crate) fn forced_import(
db: &Database,
name: &str,
@@ -61,19 +56,6 @@ impl ComputedFromHeightRatioExtension {
};
}
macro_rules! import_sd {
($suffix:expr, $period:expr, $days:expr) => {
ComputedFromHeightStdDevExtended::forced_import(
db,
&format!("{name}_{}", $suffix),
$period,
$days,
v,
indexes,
)?
};
}
macro_rules! import_price {
($suffix:expr) => {
Price::forced_import(db, &format!("{name}_{}", $suffix), v, indexes)?
@@ -83,10 +65,6 @@ impl ComputedFromHeightRatioExtension {
Ok(Self {
ratio_sma_1w: import_ratio!("ratio_sma_1w"),
ratio_sma_1m: import_ratio!("ratio_sma_1m"),
ratio_sd: import_sd!("ratio", "", usize::MAX),
ratio_sd_1y: import_sd!("ratio", "1y", 365),
ratio_sd_2y: import_sd!("ratio", "2y", 2 * 365),
ratio_sd_4y: import_sd!("ratio", "4y", 4 * 365),
ratio_pct99: import_ratio!("ratio_pct99"),
ratio_pct98: import_ratio!("ratio_pct98"),
ratio_pct95: import_ratio!("ratio_pct95"),
@@ -103,14 +81,14 @@ impl ComputedFromHeightRatioExtension {
})
}
pub(crate) fn compute_rest(
pub(crate) fn compute(
&mut self,
blocks: &blocks::Vecs,
starting_indexes: &Indexes,
exit: &Exit,
ratio_source: &impl ReadableVec<Height, StoredF32>,
metric_price: &impl ReadableVec<Height, Cents>,
) -> Result<()> {
// SMA using lookback vecs
self.ratio_sma_1w.bps.height.compute_rolling_average(
starting_indexes.height,
&blocks.count.height_1w_ago,
@@ -151,7 +129,6 @@ impl ComputedFromHeightRatioExtension {
}
}
// Process new blocks [start, ratio_len)
let new_ratios = ratio_source.collect_range_at(start, ratio_len);
let mut pct_vecs: [&mut EagerVec<PcoVec<Height, BasisPoints32>>; 6] = [
&mut self.ratio_pct1.bps.height,
@@ -179,25 +156,7 @@ impl ComputedFromHeightRatioExtension {
self.mut_pct_vecs().try_for_each(|v| v.flush())?;
}
// Compute stddev at height level
for sd in [
&mut self.ratio_sd,
&mut self.ratio_sd_4y,
&mut self.ratio_sd_2y,
&mut self.ratio_sd_1y,
] {
sd.compute_all(blocks, starting_indexes, exit, ratio_source)?;
}
Ok(())
}
pub(crate) fn compute_cents_bands(
&mut self,
starting_indexes: &Indexes,
metric_price: &impl ReadableVec<Height, Cents>,
exit: &Exit,
) -> Result<()> {
// Cents bands
macro_rules! compute_band {
($usd_field:ident, $band_source:expr) => {
self.$usd_field
@@ -218,16 +177,6 @@ impl ComputedFromHeightRatioExtension {
compute_band!(ratio_pct2_price, &self.ratio_pct2.bps.height);
compute_band!(ratio_pct1_price, &self.ratio_pct1.bps.height);
// Stddev cents bands
for sd in [
&mut self.ratio_sd,
&mut self.ratio_sd_4y,
&mut self.ratio_sd_2y,
&mut self.ratio_sd_1y,
] {
sd.compute_cents_bands(starting_indexes, metric_price, exit)?;
}
Ok(())
}
crates/brk_computer/src/internal/from_height/ratio/std_dev_bands.rs
+68
View File
@@ -0,0 +1,68 @@
use brk_error::Result;
use brk_traversable::Traversable;
use brk_types::{Cents, Height, Indexes, StoredF32, Version};
use vecdb::{Database, Exit, ReadableVec, Rw, StorageMode};
use crate::{blocks, indexes, internal::ComputedFromHeightStdDevExtended};
#[derive(Traversable)]
pub struct ComputedFromHeightRatioStdDevBands<M: StorageMode = Rw> {
pub ratio_sd: ComputedFromHeightStdDevExtended<M>,
pub ratio_sd_4y: ComputedFromHeightStdDevExtended<M>,
pub ratio_sd_2y: ComputedFromHeightStdDevExtended<M>,
pub ratio_sd_1y: ComputedFromHeightStdDevExtended<M>,
}
const VERSION: Version = Version::new(4);
impl ComputedFromHeightRatioStdDevBands {
pub(crate) fn forced_import(
db: &Database,
name: &str,
version: Version,
indexes: &indexes::Vecs,
) -> Result<Self> {
let v = version + VERSION;
macro_rules! import_sd {
($suffix:expr, $period:expr, $days:expr) => {
ComputedFromHeightStdDevExtended::forced_import(
db,
&format!("{name}_{}", $suffix),
$period,
$days,
v,
indexes,
)?
};
}
Ok(Self {
ratio_sd: import_sd!("ratio", "", usize::MAX),
ratio_sd_1y: import_sd!("ratio", "1y", 365),
ratio_sd_2y: import_sd!("ratio", "2y", 2 * 365),
ratio_sd_4y: import_sd!("ratio", "4y", 4 * 365),
})
}
pub(crate) fn compute(
&mut self,
blocks: &blocks::Vecs,
starting_indexes: &Indexes,
exit: &Exit,
ratio_source: &impl ReadableVec<Height, StoredF32>,
metric_price: &impl ReadableVec<Height, Cents>,
) -> Result<()> {
for sd in [
&mut self.ratio_sd,
&mut self.ratio_sd_4y,
&mut self.ratio_sd_2y,
&mut self.ratio_sd_1y,
] {
sd.compute_all(blocks, starting_indexes, exit, ratio_source)?;
sd.compute_cents_bands(starting_indexes, metric_price, exit)?;
}
Ok(())
}
}