brk/_src/parser/datasets/subs/price_paid.rs

use allocative::Allocative;
use struct_iterable::Iterable;

use crate::{
    parser::{
        datasets::{AnyDataset, InsertData, MinInitialStates},
        states::PricePaidState,
    },
    structs::{BiMap, Config, Date, Height, MapKind, MapPath},
};

#[derive(Allocative, Iterable)]
pub struct PricePaidSubDataset {
    min_initial_states: MinInitialStates,

    pp_median: BiMap<f32>,
    pp_95p: BiMap<f32>,
    pp_90p: BiMap<f32>,
    pp_85p: BiMap<f32>,
    pp_80p: BiMap<f32>,
    pp_75p: BiMap<f32>,
    pp_70p: BiMap<f32>,
    pp_65p: BiMap<f32>,
    pp_60p: BiMap<f32>,
    pp_55p: BiMap<f32>,
    pp_45p: BiMap<f32>,
    pp_40p: BiMap<f32>,
    pp_35p: BiMap<f32>,
    pp_30p: BiMap<f32>,
    pp_25p: BiMap<f32>,
    pp_20p: BiMap<f32>,
    pp_15p: BiMap<f32>,
    pp_10p: BiMap<f32>,
    pp_05p: BiMap<f32>,
}

impl PricePaidSubDataset {
    pub fn import(
        path: &MapPath,
        name: &Option<String>,
        config: &Config,
    ) -> color_eyre::Result<Self> {
        let f = |s: &str| {
            if let Some(name) = name {
                path.join(&format!("{name}/{s}"))
            } else {
                path.join(s)
            }
        };

        let mut s = Self {
            min_initial_states: MinInitialStates::default(),

            // ---
            // Inserted
            // ---
            pp_median: BiMap::new_bin(1, MapKind::Inserted, &f("median_price_paid")),
            pp_95p: BiMap::new_bin(1, MapKind::Inserted, &f("95p_price_paid")),
            pp_90p: BiMap::new_bin(1, MapKind::Inserted, &f("90p_price_paid")),
            pp_85p: BiMap::new_bin(1, MapKind::Inserted, &f("85p_price_paid")),
            pp_80p: BiMap::new_bin(1, MapKind::Inserted, &f("80p_price_paid")),
            pp_75p: BiMap::new_bin(1, MapKind::Inserted, &f("75p_price_paid")),
            pp_70p: BiMap::new_bin(1, MapKind::Inserted, &f("70p_price_paid")),
            pp_65p: BiMap::new_bin(1, MapKind::Inserted, &f("65p_price_paid")),
            pp_60p: BiMap::new_bin(1, MapKind::Inserted, &f("60p_price_paid")),
            pp_55p: BiMap::new_bin(1, MapKind::Inserted, &f("55p_price_paid")),
            pp_45p: BiMap::new_bin(1, MapKind::Inserted, &f("45p_price_paid")),
            pp_40p: BiMap::new_bin(1, MapKind::Inserted, &f("40p_price_paid")),
            pp_35p: BiMap::new_bin(1, MapKind::Inserted, &f("35p_price_paid")),
            pp_30p: BiMap::new_bin(1, MapKind::Inserted, &f("30p_price_paid")),
            pp_25p: BiMap::new_bin(1, MapKind::Inserted, &f("25p_price_paid")),
            pp_20p: BiMap::new_bin(1, MapKind::Inserted, &f("20p_price_paid")),
            pp_15p: BiMap::new_bin(1, MapKind::Inserted, &f("15p_price_paid")),
            pp_10p: BiMap::new_bin(1, MapKind::Inserted, &f("10p_price_paid")),
            pp_05p: BiMap::new_bin(1, MapKind::Inserted, &f("05p_price_paid")),
        };

        s.min_initial_states
            .consume(MinInitialStates::compute_from_dataset(&s, config));

        Ok(s)
    }

    pub fn insert(
        &mut self,
        &InsertData {
            height,
            is_date_last_block,
            date,
            ..
        }: &InsertData,
        state: &PricePaidState,
    ) {
        let pp_05p = state.pp_05p();
        let pp_10p = state.pp_10p();
        let pp_15p = state.pp_15p();
        let pp_20p = state.pp_20p();
        let pp_25p = state.pp_25p();
        let pp_30p = state.pp_30p();
        let pp_35p = state.pp_35p();
        let pp_40p = state.pp_40p();
        let pp_45p = state.pp_45p();
        let pp_median = state.pp_median();
        let pp_55p = state.pp_55p();
        let pp_60p = state.pp_60p();
        let pp_65p = state.pp_65p();
        let pp_70p = state.pp_70p();
        let pp_75p = state.pp_75p();
        let pp_80p = state.pp_80p();
        let pp_85p = state.pp_85p();
        let pp_90p = state.pp_90p();
        let pp_95p = state.pp_95p();

        // Check if iter was empty
        if pp_05p.is_none() {
            self.insert_height_default(height);

            if is_date_last_block {
                self.insert_date_default(date);
            }

            return;
        }

        let pp_05p = self
            .pp_05p
            .height
            .insert(height, pp_05p.unwrap().to_dollar() as f32);
        let pp_10p = self
            .pp_10p
            .height
            .insert(height, pp_10p.unwrap().to_dollar() as f32);
        let pp_15p = self
            .pp_15p
            .height
            .insert(height, pp_15p.unwrap().to_dollar() as f32);
        let pp_20p = self
            .pp_20p
            .height
            .insert(height, pp_20p.unwrap().to_dollar() as f32);
        let pp_25p = self
            .pp_25p
            .height
            .insert(height, pp_25p.unwrap().to_dollar() as f32);
        let pp_30p = self
            .pp_30p
            .height
            .insert(height, pp_30p.unwrap().to_dollar() as f32);
        let pp_35p = self
            .pp_35p
            .height
            .insert(height, pp_35p.unwrap().to_dollar() as f32);
        let pp_40p = self
            .pp_40p
            .height
            .insert(height, pp_40p.unwrap().to_dollar() as f32);
        let pp_45p = self
            .pp_45p
            .height
            .insert(height, pp_45p.unwrap().to_dollar() as f32);
        let pp_median = self
            .pp_median
            .height
            .insert(height, pp_median.unwrap().to_dollar() as f32);
        let pp_55p = self
            .pp_55p
            .height
            .insert(height, pp_55p.unwrap().to_dollar() as f32);
        let pp_60p = self
            .pp_60p
            .height
            .insert(height, pp_60p.unwrap().to_dollar() as f32);
        let pp_65p = self
            .pp_65p
            .height
            .insert(height, pp_65p.unwrap().to_dollar() as f32);
        let pp_70p = self
            .pp_70p
            .height
            .insert(height, pp_70p.unwrap().to_dollar() as f32);
        let pp_75p = self
            .pp_75p
            .height
            .insert(height, pp_75p.unwrap().to_dollar() as f32);
        let pp_80p = self
            .pp_80p
            .height
            .insert(height, pp_80p.unwrap().to_dollar() as f32);
        let pp_85p = self
            .pp_85p
            .height
            .insert(height, pp_85p.unwrap().to_dollar() as f32);
        let pp_90p = self
            .pp_90p
            .height
            .insert(height, pp_90p.unwrap().to_dollar() as f32);
        let pp_95p = self
            .pp_95p
            .height
            .insert(height, pp_95p.unwrap().to_dollar() as f32);

        if is_date_last_block {
            self.pp_05p.date.insert(date, pp_05p);
            self.pp_10p.date.insert(date, pp_10p);
            self.pp_15p.date.insert(date, pp_15p);
            self.pp_20p.date.insert(date, pp_20p);
            self.pp_25p.date.insert(date, pp_25p);
            self.pp_30p.date.insert(date, pp_30p);
            self.pp_35p.date.insert(date, pp_35p);
            self.pp_40p.date.insert(date, pp_40p);
            self.pp_45p.date.insert(date, pp_45p);
            self.pp_median.date.insert(date, pp_median);
            self.pp_55p.date.insert(date, pp_55p);
            self.pp_60p.date.insert(date, pp_60p);
            self.pp_65p.date.insert(date, pp_65p);
            self.pp_70p.date.insert(date, pp_70p);
            self.pp_75p.date.insert(date, pp_75p);
            self.pp_80p.date.insert(date, pp_80p);
            self.pp_85p.date.insert(date, pp_85p);
            self.pp_90p.date.insert(date, pp_90p);
            self.pp_95p.date.insert(date, pp_95p);
        }
    }

    fn insert_height_default(&mut self, height: Height) {
        self.inserted_as_mut_vec().into_iter().for_each(|bi| {
            bi.height.insert_default(height);
        })
    }

    fn insert_date_default(&mut self, date: Date) {
        self.inserted_as_mut_vec().into_iter().for_each(|bi| {
            bi.date.insert_default(date);
        })
    }

    pub fn inserted_as_mut_vec(&mut self) -> Vec<&mut BiMap<f32>> {
        vec![
            &mut self.pp_95p,
            &mut self.pp_90p,
            &mut self.pp_85p,
            &mut self.pp_80p,
            &mut self.pp_75p,
            &mut self.pp_70p,
            &mut self.pp_65p,
            &mut self.pp_60p,
            &mut self.pp_55p,
            &mut self.pp_median,
            &mut self.pp_45p,
            &mut self.pp_40p,
            &mut self.pp_35p,
            &mut self.pp_30p,
            &mut self.pp_25p,
            &mut self.pp_20p,
            &mut self.pp_15p,
            &mut self.pp_10p,
            &mut self.pp_05p,
        ]
    }
}

impl AnyDataset for PricePaidSubDataset {
    fn get_min_initial_states(&self) -> &MinInitialStates {
        &self.min_initial_states
    }
}