ATLAS Offline Software
Public Member Functions | Public Attributes | List of all members
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
 

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
 
std::vector< OutputColumnInfooutputColumns
 

Detailed Description

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

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

504  {
505  outputColumns.push_back ({.name = branchReader.columnName()});
506  outputColumns.push_back ({.name = branchReader.containerName(), .isOffset = true, .primary = false});
507  }

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 538 of file ColumnarPhysliteTest.cxx.

539  {
540  offsets.clear ();
541  offsets.push_back (0);
542  outData.clear ();
543  }

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

510  {
511  auto iter = requestedColumns.find (outputColumns.at(0).name);
512  if (iter == requestedColumns.end())
513  return false;
514  outputColumns.at(0).enabled = true;
515 
517 
518  if (iter->second.offsetName != outputColumns.at(1).name)
519  throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
520 
521  requestedColumns.erase (iter);
522 
523  if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
524  offsetColumn = offsetIter->second;
525  else
526  offsetColumns.emplace (outputColumns.at(1).name, &offsets);
527 
528  iter = requestedColumns.find (outputColumns.at(1).name);
529  if (iter != requestedColumns.end())
530  {
531  requestedColumns.erase (iter);
532  outputColumns.at(1).enabled = true;
533  }
534 
535  return true;
536  }

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 545 of file ColumnarPhysliteTest.cxx.

546  {
547  benchmark.startTimer ();
548  const auto& branchData = branchReader.getEntry (entry);
549  benchmark.stopTimer ();
550  benchmarkUnpack.startTimer ();
551  outData.insert (outData.end(), branchData.begin(), branchData.end());
552  offsets.push_back (outData.size());
553  benchmarkUnpack.stopTimer ();
554  }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 571 of file ColumnarPhysliteTest.cxx.

572  {
573  BranchPerfData result;
574  result.name = branchReader.columnName();
575  result.timeRead = benchmark.getEntryTime(emptyTime);
576  result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
577  benchmark.setSilence();
578  benchmarkUnpack.setSilence();
579  result.entrySize = branchReader.entrySize();
580  result.uncompressedSize = branchReader.uncompressedSize();
581  result.numBaskets = branchReader.numBaskets();
582  return result;
583  }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 556 of file ColumnarPhysliteTest.cxx.

557  {
558  if (outputColumns.at(0).enabled)
559  tool.setColumn (outputColumns.at(0).name, outData.size(), outData.data());
560  if (outputColumns.at(1).enabled)
561  tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
562  if (offsetColumn)
563  {
564  if (offsetColumn->size() != offsets.size())
565  throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
566  if (offsetColumn->back() != offsets.back())
567  throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
568  }
569  }

Member Data Documentation

◆ benchmark

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

Definition at line 500 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 499 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 495 of file ColumnarPhysliteTest.cxx.

◆ offsetColumn

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

Definition at line 496 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 497 of file ColumnarPhysliteTest.cxx.

◆ outData

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

Definition at line 498 of file ColumnarPhysliteTest.cxx.

◆ outputColumns

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

Definition at line 369 of file ColumnarPhysliteTest.cxx.


The documentation for this struct was generated from the following file:
columnar::TestUtils::ColumnDataVector::branchReader
BranchReader< std::vector< T > > branchReader
Definition: ColumnarPhysliteTest.cxx:495
createLinkingScheme.iter
iter
Definition: createLinkingScheme.py:62
columnar::TestUtils::BranchReader::getEntry
const T & getEntry(Long64_t entry)
Definition: ColumnarPhysliteTest.cxx:230
get_generator_info.result
result
Definition: get_generator_info.py:21
columnar::TestUtils::BranchReader::connectTree
void connectTree(TTree *tree)
Definition: ColumnarPhysliteTest.cxx:218
columnar::TestUtils::BranchReader::containerName
std::string containerName() const
Definition: ColumnarPhysliteTest.cxx:206
tree
TChain * tree
Definition: tile_monitor.h:30
columnar::TestUtils::ColumnDataVector::benchmarkUnpack
Benchmark benchmarkUnpack
Definition: ColumnarPhysliteTest.cxx:499
columnar::TestUtils::BranchReader::entrySize
std::optional< float > entrySize() const
Definition: ColumnarPhysliteTest.cxx:246
StateLessPT_NewConfig.primary
primary
Definition: StateLessPT_NewConfig.py:234
columnar::TestUtils::BranchReader::numBaskets
std::optional< unsigned > numBaskets()
Definition: ColumnarPhysliteTest.cxx:262
columnar::TestUtils::ColumnDataVector::offsetColumn
const std::vector< ColumnarOffsetType > * offsetColumn
Definition: ColumnarPhysliteTest.cxx:496
columnar::TestUtils::ColumnDataVector::benchmark
Benchmark benchmark
Definition: ColumnarPhysliteTest.cxx:500
columnar::TestUtils::ColumnDataVector::offsets
std::vector< ColumnarOffsetType > offsets
Definition: ColumnarPhysliteTest.cxx:497
GetAllXsec.entry
list entry
Definition: GetAllXsec.py:132
AtlCoolConsole.tool
tool
Definition: AtlCoolConsole.py:452
columnar::TestUtils::IColumnData::outputColumns
std::vector< OutputColumnInfo > outputColumns
Definition: ColumnarPhysliteTest.cxx:369
columnar::TestUtils::BranchReader::columnName
std::string columnName() const
Definition: ColumnarPhysliteTest.cxx:194
columnar::TestUtils::BranchReader::uncompressedSize
std::optional< float > uncompressedSize() const
Definition: ColumnarPhysliteTest.cxx:253
columnar::TestUtils::ColumnDataVector::outData
std::vector< T > outData
Definition: ColumnarPhysliteTest.cxx:498