GeneratorInterface/TauolaInterface/interface/TauolappInterface.h - master - githubmirror/cmssw

GeneratorInterface/TauolaInterface/interface/TauolappInterface.h
97 строк3 KB
Christopher Jones
Added ConsumesCollector usage to ExternalDecayDriver
14 окт 2021, 16:49
14 окт 2021, 16:49b35619f
#ifndef gen_TauolaInterface_TauolappInterface_h
#define gen_TauolaInterface_TauolappInterface_h

#include "HepPDT/ParticleDataTable.hh"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "GeneratorInterface/TauolaInterface/interface/TauolaInterfaceBase.h"
#include "TLorentzVector.h"
#include "TVector.h"

namespace HepMC {
  class GenEvent;
}

namespace CLHEP {
  class HepRandomEngine;
}

namespace gen {
  extern "C" {
  void ranmar_(float* rvec, int* lenv);
  void rmarin_(int*, int*, int*);
  }

  class TauolappInterface : public TauolaInterfaceBase {
  public:
    // ctor & dtor
    TauolappInterface(const edm::ParameterSet&, edm::ConsumesCollector);
    ~TauolappInterface() override;

    void enablePolarization() override {
      fPolarization = true;
      return;
    }
    void disablePolarization() override {
      fPolarization = false;
      return;
    }
    void init(const edm::EventSetup&) override;
    const std::vector<int>& operatesOnParticles() override { return fPDGs; }
    HepMC::GenEvent* decay(HepMC::GenEvent*) override;
    void statistics() override;
    void SetLHE(lhef::LHEEvent* l) override { lhe = l; }
    void setRandomEngine(CLHEP::HepRandomEngine* v) override { fRandomEngine = v; }
    static double flat();

  private:
    // member function(s)
    void decodeMDTAU(int);
    void selectDecayByMDTAU();
    int selectLeptonic();
    int selectHadronic();

    HepMC::GenEvent* make_simple_tau_event(const TLorentzVector& l, int pdgid, int status);
    void update_particles(HepMC::GenParticle* partHep,
                          HepMC::GenEvent* theEvent,
                          HepMC::GenParticle* p,
                          TVector3& boost);
    bool isLastTauInChain(const HepMC::GenParticle* tau);
    HepMC::GenParticle* GetMother(HepMC::GenParticle* tau);
    double MatchedLHESpinUp(HepMC::GenParticle* tau,
                            std::vector<HepMC::GenParticle>& p,
                            std::vector<double>& spinup,
                            std::vector<int>& m_idx);
    HepMC::GenParticle* FirstTauInChain(HepMC::GenParticle* tau);
    void BoostProdToLabLifeTimeInDecays(HepMC::GenParticle* p, TLorentzVector& lab, TLorentzVector& prod);

    //
    static CLHEP::HepRandomEngine* fRandomEngine;
    std::vector<int> fPDGs;
    bool fPolarization;
    edm::ESHandle<HepPDT::ParticleDataTable> fPDGTable;
    edm::ESGetToken<HepPDT::ParticleDataTable, edm::DefaultRecord> fPDGTableToken;
    edm::ParameterSet* fPSet;
    bool fIsInitialized;

    int fMDTAU;
    bool fSelectDecayByEvent;
    std::vector<int> fLeptonModes;
    std::vector<int> fHadronModes;
    std::vector<double> fScaledLeptonBrRatios;
    std::vector<double> fScaledHadronBrRatios;
    lhef::LHEEvent* lhe;

    double dmMatch;
    bool dolhe;
    bool dolheBosonCorr;
    int ntries;
    double lifetime;
  };

}  // namespace gen

#endif