Inheritance diagram for columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >:

Collaboration diagram for columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >:

Public Member Functions
	ColumnDataVectorVectorLink (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 (ColumnarToolWrapperData &tool) override

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

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

std::vector< ColumnarOffsetType >	columnData

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

SG::sgkey_t	targetKey = 0

std::string	targetContainerName

Benchmark	benchmarkUnpack

Benchmark	benchmark

std::vector< OutputColumnInfo >	outputColumns

Detailed Description

template<typename T>
struct columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >

Definition at line 498 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorVectorLink()

template<typename T >

columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >::ColumnDataVectorVectorLink ( const std::string & val_branchName )

inlineexplicit

Definition at line 509 of file ColumnarPhysliteTest.cxx.

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

Member Function Documentation

◆ clearColumns()

template<typename T >

virtual void columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >::clearColumns ( )

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 546 of file ColumnarPhysliteTest.cxx.

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

◆ connect()

template<typename T >

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

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 516 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);
  
         targetContainerName = iter->second.linkToName;
         if (auto keyIter = knownKeys.find (targetContainerName); keyIter != knownKeys.end())
           targetKey = keyIter->second;
         if (auto offsetIter = offsetColumns.find (iter->second.linkToName); offsetIter != offsetColumns.end())
           targetOffsetColumn = offsetIter->second;
         else
           throw std::runtime_error ("missing offset column: " + iter->second.linkToName);
  
         requestedColumns.erase (iter);
  
         iter = requestedColumns.find (outputColumns.at(1).name);
         if (iter == requestedColumns.end())
           return true;
         requestedColumns.erase (iter);
         outputColumns.at(1).enabled = true;
         return true;
       }

◆ getEntry()

template<typename T >

virtual void columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >::getEntry ( Long64_t entry )

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 553 of file ColumnarPhysliteTest.cxx.

       {
         benchmark.startTimer ();
         const auto& branchData = branchReader.getEntry (entry);
         benchmark.stopTimer ();
         benchmarkUnpack.startTimer ();
         if (targetOffsetColumn->size() < 2)
           throw std::runtime_error ("target offset column not yet filled for: " + outputColumns.at(0).name);
         for (auto& data : branchData)
         {
           for (auto& element : data)
           {
             if (element.isDefault() || (element.key() == 0 && element.index() == 0))
               columnData.push_back (invalidObjectIndex);
             else
             {
               columnData.push_back (element.index() + targetOffsetColumn->at (targetOffsetColumn->size()-2));
               if (element.key() != targetKey)
               {
                 if (targetKey == 0)
                 {
                   targetKey = element.key();
                   std::cout << "assume target key for " << targetContainerName << " is " << std::hex << targetKey << std::dec << std::endl;
                 } else
                 {
                   throw std::runtime_error(
                       std::format("target key mismatch: {:x} != {:x} for {} with element index {}",
                                   element.key(), targetKey, outputColumns.at(0).name, element.index())
                   );
                 }
               }
             }
           }
           offsets.push_back (columnData.size());
         }
         benchmarkUnpack.stopTimer ();
       }

◆ setData()

template<typename T >

virtual void columnar::PhysliteTestHelpers::ColumnDataVectorVectorLink< T >::setData ( ColumnarToolWrapperData & tool )

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 591 of file ColumnarPhysliteTest.cxx.

       {
         if (outputColumns.at(0).enabled)
           tool.setColumn (outputColumns.at(0).name, columnData.size(), columnData.data());
         if (outputColumns.at(1).enabled)
           tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
       } 