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 86 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 101 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 114 of file ColumnarPhysliteTest.cxx.

115 {
116 return m_branchName;
117 }

◆ columnName()

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

Definition at line 119 of file ColumnarPhysliteTest.cxx.

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

◆ connectTree() [1/2]

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

Definition at line 155 of file ColumnarPhysliteTest.cxx.

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

◆ connectTree() [2/2]

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

Definition at line 143 of file ColumnarPhysliteTest.cxx.

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

◆ containerName()

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

Definition at line 131 of file ColumnarPhysliteTest.cxx.

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

◆ entrySize()

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

Definition at line 177 of file ColumnarPhysliteTest.cxx.

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

◆ getCachedEntry()

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

Definition at line 172 of file ColumnarPhysliteTest.cxx.

173 {
174 return *m_data;
175 }

◆ getEntry()

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

Definition at line 161 of file ColumnarPhysliteTest.cxx.

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

◆ numBaskets()

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

Definition at line 193 of file ColumnarPhysliteTest.cxx.

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

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

110 {
112 }

◆ uncompressedSize()

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

Definition at line 184 of file ColumnarPhysliteTest.cxx.

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

Member Data Documentation

◆ m_branch

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

Definition at line 89 of file ColumnarPhysliteTest.cxx.

◆ m_branchName

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

Definition at line 88 of file ColumnarPhysliteTest.cxx.

◆ m_data

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

Definition at line 91 of file ColumnarPhysliteTest.cxx.

91{new T()};

◆ m_isStatic

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

Definition at line 90 of file ColumnarPhysliteTest.cxx.


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