ATLAS Offline Software
Loading...
Searching...
No Matches
columnar::TestUtils::ColumnReaderVector< T, Reader > Struct Template Referencefinal

#include <ColumnReaderBranch.h>

Inheritance diagram for columnar::TestUtils::ColumnReaderVector< T, Reader >:
Collaboration diagram for columnar::TestUtils::ColumnReaderVector< T, Reader >:

Public Member Functions

 ColumnReaderVector (const std::string &val_branchName)
virtual bool connect (Backend source, 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 (ColumnVectorData &columnData) 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

Reader< std::vector< T > > branchReader
const std::vector< ColumnarOffsetType > * offsetColumn = nullptr
std::vector< ColumnarOffsetType > offsets = {0}
std::vector< T > outData
Benchmark benchmarkUnpack
Benchmark benchmark
unsigned entries = 0
std::vector< OutputColumnInfo > outputColumns

Detailed Description

template<typename T, template< typename > class Reader>
struct columnar::TestUtils::ColumnReaderVector< T, Reader >

Definition at line 346 of file ColumnReaderBranch.h.

Constructor & Destructor Documentation

◆ ColumnReaderVector()

template<typename T, template< typename > class Reader>
columnar::TestUtils::ColumnReaderVector< T, Reader >::ColumnReaderVector ( const std::string & val_branchName)
inlineexplicit

Definition at line 356 of file ColumnReaderBranch.h.

357 : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
358 {
359 outputColumns.push_back ({.name = branchReader.columnName()});
360 outputColumns.push_back ({.name = branchReader.containerName(), .isOffset = true, .primary = false});
361 }
std::vector< OutputColumnInfo > outputColumns

Member Function Documentation

◆ clearColumns()

template<typename T, template< typename > class Reader>
virtual void columnar::TestUtils::ColumnReaderVector< T, Reader >::clearColumns ( )
inlineoverridevirtual

Implements columnar::TestUtils::ColumnReader.

Definition at line 392 of file ColumnReaderBranch.h.

393 {
394 offsets.clear ();
395 offsets.push_back (0);
396 outData.clear ();
397 }
std::vector< ColumnarOffsetType > offsets

◆ collectColumnData()

template<typename T, template< typename > class Reader>
virtual void columnar::TestUtils::ColumnReaderVector< T, Reader >::collectColumnData ( )
inlineoverridevirtual

Implements columnar::TestUtils::ColumnReader.

Definition at line 440 of file ColumnReaderBranch.h.

441 {
442 entries += outData.size();
443 }

◆ connect()

template<typename T, template< typename > class Reader>
virtual bool columnar::TestUtils::ColumnReaderVector< T, Reader >::connect ( Backend source,
std::unordered_map< std::string, const std::vector< ColumnarOffsetType > * > & offsetColumns,
std::unordered_map< std::string, ColumnInfo > & requestedColumns )
inlineoverridevirtual

Implements columnar::TestUtils::ColumnReader.

Definition at line 363 of file ColumnReaderBranch.h.

364 {
365 auto iter = requestedColumns.find (outputColumns.at(0).name);
366 if (iter == requestedColumns.end())
367 return false;
368 outputColumns.at(0).enabled = true;
369
370 branchReader.connectTree(source);
371
372 if (iter->second.offsetName != outputColumns.at(1).name)
373 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
374
375 requestedColumns.erase (iter);
376
377 if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
378 offsetColumn = offsetIter->second;
379 else
380 offsetColumns.emplace (outputColumns.at(1).name, &offsets);
381
382 iter = requestedColumns.find (outputColumns.at(1).name);
383 if (iter != requestedColumns.end())
384 {
385 requestedColumns.erase (iter);
386 outputColumns.at(1).enabled = true;
387 }
388
389 return true;
390 }
const std::vector< ColumnarOffsetType > * offsetColumn

◆ connectColumnIndices()

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

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

Definition at line 66 of file ColumnReader.h.

67 {
68 for (auto& col : outputColumns)
69 {
70 if (col.enabled)
71 col.columnIndex = header.getColumnIndex (col.name);
72 }
73 }

◆ getEntry()

template<typename T, template< typename > class Reader>
virtual void columnar::TestUtils::ColumnReaderVector< T, Reader >::getEntry ( Long64_t entry)
inlineoverridevirtual

Implements columnar::TestUtils::ColumnReader.

Definition at line 399 of file ColumnReaderBranch.h.

400 {
401 benchmark.startTimer ();
402 const auto& branchData = branchReader.getEntry (entry);
403 benchmark.stopTimer ();
404 benchmarkUnpack.startTimer ();
405 outData.insert (outData.end(), branchData.begin(), branchData.end());
406 offsets.push_back (outData.size());
407 benchmarkUnpack.stopTimer ();
408 }

◆ getPerfData()

template<typename T, template< typename > class Reader>
virtual BranchPerfData columnar::TestUtils::ColumnReaderVector< T, Reader >::getPerfData ( float emptyTime)
inlinenodiscardoverridevirtual

Implements columnar::TestUtils::ColumnReader.

Definition at line 425 of file ColumnReaderBranch.h.

426 {
428 result.name = branchReader.columnName();
429 result.timeRead = benchmark.getEntryTime(emptyTime);
430 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
431 benchmark.setSilence();
432 benchmarkUnpack.setSilence();
433 result.entrySize = branchReader.entrySize();
434 result.uncompressedSize = branchReader.uncompressedSize();
435 result.numBaskets = branchReader.numBaskets();
437 return result;
438 }

◆ setData()

template<typename T, template< typename > class Reader>
virtual void columnar::TestUtils::ColumnReaderVector< T, Reader >::setData ( ColumnVectorData & columnData)
inlineoverridevirtual

Implements columnar::TestUtils::ColumnReader.

Definition at line 410 of file ColumnReaderBranch.h.

411 {
412 if (outputColumns.at(0).columnIndex != ColumnVectorHeader::nullIndex)
413 columnData.setColumn (outputColumns.at(0).columnIndex, outData.size(), outData.data());
414 if (outputColumns.at(1).columnIndex != ColumnVectorHeader::nullIndex)
415 columnData.setColumn (outputColumns.at(1).columnIndex, offsets.size(), offsets.data());
416 if (offsetColumn)
417 {
418 if (offsetColumn->size() != offsets.size())
419 throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
420 if (offsetColumn->back() != offsets.back())
421 throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
422 }
423 }

Member Data Documentation

◆ benchmark

template<typename T, template< typename > class Reader>
Benchmark columnar::TestUtils::ColumnReaderVector< T, Reader >::benchmark

Definition at line 353 of file ColumnReaderBranch.h.

◆ benchmarkUnpack

template<typename T, template< typename > class Reader>
Benchmark columnar::TestUtils::ColumnReaderVector< T, Reader >::benchmarkUnpack

Definition at line 352 of file ColumnReaderBranch.h.

◆ branchReader

template<typename T, template< typename > class Reader>
Reader<std::vector<T> > columnar::TestUtils::ColumnReaderVector< T, Reader >::branchReader

Definition at line 348 of file ColumnReaderBranch.h.

◆ entries

template<typename T, template< typename > class Reader>
unsigned columnar::TestUtils::ColumnReaderVector< T, Reader >::entries = 0

Definition at line 354 of file ColumnReaderBranch.h.

◆ offsetColumn

template<typename T, template< typename > class Reader>
const std::vector<ColumnarOffsetType>* columnar::TestUtils::ColumnReaderVector< T, Reader >::offsetColumn = nullptr

Definition at line 349 of file ColumnReaderBranch.h.

◆ offsets

template<typename T, template< typename > class Reader>
std::vector<ColumnarOffsetType> columnar::TestUtils::ColumnReaderVector< T, Reader >::offsets = {0}

Definition at line 350 of file ColumnReaderBranch.h.

350{0};

◆ outData

template<typename T, template< typename > class Reader>
std::vector<T> columnar::TestUtils::ColumnReaderVector< T, Reader >::outData

Definition at line 351 of file ColumnReaderBranch.h.

◆ outputColumns

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

Definition at line 59 of file ColumnReader.h.


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