edm4hep/utils/dataframe.h

0001 #ifndef EDM4HEP_UTILS_DATAFRAME_H
0002 #define EDM4HEP_UTILS_DATAFRAME_H
0003
0004 #include "ROOT/RVec.hxx"
0005
0006 namespace edm4hep::utils {
0007
0008 /// Get the pt of the passed particle / datatypes
0009 template <typename T>
0010 ROOT::VecOps::RVec<float> pt(ROOT::VecOps::RVec<T> const& in);
0011
0012 /// Get eta of the passed particles / datatype momenta
0013 template <typename T>
0014 ROOT::VecOps::RVec<float> eta(ROOT::VecOps::RVec<T> const& in);
0015
0016 /// Get cos(theta) of the passed particles / datatype momenta
0017 template <typename T>
0018 ROOT::VecOps::RVec<float> cos_theta(ROOT::VecOps::RVec<T> const& in);
0019
0020 /// Get r of the passed particle / datatype positions
0021 template <typename T>
0022 ROOT::VecOps::RVec<float> r(ROOT::VecOps::RVec<T> const& in);
0023
0024 } // namespace edm4hep::utils
0025
0026 #endif // EDM4HEP_UTILS_DATAFRAME_H