Method find_or_create_node

src/models/inverted_index.rs:141–166

Finds or creates the node where the data should be inserted. Traverses the tree iteratively and returns a reference to the node.

(&self, path: &[u8], mut offset_fn: impl FnMut() -> u32)

    /// Finds or creates the node where the data should be inserted.
    /// Traverses the tree iteratively and returns a reference to the node.
    pub fn find_or_create_node(&self, path: &[u8], mut offset_fn: impl FnMut() -> u32) -> &Self {
        let mut current_node = self;
        for &child_index in path {
            let new_dim_index = current_node.dim_index + (1u32 << (child_index * 2));
            if let Some(child) = current_node.children.get(child_index as usize) {
                let res = unsafe { &*child };
                current_node = res;
                continue;
            }
            let (new_child, _is_newly_created) =
                current_node
                    .children
                    .get_or_insert(child_index as usize, || {
                        Box::into_raw(Box::new(Self::new(
                            new_dim_index,
                            true,
                            self.quantization_bits,
                            FileOffset(offset_fn()),
                        )))
                    });
            let res = unsafe { &*new_child };
            current_node = res;
        }

        current_node
    }

    pub fn quantize(&self, value: f32, values_upper_bound: f32) -> u8 {
        let quantization = ((1u32 << self.quantization_bits) - 1) as u8;