PhysicsTools/BPHNano/python/tracks_cff.py - master - githubmirror/cmssw

PhysicsTools/BPHNano/python/tracks_cff.py
87 строк5 KB
Georgios
add missing variables for Kshort to pipi, updates to include B+ to mumuK*+ and Xib to mumuXi modes, increase precision to 12, store all muons with pT>3 (remove trigger requirement) for BPH collection
29 май 2025, 22:05
29 май 2025, 22:05eb7975f
import FWCore.ParameterSet.Config as cms
from PhysicsTools.NanoAOD.common_cff import *

tracksBPH = cms.EDProducer(
    "BPHTrackMerger",
    beamSpot        = cms.InputTag("offlineBeamSpot"),
    dileptons       = cms.InputTag("MuMu:SelectedDiLeptons"),
    tracks          = cms.InputTag("packedPFCandidates"),
    lostTracks      = cms.InputTag("lostTracks"),
    trackSelection  = cms.string("pt>0.7 && abs(eta)<3.0"),  # We need all tracks for tagging, no cuts here for now
    muons           = cms.InputTag("slimmedMuons"),
    electrons       = cms.InputTag("slimmedElectrons"),
    pvSrc           = cms.InputTag("offlineSlimmedPrimaryVertices"),
    maxDzDilep      = cms.double(1.0),
    dcaSig          = cms.double(-100000),
)


trackBPHTable = cms.EDProducer(
    "SimpleCompositeCandidateFlatTableProducer",
    src  = cms.InputTag("tracksBPH:SelectedTracks"),
    cut  = cms.string(""),
    name = cms.string("Track"),
    doc  = cms.string("track collection"),
    singleton = cms.bool(False),
    extension = cms.bool(False), 
    variables = cms.PSet(
        CandVars,
        vx = Var("vx()", float, doc="x coordinate of of ref point [cm]", precision=6),
        vy = Var("vy()", float, doc="y coordinate of of ref point [cm]", precision=6),
        vz = Var("vz()", float, doc="z coordinate of of ref point [cm]", precision=6),
        # User variables defined in plugins/BPHTrackMerger.cc
        isPacked  = Var("userInt('isPacked')", bool, doc="track from packedCandidate collection"),
        isLostTrk = Var("userInt('isLostTrk')", bool, doc="track from lostTrack collection"),
        dz      = Var("userFloat('dz')", float, doc="dz signed wrt PV[0] [cm]", precision=6),
        dxy     = Var("userFloat('dxy')", float, doc="dxy (with sign) wrt PV associated with the track [cm]", precision=6),
        dzS     = Var("userFloat('dzS')", float, doc="dz/err (with sign) wrt PV[0] [cm]", precision=6),
        dxyS    = Var("userFloat('dxyS')", float, doc="dxy/err (with sign) wrt PV associated with the track [cm]", precision=6),
        DCASig  = Var("userFloat('DCASig')", float, doc="significance of xy-distance of closest approach wrt beamspot", precision=6),
        dzTrg   = Var("userFloat('dzTrg')", float, doc="dz from the corresponding trigger muon [cm]", precision=6),
        isMatchedToMuon = Var("userInt('isMatchedToMuon')", bool, doc="track was used to build a muon"),
        isMatchedToMuon_Idx = Var("userInt('MatchedMuonIdx')", int, doc="index to matched muon"),
        isMatchedToEle  = Var("userInt('isMatchedToEle')", bool, doc="track was used to build a PF ele"),
        nValidHits      = Var("userInt('nValidHits')", "uint8", doc="Number of valid hits"),
        # Covariance matrix elements for helix parameters for decay time uncertainty
        covQopQop = Var("userFloat('covQopQop')", float, doc="Cov. of q/p with q/p", precision=12),
        covQopLam = Var("userFloat('covQopLam')", float, doc="Cov. of q/p with lambda", precision=12),
        covQopPhi = Var("userFloat('covQopPhi')", float, doc="Cov. of q/p with phi", precision=12),
        covLamLam = Var("userFloat('covLamLam')", float, doc="Cov. of lambda with lambda", precision=12),
        covLamPhi = Var("userFloat('covLamPhi')", float, doc="Cov. of lambda with phi", precision=12),
        covPhiPhi = Var("userFloat('covPhiPhi')", float, doc="Cov. of phi with phi", precision=12),
        # Additional track parameters for tagging
        ptErr      = Var("userFloat('ptErr')", float, doc="Pt uncertainty", precision=6),
        normChi2   = Var("userFloat('normChi2')", float, doc="Track fit chi-squared divided by n.d.o.f.", precision=6),
        nValidPixelHits = Var("userInt('nValidPixelHits')", "uint8", doc="Number of pixel hits"),
        # TODO: check impact parameters
        ),
)


tracksBPHMCMatch = cms.EDProducer("MCMatcher",              # cut on deltaR, deltaPt/Pt; pick best by deltaR
    src         = trackBPHTable.src,                        # final reco collection
    matched     = cms.InputTag("finalGenParticlesBPH"),     # final mc-truth particle collection
    mcPdgId     = cms.vint32(321, 211),                     # one or more PDG ID (321 = charged kaon, 211 = charged pion); absolute values (see below)
    checkCharge = cms.bool(False),                          # True = require RECO and MC objects to have the same charge
    mcStatus    = cms.vint32(1),                            # PYTHIA status code (1 = stable, 2 = shower, 3 = hard scattering)
    maxDeltaR   = cms.double(0.05),                         # Minimum deltaR for the match
    maxDPtRel   = cms.double(0.5),                          # Minimum deltaPt/Pt for the match
    resolveAmbiguities    = cms.bool(True),                 # Forbid two RECO objects to match to the same GEN object
    resolveByMatchQuality = cms.bool(True),                 # False = just match input in order; True = pick lowest deltaR pair first
)


tracksBPHMCTable = cms.EDProducer("CandMCMatchTableProducer",
    src   = tracksBPHMCMatch.src,
    mcMap         = cms.InputTag("tracksBPHMCMatch"),
    objName       = trackBPHTable.name,
    objType       = trackBPHTable.name,
    branchName = cms.string("genPart"),
    docString     = cms.string("MC matching to status==1 kaons or pions"),
)


tracksBPHSequence   = cms.Sequence(tracksBPH)
tracksBPHSequenceMC = cms.Sequence(tracksBPH + tracksBPHMCMatch)
tracksBPHTables     = cms.Sequence(trackBPHTable)
tracksBPHTablesMC   = cms.Sequence(trackBPHTable + tracksBPHMCTable)