/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/TauAnalysis/MCEmbeddingTools/interface/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_;
};