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 "TVector3.h"

#include <array>
#include <map>
#include <vector>

namespace NAIA {

struct MCParticle {
  int PDGCode = 0; 
  int Z = 0;       
  int A = 0;       
  // clang-format off
  std::vector<TVector3> Momentum; 
  std::vector<TVector3> Position; 
  // clang-format on
  int partID = -1;   
  int parentID = -1; 
  int Proc1 = -1;    
  int Proc2 = -1;    

  double GetGenMomentum() const {
    if (!Momentum.empty()) {
      return Momentum[MCTruth::MCHeight::GenerationPoint].Mag();
    } else {
      throw std::runtime_error("No generation momentum available");
    }
  }

  TVector3 GetGenPosition() const {
    if (!Position.empty()) {
      return Position[MCTruth::MCHeight::GenerationPoint];
    } else {
      throw std::runtime_error("No generation position available");
    }
  }
};

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 {
    unsigned 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);
#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