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

Public Types

using CM = ColumnarModeArray

Public Member Functions

 ColumnDataVectorLink (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 (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< ElementLink< T > > > branchReader
const std::vector< ColumnarOffsetType > * offsetColumn = nullptr
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
unsigned entries = 0
unsigned nullEntries = 0
std::vector< OutputColumnInfooutputColumns

Detailed Description

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

Definition at line 942 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

template<typename T>
using columnar::TestUtils::ColumnDataVectorLink< T >::CM = ColumnarModeArray

Definition at line 944 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorLink()

template<typename T>
columnar::TestUtils::ColumnDataVectorLink< T >::ColumnDataVectorLink ( const std::string & val_branchName)
inline

Definition at line 957 of file ColumnarPhysliteTest.cxx.

958 : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
959 {
960 outputColumns.push_back ({.name = branchReader.columnName()});
961 outputColumns.push_back ({.name = branchReader.containerName(), .isOffset = true, .primary = false});
962 }
std::vector< OutputColumnInfo > outputColumns

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1003 of file ColumnarPhysliteTest.cxx.

1004 {
1005 columnData.clear();
1006 offsets.clear();
1007 offsets.push_back (0);
1008 }

◆ collectColumnData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1073 of file ColumnarPhysliteTest.cxx.

1074 {
1075 entries += columnData.size();
1076 for (const auto& index : columnData)
1077 {
1079 nullEntries += 1;
1080 }
1081 }

◆ connect()

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

965 {
966 auto iter = requestedColumns.find (outputColumns.at(0).name);
967 if (iter == requestedColumns.end())
968 return false;
969 outputColumns.at(0).enabled = true;
970
971 branchReader.connectTree (tree);
972
973 if (iter->second.offsetName != outputColumns.at(1).name)
974 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
975
976 if (iter->second.linkTargetNames.size() != 1)
977 throw std::runtime_error ("expected exactly one link target name for: " + outputColumns.at(0).name);
978 targetContainerName = iter->second.linkTargetNames.at(0);
979 if (auto keyIter = knownKeys.find (targetContainerName); keyIter != knownKeys.end())
980 targetKey = keyIter->second;
981 if (auto targetOffsetIter = offsetColumns.find (iter->second.linkTargetNames.at(0)); targetOffsetIter != offsetColumns.end())
983 else
984 throw std::runtime_error ("missing offset column(vector-link): " + iter->second.linkTargetNames.at(0));
985
986 requestedColumns.erase (iter);
987
988 if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
989 offsetColumn = offsetIter->second;
990 else
991 offsetColumns.emplace (outputColumns.at(1).name, &offsets);
992
993 iter = requestedColumns.find (outputColumns.at(1).name);
994 if (iter != requestedColumns.end())
995 {
996 outputColumns.at(1).enabled = true;
997 requestedColumns.erase (iter);
998 }
999
1000 return true;
1001 }

◆ 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::ColumnDataVectorLink< T >::getEntry ( Long64_t entry)
inlineoverridevirtual

Implements columnar::TestUtils::IColumnData.

Definition at line 1010 of file ColumnarPhysliteTest.cxx.

1011 {
1012 benchmark.startTimer ();
1013 const auto& branchData = branchReader.getEntry (entry);
1014 benchmark.stopTimer ();
1015 benchmarkUnpack.startTimer ();
1016 if (targetOffsetColumn->size() < 2)
1017 throw std::runtime_error ("target offset column not yet filled for: " + outputColumns.at(0).name);
1018 for (auto& element : branchData)
1019 {
1020 if (element.isDefault() || (element.key() == 0 && element.index() == 0))
1021 columnData.push_back (invalidObjectIndex);
1022 else
1023 {
1024 columnData.push_back (element.index() + targetOffsetColumn->at (targetOffsetColumn->size()-2));
1025 if (element.key() != targetKey)
1026 {
1027 if (targetKey == 0)
1028 {
1029 targetKey = element.key();
1030 std::cout << "assume target key for " << targetContainerName << " is " << std::hex << targetKey << std::dec << std::endl;
1031 } else
1032 {
1033 throw std::runtime_error ("target key mismatch: " + std::to_string (element.key()) + " != " + std::to_string (targetKey) + " for " + outputColumns.at(0).name);
1034 }
1035 }
1036 }
1037 }
1038 offsets.push_back (columnData.size());
1039 if (offsetColumn)
1040 {
1041 if (offsetColumn->size() != offsets.size())
1042 throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
1043 if (offsetColumn->back() != offsets.back())
1044 throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
1045 }
1046 benchmarkUnpack.stopTimer ();
1047 }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1057 of file ColumnarPhysliteTest.cxx.

1058 {
1060 result.name = branchReader.columnName();
1061 result.timeRead = benchmark.getEntryTime(emptyTime);
1062 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
1063 benchmark.setSilence();
1064 benchmarkUnpack.setSilence();
1065 result.entrySize = branchReader.entrySize();
1066 result.uncompressedSize = branchReader.uncompressedSize();
1067 result.numBaskets = branchReader.numBaskets();
1070 return result;
1071 }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1049 of file ColumnarPhysliteTest.cxx.

1050 {
1051 if (outputColumns.at(0).columnIndex != ColumnVectorHeader::nullIndex)
1052 colData.setColumn (outputColumns.at(0).columnIndex, columnData.size(), columnData.data());
1053 if (outputColumns.at(1).columnIndex != ColumnVectorHeader::nullIndex)
1054 colData.setColumn (outputColumns.at(1).columnIndex, offsets.size(), offsets.data());
1055 }

Member Data Documentation

◆ benchmark

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

Definition at line 953 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 952 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 945 of file ColumnarPhysliteTest.cxx.

◆ columnData

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

Definition at line 948 of file ColumnarPhysliteTest.cxx.

◆ entries

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

Definition at line 954 of file ColumnarPhysliteTest.cxx.

◆ nullEntries

template<typename T>
unsigned columnar::TestUtils::ColumnDataVectorLink< T >::nullEntries = 0

Definition at line 955 of file ColumnarPhysliteTest.cxx.

◆ offsetColumn

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

Definition at line 946 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 947 of file ColumnarPhysliteTest.cxx.

947{0};

◆ outputColumns

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

Definition at line 296 of file ColumnarPhysliteTest.cxx.

◆ targetContainerName

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

Definition at line 951 of file ColumnarPhysliteTest.cxx.

◆ targetKey

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

Definition at line 950 of file ColumnarPhysliteTest.cxx.

◆ targetOffsetColumn

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

Definition at line 949 of file ColumnarPhysliteTest.cxx.


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