GENIE2ART.h

Go to the documentation of this file.

#ifndef EVGB_GENIE2ART_H
#define EVGB_GENIE2ART_H

#include <string>
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include <unordered_map>

namespace genie {
  class EventRecord;
  class GFluxI;
  namespace flux {
    class GNuMIFlux;
    class GNuMIFluxPassThroughInfo;
    class GSimpleNtpFlux;
    class GSimpleNtpEntry;
    class GSimpleNtpNuMI;
    class GSimpleNtpAux;
    class GSimpleNtpMeta;
    class GDk2NuFlux;
  }
}
namespace bsim {
  class Dk2Nu;
  class NuChoice;
  class DkMeta;
}

namespace simb {
  class MCTruth;
  class GTruth;
  class MCFlux;
}

class TLorentzVector;

namespace evgb {


  // utility function: if s starts w/ $, get env var, otherwise return as is
  std::string ExpandEnvVar(const std::string& s);

  void SetEventGeneratorListAndTune(const std::string& evtlistname = "",
                           const std::string& tunename = "${GENIE_XSEC_TUNE}");

  // adapted from GENIEHelper
  void FillMCTruth(const genie::EventRecord* grec,
                   double spillTime,
                   simb::MCTruth& mctruth,
                   const std::string & genieVersion="unknown",
                   const std::string & genieTune="unknown",
                   bool addGenieVtxTime = false,
                   const std::unordered_map<std::string, std::string> genConfig = {});
  void FillMCTruth(const genie::EventRecord* grec,
                   TLorentzVector& vtxOffset,
                   simb::MCTruth& mctruth,
                   const std::string & genieVersion="unknown",
                   const std::string & genieTune="unknown",
                   bool addGenieVtxTime = false,
                   const std::unordered_map<std::string, std::string> genConfig = {});
  void FillGTruth(const genie::EventRecord* grec,
                  simb::GTruth& gtruth);

  // adapted from NuReweight
  //
  genie::EventRecord* RetrieveGHEP(const simb::MCTruth& truth,
                                   const simb::GTruth&  gtruth,
                                   bool useFirstTrajPosition = true);

  std::vector<std::unique_ptr<genie::EventRecord>> RetrieveGHEPs(
    const art::Handle<std::vector<simb::MCTruth>>& mcTruthHandle,
    const art::Handle<std::vector<simb::GTruth>>& gTruthHandle
  );
  std::vector<std::unique_ptr<genie::EventRecord>> RetrieveGHEPs(
    const art::Event& e,
    std::string mcTruthLabel,
    std::string gTruthLabel
  );

  void FillMCFlux(genie::GFluxI* fdriver, simb::MCFlux& mcflux);

  void FillMCFlux(genie::flux::GNuMIFlux* gnumi,
                  simb::MCFlux& mcflux);
  void FillMCFlux(const genie::flux::GNuMIFluxPassThroughInfo* nflux,
                  double dk2gen,
                  simb::MCFlux& flux);

  void FillMCFlux(genie::flux::GSimpleNtpFlux* gsimple,
                  simb::MCFlux& mcflux);
  void FillMCFlux(const genie::flux::GSimpleNtpEntry* nflux_entry,
                  const genie::flux::GSimpleNtpNuMI*  nflux_numi,
                  const genie::flux::GSimpleNtpAux*   nflux_aux,
                  const genie::flux::GSimpleNtpMeta*  nflux_meta,
                  simb::MCFlux& flux);

  void FillMCFlux(genie::flux::GDk2NuFlux* gdk2nu,
                  simb::MCFlux& mcflux);
  void FillMCFlux(const bsim::Dk2Nu*    dk2nu,
                  const bsim::NuChoice* nuchoice,
                  simb::MCFlux& flux);

} // end-of-namespace evgb

#endif  // EVGB_GENIE2ART_H