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

  const TVector3 &GetCOG() const { return m_pointAndDir.at(0); }

  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__