ATLAS Offline Software
Loading...
Searching...
No Matches
columnar::TestUtils::ColumnReaderVectorRLink< T, Reader > Struct Template Referencefinal

#include <ColumnReaderBranch.h>

Inheritance diagram for columnar::TestUtils::ColumnReaderVectorRLink< T, Reader >:
Collaboration diagram for columnar::TestUtils::ColumnReaderVectorRLink< T, Reader >:

Public Types

using CM = ColumnarModeArray

Public Member Functions

 ColumnReaderVectorRLink (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::ColumnReaderVectorRLink< T, Reader >

Definition at line 883 of file ColumnReaderBranch.h.

Member Typedef Documentation

◆ CM

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

Definition at line 885 of file ColumnReaderBranch.h.

Constructor & Destructor Documentation

◆ ColumnReaderVectorRLink()

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

Definition at line 895 of file ColumnReaderBranch.h.

896 : branchReader(val_branchName), benchmarkUnpack(branchReader.columnName() + "(unpack)"), benchmark(branchReader.columnName())
897 {
898 outputColumns.push_back({.name = branchReader.columnName()});
899 outputColumns.push_back({.name = branchReader.containerName(), .isOffset = true, .primary = false});
900 }
std::vector< OutputColumnInfo > outputColumns

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::ColumnReader.

Definition at line 935 of file ColumnReaderBranch.h.

936 {
937 columnData.clear();
938 offsets.clear();
939 offsets.push_back(0);
940 }

◆ collectColumnData()

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

Implements columnar::TestUtils::ColumnReader.

Definition at line 1003 of file ColumnReaderBranch.h.

1004 {
1005 entries += columnData.size();
1006 for (const auto& index : columnData)
1007 {
1009 nullEntries += 1;
1010 }
1011 }

◆ connect()

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

Definition at line 902 of file ColumnReaderBranch.h.

903 {
904 auto iter = requestedColumns.find(outputColumns.at(0).name);
905 if (iter == requestedColumns.end())
906 return false;
907 outputColumns.at(0).enabled = true;
908
909 branchReader.connectTree(source);
910 if (iter->second.offsetName != outputColumns.at(1).name)
911 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
912
913 for (const auto & keyColumn : columnData.connect (iter->second, offsetColumns, requestedColumns))
914 {
915 outputColumns.push_back ({.name = keyColumn, .primary = false, .enabled = true});
917 }
918 requestedColumns.erase (iter);
919
920 if (auto offsetIter = offsetColumns.find (outputColumns.at(1).name); offsetIter != offsetColumns.end())
921 offsetColumn = offsetIter->second;
922 else
923 offsetColumns.emplace (outputColumns.at(1).name, &offsets);
924
925 iter = requestedColumns.find (outputColumns.at(1).name);
926 if (iter != requestedColumns.end())
927 {
928 outputColumns.at(1).enabled = true;
929 requestedColumns.erase (iter);
930 }
931
932 return true;
933 }

◆ connectColumnIndices()

void columnar::TestUtils::ColumnReader::connectColumnIndices ( const ColumnVectorHeader & header)
inlineinherited

lookup and store column indices from the header for all enabled output columns

Definition at line 66 of file ColumnReader.h.

67 {
68 for (auto& col : outputColumns)
69 {
70 if (col.enabled)
71 col.columnIndex = header.getColumnIndex (col.name);
72 }
73 }

◆ getEntry()

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

Implements columnar::TestUtils::ColumnReader.

Definition at line 942 of file ColumnReaderBranch.h.

943 {
944 benchmark.startTimer();
945 const auto& branchData = branchReader.getEntry(entry);
946 benchmark.stopTimer();
947 benchmarkUnpack.startTimer();
948
949 columnData.checkOffsets (offsets.size() - 1);
950 for (const auto& element : branchData)
951 {
952 if (element.isDefault() || element.index() == static_cast<unsigned int>(-1))
953 columnData.addEmptyLink ();
954 else
955 columnData.addSplitLink (element.index(), element.key(), offsets.size()-1);
956 }
957
958 offsets.push_back(columnData.size());
959
960 if (offsetColumn) {
961 if (offsetColumn->size() != offsets.size())
962 {
963 throw std::runtime_error("offset column not filled yet: " + outputColumns.at(1).name);
964 }
965 if (offsetColumn->back() != offsets.back())
966 {
967 throw std::runtime_error("offset column does not match: " + outputColumns.at(1).name);
968 }
969 }
970
971 benchmarkUnpack.stopTimer();
972 }

◆ getPerfData()

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

Implements columnar::TestUtils::ColumnReader.

Definition at line 987 of file ColumnReaderBranch.h.

988 {
990 result.name = branchReader.columnName();
991 result.timeRead = benchmark.getEntryTime(emptyTime);
992 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
993 benchmark.setSilence();
994 benchmarkUnpack.setSilence();
995 result.entrySize = branchReader.entrySize();
996 result.uncompressedSize = branchReader.uncompressedSize();
997 result.numBaskets = branchReader.numBaskets();
1000 return result;
1001 }

◆ setData()

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

Implements columnar::TestUtils::ColumnReader.

Definition at line 974 of file ColumnReaderBranch.h.

975 {
976 if (outputColumns.at(0).columnIndex != ColumnVectorHeader::nullIndex)
977 colData.setColumn(outputColumns.at(0).columnIndex, columnData.size(), columnData.data());
978 if (outputColumns.at(1).columnIndex != ColumnVectorHeader::nullIndex)
979 colData.setColumn(outputColumns.at(1).columnIndex, offsets.size(), offsets.data());
980 for (std::size_t i = 2; i < outputColumns.size(); ++ i)
981 {
982 if (outputColumns.at(i).columnIndex != ColumnVectorHeader::nullIndex)
983 colData.setColumn (outputColumns.at(i).columnIndex, columnData.keysColumn(i-2).size(), columnData.keysColumn(i-2).data());
984 }
985 }

Member Data Documentation

◆ benchmark

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

Definition at line 891 of file ColumnReaderBranch.h.

◆ benchmarkUnpack

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

Definition at line 890 of file ColumnReaderBranch.h.

◆ branchReader

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

Definition at line 886 of file ColumnReaderBranch.h.

◆ columnData

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

Definition at line 889 of file ColumnReaderBranch.h.

◆ entries

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

Definition at line 892 of file ColumnReaderBranch.h.

◆ nullEntries

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

Definition at line 893 of file ColumnReaderBranch.h.

◆ offsetColumn

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

Definition at line 887 of file ColumnReaderBranch.h.

◆ offsets

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

Definition at line 888 of file ColumnReaderBranch.h.

888{0};

◆ outputColumns

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

Definition at line 59 of file ColumnReader.h.


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