ATLAS Offline Software
Loading...
Searching...
No Matches
columnar::TestUtils::ColumnDataVector< T > Struct Template Referencefinal
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
virtual BranchPerfData getPerfData (float emptyTime) override
virtual void collectColumnData () override

Public Attributes

BranchReader< std::vector< T > > branchReader
const std::vector< ColumnarOffsetType > * offsetColumn = nullptr
std::vector< ColumnarOffsetTypeoffsets = {0}
std::vector< T > outData
Benchmark benchmarkUnpack
Benchmark benchmark
unsigned entries = 0
std::vector< OutputColumnInfooutputColumns

Detailed Description

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

Definition at line 434 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 444 of file ColumnarPhysliteTest.cxx.

445 : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
446 {
447 outputColumns.push_back ({.name = branchReader.columnName()});
448 outputColumns.push_back ({.name = branchReader.containerName(), .isOffset = true, .primary = false});
449 }
std::vector< OutputColumnInfo > outputColumns
BranchReader< std::vector< T > > branchReader

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 480 of file ColumnarPhysliteTest.cxx.

481 {
482 offsets.clear ();
483 offsets.push_back (0);
484 outData.clear ();
485 }
std::vector< ColumnarOffsetType > offsets

◆ collectColumnData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 528 of file ColumnarPhysliteTest.cxx.

◆ 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 451 of file ColumnarPhysliteTest.cxx.

452 {
453 auto iter = requestedColumns.find (outputColumns.at(0).name);
454 if (iter == requestedColumns.end())
455 return false;
456 outputColumns.at(0).enabled = true;
457
458 branchReader.connectTree (tree);
459
460 if (iter->second.offsetName != outputColumns.at(1).name)
461 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
462
463 requestedColumns.erase (iter);
464
465 if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
466 offsetColumn = offsetIter->second;
467 else
468 offsetColumns.emplace (outputColumns.at(1).name, &offsets);
469
470 iter = requestedColumns.find (outputColumns.at(1).name);
471 if (iter != requestedColumns.end())
472 {
473 requestedColumns.erase (iter);
474 outputColumns.at(1).enabled = true;
475 }
476
477 return true;
478 }
const std::vector< ColumnarOffsetType > * offsetColumn

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 487 of file ColumnarPhysliteTest.cxx.

488 {
489 benchmark.startTimer ();
490 const auto& branchData = branchReader.getEntry (entry);
491 benchmark.stopTimer ();
492 benchmarkUnpack.startTimer ();
493 outData.insert (outData.end(), branchData.begin(), branchData.end());
494 offsets.push_back (outData.size());
495 benchmarkUnpack.stopTimer ();
496 }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 513 of file ColumnarPhysliteTest.cxx.

514 {
516 result.name = branchReader.columnName();
517 result.timeRead = benchmark.getEntryTime(emptyTime);
518 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
519 benchmark.setSilence();
520 benchmarkUnpack.setSilence();
521 result.entrySize = branchReader.entrySize();
522 result.uncompressedSize = branchReader.uncompressedSize();
523 result.numBaskets = branchReader.numBaskets();
525 return result;
526 }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 498 of file ColumnarPhysliteTest.cxx.

499 {
500 if (outputColumns.at(0).enabled)
501 tool.setColumn (outputColumns.at(0).name, outData.size(), outData.data());
502 if (outputColumns.at(1).enabled)
503 tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
504 if (offsetColumn)
505 {
506 if (offsetColumn->size() != offsets.size())
507 throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
508 if (offsetColumn->back() != offsets.back())
509 throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
510 }
511 }

Member Data Documentation

◆ benchmark

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

Definition at line 441 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 440 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 436 of file ColumnarPhysliteTest.cxx.

◆ entries

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

Definition at line 442 of file ColumnarPhysliteTest.cxx.

◆ offsetColumn

template<typename T>
const std::vector<ColumnarOffsetType>* columnar::TestUtils::ColumnDataVector< T >::offsetColumn = nullptr

Definition at line 437 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 438 of file ColumnarPhysliteTest.cxx.

438{0};

◆ outData

template<typename T>
std::vector<T> columnar::TestUtils::ColumnDataVector< T >::outData

Definition at line 439 of file ColumnarPhysliteTest.cxx.

◆ outputColumns

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

Definition at line 298 of file ColumnarPhysliteTest.cxx.


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