MCTruth/python/MuonAssociatorByHits_cfi.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003
0004 muonAssociatorByHitsCommonParameters = cms.PSet(
0005     dumpInputCollections = cms.untracked.bool(False),
0006     #
0007     #....... general input parameters
0008     #
0009     # include invalid muon hits
0010     includeZeroHitMuons = cms.bool(True),
0011     #
0012     # accept mismatched association of tracker/muon stub of global hit patterns
0013     acceptOneStubMatchings = cms.bool(False),
0014     # reject bad global muons made of only tracker hits
0015     rejectBadGlobal = cms.bool(True),
0016     #
0017     # switches to be set according to the input Track collection
0018     UseTracker = cms.bool(True),
0019     UseMuon = cms.bool(True),
0020     #
0021     # cuts for the muon stub
0022     AbsoluteNumberOfHits_muon = cms.bool(False),
0023     NHitCut_muon = cms.uint32(0),
0024     EfficiencyCut_muon = cms.double(0.),
0025     PurityCut_muon = cms.double(0.),
0026     #
0027     # cuts for the tracker stub
0028     AbsoluteNumberOfHits_track = cms.bool(False),
0029     NHitCut_track = cms.uint32(0),
0030     EfficiencyCut_track = cms.double(0.),
0031     PurityCut_track = cms.double(0.),
0032     #
0033     # switches for the tracker stub
0034     UsePixels = cms.bool(True),
0035     UseGrouped = cms.bool(True),
0036     UseSplitting = cms.bool(True),
0037     ThreeHitTracksAreSpecial = cms.bool(False),
0038     #
0039     # for DT Hit associator
0040     crossingframe = cms.bool(False),
0041     simtracksTag = cms.InputTag("g4SimHits"),
0042     simtracksXFTag = cms.InputTag("mix","g4SimHits"),
0043     #
0044     DTsimhitsTag = cms.InputTag("g4SimHits","MuonDTHits"),
0045     DTsimhitsXFTag = cms.InputTag("mix","g4SimHitsMuonDTHits"),
0046     DTdigiTag = cms.InputTag("simMuonDTDigis"),
0047     DTdigisimlinkTag = cms.InputTag("simMuonDTDigis"),
0048     DTrechitTag = cms.InputTag("dt1DRecHits"),
0049     #
0050     dumpDT = cms.bool(False),
0051     links_exist = cms.bool(True),
0052     associatorByWire = cms.bool(False),
0053     #
0054     # for CSC Hit associator
0055     CSCsimHitsTag = cms.InputTag("g4SimHits","MuonCSCHits"),
0056     CSCsimHitsXFTag = cms.InputTag("mix","g4SimHitsMuonCSCHits"),
0057     CSClinksTag = cms.InputTag("simMuonCSCDigis","MuonCSCStripDigiSimLinks"),
0058     CSCwireLinksTag = cms.InputTag("simMuonCSCDigis","MuonCSCWireDigiSimLinks"),
0059     #
0060     # for RPC Hit associator
0061     RPCsimhitsTag = cms.InputTag("g4SimHits","MuonRPCHits"),
0062     RPCsimhitsXFTag = cms.InputTag("mix","g4SimHitsMuonRPCHits"),
0063     RPCdigisimlinkTag = cms.InputTag("simMuonRPCDigis","RPCDigiSimLink"),
0064     #
0065     # for GEM Hit associator
0066     useGEMs = cms.bool(False),
0067     GEMsimhitsTag = cms.InputTag("g4SimHits","MuonGEMHits"),
0068     GEMsimhitsXFTag = cms.InputTag("mix","g4SimHitsMuonGEMHits"),
0069     GEMdigisimlinkTag = cms.InputTag("simMuonGEMDigis","GEM"),
0070     #
0071     # for Tracker Hit associator
0072     #
0073     associatePixel = cms.bool(True),
0074     associateStrip = cms.bool(True),
0075     usePhase2Tracker = cms.bool(False),
0076     pixelSimLinkSrc = cms.InputTag("simSiPixelDigis"),
0077     stripSimLinkSrc = cms.InputTag("simSiStripDigis"),
0078     phase2TrackerSimLinkSrc  = cms.InputTag("simSiPixelDigis","Tracker"),
0079     associateRecoTracks = cms.bool(True),
0080     #
0081     ROUList = cms.vstring('TrackerHitsTIBLowTof',
0082         'TrackerHitsTIBHighTof',
0083         'TrackerHitsTIDLowTof',
0084         'TrackerHitsTIDHighTof',
0085         'TrackerHitsTOBLowTof',
0086         'TrackerHitsTOBHighTof',
0087         'TrackerHitsTECLowTof',
0088         'TrackerHitsTECHighTof',
0089         'TrackerHitsPixelBarrelLowTof',
0090         'TrackerHitsPixelBarrelHighTof',
0091         'TrackerHitsPixelEndcapLowTof',
0092         'TrackerHitsPixelEndcapHighTof'),
0093     #
0094     # to associate to reco::Muon segments
0095     inputDTRecSegment4DCollection = cms.InputTag("dt4DSegments"),
0096     inputCSCSegmentCollection = cms.InputTag("cscSegments"),
0097 )
0098
0099 from Configuration.Eras.Modifier_run3_GEM_cff import run3_GEM
0100 from Configuration.Eras.Modifier_phase2_tracker_cff import phase2_tracker
0101 run3_GEM.toModify(muonAssociatorByHitsCommonParameters, useGEMs = True)
0102 phase2_tracker.toModify(muonAssociatorByHitsCommonParameters,
0103     usePhase2Tracker = True,
0104     pixelSimLinkSrc = "simSiPixelDigis:Pixel",
0105 )
0106
0107 from Configuration.Eras.Modifier_fastSim_cff import fastSim
0108 fastSim.toModify(muonAssociatorByHitsCommonParameters,
0109     simtracksTag = "fastSimProducer",
0110     DTsimhitsTag  = "MuonSimHits:MuonDTHits",
0111     CSCsimHitsTag = "MuonSimHits:MuonCSCHits",
0112     RPCsimhitsTag = "MuonSimHits:MuonRPCHits",
0113     simtracksXFTag = "mix:fastSimProducer",
0114     DTsimhitsXFTag  = "mix:MuonSimHitsMuonDTHits",
0115     CSCsimHitsXFTag = "mix:MuonSimHitsMuonCSCHits",
0116     RPCsimhitsXFTag = "mix:MuonSimHitsMuonRPCHits",
0117     ROUList = ['fastSimProducerTrackerHits']
0118 )
0119
0120 from Configuration.ProcessModifiers.premix_stage2_cff import premix_stage2
0121 premix_stage2.toModify(muonAssociatorByHitsCommonParameters,
0122     pixelSimLinkSrc = "mixData:PixelDigiSimLink",
0123     stripSimLinkSrc = "mixData:StripDigiSimLink",
0124     phase2TrackerSimLinkSrc = "mixData:Phase2OTDigiSimLink",
0125 )
0126
0127 muonAssociatorByHits = cms.EDProducer("MuonAssociatorEDProducer",
0128     # COMMON CONFIGURATION
0129     muonAssociatorByHitsCommonParameters,
0130     # for Muon Track association
0131     #
0132     #     input collections
0133     #
0134     # ... reco::Track collection
0135     tracksTag = cms.InputTag("globalMuons"),
0136     ignoreMissingTrackCollection = cms.untracked.bool(False),
0137     # ... TrackingParticle collection
0138     tpTag = cms.InputTag("mix","MergedTrackTruth"),
0139     tpRefVector = cms.bool(False)
0140 )
0141
0142 premix_stage2.toModify(muonAssociatorByHits, tpTag = "mixData:MergedTrackTruth")