ATLAS Offline Software
Loading...
Searching...
No Matches
columnar::TestUtils::BranchReader< T > Class Template Referencefinal
Collaboration diagram for columnar::TestUtils::BranchReader< T >:

Public Member Functions

 BranchReader (const std::string &val_branchName)
 ~BranchReader () noexcept
 BranchReader (const BranchReader &)=delete
BranchReaderoperator= (const BranchReader &)=delete
void setIsStatic (bool isStatic)
const std::string & branchName () const
std::string columnName () const
std::string containerName () const
void connectTree (TTree *tree)
void connectTree (const Backend &b)
const T & getEntry (Long64_t entry)
const T & getCachedEntry () const
std::optional< float > entrySize () const
std::optional< float > uncompressedSize () const
std::optional< unsigned > numBaskets ()

Private Attributes

std::string m_branchName
TBranch * m_branch = nullptr
bool m_isStatic = std::is_pod_v<T>
T * m_data {new T()}

Detailed Description

template<typename T>
class columnar::TestUtils::BranchReader< T >

Definition at line 84 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

◆ BranchReader() [1/2]

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

◆ ~BranchReader()

template<typename T>
columnar::TestUtils::BranchReader< T >::~BranchReader ( )
inlinenoexcept

Definition at line 99 of file ColumnarPhysliteTest.cxx.

◆ BranchReader() [2/2]

template<typename T>
columnar::TestUtils::BranchReader< T >::BranchReader ( const BranchReader< T > & )
delete

Member Function Documentation

◆ branchName()

template<typename T>
const std::string & columnar::TestUtils::BranchReader< T >::branchName ( ) const
inlinenodiscard

Definition at line 112 of file ColumnarPhysliteTest.cxx.

113 {
114 return m_branchName;
115 }

◆ columnName()

template<typename T>
std::string columnar::TestUtils::BranchReader< T >::columnName ( ) const
inlinenodiscard

Definition at line 117 of file ColumnarPhysliteTest.cxx.

118 {
120 if (auto index = columnName.find("AuxDyn."); index != std::string::npos)
121 columnName.replace(index, 6, "");
122 else if (auto index = columnName.find("Aux."); index != std::string::npos)
123 columnName.replace(index, 3, "");
124 else if (columnName.find(".") != std::string::npos)
125 throw std::runtime_error("branch name does not contain AuxDyn or Aux: " +m_branchName);
126 return columnName;
127 }

◆ connectTree() [1/2]

template<typename T>
void columnar::TestUtils::BranchReader< T >::connectTree ( const Backend & b)
inline

Definition at line 153 of file ColumnarPhysliteTest.cxx.

154 {
155 auto* tree = std::get<TTree*>(b); // throws if wrong backend
157 }

◆ connectTree() [2/2]

template<typename T>
void columnar::TestUtils::BranchReader< T >::connectTree ( TTree * tree)
inline

Definition at line 141 of file ColumnarPhysliteTest.cxx.

142 {
143 m_branch = tree->GetBranch(m_branchName.c_str());
144 if (!m_branch)
145 throw std::runtime_error("failed to get branch: " + m_branchName);
146 m_branch->SetMakeClass(1);
147 if (m_isStatic)
148 m_branch->SetAddress(m_data);
149 else
150 m_branch->SetAddress(&m_data);
151 }

◆ containerName()

template<typename T>
std::string columnar::TestUtils::BranchReader< T >::containerName ( ) const
inlinenodiscard

Definition at line 129 of file ColumnarPhysliteTest.cxx.

130 {
131 if (auto index = m_branchName.find("AuxDyn."); index != std::string::npos)
132 return m_branchName.substr(0, index);
133 else if (auto index = m_branchName.find("Aux."); index != std::string::npos)
134 return m_branchName.substr(0, index);
135 else if (m_branchName.find(".") == std::string::npos)
136 return m_branchName;
137 else
138 throw std::runtime_error("branch name does not contain AuxDyn or Aux: " +m_branchName);
139 }

◆ entrySize()

template<typename T>
std::optional< float > columnar::TestUtils::BranchReader< T >::entrySize ( ) const
inline

Definition at line 175 of file ColumnarPhysliteTest.cxx.

176 {
177 if (!m_branch)
178 return std::nullopt;
179 return static_cast<float>(m_branch->GetZipBytes()) / m_branch->GetEntries();
180 }

◆ getCachedEntry()

template<typename T>
const T & columnar::TestUtils::BranchReader< T >::getCachedEntry ( ) const
inline

Definition at line 170 of file ColumnarPhysliteTest.cxx.

171 {
172 return *m_data;
173 }

◆ getEntry()

template<typename T>
const T & columnar::TestUtils::BranchReader< T >::getEntry ( Long64_t entry)
inline

Definition at line 159 of file ColumnarPhysliteTest.cxx.

160 {
161 if (!m_branch)
162 throw std::runtime_error("branch not connected: " + m_branchName);
163 if (m_branch->GetEntry(entry) <= 0)
164 throw std::runtime_error("failed to get entry " + std::to_string(entry) + " for branch: " + m_branchName);
165 if (m_data == nullptr)
166 throw std::runtime_error("got nullptr reading data for branch: " + m_branchName);
167 return *m_data;
168 }

◆ numBaskets()

template<typename T>
std::optional< unsigned > columnar::TestUtils::BranchReader< T >::numBaskets ( )
inline

Definition at line 191 of file ColumnarPhysliteTest.cxx.

192 {
193 if (!m_branch)
194 return std::nullopt;
195 return m_branch->GetListOfBaskets()->GetSize();
196 }

◆ operator=()

template<typename T>
BranchReader & columnar::TestUtils::BranchReader< T >::operator= ( const BranchReader< T > & )
delete

◆ setIsStatic()

template<typename T>
void columnar::TestUtils::BranchReader< T >::setIsStatic ( bool isStatic)
inline

Definition at line 107 of file ColumnarPhysliteTest.cxx.

108 {
110 }

◆ uncompressedSize()

template<typename T>
std::optional< float > columnar::TestUtils::BranchReader< T >::uncompressedSize ( ) const
inline

Definition at line 182 of file ColumnarPhysliteTest.cxx.

183 {
184 if (!m_branch)
185 return std::nullopt;
186 return static_cast<float>(m_branch->GetTotBytes()) / m_branch->GetEntries();
187 }

Member Data Documentation

◆ m_branch

template<typename T>
TBranch* columnar::TestUtils::BranchReader< T >::m_branch = nullptr
private

Definition at line 87 of file ColumnarPhysliteTest.cxx.

◆ m_branchName

template<typename T>
std::string columnar::TestUtils::BranchReader< T >::m_branchName
private

Definition at line 86 of file ColumnarPhysliteTest.cxx.

◆ m_data

template<typename T>
T* columnar::TestUtils::BranchReader< T >::m_data {new T()}
private

Definition at line 89 of file ColumnarPhysliteTest.cxx.

89{new T()};

◆ m_isStatic

template<typename T>
bool columnar::TestUtils::BranchReader< T >::m_isStatic = std::is_pod_v<T>
private

Definition at line 88 of file ColumnarPhysliteTest.cxx.


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