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
virtual BranchPerfData	getPerfData (float emptyTime) override
virtual void	collectColumnData () 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
unsigned	entries = 0
unsigned	nullEntries = 0
std::vector< OutputColumnInfo >	outputColumns

Detailed Description

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

Definition at line 934 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

template<typename T>

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

Definition at line 936 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 949 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 995 of file ColumnarPhysliteTest.cxx.

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

◆ collectColumnData()

template<typename T>

virtual void columnar::TestUtils::ColumnDataVectorLink< T >::collectColumnData ( )

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 1065 of file ColumnarPhysliteTest.cxx.

      {
        entries += columnData.size();
        for (const auto& index : columnData)
        {
          if (index == invalidObjectIndex)
            nullEntries += 1;
        }
      }

◆ 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 956 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(vector-link): " + 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 1002 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() || (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 ("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 ();
      }

◆ getPerfData()

template<typename T>

virtual BranchPerfData columnar::TestUtils::ColumnDataVectorLink< T >::getPerfData ( float emptyTime )

inlinenodiscardoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 1049 of file ColumnarPhysliteTest.cxx.

      {
        BranchPerfData result;
        result.name = branchReader.columnName();
        result.timeRead = benchmark.getEntryTime(emptyTime);
        result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
        benchmark.setSilence();
        benchmarkUnpack.setSilence();
        result.entrySize = branchReader.entrySize();
        result.uncompressedSize = branchReader.uncompressedSize();
        result.numBaskets = branchReader.numBaskets();
        result.entries = entries;
        result.nullEntries = nullEntries;
        return result;
      }

◆ setData()

template<typename T>

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

inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

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