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

Public Types

using CM = ColumnarModeArray

Public Member Functions

 ColumnDataVectorLink (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 &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

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

Detailed Description

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

Definition at line 1293 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

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

Definition at line 1295 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorLink()

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

Definition at line 1305 of file ColumnarPhysliteTest.cxx.

1306 : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
1307 {
1308 outputColumns.push_back ({.name = branchReader.columnName()});
1309 outputColumns.push_back ({.name = branchReader.containerName(), .isOffset = true, .primary = false});
1310 }
std::vector< OutputColumnInfo > outputColumns

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1347 of file ColumnarPhysliteTest.cxx.

1348 {
1349 columnData.clear();
1350 offsets.clear();
1351 offsets.push_back (0);
1352 }

◆ collectColumnData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1403 of file ColumnarPhysliteTest.cxx.

1404 {
1405 entries += columnData.size();
1406 for (const auto& index : columnData)
1407 {
1409 nullEntries += 1;
1410 }
1411 }

◆ connect()

template<typename T, template< typename > class Reader>
virtual bool columnar::TestUtils::ColumnDataVectorLink< 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::IColumnData.

Definition at line 1313 of file ColumnarPhysliteTest.cxx.

1314 {
1315 auto iter = requestedColumns.find (outputColumns.at(0).name);
1316 if (iter == requestedColumns.end())
1317 return false;
1318 outputColumns.at(0).enabled = true;
1319
1320 branchReader.connectTree(source);
1321
1322 if (iter->second.offsetName != outputColumns.at(1).name)
1323 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
1324 for (const auto & keyColumn : columnData.connect (iter->second, offsetColumns, requestedColumns))
1325 {
1326 outputColumns.push_back ({.name = keyColumn, .primary = false, .enabled = true});
1328 }
1329
1330 requestedColumns.erase (iter);
1331
1332 if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
1333 offsetColumn = offsetIter->second;
1334 else
1335 offsetColumns.emplace (outputColumns.at(1).name, &offsets);
1336
1337 iter = requestedColumns.find (outputColumns.at(1).name);
1338 if (iter != requestedColumns.end())
1339 {
1340 outputColumns.at(1).enabled = true;
1341 requestedColumns.erase (iter);
1342 }
1343
1344 return true;
1345 }

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

674 {
675 for (auto& col : outputColumns)
676 {
677 if (col.enabled)
678 col.columnIndex = header.getColumnIndex (col.name);
679 }
680 }

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1354 of file ColumnarPhysliteTest.cxx.

1355 {
1356 benchmark.startTimer ();
1357 const auto& branchData = branchReader.getEntry (entry);
1358 benchmark.stopTimer ();
1359 benchmarkUnpack.startTimer ();
1360 columnData.checkOffsets (offsets.size() - 1);
1361 for (auto& element : branchData)
1362 columnData.addLink (element, offsets.size()-1);
1363 offsets.push_back (columnData.size());
1364 if (offsetColumn)
1365 {
1366 if (offsetColumn->size() != offsets.size())
1367 throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
1368 if (offsetColumn->back() != offsets.back())
1369 throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
1370 }
1371 benchmarkUnpack.stopTimer ();
1372 }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1387 of file ColumnarPhysliteTest.cxx.

1388 {
1390 result.name = branchReader.columnName();
1391 result.timeRead = benchmark.getEntryTime(emptyTime);
1392 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
1393 benchmark.setSilence();
1394 benchmarkUnpack.setSilence();
1395 result.entrySize = branchReader.entrySize();
1396 result.uncompressedSize = branchReader.uncompressedSize();
1397 result.numBaskets = branchReader.numBaskets();
1400 return result;
1401 }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1374 of file ColumnarPhysliteTest.cxx.

1375 {
1376 if (outputColumns.at(0).columnIndex != ColumnVectorHeader::nullIndex)
1377 colData.setColumn (outputColumns.at(0).columnIndex, columnData.size(), columnData.data());
1378 if (outputColumns.at(1).columnIndex != ColumnVectorHeader::nullIndex)
1379 colData.setColumn (outputColumns.at(1).columnIndex, offsets.size(), offsets.data());
1380 for (std::size_t i = 2; i < outputColumns.size(); ++ i)
1381 {
1382 if (outputColumns.at(i).columnIndex != ColumnVectorHeader::nullIndex)
1383 colData.setColumn (outputColumns.at(i).columnIndex, columnData.keysColumn(i-2).size(), columnData.keysColumn(i-2).data());
1384 }
1385 }

Member Data Documentation

◆ benchmark

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

Definition at line 1301 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 1300 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 1296 of file ColumnarPhysliteTest.cxx.

◆ columnData

template<typename T, template< typename > class Reader>
LinkColumnVector columnar::TestUtils::ColumnDataVectorLink< T, Reader >::columnData

Definition at line 1299 of file ColumnarPhysliteTest.cxx.

◆ entries

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

Definition at line 1302 of file ColumnarPhysliteTest.cxx.

◆ nullEntries

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

Definition at line 1303 of file ColumnarPhysliteTest.cxx.

◆ offsetColumn

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

Definition at line 1297 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 1298 of file ColumnarPhysliteTest.cxx.

1298{0};

◆ outputColumns

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

Definition at line 666 of file ColumnarPhysliteTest.cxx.


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