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

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

Public Types
using	CM = ColumnarModeArray

Public Member Functions
	ColumnDataVectorLink (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< ElementLink< T > > >	branchReader

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

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

std::vector< typename CM::LinkIndexType >	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::TestUtils::ColumnDataVectorLink< T >

Definition at line 781 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

template<typename T >

using columnar::TestUtils::ColumnDataVectorLink< T >::CM = ColumnarModeArray

Definition at line 783 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorLink()

template<typename T >

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

inline

Definition at line 794 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::ColumnDataVectorLink< T >::clearColumns ( )

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 840 of file ColumnarPhysliteTest.cxx.

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

◆ connect()

template<typename T >

virtual bool columnar::TestUtils::ColumnDataVectorLink< 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 801 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);
  
         if (iter->second.linkTargetNames.size() != 1)
           throw std::runtime_error ("expected exactly one link target name for: " + outputColumns.at(0).name);
         targetContainerName = iter->second.linkTargetNames.at(0);
         if (auto keyIter = knownKeys.find (targetContainerName); keyIter != knownKeys.end())
           targetKey = keyIter->second;
         if (auto targetOffsetIter = offsetColumns.find (iter->second.linkTargetNames.at(0)); targetOffsetIter != offsetColumns.end())
           targetOffsetColumn = targetOffsetIter->second;
         else
           throw std::runtime_error ("missing offset column: " + iter->second.linkTargetNames.at(0));
  
         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())
         {
           outputColumns.at(1).enabled = true;
           requestedColumns.erase (iter);
         }
  
         return true;
       }

◆ getEntry()

template<typename T >

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

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 847 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& element : branchData)
         {
           if (element.isDefault())
             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 ("target key mismatch: " + std::to_string (element.key()) + " != " + std::to_string (targetKey) + " for " + outputColumns.at(0).name);
               }
             }
           }
         }
         offsets.push_back (columnData.size());
         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);
         }
         benchmarkUnpack.stopTimer ();
       }

◆ setData()

template<typename T >

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

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 886 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());
       }