Add configurable pT cut label to LSTProducer

SegmentLinking · Jun 26, 2024 · 0c5c484 · 0c5c484
1 parent 959da59
commit 0c5c484
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/RecoTracker/LST/plugins/alpaka/LSTModulesDevESProducer.cc b/RecoTracker/LST/plugins/alpaka/LSTModulesDevESProducer.cc
@@ -30,8 +30,8 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
 
   LSTModulesDevESProducer::LSTModulesDevESProducer(const edm::ParameterSet& iConfig)
       : ESProducer(iConfig), ptCutLabel_(iConfig.getParameter<std::string>("ptCutLabel")) {
-    setWhatProduced(this, &LSTModulesDevESProducer::produceHost);
-    auto cc = setWhatProduced(this, &LSTModulesDevESProducer::produceDevice);
+    setWhatProduced(this, &LSTModulesDevESProducer::produceHost, ptCutLabel_);
+    auto cc = setWhatProduced(this, &LSTModulesDevESProducer::produceDevice, ptCutLabel_);
     lstESHostToken_ = cc.consumes();
   }
 

diff --git a/RecoTracker/LST/plugins/alpaka/LSTProducer.cc b/RecoTracker/LST/plugins/alpaka/LSTProducer.cc
@@ -29,7 +29,7 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
         : lstPixelSeedInputToken_{consumes<LSTPixelSeedInput>(config.getParameter<edm::InputTag>("pixelSeedInput"))},
           lstPhase2OTHitsInputToken_{
               consumes<LSTPhase2OTHitsInput>(config.getParameter<edm::InputTag>("phase2OTHitsInput"))},
-          lstESToken_{esConsumes()},
+          lstESToken_{esConsumes(edm::ESInputTag("", config.getParameter<std::string>("ptCutLabel")))},
           verbose_(config.getParameter<bool>("verbose")),
           ptCut_(config.getParameter<double>("ptCut")),
           nopLSDupClean_(config.getParameter<bool>("nopLSDupClean")),
@@ -84,6 +84,7 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
       desc.add<edm::InputTag>("phase2OTHitsInput", edm::InputTag{"lstPhase2OTHitsInputProducer"});
       desc.add<bool>("verbose", false);
       desc.add<double>("ptCut", 0.8);
+      desc.add<std::string>("ptCutLabel", "0.8");
       desc.add<bool>("nopLSDupClean", false);
       desc.add<bool>("tcpLSTriplets", false);
       descriptions.addWithDefaultLabel(desc);

diff --git a/RecoTracker/LSTCore/src/alpaka/MiniDoublet.h b/RecoTracker/LSTCore/src/alpaka/MiniDoublet.h
@@ -336,7 +336,7 @@ namespace SDL {
                                                      float rt,
                                                      struct SDL::modules& modulesInGPU,
                                                      uint16_t& moduleIndex,
-                                                     const float& ptCut,
+                                                     const float ptCut,
                                                      float dPhi = 0,
                                                      float dz = 0) {
     // =================================================================