SiStripClusterizer/python/customizeStripClustersFromRaw.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # replace the standard SiStripClusterizer with the switch producer
0004 # meant primarily for testing
0005 def customizeStripClustersFromRaw(process):
0006     if hasattr(process, 'striptrackerlocalrecoTask'):
0007         process.striptrackerlocalrecoTask.remove(process.siStripClusters)
0008         process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi")
0009         # CPU should emulate the full detector clusterizer
0010         process.siStripClusters.cpu.onDemand = cms.bool(False)
0011         process.striptrackerlocalrecoTask.add(process.siStripClustersTask)
0012
0013     return process
0014
0015 def customizeHLTStripClustersFromRaw(process):
0016     if hasattr(process, 'hltSiStripRawToClustersFacility'):
0017         import RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi as SiStripClusterizerOnDemand_cfi
0018
0019         process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi")
0020         process.hltSiStripRawToClustersFacility = SiStripClusterizerOnDemand_cfi.siStripClusters.clone()
0021         process.HLTDoLocalStripSequence.replace(process.hltSiStripRawToClustersFacility,
0022                                    cms.Sequence(process.hltSiStripRawToClustersFacility, process.siStripClustersTaskCUDA))
0023
0024     return process