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

Public Types

using CM = ColumnarModeArray
 

Public Member Functions

 ColumnDataVectorVectorVariantLink (const std::string &val_branchName)
 
 ~ColumnDataVectorVectorVariantLink ()
 
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
 
std::vector< typename CM::LinkKeyTypekeysColumn
 
std::vector< std::string > containers
 
std::vector< SG::sgkey_tcontainerKeys
 
std::vector< const std::vector< ColumnarOffsetType > * > containerOffsets
 
Benchmark benchmarkUnpack
 
Benchmark benchmark
 
bool checkUnknownKeys = false
 
std::unordered_map< SG::sgkey_t, std::unordered_set< std::string > > unknownKeys
 
std::vector< OutputColumnInfooutputColumns
 

Detailed Description

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

Definition at line 1251 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

Definition at line 1253 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorVectorVariantLink()

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

Definition at line 1267 of file ColumnarPhysliteTest.cxx.

1268  : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
1269  {
1270  outputColumns.push_back ({.name = branchReader.columnName() + ".data"});
1271  outputColumns.push_back ({.name = branchReader.columnName() + ".offset", .isOffset = true});
1272  outputColumns.push_back ({.name = branchReader.columnName() + ".keys"});
1273  }

◆ ~ColumnDataVectorVectorVariantLink()

Definition at line 1275 of file ColumnarPhysliteTest.cxx.

1276  {
1277  // print unknown keys and containers they may be associated
1278  // with, based on whether they were always within the range of
1279  // elements allowed for the container.
1280  for (auto& [key, forbiddenContainer] : unknownKeys)
1281  {
1282  std::cout << "unknown key: " << std::hex << key << std::dec << ", allowed containers:";
1283  for (const auto& container : containers)
1284  {
1285  if (forbiddenContainer.find (container) == forbiddenContainer.end())
1286  std::cout << " " << container;
1287  }
1288  std::cout << std::endl;
1289  }
1290  }

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1343 of file ColumnarPhysliteTest.cxx.

1344  {
1345  columnData.clear();
1346  offsets.clear();
1347  offsets.push_back (0);
1348  }

◆ connect()

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

1293  {
1294  auto iter = requestedColumns.find (outputColumns.at(0).name);
1295  if (iter == requestedColumns.end())
1296  return false;
1297  outputColumns.at(0).enabled = true;
1298 
1300 
1301  if (iter->second.offsetName != outputColumns.at(1).name)
1302  throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
1303  containers = iter->second.linkTargetNames;
1304  if (containers.empty() || iter->second.variantLinkKeyColumn.empty())
1305  throw std::runtime_error ("no variant link containers for: " + outputColumns.at(0).name);
1306  if (iter->second.variantLinkKeyColumn != outputColumns.at(2).name)
1307  throw std::runtime_error ("variant link key column mismatch: " + iter->second.variantLinkKeyColumn + " != " + outputColumns.at(2).name);
1308 
1309  for ([[maybe_unused]] auto& container : containers)
1310  {
1311  keysColumn.push_back (keysColumn.size()+1);
1312  if (!offsetColumns.contains (container))
1313  throw std::runtime_error ("missing offset column(variant-link): " + container);
1314  containerOffsets.push_back (offsetColumns.at (container));
1315  if (auto iter = knownKeys.find (container); iter != knownKeys.end())
1316  {
1317  containerKeys.push_back (iter->second);
1318  } else
1319  {
1320  checkUnknownKeys = true;
1321  containerKeys.push_back (0u);
1322  }
1323  }
1324 
1325  requestedColumns.erase (iter);
1326 
1327  iter = requestedColumns.find (outputColumns.at(1).name);
1328  if (iter != requestedColumns.end())
1329  {
1330  outputColumns.at(1).enabled = true;
1331  requestedColumns.erase (iter);
1332  }
1333 
1334  iter = requestedColumns.find (outputColumns.at(2).name);
1335  if (iter != requestedColumns.end())
1336  {
1337  outputColumns.at(2).enabled = true;
1338  requestedColumns.erase (iter);
1339  }
1340  return true;
1341  }

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1350 of file ColumnarPhysliteTest.cxx.

1351  {
1352  benchmark.startTimer ();
1353  const auto& branchData = branchReader.getEntry (entry);
1354  benchmark.stopTimer ();
1355  benchmarkUnpack.startTimer ();
1356  for (auto& data : branchData)
1357  {
1358  for (auto& element : data)
1359  {
1360  if (element.isDefault())
1361  columnData.push_back (invalidObjectIndex);
1362  else
1363  {
1364  typename CM::LinkIndexType key = 0xff;
1365  typename CM::LinkIndexType index = 0;
1366  for (std::size_t i = 0; i < containers.size(); ++i)
1367  {
1368  if (element.key() == containerKeys[i])
1369  {
1370  if (containerOffsets[i]->back() <= element.index())
1371  throw std::runtime_error ("invalid index: " + std::to_string (element.index()) + " in container: " + containers[i] + " with size: " + std::to_string (containerOffsets[i]->back()));
1372  key = keysColumn[i];
1373  if (containerOffsets[i]->size() < 2)
1374  throw std::runtime_error ("container offset not yet filled for: " + containers[i]);
1375  index = containerOffsets[i]->at (containerOffsets[i]->size()-2) + element.index();
1376  break;
1377  }
1378  }
1379  if (key == 0xff && checkUnknownKeys)
1380  {
1381  // this records which containers the unknown key is
1382  // compatible with, so that I may figure out which
1383  // container it is and hard-code it above.
1384  auto& forbiddenContainers = unknownKeys[element.key()];
1385  for (std::size_t i = 0; i < containers.size(); ++i)
1386  {
1387  if (containerOffsets[i]->back() <= containerOffsets[i]->at (containerOffsets[i]->size()-2) + element.index())
1388  forbiddenContainers.insert (containers[i]);
1389  }
1390  }
1391  columnData.push_back (CM::mergeLinkKeyIndex (key, index));
1392  }
1393  }
1394  offsets.push_back (columnData.size());
1395  }
1396  benchmarkUnpack.stopTimer ();
1397  }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1409 of file ColumnarPhysliteTest.cxx.

1410  {
1411  BranchPerfData result;
1412  result.name = branchReader.columnName();
1413  result.timeRead = benchmark.getEntryTime(emptyTime);
1414  result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
1415  benchmark.setSilence();
1416  benchmarkUnpack.setSilence();
1417  result.entrySize = branchReader.entrySize();
1418  result.uncompressedSize = branchReader.uncompressedSize();
1419  result.numBaskets = branchReader.numBaskets();
1420  return result;
1421  }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1399 of file ColumnarPhysliteTest.cxx.

1400  {
1401  if (outputColumns.at(0).enabled)
1402  tool.setColumn (outputColumns.at(0).name, columnData.size(), columnData.data());
1403  if (outputColumns.at(1).enabled)
1404  tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
1405  if (outputColumns.at(2).enabled)
1406  tool.setColumn (outputColumns.at(2).name, keysColumn.size(), keysColumn.data());
1407  }

Member Data Documentation

◆ benchmark

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

Definition at line 1262 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 1261 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 1254 of file ColumnarPhysliteTest.cxx.

◆ checkUnknownKeys

template<typename T >
bool columnar::TestUtils::ColumnDataVectorVectorVariantLink< T >::checkUnknownKeys = false

Definition at line 1264 of file ColumnarPhysliteTest.cxx.

◆ columnData

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

Definition at line 1256 of file ColumnarPhysliteTest.cxx.

◆ containerKeys

template<typename T >
std::vector<SG::sgkey_t> columnar::TestUtils::ColumnDataVectorVectorVariantLink< T >::containerKeys

Definition at line 1259 of file ColumnarPhysliteTest.cxx.

◆ containerOffsets

template<typename T >
std::vector<const std::vector<ColumnarOffsetType>*> columnar::TestUtils::ColumnDataVectorVectorVariantLink< T >::containerOffsets

Definition at line 1260 of file ColumnarPhysliteTest.cxx.

◆ containers

template<typename T >
std::vector<std::string> columnar::TestUtils::ColumnDataVectorVectorVariantLink< T >::containers

Definition at line 1258 of file ColumnarPhysliteTest.cxx.

◆ keysColumn

template<typename T >
std::vector<typename CM::LinkKeyType> columnar::TestUtils::ColumnDataVectorVectorVariantLink< T >::keysColumn

Definition at line 1257 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 1255 of file ColumnarPhysliteTest.cxx.

◆ outputColumns

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

Definition at line 369 of file ColumnarPhysliteTest.cxx.

◆ unknownKeys

template<typename T >
std::unordered_map<SG::sgkey_t,std::unordered_set<std::string> > columnar::TestUtils::ColumnDataVectorVectorVariantLink< T >::unknownKeys

Definition at line 1265 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
index
Definition: index.py:1
columnar::TestUtils::BranchReader::connectTree
void connectTree(TTree *tree)
Definition: ColumnarPhysliteTest.cxx:218
tree
TChain * tree
Definition: tile_monitor.h:30
MuonR4::to_string
std::string to_string(const SectorProjector proj)
Definition: MsTrackSeeder.cxx:74
columnar::TestUtils::BranchReader::entrySize
std::optional< float > entrySize() const
Definition: ColumnarPhysliteTest.cxx:246
Trk::u
@ u
Enums for curvilinear frames.
Definition: ParamDefs.h:77
columnar::ColumnarModeArray::LinkIndexType
std::size_t LinkIndexType
the type used for columns that represent element links
Definition: ColumnarDef.h:68
columnar::TestUtils::BranchReader::numBaskets
std::optional< unsigned > numBaskets()
Definition: ColumnarPhysliteTest.cxx:262
lumiFormat.i
int i
Definition: lumiFormat.py:85
columnar::size
std::size_t size() const noexcept
Definition: ObjectRange.h:170
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::ColumnarModeArray::mergeLinkKeyIndex
static LinkIndexType mergeLinkKeyIndex(LinkIndexType key, LinkIndexType index)
merge a key and index value into a link value
Definition: ColumnarDef.h:96
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
mapkey::key
key
Definition: TElectronEfficiencyCorrectionTool.cxx:37