Struct EpisodeTracker 

pub struct EpisodeTracker { /* private fields */ }

Tracks episode boundaries within a ring buffer.

Maintains a list of EpisodeMeta entries, invalidating episodes whose data has been overwritten by the ring buffer’s write pointer. Provides efficient sampling of contiguous windows for sequence models.

Implementations

impl EpisodeTracker

pub fn new(ring_capacity: usize) -> Self

Create a new episode tracker for a ring buffer with the given capacity.

pub fn notify_push(&mut self, write_pos: usize, done: bool)

Record a push at write_pos. If done, the current episode is finalized.

pub fn invalidate_overwritten(&mut self, write_pos: usize, count: usize)

Remove any episodes whose transitions have been overwritten.

Called when the ring buffer wraps. An episode is invalidated if any of its positions overlap with the region [write_pos, write_pos + count) modulo the ring capacity.

pub fn num_complete_episodes(&self) -> usize

Number of complete episodes currently tracked.

pub fn episodes(&self) -> &[EpisodeMeta]

All currently tracked episodes (complete and in-progress).

pub fn eligible_episodes(&self, min_length: usize) -> Vec<usize>

Indices of episodes long enough for a given sequence length.

pub fn sample_windows( &self, batch_size: usize, seq_len: usize, seed: u64, ) -> Result<Vec<EpisodeWindow>, RloxError>

Sample batch_size windows of seq_len consecutive transitions, each entirely within a single complete episode.

Uses ChaCha8Rng seeded with seed.

Trait Implementations

impl Debug for EpisodeTracker

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

Formats the value using the given formatter.

impl EpisodeAware for EpisodeTracker

fn notify_push(&mut self, write_pos: usize, done: bool)

Notify that a transition was pushed at write_pos with done flag.

fn invalidate_overwritten(&mut self, write_pos: usize, count: usize)

Invalidate any episodes that overlap with the overwritten region.

fn num_complete_episodes(&self) -> usize

Number of complete (terminated/truncated) episodes currently tracked.