Struct ContextNode 

pub struct ContextNode { /* private fields */ }

A node in the context tree that stores transition statistics

Uses optimized storage for small collections and StateId for memory efficiency Implements lazy computation with caching for entropy and KL divergence

Implementations

impl ContextNode

pub fn new(interner: Arc<StringInterner>) -> Self

Create a new empty context node with string interner

pub fn add_transition(&mut self, next_state: &str)

Add a transition to this context using string interning

pub fn add_transition_by_id(&mut self, state_id: StateId)

Add a transition using StateId directly (internal use)

pub fn total_count(&self) -> usize

Get the total number of transitions from this context

pub fn get_count(&self, next_state: &str) -> usize

Get the count for a specific next state

pub fn get_count_by_id(&self, state_id: StateId) -> usize

Get the count for a StateId directly (internal use)

pub fn vocab_size(&self) -> usize

Get the number of unique next states

pub fn get_state_counts(&self) -> impl Iterator<Item = (StateId, usize)> + '_

Get all state IDs with their counts (internal use)

pub fn total_transitions(&self) -> usize

Get the sum of all transition counts (for compatibility)

pub fn get_string_counts(&self) -> HashMap<String, usize>

Get all counts as strings for compatibility with performance module

pub fn counts(&self) -> HashMap<String, usize>

Get counts for debugging (returns string representation)

pub fn get_probability( &self, next_state: &str, config: &AnomalyGridConfig, ) -> f64

Get the probability for a specific next state using Laplace smoothing

Computes probability on-demand: P(state) = (count + α) / (total + α * |V|)

pub fn get_probability_by_id( &self, state_id: StateId, config: &AnomalyGridConfig, ) -> f64

Get probability for a StateId directly (internal use)

pub fn get_probability_normalized( &self, next_state: &str, config: &AnomalyGridConfig, global_vocab_size: usize, ) -> f64

Get probability with proper normalization using global vocabulary size

pub fn get_probability_normalized_by_id( &self, state_id: StateId, config: &AnomalyGridConfig, global_vocab_size: usize, ) -> f64

Get probability with proper normalization using global vocabulary by StateId

pub fn calculate_entropy(&mut self, config: &AnomalyGridConfig) -> f64

Calculate Shannon entropy with lazy computation and caching: H(X) = -∑ P(x) log₂ P(x)

pub fn compute_entropy(&self, config: &AnomalyGridConfig) -> f64

Calculate Shannon entropy without caching (for immutable access)

pub fn calculate_kl_divergence(&mut self, config: &AnomalyGridConfig) -> f64

Calculate KL divergence from uniform distribution with lazy computation and caching

pub fn compute_kl_divergence(&self, config: &AnomalyGridConfig) -> f64

Calculate KL divergence from uniform distribution without caching (for immutable access)

pub fn get_all_probabilities( &self, config: &AnomalyGridConfig, ) -> HashMap<String, f64>

Get all probabilities as a HashMap (for compatibility with existing code)

Note: This creates temporary storage and should be used sparingly

pub fn reset(&mut self, interner: Arc<StringInterner>)

Reset the context node for reuse in memory pool

pub fn clear(&mut self)

Clear the context node data for memory pool return

pub fn cache_stats(&self) -> (bool, bool)

Get cache hit statistics for monitoring

Trait Implementations

impl Clone for ContextNode

fn clone(&self) -> ContextNode

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ContextNode

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ContextNode

fn default() -> Self

Returns the “default value” for a type.