MCTruth.h

Go to the documentation of this file.

#ifndef SIMB_MCTRUTH_H
#define SIMB_MCTRUTH_H

#include <string>
#include <vector>
#include "nusimdata/SimulationBase/MCGeneratorInfo.h"
#include "nusimdata/SimulationBase/MCNeutrino.h"

namespace simb {

  class MCParticle;

  typedef enum _ev_origin{
    kUnknown,           
    kBeamNeutrino,      
    kCosmicRay,         
    kSuperNovaNeutrino, 
    kSingleParticle     
  } Origin_t;

  //......................................................................

  class MCTruth {
  public:
    MCTruth();

  private:

    std::vector<simb::MCParticle> fPartList;    
    simb::MCNeutrino              fMCNeutrino;  
    simb::Origin_t                fOrigin;      
    simb::MCGeneratorInfo         fGenInfo;     
    bool                          fNeutrinoSet; 

  public:
    const simb::MCGeneratorInfo&  GeneratorInfo()     const;
    simb::Origin_t                Origin()            const;
    int                           NParticles()        const;
    const simb::MCParticle&       GetParticle(int i)  const;
    const simb::MCNeutrino&       GetNeutrino()       const;
    bool                          NeutrinoSet()       const;

    void             Add(simb::MCParticle const& part);
    void             Add(simb::MCParticle&& part);
    void             SetGeneratorInfo(simb::Generator_t generator,
                                      const std::string & genVersion,
                                      const std::unordered_map<std::string, std::string>& genConfig);
    void             SetOrigin(simb::Origin_t origin);
    void             SetNeutrino(int CCNC, 
                                 int mode,
                                 int interactionType,
                                 int target,
                                 int nucleon,
                                 int quark, 
                                 double w, 
                                 double x, 
                                 double y, 
                                 double qsqr);                      
 
    friend std::ostream&  operator<< (std::ostream& o, simb::MCTruth const& a);
  };
}

inline const simb::MCGeneratorInfo& simb::MCTruth::GeneratorInfo()     const { return fGenInfo;              }
inline simb::Origin_t               simb::MCTruth::Origin()            const { return fOrigin;               }
inline int                          simb::MCTruth::NParticles()        const { return (int)fPartList.size(); }
inline const simb::MCParticle&      simb::MCTruth::GetParticle(int i)  const { return fPartList[i];          }
inline const simb::MCNeutrino&      simb::MCTruth::GetNeutrino()       const { return fMCNeutrino;           }
inline bool                         simb::MCTruth::NeutrinoSet()       const { return fNeutrinoSet;          }

inline void                         simb::MCTruth::Add(simb::MCParticle const& part) { fPartList.push_back(part); }
inline void                         simb::MCTruth::Add(simb::MCParticle&& part)      { fPartList.push_back(std::move(part)); }
inline void                         simb::MCTruth::SetOrigin(simb::Origin_t origin)  { fOrigin = origin;             }

inline void simb::MCTruth::SetGeneratorInfo(simb::Generator_t generator,
                                            const std::string &genVersion,
                                            const std::unordered_map<std::string, std::string>& genConfig)
{
  fGenInfo = simb::MCGeneratorInfo(generator, genVersion, genConfig);
}

#endif //SIMB_MCTRUTH_H