brk: first commit

crates/brk_computer/src/lib.rs

@@ -0,0 +1,114 @@
+use std::path::{Path, PathBuf};
+
+use brk_indexer::Indexer;
+pub use brk_parser::rpc;
+use hodor::Exit;
+
+mod storage;
+mod structs;
+
+use brk_fetcher::Date;
+use storable_vec::SINGLE_THREAD;
+use storage::{Fjalls, StorableVecs};
+pub use structs::*;
+
+pub struct Computer<const MODE: u8> {
+    path: PathBuf,
+    pub vecs: StorableVecs<MODE>,
+    pub trees: Fjalls,
+}
+
+impl<const MODE: u8> Computer<MODE> {
+    pub fn import(computed_dir: &Path) -> color_eyre::Result<Self> {
+        let vecs = StorableVecs::import(&computed_dir.join("vecs"))?;
+        let trees = Fjalls::import(&computed_dir.join("fjall"))?;
+        Ok(Self {
+            path: computed_dir.to_owned(),
+            vecs,
+            trees,
+        })
+    }
+}
+
+impl Computer<SINGLE_THREAD> {
+    pub fn compute(&mut self, mut indexer: Indexer<SINGLE_THREAD>, exit: &Exit) -> color_eyre::Result<()> {
+        let height_count = indexer.vecs.height_to_size.len();
+        let txindexes_count = indexer.vecs.txindex_to_txid.len();
+        let txinindexes_count = indexer.vecs.txinindex_to_txoutindex.len();
+        let txoutindexes_count = indexer.vecs.txoutindex_to_addressindex.len();
+
+        // TODO: Remove all outdated
+
+        self.vecs
+            .txindex_to_last_txinindex
+            .compute_last_index_from_first(&mut indexer.vecs.txindex_to_first_txinindex, txinindexes_count)?;
+
+        self.vecs.txindex_to_inputs_count.compute_count_from_indexes(
+            &mut indexer.vecs.txindex_to_first_txinindex,
+            &mut self.vecs.txindex_to_last_txinindex,
+        )?;
+
+        self.vecs
+            .txindex_to_last_txoutindex
+            .compute_last_index_from_first(&mut indexer.vecs.txindex_to_first_txoutindex, txoutindexes_count)?;
+
+        self.vecs.txindex_to_outputs_count.compute_count_from_indexes(
+            &mut indexer.vecs.txindex_to_first_txoutindex,
+            &mut self.vecs.txindex_to_last_txoutindex,
+        )?;
+
+        self.vecs
+            .height_to_date
+            .compute_transform(&mut indexer.vecs.height_to_timestamp, |timestamp| {
+                Date::from(*timestamp)
+            })?;
+
+        self.vecs
+            .height_to_last_txindex
+            .compute_last_index_from_first(&mut indexer.vecs.height_to_first_txindex, height_count)?;
+
+        self.vecs.txindex_to_height.compute_inverse_less_to_more(
+            &mut indexer.vecs.height_to_first_txindex,
+            &mut self.vecs.height_to_last_txindex,
+        )?;
+
+        self.vecs.txindex_to_is_coinbase.compute_is_first_ordered(
+            &mut self.vecs.txindex_to_height,
+            &mut indexer.vecs.height_to_first_txindex,
+        )?;
+
+        // self.vecs.txindex_to_fee.compute_transform(
+        //     &mut self.vecs.txindex_to_height,
+        //     &mut indexer.vecs.height_to_first_txindex,
+        // )?;
+
+        let date_count = self.vecs.height_to_date.len();
+
+        // self.vecs.height_to_dateindex.compute(...)
+
+        self.vecs
+            .dateindex_to_first_height
+            .compute_inverse_more_to_less(&mut self.vecs.height_to_dateindex)?;
+
+        // ---
+        // Date to X
+        // ---
+        // ...
+
+        // ---
+        // Month to X
+        // ---
+        // ...
+
+        // ---
+        // Year to X
+        // ---
+        // ...
+
+        Ok(())
+    }
+
+    pub fn path(&self) -> &Path {
+        &self.path
+    }
+}