RecoLocalTracker/SiStripZeroSuppression/python/customiseHybrid.py - master - githubmirror/cmssw

RecoLocalTracker/SiStripZeroSuppression/python/customiseHybrid.py

50 строк2 KB

mmusich

update customiseHybrid to comply with new definition of siStripClusters

08 янв 2025, 12:27

08 янв 2025, 12:27974bebf

import FWCore.ParameterSet.Config as cms

## Adjust the local reco sequence for running on hybrid zero-suppressed data
def runOnHybridZS(process):
    process.load("RecoLocalTracker.SiStripZeroSuppression.SiStripZeroSuppression_cfi")
    process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizer_cfi")
    process.siStripZeroSuppression.Algorithms.APVInspectMode = "Hybrid"
    zsInputs = process.siStripZeroSuppression.RawDigiProducersList
    clusInputs = process.siStripClusters.DigiProducersList
    unpackedZS = cms.InputTag("siStripDigis", "ZeroSuppressed")

    # Convert string elements to cms.InputTag objects if necessary
    clusInputs = [cms.InputTag(x) if isinstance(x, str) else x for x in clusInputs]

    zsInputs.append(unpackedZS)
    if unpackedZS in clusInputs:
        clusInputs.remove(unpackedZS)
    clusInputs.append(cms.InputTag("siStripZeroSuppression", "ZeroSuppressed"))

    # for on-demand clusterizer
    from FWCore.ParameterSet.MassReplace import massReplaceParameter
    massReplaceParameter(process, "HybridZeroSuppressed", cms.bool(False), cms.bool(True))
    return process

## Change the (normal, ZS) repacker to use zero-suppressed hybrid data
def repackZSHybrid(process):
    process.SiStripDigiToZSRaw.InputDigis = cms.InputTag("siStripZeroSuppression", "ZeroSuppressed")

    process.DigiToRawRepack.insert(0, process.siStripZeroSuppression)

    return process

## Add the ZS algorithm (in hybrid emulation mode) before repacking, to produce emulated hybrid samples with
##   cmsDriver --step RAW2DIGI,REPACK:DigiToHybridRawRepack --customiseRecoLocalTracker/SiStripZeroSuppression/customiseHybrid.addHybridEmulationBeforeRepack ...
def addHybridEmulationBeforeRepack(process):
    process.load("RecoLocalTracker.SiStripZeroSuppression.SiStripZeroSuppression_cfi")
    zs = process.siStripZeroSuppression
    zs.produceRawDigis = False
    zs.produceHybridFormat = True
    zs.Algorithms.APVInspectMode = "HybridEmulation"
    zs.Algorithms.APVRestoreMode = ""
    zs.Algorithms.CommonModeNoiseSubtractionMode = 'Median'
    zs.Algorithms.MeanCM = 0
    zs.Algorithms.DeltaCMThreshold = 20
    zs.Algorithms.Use10bitsTruncation = True
    zs.RawDigiProducersList = cms.VInputTag(cms.InputTag("siStripDigis", "VirginRaw"))

    process.DigiToHybridRawRepack.insert(0, zs) ## insert before repacking

    return process