Struct NDArrayCollector 

pub struct NDArrayCollector<F> { /* private fields */ }

Collects replay frames into a dense 2D feature matrix.

Implementations

impl<F> NDArrayCollector<F>

pub fn new( feature_adders: FeatureAdders<F>, player_feature_adders: PlayerFeatureAdders<F>, ) -> Self

Creates a collector from explicit global and per-player feature adders.

pub fn get_column_headers(&self) -> NDArrayColumnHeaders

Returns the column headers implied by the configured feature adders.

pub fn get_ndarray(self) -> SubtrActorResult<Array2<F>>

Finalizes collection and returns only the ndarray payload.

pub fn get_meta_and_ndarray( self, ) -> SubtrActorResult<(ReplayMetaWithHeaders, Array2<F>)>

Finalizes collection and returns replay metadata alongside the ndarray.

pub fn process_and_get_meta_and_headers( &mut self, replay: &Replay, ) -> SubtrActorResult<ReplayMetaWithHeaders>

Processes enough of a replay to determine metadata and column headers.

impl<F> NDArrayCollector<F>

where F: TryFrom<f32> + Send + Sync + 'static, <F as TryFrom<f32>>::Error: Debug,

pub fn from_strings_typed( fa_names: &[&str], pfa_names: &[&str], ) -> SubtrActorResult<Self>

Builds a collector from the registered string names of feature adders.

impl NDArrayCollector<f32>

pub fn from_strings( fa_names: &[&str], pfa_names: &[&str], ) -> SubtrActorResult<Self>

Builds an f32 collector from the registered string names of feature adders.

Trait Implementations

impl<F> Collector for NDArrayCollector<F>

fn process_frame( &mut self, processor: &ReplayProcessor<'_>, frame: &Frame, frame_number: usize, current_time: f32, ) -> SubtrActorResult<TimeAdvance>

Process a single frame from a replay.

fn process_replay(self, replay: &Replay) -> SubtrActorResult<Self>

where Self: Sized,

Process an entire replay.

fn finish_replay( &mut self, _processor: &ReplayProcessor<'_>, ) -> SubtrActorResult<()>

Finalize replay-derived state after the last frame has been processed.

impl<F: TryFrom<f32> + Send + Sync + 'static> Default for NDArrayCollector<F>

where <F as TryFrom<f32>>::Error: Debug,

fn default() -> Self

Returns the “default value” for a type.