Tof.h

Go to the documentation of this file.

 
#ifndef NAIA_TOF_H
#define NAIA_TOF_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/Utils.h"
 
#include "TVector3.h"
 
#include <array>
#include <map>
#include <vector>
 
namespace NAIA {
 
template <class T> using TofChargeVariable = std::map<Tof::ChargeType, T>;
template <class T> using TofBetaVariable = std::map<Tof::BetaType, T>;
template <class T> using TofClusterTypeVariable = std::map<Tof::BetaClusterType, T>;
 
class TofBaseData : public TObject {
public:
  // clang-format off
  TofBetaVariable<float> Beta;     
  TofChargeVariable<float> Charge; 
  TofChargeVariable<float> ChargeNoPL; 
  // clang-format on
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(BetaR *betaPtr, BetaHR *betahPtr, bool standalone = false, bool use_mc_beta = false);
 
  void SetMC(bool isMC) { _isMC = isMC; }
#endif
 
  void Clear();
 
  void Dump() const;
 
  [[nodiscard]] TVector3 InterpolateAtZ(float z) const;
 
  // ugly but better than saving 6 doubles for every event
  // NB: it's gonna throw if the Tof object is not valid
  // NB2: thanks ROOT for not supporting std::vector<std::array>.
 
  [[nodiscard]] const TVector3 &GetLinePoint() const { return m_pointAndDir.at(0); }
  [[nodiscard]] const TVector3 &GetLineDirection() const { return m_pointAndDir.at(1); }
 
private:
  // this is used to store point and direction of the TOF reconstruction
  std::vector<TVector3> m_pointAndDir;
 
  bool _isMC = false; 
 
  ClassDef(TofBaseData, 2)
};
 
class TofPlusData : public TObject {
public:
  // clang-format off
  float Chi2Coo{0};  
  float Chi2Time{0}; 
 
  short NTrkClusters{0};    
  short NBetaClusters{0};   
  short NChargeClusters{0}; 
 
  float ChargeLikelihoodEst{0}; 
  float ChargeLikelihoodProb{0}; 
 
  TofBetaVariable<short> BetaPattern; 
  std::vector<bool> LayerGoodPathl;  
  LayerVariable<float> LayerCharge;  
  LayerVariable<float> LayerEdep;    
  LayerVariable<TofClusterTypeVariable<unsigned int>> NClusters; 
  LayerVariable<TofClusterTypeVariable<float>> AverageDTime;     
  LayerVariable<TofClusterTypeVariable<float>> AverageEdep;      
  // clang-format on
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(BetaR *betaPtr, BetaHR *betahPtr, bool standalone = false);
#endif
 
  void Clear();
 
  void Dump() const;
 
private:
  void ResizeMembers();
 
  ClassDef(TofPlusData, 2)
};
 
class TofBase : public OnDemandContainer<TofBase>, public TofBaseData {
public:
  static const std::string BranchName;
 
  TofBase() : TofBaseData(), OnDemandContainer() {}
};
 
class TofPlus : public OnDemandContainer<TofPlus>, public TofPlusData {
public:
  static const std::string BranchName;
 
  TofPlus() : TofPlusData(), OnDemandContainer() {}
};
 
class TofBaseStandalone : public OnDemandContainer<TofBaseStandalone>, public TofBaseData {
public:
  static const std::string BranchName;
 
  TofBaseStandalone() : TofBaseData(), OnDemandContainer() {}
};
 
class TofPlusStandalone : public OnDemandContainer<TofPlusStandalone>, public TofPlusData {
public:
  static const std::string BranchName;
 
  TofPlusStandalone() : TofPlusData(), OnDemandContainer() {}
};
 
} // namespace NAIA
 
#endif