NanoAOD/python/particlelevel_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from  PhysicsTools.NanoAOD.common_cff import *
0003 from PhysicsTools.NanoAOD.simpleCandidateFlatTableProducer_cfi import simpleCandidateFlatTableProducer
0004 from PhysicsTools.NanoAOD.simpleSingletonCandidateFlatTableProducer_cfi import simpleSingletonCandidateFlatTableProducer
0005 from PhysicsTools.NanoAOD.simpleHTXSFlatTableProducer_cfi import simpleHTXSFlatTableProducer
0006
0007 ##################### User floats producers, selectors ##########################
0008
0009 mergedGenParticles = cms.EDProducer("MergedGenParticleProducer",
0010     inputPruned = cms.InputTag("prunedGenParticles"),
0011     inputPacked = cms.InputTag("packedGenParticles"),
0012 )
0013
0014 genParticles2HepMC = cms.EDProducer("GenParticles2HepMCConverter",
0015     genParticles = cms.InputTag("mergedGenParticles"),
0016     genEventInfo = cms.InputTag("generator"),
0017     signalParticlePdgIds = cms.vint32(),
0018 )
0019
0020 genParticles2HepMCHiggsVtx = cms.EDProducer("GenParticles2HepMCConverter",
0021      genParticles = cms.InputTag("mergedGenParticles"),
0022      genEventInfo = cms.InputTag("generator"),
0023      signalParticlePdgIds = cms.vint32(25), ## for the Higgs analysis
0024 )
0025
0026
0027 particleLevel = cms.EDProducer("ParticleLevelProducer",
0028     src = cms.InputTag("genParticles2HepMC:unsmeared"),
0029
0030     doJetClustering = cms.bool(False), # Not needed as Rivet jets aren't used currently
0031     usePromptFinalStates = cms.bool(True), # for leptons, photons, neutrinos
0032     excludePromptLeptonsFromJetClustering = cms.bool(False),
0033     excludeNeutrinosFromJetClustering = cms.bool(True),
0034
0035     particleMinPt  = cms.double(0.),
0036     particleMaxEta = cms.double(5.), # HF range. Maximum 6.0 on MiniAOD
0037
0038     lepConeSize = cms.double(0.1), # for photon dressing
0039     lepMinPt    = cms.double(1.),
0040     lepMaxEta   = cms.double(2.5),
0041
0042     jetConeSize = cms.double(0.4),
0043     jetMinPt    = cms.double(10.),
0044     jetMaxEta   = cms.double(999.),
0045
0046     fatJetConeSize = cms.double(0.8),
0047     fatJetMinPt    = cms.double(170.),
0048     fatJetMaxEta   = cms.double(999.),
0049
0050     phoIsoConeSize = cms.double(0.4),
0051     phoMaxRelIso = cms.double(0.5),
0052     phoMinPt = cms.double(1.),
0053     phoMaxEta = cms.double(2.5),
0054 )
0055
0056 rivetProducerHTXS = cms.EDProducer('HTXSRivetProducer',
0057    HepMCCollection = cms.InputTag('genParticles2HepMCHiggsVtx','unsmeared'),
0058    LHERunInfo = cms.InputTag('externalLHEProducer'),
0059    ProductionMode = cms.string('AUTO'),
0060 )
0061
0062
0063 ##################### Tables for final output and docs ##########################
0064 rivetLeptonTable = simpleCandidateFlatTableProducer.clone(
0065     src = cms.InputTag("particleLevel:leptons"),
0066     cut = cms.string("pt > 15"),
0067     name= cms.string("GenDressedLepton"),
0068     doc = cms.string("Dressed leptons from Rivet-based ParticleLevelProducer"),
0069     externalVariables = cms.PSet(
0070         hasTauAnc = ExtVar(cms.InputTag("tautagger"),bool, doc="true if Dressed lepton has a tau as ancestor"),
0071         ),
0072     variables = cms.PSet(
0073         P4Vars,
0074         pdgId = Var("pdgId", int, doc="PDG id"),
0075     )
0076 )
0077
0078 rivetPhotonTable = simpleCandidateFlatTableProducer.clone(
0079     src = cms.InputTag("particleLevel:photons"),
0080     cut = cms.string("pt > 10"),
0081     name= cms.string("GenIsolatedPhoton"),
0082     doc = cms.string("Isolated photons from Rivet-based ParticleLevelProducer"),
0083     variables = cms.PSet(
0084         P4Vars
0085     )
0086 )
0087
0088 tautagger = cms.EDProducer("GenJetTauTaggerProducer",
0089     src = rivetLeptonTable.src,
0090 )
0091
0092 rivetMetTable = simpleSingletonCandidateFlatTableProducer.clone(
0093     src = cms.InputTag("particleLevel:mets"),
0094     name = cms.string("MET"),
0095     doc = cms.string("MET from Rivet-based ParticleLevelProducer in fiducial volume abs(eta)<5"),
0096     extension = cms.bool(True),
0097     variables = cms.PSet(
0098        fiducialGenPt  = Var("pt",  float, precision=10),
0099        fiducialGenPhi = Var("phi", float, precision=10),
0100     ),
0101 )
0102
0103 HTXSCategoryTable = simpleHTXSFlatTableProducer.clone(
0104     src = cms.InputTag("rivetProducerHTXS","HiggsClassification"),
0105     name = cms.string("HTXS"),
0106     doc = cms.string("HTXS classification"),
0107     variables=cms.PSet(
0108         stage_0 = Var("stage0_cat",int, doc="HTXS stage-0 category"),
0109         stage_1_pTjet30 = Var("stage1_cat_pTjet30GeV",int, doc="HTXS stage-1 category (jet pt>30 GeV)"),
0110         stage_1_pTjet25 = Var("stage1_cat_pTjet25GeV",int, doc="HTXS stage-1 category (jet pt>25 GeV)"),
0111         stage1_1_cat_pTjet30GeV = Var("stage1_1_cat_pTjet30GeV",int,doc="HTXS stage-1.1 category(jet pt>30 GeV)"),
0112         stage1_1_cat_pTjet25GeV = Var("stage1_1_cat_pTjet25GeV",int,doc="HTXS stage-1.1 category(jet pt>25 GeV)"),
0113         stage1_1_fine_cat_pTjet30GeV = Var("stage1_1_fine_cat_pTjet30GeV",int,doc="HTXS stage-1.1-fine category(jet pt>30 GeV)"),
0114         stage1_1_fine_cat_pTjet25GeV = Var("stage1_1_fine_cat_pTjet25GeV",int,doc="HTXS stage-1.1-fine category(jet pt>25 GeV)"),
0115         stage1_2_cat_pTjet30GeV = Var("stage1_2_cat_pTjet30GeV",int,doc="HTXS stage-1.2 category(jet pt>30 GeV)"),
0116         stage1_2_cat_pTjet25GeV = Var("stage1_2_cat_pTjet25GeV",int,doc="HTXS stage-1.2 category(jet pt>25 GeV)"),
0117         stage1_2_fine_cat_pTjet30GeV = Var("stage1_2_fine_cat_pTjet30GeV",int,doc="HTXS stage-1.2-fine category(jet pt>30 GeV)"),
0118         stage1_2_fine_cat_pTjet25GeV = Var("stage1_2_fine_cat_pTjet25GeV",int,doc="HTXS stage-1.2-fine category(jet pt>25 GeV)"),
0119         Higgs_pt = Var("higgs.Pt()",float, doc="pt of the Higgs boson as identified in HTXS", precision=14),
0120         Higgs_y = Var("higgs.Rapidity()",float, doc="rapidity of the Higgs boson as identified in HTXS", precision=12),
0121         njets30 = Var("jets30.size()","uint8", doc="number of jets with pt>30 GeV as identified in HTXS"),
0122         njets25 = Var("jets25.size()","uint8", doc="number of jets with pt>25 GeV as identified in HTXS"),
0123    )
0124 )
0125
0126 lheInfoTable = cms.EDProducer("LHETablesProducer",
0127      lheInfo = cms.VInputTag(cms.InputTag("externalLHEProducer"), cms.InputTag("source")),
0128      precision = cms.int32(14),
0129      storeLHEParticles = cms.bool(True)
0130  )
0131
0132 particleLevelTask = cms.Task(mergedGenParticles,genParticles2HepMC,particleLevel,tautagger,genParticles2HepMCHiggsVtx,rivetProducerHTXS)
0133 particleLevelTablesTask = cms.Task(rivetLeptonTable,rivetPhotonTable,rivetMetTable,HTXSCategoryTable,lheInfoTable)