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

Public Types

using CM = ColumnarModeArray

Public Member Functions

 ColumnDataVectorVectorLink (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 (TestUtils::ToolWrapperData &tool) override
virtual BranchPerfData getPerfData (float emptyTime) override
virtual void collectColumnData () override

Public Attributes

BranchReader< std::vector< std::vector< ElementLink< T > > > > branchReader
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::ColumnDataVectorVectorLink< T >

Definition at line 703 of file ColumnarPhysliteTest.cxx.

Member Typedef Documentation

◆ CM

Definition at line 705 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ ColumnDataVectorVectorLink()

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

Definition at line 717 of file ColumnarPhysliteTest.cxx.

718 : branchReader (val_branchName), benchmarkUnpack (branchReader.columnName()+"(unpack)"), benchmark (branchReader.columnName())
719 {
720 outputColumns.push_back ({.name = branchReader.columnName() + ".data"});
721 outputColumns.push_back ({.name = branchReader.columnName() + ".offset", .isOffset = true});
722 }
std::vector< OutputColumnInfo > outputColumns

Member Function Documentation

◆ clearColumns()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 756 of file ColumnarPhysliteTest.cxx.

757 {
758 columnData.clear();
759 offsets.clear();
760 offsets.push_back (0);
761 }

◆ collectColumnData()

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

◆ connect()

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

725 {
726 auto iter = requestedColumns.find (outputColumns.at(0).name);
727 if (iter == requestedColumns.end())
728 return false;
729 outputColumns.at(0).enabled = true;
730
731 branchReader.connectTree (tree);
732
733 if (iter->second.offsetName != outputColumns.at(1).name)
734 throw std::runtime_error ("offset name mismatch: " + iter->second.offsetName + " != " + outputColumns.at(1).name);
735
736 if (iter->second.linkTargetNames.size() != 1)
737 throw std::runtime_error ("expected exactly one link target name for: " + outputColumns.at(0).name);
738 targetContainerName = iter->second.linkTargetNames.at(0);
739 if (auto keyIter = knownKeys.find (targetContainerName); keyIter != knownKeys.end())
740 targetKey = keyIter->second;
741 if (auto offsetIter = offsetColumns.find (iter->second.linkTargetNames.at(0)); offsetIter != offsetColumns.end())
743 else
744 throw std::runtime_error ("missing offset column: " + iter->second.linkTargetNames.at(0));
745
746 requestedColumns.erase (iter);
747
748 iter = requestedColumns.find (outputColumns.at(1).name);
749 if (iter == requestedColumns.end())
750 return true;
751 requestedColumns.erase (iter);
752 outputColumns.at(1).enabled = true;
753 return true;
754 }

◆ getEntry()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 763 of file ColumnarPhysliteTest.cxx.

764 {
765 benchmark.startTimer ();
766 const auto& branchData = branchReader.getEntry (entry);
767 benchmark.stopTimer ();
768 benchmarkUnpack.startTimer ();
769 if (targetOffsetColumn->size() < 2)
770 throw std::runtime_error ("target offset column not yet filled for: " + outputColumns.at(0).name);
771 for (auto& data : branchData)
772 {
773 for (auto& element : data)
774 {
775 if (element.isDefault() || (element.key() == 0 && element.index() == 0))
776 columnData.push_back (invalidObjectIndex);
777 else
778 {
779 columnData.push_back (element.index() + targetOffsetColumn->at (targetOffsetColumn->size()-2));
780 if (element.key() != targetKey)
781 {
782 if (targetKey == 0)
783 {
784 targetKey = element.key();
785 std::cout << "assume target key for " << targetContainerName << " is " << std::hex << targetKey << std::dec << std::endl;
786 } else
787 {
788 throw std::runtime_error(
789 std::format("target key mismatch: {:x} != {:x} for {} with element index {}",
790 element.key(), targetKey, outputColumns.at(0).name, element.index())
791 );
792 }
793 }
794 }
795 }
796 offsets.push_back (columnData.size());
797 }
798 benchmarkUnpack.stopTimer ();
799 }

◆ getPerfData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 809 of file ColumnarPhysliteTest.cxx.

810 {
812 result.name = branchReader.columnName();
813 result.timeRead = benchmark.getEntryTime(emptyTime);
814 result.timeUnpack = benchmarkUnpack.getEntryTime(emptyTime);
815 benchmark.setSilence();
816 benchmarkUnpack.setSilence();
817 result.entrySize = branchReader.entrySize();
818 result.uncompressedSize = branchReader.uncompressedSize();
819 result.numBaskets = branchReader.numBaskets();
822 return result;
823 }

◆ setData()

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

Implements columnar::TestUtils::IColumnData.

Definition at line 801 of file ColumnarPhysliteTest.cxx.

802 {
803 if (outputColumns.at(0).enabled)
804 tool.setColumn (outputColumns.at(0).name, columnData.size(), columnData.data());
805 if (outputColumns.at(1).enabled)
806 tool.setColumn (outputColumns.at(1).name, offsets.size(), offsets.data());
807 }

Member Data Documentation

◆ benchmark

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

Definition at line 713 of file ColumnarPhysliteTest.cxx.

◆ benchmarkUnpack

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

Definition at line 712 of file ColumnarPhysliteTest.cxx.

◆ branchReader

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

Definition at line 706 of file ColumnarPhysliteTest.cxx.

◆ columnData

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

Definition at line 708 of file ColumnarPhysliteTest.cxx.

◆ entries

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

Definition at line 714 of file ColumnarPhysliteTest.cxx.

◆ nullEntries

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

Definition at line 715 of file ColumnarPhysliteTest.cxx.

◆ offsets

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

Definition at line 707 of file ColumnarPhysliteTest.cxx.

707{0};

◆ outputColumns

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

Definition at line 311 of file ColumnarPhysliteTest.cxx.

◆ targetContainerName

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

Definition at line 711 of file ColumnarPhysliteTest.cxx.

◆ targetKey

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

Definition at line 710 of file ColumnarPhysliteTest.cxx.

◆ targetOffsetColumn

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

Definition at line 709 of file ColumnarPhysliteTest.cxx.


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