Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2022-08-09 22:38:34

0001 import FWCore.ParameterSet.Config as cms
0002 
0003 # define the b-tag squences for offline reconstruction
0004 from RecoBTag.SoftLepton.softLepton_cff import *
0005 from RecoBTag.ImpactParameter.impactParameter_cff import *
0006 from RecoBTag.SecondaryVertex.secondaryVertex_cff import *
0007 from RecoBTag.Combined.combinedMVA_cff import *
0008 from RecoBTag.CTagging.RecoCTagging_cff import *
0009 from RecoBTag.Combined.deepFlavour_cff import *
0010 from RecoBTag.ONNXRuntime.pfDeepFlavour_cff import *
0011 from RecoBTag.ONNXRuntime.pfDeepDoubleX_cff import *
0012 from RecoBTag.ONNXRuntime.pfDeepBoostedJet_cff import *
0013 from RecoBTag.ONNXRuntime.pfHiggsInteractionNet_cff import *
0014 from RecoBTag.ONNXRuntime.pfParticleNet_cff import *
0015 from RecoBTag.ONNXRuntime.pfParticleNetAK4_cff import *
0016 from RecoVertex.AdaptiveVertexFinder.inclusiveVertexing_cff import *
0017 from RecoBTag.PixelCluster.pixelClusterTagInfos_cfi import *
0018 
0019 legacyBTaggingTask = cms.Task(
0020     # impact parameters and IP-only algorithms
0021     impactParameterTagInfos,
0022     trackCountingHighEffBJetTags,
0023     jetProbabilityBJetTags,
0024     jetBProbabilityBJetTags,
0025 
0026     # SV tag infos depending on IP tag infos, and SV (+IP) based algos
0027     secondaryVertexTagInfos,
0028     simpleSecondaryVertexHighEffBJetTags,
0029     combinedSecondaryVertexV2BJetTags,
0030     inclusiveSecondaryVertexFinderTagInfos,
0031     combinedInclusiveSecondaryVertexV2BJetTags,
0032     ghostTrackVertexTagInfos,
0033     ghostTrackBJetTags,
0034 
0035     # soft lepton tag infos and algos
0036     softPFMuonsTagInfos,
0037     softPFMuonBJetTags,
0038     softPFElectronsTagInfos,
0039     softPFElectronBJetTags,
0040 
0041     # overall combined taggers
0042     combinedMVAV2BJetTags,
0043     
0044     # pixel cluster
0045     pixelClusterTagInfos,
0046 )
0047 legacyBTagging = cms.Sequence(legacyBTaggingTask)
0048 
0049 # new candidate-based fwk, with PF inputs
0050 pfBTaggingTask = cms.Task(
0051     # impact parameters and IP-only algorithms
0052     pfImpactParameterTagInfos,
0053     pfTrackCountingHighEffBJetTags,
0054     pfJetProbabilityBJetTags,
0055     pfJetBProbabilityBJetTags,
0056 
0057     # SV tag infos depending on IP tag infos, and SV (+IP) based algos
0058     pfSecondaryVertexTagInfos,
0059     pfSimpleSecondaryVertexHighEffBJetTags,
0060     pfCombinedSecondaryVertexV2BJetTags,
0061     inclusiveCandidateVertexingTask,
0062     pfInclusiveSecondaryVertexFinderTagInfos,
0063     pfSimpleInclusiveSecondaryVertexHighEffBJetTags,
0064     pfCombinedInclusiveSecondaryVertexV2BJetTags,
0065     pfGhostTrackVertexTagInfos,
0066     pfGhostTrackBJetTags,
0067     pfDeepCSVTask,
0068 
0069     # soft lepton tag infos and algos
0070     softPFMuonsTagInfos,
0071     softPFMuonBJetTags,
0072     softPFElectronsTagInfos,
0073     softPFElectronBJetTags,
0074 
0075     # overall combined taggers
0076     #CSV + soft-lepton + jet probability discriminators combined
0077     pfCombinedMVAV2BJetTags,
0078     pfChargeBJetTags,
0079     
0080     # pixel cluster
0081     pixelClusterTagInfos,
0082 
0083 )
0084 
0085 pfBTagging = cms.Sequence(pfBTaggingTask)
0086 
0087 btaggingTask = cms.Task(
0088     pfBTaggingTask,
0089     pfCTaggingTask
0090 )
0091 btagging = cms.Sequence(btaggingTask)
0092 
0093 ## modifying b-tagging task in Run3 adding ParticleNet inferece
0094 from Configuration.Eras.Modifier_run3_common_cff import run3_common
0095 _pfBTaggingTask_particleNet = pfBTaggingTask.copy()
0096 _pfBTaggingTask_particleNet.add( pfParticleNetAK4TaskForRECO, pfParticleNetTask )
0097 run3_common.toReplaceWith( pfBTaggingTask, _pfBTaggingTask_particleNet)
0098