Ecal.h

Go to the documentation of this file.

 
#ifndef NAIA_ECAL_H
#define NAIA_ECAL_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/Utils.h"
 
#include "TVector3.h"
 
namespace NAIA {
 
template <class T> using EcalEnergyVariable = std::vector<T>;
template <class T> using EcalLikelihoodVariable = std::vector<T>;
template <class T> using EcalBDTVariable = std::vector<T>;
 
class EcalBaseData : public TObject {
public:
  // hit-related
  short TotalHits;  
  short ShowerHits; 
 
  // energy-related
  // clang-format off
  EcalEnergyVariable<float> Energy; 
 
  // shower reconstruction related
  EcalBDTVariable<float> BDT; 
  // clang-format on
 
  void Clear();
 
  void Dump() const;
 
#ifdef ENABLE_PRODUCTION_CODE
 
  bool Fill(EcalShowerR *showerPtr);
#endif
 
  // ugly but better than saving 6 doubles for every event
  // NB: it's gonna throw if the Ecal object is not valid
  // NB2: thanks ROOT for not supporting std::vector<std::array>. Hopefully this is the only case where we have
  // to use TVector3
 
  [[nodiscard]] const TVector3 &GetCOG() const { return m_pointAndDir.at(0); }
 
  [[nodiscard]] const TVector3 &GetDirection() const { return m_pointAndDir.at(1); }
 
private:
  void ResizeMembers();
 
  // this is used to store COG and direction of the Ecal axis reconstruction
  std::vector<TVector3> m_pointAndDir;
 
  float _beta = 1.0; 
 
  ClassDef(EcalBaseData, 1);
};
 
class EcalPlusData : public TObject {
public:
  // shower reconstruction related
  int EcalKNShowers;     
  int EcalKHadronicFlag; 
  // clang-format off
  EcalLikelihoodVariable<float> Likelihood; 
  // clang-format on
  LayerVariable<float> MaxEnergyFraction; 
 
  void Clear();
 
  void Dump() const;
 
#ifdef ENABLE_PRODUCTION_CODE
 
  bool Fill(EcalShowerR *showerPtr);
#endif
 
private:
  void ResizeMembers();
 
  ClassDef(EcalPlusData, 1);
};
 
class EcalBase : public OnDemandContainer<EcalBase>, public EcalBaseData {
public:
  static const std::string BranchName;
 
  EcalBase() : EcalBaseData(), OnDemandContainer(){};
};
 
class EcalPlus : public OnDemandContainer<EcalPlus>, public EcalPlusData {
public:
  static const std::string BranchName;
 
  EcalPlus() : EcalPlusData(), OnDemandContainer(){};
};
 
} // namespace NAIA
#endif // __ECAL_H__