Inheritance diagram for columnar::TestUtils::ColumnDataVector< T >:

Collaboration diagram for columnar::TestUtils::ColumnDataVector< T >:

Public Member Functions
	ColumnDataVector (const std::string &val_branchName)

virtual bool	connect (TTree tree, std::unordered_map< std::string, const std::vector< ColumnarOffsetType > > &offsetColumns, std::unordered_map< std::string, ColumnInfo > &requestedColumns) override

virtual void	clearColumns () override

virtual void	getEntry (Long64_t entry) override

virtual void	setData (TestUtils::ToolWrapperData &tool) override

Public Attributes
BranchReader< std::vector< T > >	branchReader

const std::vector< ColumnarOffsetType > *	offsetColumn = nullptr

std::vector< ColumnarOffsetType >	offsets = {0}

std::vector< T >	outData

Benchmark	benchmarkUnpack

Benchmark	benchmark

std::vector< OutputColumnInfo >	outputColumns

Detailed Description

template<typename T>
struct columnar::TestUtils::ColumnDataVector< T >

Definition at line 386 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVector()

template<typename T >

columnar::TestUtils::ColumnDataVector< T >::ColumnDataVector ( const std::string & val_branchName )

inlineexplicit

Definition at line 395 of file ColumnarPhysliteTest.cxx.

         : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
       {
         outputColumns.push_back ({.name = branchReader.columnName()});
         outputColumns.push_back ({.name = branchReader.containerName(), .isOffset = true, .primary = false});
       }

Member Function Documentation

◆ clearColumns()

template<typename T >

virtual void columnar::TestUtils::ColumnDataVector< T >::clearColumns ( )

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 431 of file ColumnarPhysliteTest.cxx.

       {
         offsets.clear ();
         offsets.push_back (0);
         outData.clear ();
       }

◆ connect()

template<typename T >

virtual bool columnar::TestUtils::ColumnDataVector< T >::connect	(	TTree *	tree,
		std::unordered_map< std::string, const std::vector< ColumnarOffsetType > * > &	offsetColumns,
		std::unordered_map< std::string, ColumnInfo > &	requestedColumns
	)

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 402 of file ColumnarPhysliteTest.cxx.

       {
         auto iter = requestedColumns.find (outputColumns.at(0).name);
         if (iter == requestedColumns.end())
           return false;
         outputColumns.at(0).enabled = true;
  
         branchReader.connectTree (tree);
  
         if (iter->second.offsetName != outputColumns.at(1).name)
           throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
  
         requestedColumns.erase (iter);
  
         if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
           offsetColumn = offsetIter->second;
         else
           offsetColumns.emplace (outputColumns.at(1).name, &offsets);
  
         iter = requestedColumns.find (outputColumns.at(1).name);
         if (iter != requestedColumns.end())
         {
           requestedColumns.erase (iter);
           outputColumns.at(1).enabled = true;
         }
  
         return true;
       }

◆ getEntry()

template<typename T >

virtual void columnar::TestUtils::ColumnDataVector< T >::getEntry ( Long64_t entry )

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 438 of file ColumnarPhysliteTest.cxx.

       {
         benchmark.startTimer ();
         const auto& branchData = branchReader.getEntry (entry);
         benchmark.stopTimer ();
         benchmarkUnpack.startTimer ();
         outData.insert (outData.end(), branchData.begin(), branchData.end());
         offsets.push_back (outData.size());
         benchmarkUnpack.stopTimer ();
       }

◆ setData()

template<typename T >

virtual void columnar::TestUtils::ColumnDataVector< T >::setData ( TestUtils::ToolWrapperData & tool )

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 449 of file ColumnarPhysliteTest.cxx.

       {
         if (outputColumns.at(0).enabled)
           tool.setColumn (outputColumns.at(0).name, outData.size(), outData.data());
         if (outputColumns.at(1).enabled)
           tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
         if (offsetColumn)
         {
           if (offsetColumn->size() != offsets.size())
             throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
           if (offsetColumn->back() != offsets.back())
             throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
         }
       }