Configuration/python/RecoBTag_cff.py

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 RecoBTag.ONNXRuntime.pfParticleTransformerAK4_cff import *
0017 from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4_cff import *
0018 from RecoVertex.AdaptiveVertexFinder.inclusiveVertexing_cff import *
0019 from RecoBTag.PixelCluster.pixelClusterTagInfos_cfi import *
0020
0021 legacyBTaggingTask = cms.Task(
0022     # impact parameters and IP-only algorithms
0023     impactParameterTagInfos,
0024     trackCountingHighEffBJetTags,
0025     jetProbabilityBJetTags,
0026     jetBProbabilityBJetTags,
0027
0028     # SV tag infos depending on IP tag infos, and SV (+IP) based algos
0029     secondaryVertexTagInfos,
0030     simpleSecondaryVertexHighEffBJetTags,
0031     combinedSecondaryVertexV2BJetTags,
0032     inclusiveSecondaryVertexFinderTagInfos,
0033     combinedInclusiveSecondaryVertexV2BJetTags,
0034     ghostTrackVertexTagInfos,
0035     ghostTrackBJetTags,
0036
0037     # soft lepton tag infos and algos
0038     softPFMuonsTagInfos,
0039     softPFMuonBJetTags,
0040     softPFElectronsTagInfos,
0041     softPFElectronBJetTags,
0042
0043     # overall combined taggers
0044     combinedMVAV2BJetTags,
0045
0046     # pixel cluster
0047     pixelClusterTagInfos,
0048 )
0049 legacyBTagging = cms.Sequence(legacyBTaggingTask)
0050
0051 # new candidate-based fwk, with PF inputs
0052 pfBTaggingTask = cms.Task(
0053     # impact parameters and IP-only algorithms
0054     pfImpactParameterTagInfos,
0055     pfTrackCountingHighEffBJetTags,
0056     pfJetProbabilityBJetTags,
0057     pfJetBProbabilityBJetTags,
0058
0059     # SV tag infos depending on IP tag infos, and SV (+IP) based algos
0060     pfSecondaryVertexTagInfos,
0061     pfSimpleSecondaryVertexHighEffBJetTags,
0062     pfCombinedSecondaryVertexV2BJetTags,
0063     inclusiveCandidateVertexingTask,
0064     pfInclusiveSecondaryVertexFinderTagInfos,
0065     pfSimpleInclusiveSecondaryVertexHighEffBJetTags,
0066     pfCombinedInclusiveSecondaryVertexV2BJetTags,
0067     pfGhostTrackVertexTagInfos,
0068     pfGhostTrackBJetTags,
0069     pfDeepCSVTask,
0070     pfDeepFlavourTask,
0071
0072     # soft lepton tag infos and algos
0073     softPFMuonsTagInfos,
0074     softPFMuonBJetTags,
0075     softPFElectronsTagInfos,
0076     softPFElectronBJetTags,
0077
0078     # overall combined taggers
0079     #CSV + soft-lepton + jet probability discriminators combined
0080     pfCombinedMVAV2BJetTags,
0081     pfChargeBJetTags,
0082
0083     # pixel cluster
0084     pixelClusterTagInfos,
0085
0086 )
0087
0088 pfBTagging = cms.Sequence(pfBTaggingTask)
0089
0090 btaggingTask = cms.Task(
0091     pfBTaggingTask,
0092     pfCTaggingTask
0093 )
0094 btagging = cms.Sequence(btaggingTask)
0095
0096 ## modifying b-tagging task in Run3 adding ParticleNet inferece
0097 from Configuration.Eras.Modifier_run3_common_cff import run3_common
0098 _pfBTaggingTask_run3 = cms.Task(
0099     # Keep all the infos and DeepCSV and DeepFlavour
0100     pfImpactParameterTagInfos,
0101     pfTrackCountingHighEffBJetTags,
0102     pfJetProbabilityBJetTags,
0103     pfJetBProbabilityBJetTags,
0104
0105     pfSecondaryVertexTagInfos,
0106     inclusiveCandidateVertexingTask,
0107     pfInclusiveSecondaryVertexFinderTagInfos,
0108     pfGhostTrackVertexTagInfos,
0109     pfDeepCSVTask,
0110     pfDeepFlavourTask,
0111
0112     softPFMuonsTagInfos,
0113     softPFElectronsTagInfos,
0114     pixelClusterTagInfos,
0115
0116     pfParticleNetAK4TaskForRECO,
0117     pfParticleNetTask
0118 )
0119 _pfCTaggingTask_run3 = cms.Task(
0120     inclusiveCandidateVertexingCvsLTask,
0121     pfInclusiveSecondaryVertexFinderCvsLTagInfos,
0122 )
0123 run3_common.toReplaceWith( pfBTaggingTask, _pfBTaggingTask_run3 )
0124 run3_common.toReplaceWith( pfCTaggingTask, _pfCTaggingTask_run3 )
0125