ATLAS Offline Software
Public Types | Public Member Functions | Public Attributes | List of all members
columnar::TestUtils::ColumnDataVectorVectorLink< T > Struct Template Referencefinal
Inheritance diagram for columnar::TestUtils::ColumnDataVectorVectorLink< T >:
Collaboration diagram for columnar::TestUtils::ColumnDataVectorVectorLink< T >:

Public Types

using CM = ColumnarModeArray
 

Public Member Functions

 ColumnDataVectorVectorLink (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< std::vector< ElementLink< T > > > > branchReader
 
std::vector< ColumnarOffsetTypeoffsets = {0}
 
std::vector< typename CM::LinkIndexTypecolumnData
 
const std::vector< ColumnarOffsetType > * targetOffsetColumn = nullptr
 
SG::sgkey_t targetKey = 0
 
std::string targetContainerName
 
Benchmark benchmarkUnpack
 
Benchmark benchmark
 
std::vector< OutputColumnInfooutputColumns
 

Detailed Description

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

Definition at line 728 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

Definition at line 730 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorVectorLink()

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

Definition at line 740 of file ColumnarPhysliteTest.cxx.

742  {
743  outputColumns.push_back ({.name = branchReader.columnName() + ".data"});
744  outputColumns.push_back ({.name = branchReader.columnName() + ".offset", .isOffset = true});
745  }

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 779 of file ColumnarPhysliteTest.cxx.

780  {
781  columnData.clear();
782  offsets.clear();
783  offsets.push_back (0);
784  }

◆ connect()

template<typename T >
virtual bool columnar::TestUtils::ColumnDataVectorVectorLink< 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 747 of file ColumnarPhysliteTest.cxx.

748  {
749  auto iter = requestedColumns.find (outputColumns.at(0).name);
750  if (iter == requestedColumns.end())
751  return false;
752  outputColumns.at(0).enabled = true;
753 
755 
756  if (iter->second.offsetName != outputColumns.at(1).name)
757  throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
758 
759  if (iter->second.linkTargetNames.size() != 1)
760  throw std::runtime_error ("expected exactly one link target name for: " + outputColumns.at(0).name);
761  targetContainerName = iter->second.linkTargetNames.at(0);
762  if (auto keyIter = knownKeys.find (targetContainerName); keyIter != knownKeys.end())
763  targetKey = keyIter->second;
764  if (auto offsetIter = offsetColumns.find (iter->second.linkTargetNames.at(0)); offsetIter != offsetColumns.end())
765  targetOffsetColumn = offsetIter->second;
766  else
767  throw std::runtime_error ("missing offset column: " + iter->second.linkTargetNames.at(0));
768 
769  requestedColumns.erase (iter);
770 
771  iter = requestedColumns.find (outputColumns.at(1).name);
772  if (iter == requestedColumns.end())
773  return true;
774  requestedColumns.erase (iter);
775  outputColumns.at(1).enabled = true;
776  return true;
777  }

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 786 of file ColumnarPhysliteTest.cxx.

787  {
788  benchmark.startTimer ();
789  const auto& branchData = branchReader.getEntry (entry);
790  benchmark.stopTimer ();
791  benchmarkUnpack.startTimer ();
792  if (targetOffsetColumn->size() < 2)
793  throw std::runtime_error ("target offset column not yet filled for: " + outputColumns.at(0).name);
794  for (auto& data : branchData)
795  {
796  for (auto& element : data)
797  {
798  if (element.isDefault() || (element.key() == 0 && element.index() == 0))
799  columnData.push_back (invalidObjectIndex);
800  else
801  {
802  columnData.push_back (element.index() + targetOffsetColumn->at (targetOffsetColumn->size()-2));
803  if (element.key() != targetKey)
804  {
805  if (targetKey == 0)
806  {
807  targetKey = element.key();
808  std::cout << "assume target key for " << targetContainerName << " is " << std::hex << targetKey << std::dec << std::endl;
809  } else
810  {
811  throw std::runtime_error(
812  std::format("target key mismatch: {:x} != {:x} for {} with element index {}",
813  element.key(), targetKey, outputColumns.at(0).name, element.index())
814  );
815  }
816  }
817  }
818  }
819  offsets.push_back (columnData.size());
820  }
821  benchmarkUnpack.stopTimer ();
822  }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 832 of file ColumnarPhysliteTest.cxx.

833  {
834  BranchPerfData result;
835  result.name = branchReader.columnName();
836  result.timeRead = benchmark.getEntryTime(emptyTime);
837  result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
838  benchmark.setSilence();
839  benchmarkUnpack.setSilence();
840  result.entrySize = branchReader.entrySize();
841  result.uncompressedSize = branchReader.uncompressedSize();
842  result.numBaskets = branchReader.numBaskets();
843  return result;
844  }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 824 of file ColumnarPhysliteTest.cxx.

825  {
826  if (outputColumns.at(0).enabled)
827  tool.setColumn (outputColumns.at(0).name, columnData.size(), columnData.data());
828  if (outputColumns.at(1).enabled)
829  tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
830  }

Member Data Documentation

◆ benchmark

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

Definition at line 738 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 737 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 731 of file ColumnarPhysliteTest.cxx.

◆ columnData

template<typename T >
std::vector<typename CM::LinkIndexType> columnar::TestUtils::ColumnDataVectorVectorLink< T >::columnData

Definition at line 733 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 732 of file ColumnarPhysliteTest.cxx.

◆ outputColumns

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

Definition at line 369 of file ColumnarPhysliteTest.cxx.

◆ targetContainerName

template<typename T >
std::string columnar::TestUtils::ColumnDataVectorVectorLink< T >::targetContainerName

Definition at line 736 of file ColumnarPhysliteTest.cxx.

◆ targetKey

template<typename T >
SG::sgkey_t columnar::TestUtils::ColumnDataVectorVectorLink< T >::targetKey = 0

Definition at line 735 of file ColumnarPhysliteTest.cxx.

◆ targetOffsetColumn

template<typename T >
const std::vector<ColumnarOffsetType>* columnar::TestUtils::ColumnDataVectorVectorLink< T >::targetOffsetColumn = nullptr

Definition at line 734 of file ColumnarPhysliteTest.cxx.


The documentation for this struct was generated from the following file:
data
char data[hepevt_bytes_allocation_ATLAS]
Definition: HepEvt.cxx:11
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
vtune_athena.format
format
Definition: vtune_athena.py:14
columnar::TestUtils::BranchReader::connectTree
void connectTree(TTree *tree)
Definition: ColumnarPhysliteTest.cxx:218
tree
TChain * tree
Definition: tile_monitor.h:30
columnar::TestUtils::BranchReader::entrySize
std::optional< float > entrySize() const
Definition: ColumnarPhysliteTest.cxx:246
columnar::TestUtils::BranchReader::numBaskets
std::optional< unsigned > numBaskets()
Definition: ColumnarPhysliteTest.cxx:262
GetAllXsec.entry
list entry
Definition: GetAllXsec.py:132
AtlCoolConsole.tool
tool
Definition: AtlCoolConsole.py:452
columnar::invalidObjectIndex
constexpr ColumnarOffsetType invalidObjectIndex
the value for an invalid element index
Definition: IColumnarTool.h:25
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