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

Public Types

using CM = ColumnarModeArray

Public Member Functions

 ColumnDataVectorSplitLink (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

BranchReader< Int_t > branchReaderSize
BranchReaderArray< UInt_t > branchReaderKey
BranchReaderArray< UInt_t > branchReaderIndex
const std::vector< ColumnarOffsetType > * offsetColumn = nullptr
std::vector< ColumnarOffsetTypeoffsets = {0}
LinkColumnVector columnData
Benchmark benchmarkUnpack
Benchmark benchmark
unsigned entries = 0
unsigned nullEntries = 0
std::vector< OutputColumnInfooutputColumns

Detailed Description

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

Definition at line 1550 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

Definition at line 1552 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorSplitLink()

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

Definition at line 1564 of file ColumnarPhysliteTest.cxx.

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1607 of file ColumnarPhysliteTest.cxx.

1608 {
1609 columnData.clear();
1610 offsets.clear();
1611 offsets.push_back (0);
1612 }

◆ collectColumnData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1670 of file ColumnarPhysliteTest.cxx.

1671 {
1672 entries += columnData.size();
1673 for (const auto& index : columnData)
1674 {
1676 nullEntries += 1;
1677 }
1678 }

◆ connect()

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

1572 {
1573 auto iter = requestedColumns.find (outputColumns.at(0).name);
1574 if (iter == requestedColumns.end())
1575 return false;
1576 outputColumns.at(0).enabled = true;
1577 auto* tree = std::get<TTree*>(source);
1578 branchReaderSize.connectTree (tree);
1579 branchReaderKey.connectTree (tree);
1580 branchReaderIndex.connectTree (tree);
1581
1582 if (iter->second.offsetName != outputColumns.at(1).name)
1583 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
1584
1585 for (auto keyColumn : columnData.connect (iter->second, offsetColumns, requestedColumns))
1586 {
1587 outputColumns.push_back ({.name = keyColumn, .primary = false, .enabled = true});
1589 }
1590 requestedColumns.erase (iter);
1591
1592 if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
1593 offsetColumn = offsetIter->second;
1594 else
1595 offsetColumns.emplace (outputColumns.at(1).name, &offsets);
1596
1597 iter = requestedColumns.find (outputColumns.at(1).name);
1598 if (iter != requestedColumns.end())
1599 {
1600 outputColumns.at(1).enabled = true;
1601 requestedColumns.erase (iter);
1602 }
1603
1604 return true;
1605 }

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

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

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1614 of file ColumnarPhysliteTest.cxx.

1615 {
1616 benchmark.startTimer ();
1620 benchmark.stopTimer ();
1621 benchmarkUnpack.startTimer ();
1622 columnData.checkOffsets (offsets.size() - 1);
1624 {
1625 if (branchDataIndex[index] == static_cast<UInt_t>(-1))
1626 columnData.addEmptyLink ();
1627 else
1628 columnData.addSplitLink (branchDataIndex[index], branchDataKey[index], offsets.size()-1);
1629 }
1630 offsets.push_back (columnData.size());
1631 if (offsetColumn)
1632 {
1633 if (offsetColumn->size() != offsets.size())
1634 throw std::runtime_error ("offset column not filled yet: " + outputColumns.at(1).name);
1635 if (offsetColumn->back() != offsets.back())
1636 throw std::runtime_error ("offset column does not match: " + outputColumns.at(1).name);
1637 }
1638 benchmarkUnpack.stopTimer ();
1639 }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1654 of file ColumnarPhysliteTest.cxx.

1655 {
1657 result.name = branchReaderSize.columnName();
1658 result.timeRead = benchmark.getEntryTime(emptyTime);
1659 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
1660 benchmark.setSilence();
1661 benchmarkUnpack.setSilence();
1662 result.entrySize = branchReaderSize.entrySize().value() + branchReaderKey.entrySize().value() + branchReaderIndex.entrySize().value();
1663 result.uncompressedSize = branchReaderSize.uncompressedSize().value() + branchReaderKey.uncompressedSize().value() + branchReaderIndex.uncompressedSize().value();
1664 result.numBaskets = branchReaderSize.numBaskets().value() + branchReaderKey.numBaskets().value() + branchReaderIndex.numBaskets().value();
1667 return result;
1668 }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 1641 of file ColumnarPhysliteTest.cxx.

1642 {
1643 if (outputColumns.at(0).columnIndex != ColumnVectorHeader::nullIndex)
1644 colData.setColumn (outputColumns.at(0).columnIndex, columnData.size(), columnData.data());
1645 if (outputColumns.at(1).columnIndex != ColumnVectorHeader::nullIndex)
1646 colData.setColumn (outputColumns.at(1).columnIndex, offsets.size(), offsets.data());
1647 for (std::size_t i = 2; i < outputColumns.size(); ++ i)
1648 {
1649 if (outputColumns.at(i).columnIndex != ColumnVectorHeader::nullIndex)
1650 colData.setColumn (outputColumns.at(i).columnIndex, columnData.keysColumn(i-2).size(), columnData.keysColumn(i-2).data());
1651 }
1652 }

Member Data Documentation

◆ benchmark

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

Definition at line 1560 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 1559 of file ColumnarPhysliteTest.cxx.

◆ branchReaderIndex

template<typename T>
BranchReaderArray<UInt_t> columnar::TestUtils::ColumnDataVectorSplitLink< T >::branchReaderIndex

Definition at line 1555 of file ColumnarPhysliteTest.cxx.

◆ branchReaderKey

template<typename T>
BranchReaderArray<UInt_t> columnar::TestUtils::ColumnDataVectorSplitLink< T >::branchReaderKey

Definition at line 1554 of file ColumnarPhysliteTest.cxx.

◆ branchReaderSize

template<typename T>
BranchReader<Int_t> columnar::TestUtils::ColumnDataVectorSplitLink< T >::branchReaderSize

Definition at line 1553 of file ColumnarPhysliteTest.cxx.

◆ columnData

Definition at line 1558 of file ColumnarPhysliteTest.cxx.

◆ entries

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

Definition at line 1561 of file ColumnarPhysliteTest.cxx.

◆ nullEntries

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

Definition at line 1562 of file ColumnarPhysliteTest.cxx.

◆ offsetColumn

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

Definition at line 1556 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 1557 of file ColumnarPhysliteTest.cxx.

1557{0};

◆ outputColumns

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

Definition at line 668 of file ColumnarPhysliteTest.cxx.


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