MCTruth.h

Go to the documentation of this file.

 
#ifndef NAIA_MCTRUTH_H
#define NAIA_MCTRUTH_H
 
// gbatch headers
#ifdef ENABLE_PRODUCTION_CODE
#include "root.h"
#endif
 
#include "Containers/OnDemandContainer.h"
#include "Containers/Utils.h"
#include "Utility/Logging/Logging.h"
 
#include "TVector3.h"
 
#include <array>
#include <map>
#include <vector>
 
namespace NAIA {
 
struct MCParticle {
  int PDGCode = 0;    
  int Z = 0;          
  unsigned int A = 0; 
  float Mass = 0;     
  // clang-format off
  std::vector<TVector3> Position; 
  std::vector<TVector3> Momentum; 
  // clang-format on
  int partID = -1;   
  int parentID = -1; 
  int Proc1 = -1;    
  int Proc2 = -1;    
 
  [[nodiscard]] double GetGenMomentum() const {
    static constexpr std::string_view routine_name{"MCTruth::GetGenMomentum"};
 
    if (!Momentum.empty()) {
      return Momentum[MCTruth::MCHeight::GenerationPoint].Mag();
    } else {
      getLogger(routine_name)->error("No generation momentum available");
      throw std::runtime_error("No generation momentum available");
    }
  }
 
  [[nodiscard]] TVector3 GetGenPosition() const {
    static constexpr std::string_view routine_name{"MCTruth::GetGenPosition"};
 
    if (!Position.empty()) {
      return Position[MCTruth::MCHeight::GenerationPoint];
    } else {
      getLogger(routine_name)->error("No generation position available");
      throw std::runtime_error("No generation position available");
    }
  }
 
  [[nodiscard]] int NSkip() const { return (Proc1 << 24) | Proc2; }
};
 
class MCTruthBaseData : public TObject {
public:
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(AMSEventR *evPtr);
#endif
 
  void Clear();
 
  void Dump() const;
 
  MCParticle Primary; 
 
private:
  ClassDef(MCTruthBaseData, 2)
};
 
class MCTruthPlusData : public TObject {
public:
  struct TrMCHit {
    int pID = 0;              
    std::array<float, 3> Coo; 
    std::array<float, 3> Mom; 
  };
 
#ifdef ENABLE_PRODUCTION_CODE
  bool Fill(AMSEventR *evPtr);
  bool FillTrMCHit(TrTrackR *pTrTrack, float tof_beta);
#endif
 
  void Clear();
 
  void Dump() const;
 
  // clang-format off
  std::vector<MCParticle> Secondaries; 
  LayerVariable<TrMCHit> TrackMCHits;  
  // clang-format on
 
private:
  ClassDef(MCTruthPlusData, 1)
};
 
class MCTruthBase : public OnDemandContainer<MCTruthBase>, public MCTruthBaseData {
public:
  static const std::string BranchName;
 
  MCTruthBase() : MCTruthBaseData(), OnDemandContainer() {}
};
 
class MCTruthPlus : public OnDemandContainer<MCTruthPlus>, public MCTruthPlusData {
public:
  static const std::string BranchName;
 
  MCTruthPlus() : MCTruthPlusData(), OnDemandContainer() {}
};
 
} // namespace NAIA
#endif