ParticleReplacerClass.h

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    Replacer
// Class:      Replacer
//
//
// Original Author:  Manuel Zeise
//
//


// system include files
#include "TauAnalysis/MCEmbeddingTools/interface/ParticleReplacerBase.h"

#include <stack>
#include <queue>

#include "Math/LorentzVector.h"
#include "Math/VectorUtil.h"
#include "DataFormats/Math/interface/deltaR.h"

#include "DataFormats/Candidate/interface/Particle.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

/*
#include "HepMC/IO_HEPEVT.h"
#include "HepMC/GenParticle.h"
#include "HepMC/GenVertex.h"
#include "HepMC/PythiaWrapper.h"
*/

#include "GeneratorInterface/Pythia6Interface/interface/Pythia6Service.h"
#include "GeneratorInterface/ExternalDecays/interface/TauolaInterface.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "FWCore/Framework/interface/Event.h"

#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "HepPDT/ParticleDataTable.hh"

#include "DataFormats/MuonReco/interface/Muon.h"

#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "TTree.h"

#include<string>

class ParticleReplacerClass : public ParticleReplacerBase
{
public:
    explicit ParticleReplacerClass(const edm::ParameterSet&, bool);
    ~ParticleReplacerClass();

        virtual std::auto_ptr<HepMC::GenEvent> produce(const reco::MuonCollection& muons, const reco::Vertex *pvtx=0, const HepMC::GenEvent *genEvt=0);
    virtual void beginRun(const edm::Run& iRun,const edm::EventSetup& iSetup);
    virtual void endJob();

private:
    void transformMuMu2TauTau(reco::Particle * muon1, reco::Particle * muon2);
    void transformMuMu2TauNu(reco::Particle * muon1, reco::Particle * muon2);

    HepMC::GenEvent * processEventWithTauola(HepMC::GenEvent * evt);
    HepMC::GenEvent * processEventWithPythia(HepMC::GenEvent * evt);
    
    bool testEvent(HepMC::GenEvent * evt);  

    void cleanEvent(HepMC::GenEvent * evt, HepMC::GenVertex * vtx);
    void repairBarcodes(HepMC::GenEvent * evt);

    std::string generatorMode_;
    bool noInitialisation_;

    // this variable defines the type of decay to simulate
    // 0 - mumu->mumu (i.e. no transformation)
    // 1 - mumu->tautau (default value)
    unsigned int transformationMode_;
    
    int motherParticleID_;
    bool useExternalGenerators_ ;
    bool useTauola_ ;
    bool useTauolaPolarization_ ;
    
    gen::TauolaInterface* tauola_;

    bool printEvent_;

    struct MinVisPtCut { enum { ELEC, MU, HAD, TAU } type_; unsigned int index_; double pt_; };
    std::vector<std::vector<MinVisPtCut> > minVisPtCuts_;
//  double minVisibleTransverseMomentum_;
    
    double targetParticleMass_;
    int targetParticlePdgID_;
    
    TTree * outTree;
    int attempts;
    int maxNumberOfAttempts_;
};