Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ROOT::Experimental::Internal::ML::RTTreeBatchSink Class Referencefinal

Writes batches into a TTree, as one float leaf per scalar column and one std::vector<float> branch per vector column, matching the RNTuple sink's schema.

Definition at line 80 of file RBatchSink.hxx.

Public Member Functions

 RTTreeBatchSink (const RTTreeBatchSink &)=delete
 
 RTTreeBatchSink (RTTreeBatchSink &&)=delete
 
 RTTreeBatchSink (std::string_view dataset_name, std::string_view filename, std::vector< RColumnLayout > layout)
 
 ~RTTreeBatchSink () override
 
void Commit () override
 Flush everything to disk once after the last batch.
 
void FillBatch (const RFlat2DMatrix &batch)
 Write every row of batch.
 
void FillRow (const float *row) override
 Copy one row of fRowWidth floats into the writer's buffers and write the entry.
 
RTTreeBatchSinkoperator= (const RTTreeBatchSink &)=delete
 
RTTreeBatchSinkoperator= (RTTreeBatchSink &&)=delete
 

Protected Attributes

std::vector< RColumnLayoutfLayout
 
std::size_t fRowWidth {}
 

Private Attributes

std::unique_ptr< TFilefFile
 
std::vector< float > fRow
 
TTreefTree
 
std::vector< std::vector< float > > fVectorBuffers
 

#include <ROOT/ML/RBatchSink.hxx>

Inheritance diagram for ROOT::Experimental::Internal::ML::RTTreeBatchSink:
[legend]

Constructor & Destructor Documentation

◆ RTTreeBatchSink() [1/3]

ROOT::Experimental::Internal::ML::RTTreeBatchSink::RTTreeBatchSink ( std::string_view dataset_name,
std::string_view filename,
std::vector< RColumnLayout > layout )

Definition at line 36 of file RBatchSink.cxx.

◆ RTTreeBatchSink() [2/3]

ROOT::Experimental::Internal::ML::RTTreeBatchSink::RTTreeBatchSink ( const RTTreeBatchSink & )
delete

◆ RTTreeBatchSink() [3/3]

ROOT::Experimental::Internal::ML::RTTreeBatchSink::RTTreeBatchSink ( RTTreeBatchSink && )
delete

◆ ~RTTreeBatchSink()

ROOT::Experimental::Internal::ML::RTTreeBatchSink::~RTTreeBatchSink ( )
overridedefault

Member Function Documentation

◆ Commit()

void ROOT::Experimental::Internal::ML::RTTreeBatchSink::Commit ( )
overridevirtual

Flush everything to disk once after the last batch.

Implements ROOT::Experimental::Internal::ML::RBatchSink.

Definition at line 74 of file RBatchSink.cxx.

◆ FillBatch()

void ROOT::Experimental::Internal::ML::RBatchSink::FillBatch ( const RFlat2DMatrix & batch)
inherited

Write every row of batch.

Definition at line 25 of file RBatchSink.cxx.

◆ FillRow()

void ROOT::Experimental::Internal::ML::RTTreeBatchSink::FillRow ( const float * row)
overridevirtual

Copy one row of fRowWidth floats into the writer's buffers and write the entry.

Implements ROOT::Experimental::Internal::ML::RBatchSink.

Definition at line 61 of file RBatchSink.cxx.

◆ operator=() [1/2]

RTTreeBatchSink & ROOT::Experimental::Internal::ML::RTTreeBatchSink::operator= ( const RTTreeBatchSink & )
delete

◆ operator=() [2/2]

RTTreeBatchSink & ROOT::Experimental::Internal::ML::RTTreeBatchSink::operator= ( RTTreeBatchSink && )
delete

Member Data Documentation

◆ fFile

std::unique_ptr<TFile> ROOT::Experimental::Internal::ML::RTTreeBatchSink::fFile
private

Definition at line 81 of file RBatchSink.hxx.

◆ fLayout

std::vector<RColumnLayout> ROOT::Experimental::Internal::ML::RBatchSink::fLayout
protectedinherited

Definition at line 52 of file RBatchSink.hxx.

◆ fRow

std::vector<float> ROOT::Experimental::Internal::ML::RTTreeBatchSink::fRow
private

Definition at line 83 of file RBatchSink.hxx.

◆ fRowWidth

std::size_t ROOT::Experimental::Internal::ML::RBatchSink::fRowWidth {}
protectedinherited

Definition at line 53 of file RBatchSink.hxx.

◆ fTree

TTree* ROOT::Experimental::Internal::ML::RTTreeBatchSink::fTree
private

Definition at line 82 of file RBatchSink.hxx.

◆ fVectorBuffers

std::vector<std::vector<float> > ROOT::Experimental::Internal::ML::RTTreeBatchSink::fVectorBuffers
private

Definition at line 84 of file RBatchSink.hxx.

Libraries for ROOT::Experimental::Internal::ML::RTTreeBatchSink:

The documentation for this class was generated from the following files: