RecoTau/python/PFRecoTauDiscriminationAgainstElectronMVA6Phase2_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # HGCal electron stuff
0004 from RecoEgamma.EgammaTools.cleanedEcalDrivenGsfElectronsHGC_cfi import cleanedEcalDrivenGsfElectronsHGC
0005 from RecoEgamma.EgammaTools.hgcalElectronIDValueMap_cff import hgcalElectronIDValueMap
0006 # HGCal electrons cleaned against duplicates and electrons in barrel (pt>10GeV)
0007 # TauValElectronSelector defined Validation/RecoTau/plugins/Selectors.cc;
0008 # is there a more intuitive place where such a selector is defined?
0009 cleanedEcalDrivenGsfElectronsHGCnoEB = cms.EDFilter('TauValElectronSelector',
0010     cut = cms.string('!isEB && pt >= 10.'),
0011     src = cms.InputTag('cleanedEcalDrivenGsfElectronsHGC')
0012 )
0013 # Electron collection merger
0014 mergedGsfElectronsForTauId = cms.EDProducer('GsfElectronCollectionMerger',
0015     src = cms.VInputTag('gedGsfElectrons', 'cleanedEcalDrivenGsfElectronsHGCnoEB')
0016 )
0017 # HGCal EleID with merged electron collection
0018 hgcElectronIdForTauId = hgcalElectronIDValueMap.clone(
0019     electrons = "mergedGsfElectronsForTauId"
0020 )
0021 # anti-e phase-2 tauID (raw)
0022 from RecoTauTag.RecoTau.tauDiscriminationAgainstElectronMVA6Phase2_mvaDefs_cff import mvaNames_phase2, mapping_phase2, workingPoints_phase2
0023 from RecoTauTag.RecoTau.pfRecoTauDiscriminationAgainstElectronMVA6_cfi import *
0024 pfRecoTauDiscriminationAgainstElectronMVA6Phase2Raw = pfRecoTauDiscriminationAgainstElectronMVA6.clone(
0025     #Note: PFTauProducer and Prediscriminants have to be set in the final cfg
0026     srcElectrons = "mergedGsfElectronsForTauId",
0027     isPhase2 = True,
0028     vetoEcalCracks = False,
0029     hgcalElectronIDs = [cms.InputTag("hgcElectronIdForTauId", key) for key in hgcElectronIdForTauId.variables],
0030     **mvaNames_phase2
0031 )
0032 # anti-e phase-2 tauID (WPs)
0033 from RecoTauTag.RecoTau.recoTauDiscriminantCutMultiplexerDefault_cfi import recoTauDiscriminantCutMultiplexerDefault
0034 pfRecoTauDiscriminationAgainstElectronMVA6Phase2 = recoTauDiscriminantCutMultiplexerDefault.clone(
0035     #Note: PFTauProducer and Prediscriminants have to be set in the final cfg
0036     toMultiplex = "pfRecoTauDiscriminationAgainstElectronMVA6Phase2Raw",
0037     mapping = mapping_phase2,
0038     rawValues = ["discriminator", "category"],
0039     workingPoints = workingPoints_phase2
0040 )
0041
0042 electronsForTauDiscriminationAgainstElectronMVA6Phase2Task = cms.Task(
0043     cleanedEcalDrivenGsfElectronsHGC,
0044     cleanedEcalDrivenGsfElectronsHGCnoEB,
0045     mergedGsfElectronsForTauId,
0046     hgcElectronIdForTauId
0047 )
0048
0049 pfRecoTauDiscriminationAgainstElectronMVA6Phase2Task = cms.Task(
0050     electronsForTauDiscriminationAgainstElectronMVA6Phase2Task,
0051     pfRecoTauDiscriminationAgainstElectronMVA6Phase2Raw,
0052     pfRecoTauDiscriminationAgainstElectronMVA6Phase2
0053 )
0054
0055 pfRecoTauDiscriminationAgainstElectronMVA6Phase2Seq = cms.Sequence(
0056     pfRecoTauDiscriminationAgainstElectronMVA6Phase2Task
0057 )