SiStripZeroSuppression/python/customiseHybrid.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 ## Adjust the local reco sequence for running on hybrid zero-suppressed data
0004 def runOnHybridZS(process):
0005     process.load("RecoLocalTracker.SiStripZeroSuppression.SiStripZeroSuppression_cfi")
0006     process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizer_cfi")
0007     process.siStripZeroSuppression.Algorithms.APVInspectMode = "Hybrid"
0008     zsInputs = process.siStripZeroSuppression.RawDigiProducersList
0009     clusInputs = process.siStripClusters.DigiProducersList
0010     unpackedZS = cms.InputTag("siStripDigis", "ZeroSuppressed")
0011
0012     # Convert string elements to cms.InputTag objects if necessary
0013     clusInputs = [cms.InputTag(x) if isinstance(x, str) else x for x in clusInputs]
0014
0015     zsInputs.append(unpackedZS)
0016     if unpackedZS in clusInputs:
0017         clusInputs.remove(unpackedZS)
0018     clusInputs.append(cms.InputTag("siStripZeroSuppression", "ZeroSuppressed"))
0019
0020     # for on-demand clusterizer
0021     from FWCore.ParameterSet.MassReplace import massReplaceParameter
0022     massReplaceParameter(process, "HybridZeroSuppressed", cms.bool(False), cms.bool(True))
0023     return process
0024
0025 ## Change the (normal, ZS) repacker to use zero-suppressed hybrid data
0026 def repackZSHybrid(process):
0027     process.SiStripDigiToZSRaw.InputDigis = cms.InputTag("siStripZeroSuppression", "ZeroSuppressed")
0028
0029     process.DigiToRawRepack.insert(0, process.siStripZeroSuppression)
0030
0031     return process
0032
0033 ## Add the ZS algorithm (in hybrid emulation mode) before repacking, to produce emulated hybrid samples with
0034 ##   cmsDriver --step RAW2DIGI,REPACK:DigiToHybridRawRepack --customiseRecoLocalTracker/SiStripZeroSuppression/customiseHybrid.addHybridEmulationBeforeRepack ...
0035 def addHybridEmulationBeforeRepack(process):
0036     process.load("RecoLocalTracker.SiStripZeroSuppression.SiStripZeroSuppression_cfi")
0037     zs = process.siStripZeroSuppression
0038     zs.produceRawDigis = False
0039     zs.produceHybridFormat = True
0040     zs.Algorithms.APVInspectMode = "HybridEmulation"
0041     zs.Algorithms.APVRestoreMode = ""
0042     zs.Algorithms.CommonModeNoiseSubtractionMode = 'Median'
0043     zs.Algorithms.MeanCM = 0
0044     zs.Algorithms.DeltaCMThreshold = 20
0045     zs.Algorithms.Use10bitsTruncation = True
0046     zs.RawDigiProducersList = cms.VInputTag(cms.InputTag("siStripDigis", "VirginRaw"))
0047
0048     process.DigiToHybridRawRepack.insert(0, zs) ## insert before repacking
0049
0050     return process