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)
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 74 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 89 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 102 of file ColumnarPhysliteTest.cxx.

103 {
104 return m_branchName;
105 }

◆ columnName()

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

Definition at line 107 of file ColumnarPhysliteTest.cxx.

108 {
110 if (auto index = columnName.find ("AuxDyn."); index != std::string::npos)
111 columnName.replace (index, 6, "");
112 else if (auto index = columnName.find ("Aux."); index != std::string::npos)
113 columnName.replace (index, 3, "");
114 else if (columnName.find (".") != std::string::npos)
115 throw std::runtime_error ("branch name does not contain AuxDyn or Aux: " + m_branchName);
116 return columnName;
117 }

◆ connectTree()

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

Definition at line 131 of file ColumnarPhysliteTest.cxx.

132 {
133 m_branch = tree->GetBranch (m_branchName.c_str());
134 if (!m_branch)
135 throw std::runtime_error ("failed to get branch: " + m_branchName);
136 m_branch->SetMakeClass (1);
137 if (m_isStatic)
138 m_branch->SetAddress (m_data);
139 else
140 m_branch->SetAddress (&m_data);
141 }

◆ containerName()

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

Definition at line 119 of file ColumnarPhysliteTest.cxx.

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

◆ entrySize()

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

Definition at line 159 of file ColumnarPhysliteTest.cxx.

160 {
161 if (!m_branch)
162 return std::nullopt;
163 return static_cast<float>(m_branch->GetZipBytes()) / m_branch->GetEntries();
164 }

◆ getCachedEntry()

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

Definition at line 154 of file ColumnarPhysliteTest.cxx.

155 {
156 return *m_data;
157 }

◆ getEntry()

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

Definition at line 143 of file ColumnarPhysliteTest.cxx.

144 {
145 if (!m_branch)
146 throw std::runtime_error ("branch not connected: " + m_branchName);
147 if (m_branch->GetEntry (entry) <= 0)
148 throw std::runtime_error ("failed to get entry " + std::to_string (entry) + " for branch: " + m_branchName);
149 if (m_data == nullptr)
150 throw std::runtime_error ("got nullptr reading data for branch: " + m_branchName);
151 return *m_data;
152 }

◆ numBaskets()

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

Definition at line 175 of file ColumnarPhysliteTest.cxx.

176 {
177 if (!m_branch)
178 return std::nullopt;
179 return m_branch->GetListOfBaskets()->GetSize();
180 }

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

98 {
100 }

◆ uncompressedSize()

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

Definition at line 166 of file ColumnarPhysliteTest.cxx.

167 {
168 if (!m_branch)
169 return std::nullopt;
170 return static_cast<float>(m_branch->GetTotBytes()) / m_branch->GetEntries();
171 }

Member Data Documentation

◆ m_branch

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

Definition at line 77 of file ColumnarPhysliteTest.cxx.

◆ m_branchName

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

Definition at line 76 of file ColumnarPhysliteTest.cxx.

◆ m_data

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

Definition at line 79 of file ColumnarPhysliteTest.cxx.

79{new T()};

◆ m_isStatic

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

Definition at line 78 of file ColumnarPhysliteTest.cxx.


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