customize hlt

RecoLocalTracker/SiStripClusterizer/python/customizeStripClustersFromRaw.py

@@ -1,9 +1,22 @@
 import FWCore.ParameterSet.Config as cms
+import RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi as SiStripClusterizerOnDemand_cfi
 
 def customizeStripClustersFromRaw(process):
-    process.striptrackerlocalrecoTask.remove(process.siStripClusters)
-    process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi")
-    #process.siStripClusterizerFromRaw.onDemand = cms.bool(False)
-    process.striptrackerlocalrecoTask.add(process.siStripClustersTask)
+    if hasattr(process, 'striptrackerlocalrecoTask'):
+        process.striptrackerlocalrecoTask.remove(process.siStripClusters)
+        process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi")
+        # CPU should emulate the full detector clusterizer
+        process.siStripClusterizerFromRaw.onDemand = cms.bool(False)
+        process.striptrackerlocalrecoTask.add(process.siStripClustersTask)
+
+    return process
+
+def customizeHLTStripClustersFromRaw(process):
+    if hasattr(process, 'hltSiStripRawToClustersFacility'):
+        process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizerConditionsGPUESProducer_cfi")
+        process.load("RecoLocalTracker.SiStripClusterizer.SiStripClusterizerOnDemand_cfi")
+        process.hltSiStripRawToClustersFacility = SiStripClusterizerOnDemand_cfi.siStripClusters.clone()
+        process.HLTDoLocalStripSequence.replace(process.hltSiStripRawToClustersFacility,
+                                   cms.Sequence(process.hltSiStripRawToClustersFacility, process.siStripClustersTask))
 
     return process