File indexing completed on 2023-03-17 11:28:45
0001 import FWCore.ParameterSet.Config as cms
0002 from Validation.RecoTau.RecoTauValidation_cfi import *
0003
0004 from PhysicsTools.JetMCAlgos.TauGenJets_cfi import tauGenJets
0005 from PhysicsTools.HepMCCandAlgos.genParticles_cfi import *
0006 import PhysicsTools.PatAlgos.tools.helpers as helpers
0007
0008 tauGenJetsForVal = tauGenJets.clone()
0009
0010
0011 objectTypeSelectedTauValDenominatorModuleZTT = cms.EDFilter("TauGenJetDecayModeSelector",
0012 src = cms.InputTag("tauGenJetsForVal"),
0013 select = cms.vstring('oneProng0Pi0', 'oneProng1Pi0', 'oneProng2Pi0', 'oneProngOther',
0014 'threeProng0Pi0', 'threeProng1Pi0', 'threeProngOther', 'rare'),
0015 filter = cms.bool(False)
0016 )
0017
0018
0019 kinematicSelectedTauValDenominatorZTT = cms.EDFilter(
0020 "GenJetSelector",
0021 src = cms.InputTag('objectTypeSelectedTauValDenominatorModuleZTT'),
0022 cut = kinematicSelectedTauValDenominatorCut,
0023 filter = cms.bool(False)
0024 )
0025
0026 procAttributes = dir(proc)
0027 helpers.cloneProcessingSnippet( proc, proc.TauValNumeratorAndDenominator, 'ZTT')
0028 helpers.cloneProcessingSnippet( proc, proc.TauEfficiencies, 'ZTT')
0029 helpers.massSearchReplaceAnyInputTag(proc.TauValNumeratorAndDenominatorZTT, 'kinematicSelectedTauValDenominator', 'kinematicSelectedTauValDenominatorZTT')
0030
0031
0032 zttLabeler = lambda module : SetValidationExtention(module, 'ZTT')
0033 zttModifier = ApplyFunctionToSequence(zttLabeler)
0034 proc.TauValNumeratorAndDenominatorZTT.visit(zttModifier)
0035
0036
0037 discs_to_retain = ['ByDecayModeFinding', 'CombinedIsolationDBSumPtCorr3Hits', 'IsolationMVArun2v1DBoldDMwLT', 'IsolationMVArun2v1DBnewDMwLT', 'MuonRejection', 'ElectronRejection']
0038 proc.RunHPSValidationZTT.discriminators = cms.VPSet([p for p in proc.RunHPSValidationZTT.discriminators if any(disc in p.discriminator.value() for disc in discs_to_retain) ])
0039
0040
0041 proc.efficienciesZTT.plots = Utils.SetPlotSequence(proc.TauValNumeratorAndDenominatorZTT)
0042 proc.efficienciesZTTSummary = cms.EDProducer("TauDQMHistEffProducer",
0043 plots = cms.PSet(
0044 Summary = cms.PSet(
0045 denominator = cms.string('RecoTauV/hpsPFTauProducerZTT_Summary/#PAR#PlotDen'),
0046 efficiency = cms.string('RecoTauV/hpsPFTauProducerZTT_Summary/#PAR#Plot'),
0047 numerator = cms.string('RecoTauV/hpsPFTauProducerZTT_Summary/#PAR#PlotNum'),
0048 parameter = cms.vstring('summary'),
0049 stepByStep = cms.bool(True)
0050 ),
0051 )
0052 )
0053
0054
0055 newProcAttributes = [x for x in dir(proc) if (x not in procAttributes) and (x.find('ZTT') != -1)]
0056
0057
0058 for newAttr in newProcAttributes:
0059 locals()[newAttr] = getattr(proc,newAttr)
0060
0061 produceDenominatorZTT = cms.Sequence(
0062 tauGenJetsForVal
0063 +objectTypeSelectedTauValDenominatorModuleZTT
0064 +kinematicSelectedTauValDenominatorZTT
0065 )
0066
0067 produceDenominator = cms.Sequence(produceDenominatorZTT)
0068
0069 runTauValidationBatchMode = cms.Sequence(
0070 produceDenominatorZTT
0071 +TauValNumeratorAndDenominatorZTT
0072 )
0073
0074 runTauValidation = cms.Sequence(
0075 runTauValidationBatchMode
0076 +TauEfficienciesZTT
0077 )
0078