ATLAS Offline Software
Loading...
Searching...
No Matches
columnar::TestUtils::ColumnDataVectorVector< T > Struct Template Referencefinal
Inheritance diagram for columnar::TestUtils::ColumnDataVectorVector< T >:
Collaboration diagram for columnar::TestUtils::ColumnDataVectorVector< T >:

Public Member Functions

 ColumnDataVectorVector (const std::string &val_branchName)
virtual bool connect (TTree *tree, std::unordered_map< std::string, const std::vector< ColumnarOffsetType > * > &, std::unordered_map< std::string, ColumnInfo > &requestedColumns) override
virtual void clearColumns () override
virtual void getEntry (Long64_t entry) override
virtual void setData (ColumnVectorData &colData) override
virtual BranchPerfData getPerfData (float emptyTime) override
virtual void collectColumnData () override
void connectColumnIndices (const ColumnVectorHeader &header)
 lookup and store column indices from the header for all enabled output columns

Public Attributes

BranchReader< std::vector< std::vector< T > > > branchReader
std::vector< ColumnarOffsetTypeoffsets = {0}
std::vector< T > columnData
Benchmark benchmarkUnpack
Benchmark benchmark
unsigned entries = 0
std::vector< OutputColumnInfooutputColumns

Detailed Description

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

Definition at line 609 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorVector()

template<typename T>
columnar::TestUtils::ColumnDataVectorVector< T >::ColumnDataVectorVector ( const std::string & val_branchName)
inlineexplicit

Definition at line 618 of file ColumnarPhysliteTest.cxx.

619 : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
620 {
621 outputColumns.push_back ({.name = branchReader.columnName() + ".data"});
622 outputColumns.push_back ({.name = branchReader.columnName() + ".offset", .isOffset = true});
623 }
std::vector< OutputColumnInfo > outputColumns
BranchReader< std::vector< std::vector< T > > > branchReader

Member Function Documentation

◆ clearColumns()

template<typename T>
virtual void columnar::TestUtils::ColumnDataVectorVector< T >::clearColumns ( )
inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 647 of file ColumnarPhysliteTest.cxx.

648 {
649 columnData.clear();
650 offsets.clear();
651 offsets.push_back (0);
652 }

◆ collectColumnData()

template<typename T>
virtual void columnar::TestUtils::ColumnDataVectorVector< T >::collectColumnData ( )
inlineoverridevirtual

◆ connect()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 625 of file ColumnarPhysliteTest.cxx.

626 {
627 auto iter = requestedColumns.find (outputColumns.at(0).name);
628 if (iter == requestedColumns.end())
629 return false;
630 outputColumns.at(0).enabled = true;
631
632 branchReader.connectTree (tree);
633
634 if (iter->second.offsetName != outputColumns.at(1).name)
635 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
636
637 requestedColumns.erase (iter);
638
639 iter = requestedColumns.find (outputColumns.at(1).name);
640 if (iter == requestedColumns.end())
641 return true;
642 requestedColumns.erase (iter);
643 outputColumns.at(1).enabled = true;
644 return true;
645 }

◆ connectColumnIndices()

void columnar::TestUtils::IColumnData::connectColumnIndices ( const ColumnVectorHeader & header)
inlineinherited

lookup and store column indices from the header for all enabled output columns

Definition at line 303 of file ColumnarPhysliteTest.cxx.

304 {
305 for (auto& col : outputColumns)
306 {
307 if (col.enabled)
308 col.columnIndex = header.getColumnIndex (col.name);
309 }
310 }

◆ getEntry()

template<typename T>
virtual void columnar::TestUtils::ColumnDataVectorVector< T >::getEntry ( Long64_t entry)
inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 654 of file ColumnarPhysliteTest.cxx.

655 {
656 benchmark.startTimer ();
657 const auto& branchData = branchReader.getEntry (entry);
658 benchmark.stopTimer ();
659 benchmarkUnpack.startTimer ();
660 for (auto& data : branchData)
661 {
662 columnData.insert (columnData.end(), data.begin(), data.end());
663 offsets.push_back (columnData.size());
664 }
665 benchmarkUnpack.stopTimer ();
666 }

◆ getPerfData()

template<typename T>
virtual BranchPerfData columnar::TestUtils::ColumnDataVectorVector< T >::getPerfData ( float emptyTime)
inlinenodiscardoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 676 of file ColumnarPhysliteTest.cxx.

677 {
679 result.name = branchReader.columnName();
680 result.timeRead = benchmark.getEntryTime(emptyTime);
681 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
682 benchmark.setSilence();
683 benchmarkUnpack.setSilence();
684 result.entrySize = branchReader.entrySize();
685 result.uncompressedSize = branchReader.uncompressedSize();
686 result.numBaskets = branchReader.numBaskets();
688 return result;
689 }

◆ setData()

template<typename T>
virtual void columnar::TestUtils::ColumnDataVectorVector< T >::setData ( ColumnVectorData & colData)
inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 668 of file ColumnarPhysliteTest.cxx.

669 {
670 if (outputColumns.at(0).columnIndex != ColumnVectorHeader::nullIndex)
671 colData.setColumn (outputColumns.at(0).columnIndex, columnData.size(), columnData.data());
672 if (outputColumns.at(1).columnIndex != ColumnVectorHeader::nullIndex)
673 colData.setColumn (outputColumns.at(1).columnIndex, offsets.size(), offsets.data());
674 }

Member Data Documentation

◆ benchmark

template<typename T>
Benchmark columnar::TestUtils::ColumnDataVectorVector< T >::benchmark

Definition at line 615 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

template<typename T>
Benchmark columnar::TestUtils::ColumnDataVectorVector< T >::benchmarkUnpack

Definition at line 614 of file ColumnarPhysliteTest.cxx.

◆ branchReader

template<typename T>
BranchReader<std::vector<std::vector<T> > > columnar::TestUtils::ColumnDataVectorVector< T >::branchReader

Definition at line 611 of file ColumnarPhysliteTest.cxx.

◆ columnData

template<typename T>
std::vector<T> columnar::TestUtils::ColumnDataVectorVector< T >::columnData

Definition at line 613 of file ColumnarPhysliteTest.cxx.

◆ entries

template<typename T>
unsigned columnar::TestUtils::ColumnDataVectorVector< T >::entries = 0

Definition at line 616 of file ColumnarPhysliteTest.cxx.

◆ offsets

template<typename T>
std::vector<ColumnarOffsetType> columnar::TestUtils::ColumnDataVectorVector< T >::offsets = {0}

Definition at line 612 of file ColumnarPhysliteTest.cxx.

612{0};

◆ outputColumns

std::vector<OutputColumnInfo> columnar::TestUtils::IColumnData::outputColumns
inherited

Definition at line 296 of file ColumnarPhysliteTest.cxx.


The documentation for this struct was generated from the following file: