Move LSTGeometry repo into ESProducer and standalone binary

RecoTracker/LST/plugins/alpaka/LSTModulesDevESProducer.cc

@@ -1,5 +1,8 @@
+#include <sstream>
+
 // LST includes
 #include "RecoTracker/LSTCore/interface/alpaka/LST.h"
+#include "RecoTracker/LSTGeometry/interface/Geometry.h"
 
 #include "FWCore/ParameterSet/interface/ParameterSet.h"
 
@@ -14,22 +17,29 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
 
   class LSTModulesDevESProducer : public ESProducer {
   private:
-    std::string ptCutLabel_;
+    double ptCut_;
+    edm::ESGetToken<lstgeometry::Geometry, TrackerRecoGeometryRecord> lstGeoToken_;
 
   public:
     LSTModulesDevESProducer(edm::ParameterSet const& iConfig)
-        : ESProducer(iConfig), ptCutLabel_(iConfig.getParameter<std::string>("ptCutLabel")) {
-      setWhatProduced(this, ptCutLabel_);
+        : ESProducer(iConfig), ptCut_(iConfig.getParameter<double>("ptCut")) {
+      std::ostringstream ptCutOSS;
+      ptCutOSS << std::setprecision(1) << ptCut_;
+      std::string ptCutStr = ptCutOSS.str();
+
+      auto cc = setWhatProduced(this, "LSTModuleMaps_" + ptCutStr);
+      lstGeoToken_ = cc.consumes<lstgeometry::Geometry>(edm::ESInputTag("", "LSTGeometry_" + ptCutStr));
     }
 
     static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
       edm::ParameterSetDescription desc;
-      desc.add<std::string>("ptCutLabel", "0.8");
+      desc.add<double>("ptCut", 0.8);
       descriptions.addWithDefaultLabel(desc);
     }
 
     std::unique_ptr<lst::LSTESData<DevHost>> produce(TrackerRecoGeometryRecord const& iRecord) {
-      return lst::loadAndFillESHost(ptCutLabel_);
+      const auto& lstg = iRecord.get(lstGeoToken_);
+      return lst::fillESDataHost(lstg);
     }
   };
 

RecoTracker/LST/plugins/alpaka/LSTProducer.cc

@@ -1,5 +1,7 @@
 #include <alpaka/alpaka.hpp>
 
+#include <sstream>
+
 #include "RecoTracker/LSTCore/interface/alpaka/LST.h"
 
 #include "FWCore/MessageLogger/interface/MessageLogger.h"
@@ -25,13 +27,14 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
   public:
     LSTProducer(edm::ParameterSet const& config)
         : EDProducer(config),
-          lstInputToken_{consumes(config.getParameter<edm::InputTag>("lstInput"))},
-          lstESToken_{esConsumes(edm::ESInputTag("", config.getParameter<std::string>("ptCutLabel")))},
           verbose_(config.getParameter<bool>("verbose")),
           ptCut_(config.getParameter<double>("ptCut")),
+          ptCutStr_(getPtCutStr(ptCut_)),
           clustSizeCut_(static_cast<uint16_t>(config.getParameter<uint32_t>("clustSizeCut"))),
           nopLSDupClean_(config.getParameter<bool>("nopLSDupClean")),
           tcpLSTriplets_(config.getParameter<bool>("tcpLSTriplets")),
+          lstInputToken_{consumes(config.getParameter<edm::InputTag>("lstInput"))},
+          lstESToken_{esConsumes(edm::ESInputTag("", "LSTModuleMaps_" + ptCutStr_))},
           lstOutputToken_{produces()} {}
 
     void produce(edm::StreamID sid, device::Event& iEvent, const device::EventSetup& iSetup) const override {
@@ -42,7 +45,7 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
 
       lst.run(iEvent.queue(),
               verbose_,
-              static_cast<float>(ptCut_),
+              ptCut_,
               clustSizeCut_,
               &lstESDeviceData,
               &lstInputDC,
@@ -60,21 +63,27 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE {
       desc.add<bool>("verbose", false);
       desc.add<double>("ptCut", 0.8);
       desc.add<uint32_t>("clustSizeCut", 16);
-      desc.add<std::string>("ptCutLabel", "0.8");
       desc.add<bool>("nopLSDupClean", false);
       desc.add<bool>("tcpLSTriplets", false);
       descriptions.addWithDefaultLabel(desc);
     }
 
   private:
-    const device::EDGetToken<lst::LSTInputDeviceCollection> lstInputToken_;
-    const device::ESGetToken<lst::LSTESData<Device>, TrackerRecoGeometryRecord> lstESToken_;
     const bool verbose_;
     const double ptCut_;
+    const std::string ptCutStr_;
     const uint16_t clustSizeCut_;
     const bool nopLSDupClean_;
     const bool tcpLSTriplets_;
+    const device::EDGetToken<lst::LSTInputDeviceCollection> lstInputToken_;
+    const device::ESGetToken<lst::LSTESData<Device>, TrackerRecoGeometryRecord> lstESToken_;
     const device::EDPutToken<lst::TrackCandidatesBaseDeviceCollection> lstOutputToken_;
+
+    static std::string getPtCutStr(double ptCut) {
+      std::ostringstream ptCutOSS;
+      ptCutOSS << std::setprecision(1) << ptCut;
+      return ptCutOSS.str();
+    }
   };
 
 }  // namespace ALPAKA_ACCELERATOR_NAMESPACE

RecoTracker/LST/python/lstProducerTask_cff.py

@@ -4,4 +4,8 @@
 
 from RecoTracker.LST.lstModulesDevESProducer_cfi import lstModulesDevESProducer
 
-lstProducerTask = cms.Task(lstModulesDevESProducer, lstProducer)
+from RecoTracker.LST.lstInputProducer_cfi import lstInputProducer
+
+from RecoTracker.LSTGeometry.lstGeometryESProducer_cfi import lstGeometryESProducer
+
+lstProducerTask = cms.Task(lstGeometryESProducer, lstModulesDevESProducer, lstInputProducer, lstProducer)

RecoTracker/LSTCore/BuildFile.xml

@@ -5,6 +5,7 @@
 <use name="DataFormats/Common"/>
 <use name="DataFormats/TrackerRecHit2D"/>
 <use name="HeterogeneousCore/AlpakaInterface"/>
+<use name="RecoTracker/LSTGeometry"/>
 <flags ALPAKA_BACKENDS="1"/>
 <export>
   <lib name="1"/>

RecoTracker/LSTCore/interface/Common.h

@@ -4,6 +4,8 @@
 #include "DataFormats/Common/interface/StdArray.h"
 #include "HeterogeneousCore/AlpakaInterface/interface/config.h"
 
+#include "RecoTracker/LSTGeometry/interface/Common.h"
+
 #if defined(FP16_Base)
 #if defined ALPAKA_ACC_GPU_CUDA_ENABLED
 #include <cuda_fp16.h>
@@ -42,6 +44,9 @@ namespace lst {
   constexpr uint16_t kTCEmptyLowerModule = 0xFFFF;     // Sentinel for empty lowerModule index
   constexpr unsigned int kTCEmptyHitIdx = 0xFFFFFFFF;  // Sentinel for empty hit slots
 
+  constexpr float kB = lstgeometry::kB;
+  constexpr float kC = lstgeometry::kC;
+
 // Half precision wrapper functions.
 #if defined(FP16_Base)
 #define __F2H __float2half

RecoTracker/LSTCore/interface/EndcapGeometry.h

@@ -1,6 +1,9 @@
 #ifndef RecoTracker_LSTCore_interface_EndcapGeometry_h
 #define RecoTracker_LSTCore_interface_EndcapGeometry_h
 
+#include "RecoTracker/LSTGeometry/interface/Sensor.h"
+#include "RecoTracker/LSTGeometry/interface/Slope.h"
+
 #include <map>
 #include <string>
 #include <vector>
@@ -18,9 +21,11 @@ namespace lst {
     unsigned int nEndCapMap;
 
     EndcapGeometry() = default;
-    EndcapGeometry(std::string const& filename);
+    EndcapGeometry(std::string const&);
+    EndcapGeometry(lstgeometry::Slopes const&, lstgeometry::Sensors const&);
 
     void load(std::string const&);
+    void load(lstgeometry::Slopes const&, lstgeometry::Sensors const&);
     void fillGeoMapArraysExplicit();
     float getdxdy_slope(unsigned int detid) const;
   };

RecoTracker/LSTCore/interface/LSTESData.h

@@ -5,6 +5,7 @@
 #include "RecoTracker/LSTCore/interface/EndcapGeometryDevHostCollection.h"
 #include "RecoTracker/LSTCore/interface/ModulesHostCollection.h"
 #include "RecoTracker/LSTCore/interface/PixelMap.h"
+#include "RecoTracker/LSTGeometry/interface/Geometry.h"
 
 #include "HeterogeneousCore/AlpakaInterface/interface/CopyToDevice.h"
 
@@ -40,7 +41,8 @@ namespace lst {
           pixelMapping(pixelMappingIn) {}
   };
 
-  std::unique_ptr<LSTESData<alpaka_common::DevHost>> loadAndFillESHost(std::string& ptCutLabel);
+  std::unique_ptr<LSTESData<alpaka_common::DevHost>> loadAndFillESDataHost(std::string& ptCutLabel);
+  std::unique_ptr<LSTESData<alpaka_common::DevHost>> fillESDataHost(lstgeometry::Geometry const& lstg);
 
 }  // namespace lst
 

RecoTracker/LSTCore/interface/ModuleConnectionMap.h

@@ -12,10 +12,12 @@ namespace lst {
     std::map<unsigned int, std::vector<unsigned int>> moduleConnections_;
 
   public:
-    ModuleConnectionMap();
-    ModuleConnectionMap(std::string const& filename);
+    ModuleConnectionMap() = default;
+    ModuleConnectionMap(std::string const&);
+    ModuleConnectionMap(std::map<unsigned int, std::vector<unsigned int>> const&);
 
     void load(std::string const&);
+    void load(std::map<unsigned int, std::vector<unsigned int>> const&);
     void add(std::string const&);
     void print();
 

RecoTracker/LSTCore/interface/TiltedGeometry.h

@@ -1,6 +1,8 @@
 #ifndef RecoTracker_LSTCore_interface_TiltedGeometry_h
 #define RecoTracker_LSTCore_interface_TiltedGeometry_h
 
+#include "RecoTracker/LSTGeometry/interface/Slope.h"
+
 #include <map>
 #include <string>
 #include <vector>
@@ -13,9 +15,11 @@ namespace lst {
 
   public:
     TiltedGeometry() = default;
-    TiltedGeometry(std::string const& filename);
+    TiltedGeometry(std::string const&);
+    TiltedGeometry(lstgeometry::Slopes const&);
 
     void load(std::string const&);
+    void load(lstgeometry::Slopes const&);
 
     float getDrDz(unsigned int detid) const;
     float getDxDy(unsigned int detid) const;

RecoTracker/LSTCore/interface/alpaka/Common.h

@@ -32,8 +32,8 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE::lst {
       25.007152356, 37.2186993757, 52.3104270826, 68.6658656666, 85.9770373007, 108.301772384};
   HOST_DEVICE_CONSTANT float kMiniRminMeanEndcap[5] = {
       130.992832231, 154.813883559, 185.352604327, 221.635123002, 265.022076742};
-  HOST_DEVICE_CONSTANT float k2Rinv1GeVf = (2.99792458e-3 * 3.8) / 2;
-  HOST_DEVICE_CONSTANT float kR1GeVf = 1. / (2.99792458e-3 * 3.8);
+  HOST_DEVICE_CONSTANT float k2Rinv1GeVf = (kC * kB) / 2;
+  HOST_DEVICE_CONSTANT float kR1GeVf = 1. / (kC * kB);
   HOST_DEVICE_CONSTANT float kSinAlphaMax = 0.95;
   HOST_DEVICE_CONSTANT float kDeltaZLum = 15.0;
   HOST_DEVICE_CONSTANT float kPixelPSZpitch = 0.15;
@@ -43,8 +43,6 @@ namespace ALPAKA_ACCELERATOR_NAMESPACE::lst {
   HOST_DEVICE_CONSTANT float kWidthPS = 0.01;
   HOST_DEVICE_CONSTANT float kPt_betaMax = 7.0;
   HOST_DEVICE_CONSTANT int kNTripletThreshold = 1000;
-  // To be updated with std::numeric_limits<float>::infinity() in the code and data files
-  HOST_DEVICE_CONSTANT float kVerticalModuleSlope = 123456789.0;
   HOST_DEVICE_CONSTANT int kLogicalOTLayers = 11;  // logical OT layers are 1..11
 
   HOST_DEVICE_CONSTANT float kMiniDeltaTilted[3] = {0.26f, 0.26f, 0.26f};

RecoTracker/LSTCore/src/EndcapGeometry.cc

@@ -7,6 +7,10 @@
 
 lst::EndcapGeometry::EndcapGeometry(std::string const& filename) { load(filename); }
 
+lst::EndcapGeometry::EndcapGeometry(lstgeometry::Slopes const& slopes, lstgeometry::Sensors const& sensors) {
+  load(slopes, sensors);
+}
+
 void lst::EndcapGeometry::load(std::string const& filename) {
   dxdy_slope_.clear();
   centroid_phis_.clear();
@@ -39,6 +43,18 @@ void lst::EndcapGeometry::load(std::string const& filename) {
   fillGeoMapArraysExplicit();
 }
 
+void lst::EndcapGeometry::load(lstgeometry::Slopes const& slopes, lstgeometry::Sensors const& sensors) {
+  dxdy_slope_.clear();
+  centroid_phis_.clear();
+
+  for (const auto& [detId, slope] : slopes) {
+    dxdy_slope_[detId] = slope.dxdy;
+    centroid_phis_[detId] = sensors.at(detId).centerPhi;
+  }
+
+  fillGeoMapArraysExplicit();
+}
+
 void lst::EndcapGeometry::fillGeoMapArraysExplicit() {
   nEndCapMap = centroid_phis_.size();
 

RecoTracker/LSTCore/src/LSTESData.cc

@@ -7,6 +7,8 @@
 #include "ModuleMethods.h"
 
 #include <filesystem>
+#include <forward_list>
+#include <iterator>
 
 namespace {
   std::string geometryDataDir() {
@@ -79,7 +81,7 @@ namespace {
   }
 }  // namespace
 
-std::unique_ptr<lst::LSTESData<alpaka_common::DevHost>> lst::loadAndFillESHost(std::string& ptCutLabel) {
+std::unique_ptr<lst::LSTESData<alpaka_common::DevHost>> lst::loadAndFillESDataHost(std::string& ptCutLabel) {
   uint16_t nModules;
   uint16_t nLowerModules;
   unsigned int nPixels;
@@ -119,3 +121,83 @@ std::unique_ptr<lst::LSTESData<alpaka_common::DevHost>> lst::loadAndFillESHost(s
                                                              std::move(endcapGeometryDev),
                                                              pixelMappingPtr);
 }
+
+std::unique_ptr<lst::LSTESData<alpaka_common::DevHost>> lst::fillESDataHost(lstgeometry::Geometry const& lstg) {
+  uint16_t nModules;
+  uint16_t nLowerModules;
+  unsigned int nPixels;
+  MapPLStoLayer pLStoLayer;
+  EndcapGeometry endcapGeometry;
+  TiltedGeometry tiltedGeometry;
+  PixelMap pixelMapping;
+  ModuleConnectionMap moduleConnectionMap;
+
+  endcapGeometry.load(lstg.endcap_slopes, *lstg.sensors);
+  auto endcapGeometryDev =
+      std::make_shared<EndcapGeometryDevHostCollection>(cms::alpakatools::host(), endcapGeometry.nEndCapMap);
+  std::memcpy(endcapGeometryDev->view().geoMapDetId().data(),
+              endcapGeometry.geoMapDetId_buf.data(),
+              endcapGeometry.nEndCapMap * sizeof(unsigned int));
+  std::memcpy(endcapGeometryDev->view().geoMapPhi().data(),
+              endcapGeometry.geoMapPhi_buf.data(),
+              endcapGeometry.nEndCapMap * sizeof(float));
+
+  tiltedGeometry.load(lstg.barrel_slopes);
+
+  std::map<unsigned int, std::vector<unsigned int>> final_modulemap;
+  for (auto const& [detId, connections] : lstg.module_map) {
+    final_modulemap[detId] = std::vector<unsigned int>(connections.begin(), connections.end());
+  }
+  moduleConnectionMap.load(final_modulemap);
+
+  for (auto& [layersubdetcharge, map] : lstg.pixel_map) {
+    auto& [layer, subdet, charge] = layersubdetcharge;
+
+    std::map<unsigned int, std::vector<unsigned int>> final_pixelmap;
+    for (unsigned int isuperbin = 0; isuperbin < map.size(); isuperbin++) {
+      auto const& set = map.at(isuperbin);
+      final_pixelmap[isuperbin] = std::vector<unsigned int>(set.begin(), set.end());
+    }
+
+    if (charge == 0) {
+      pLStoLayer[0][layer - 1 + (subdet == Endcap ? 2 : 0)] = lst::ModuleConnectionMap(final_pixelmap);
+    } else if (charge > 0) {
+      pLStoLayer[1][layer - 1 + (subdet == Endcap ? 2 : 0)] = lst::ModuleConnectionMap(final_pixelmap);
+    } else {
+      pLStoLayer[2][layer - 1 + (subdet == Endcap ? 2 : 0)] = lst::ModuleConnectionMap(final_pixelmap);
+    }
+  }
+
+  ModuleMetaData mmd;
+  unsigned int counter = 0;
+  for (auto const& [detId, sensor] : *lstg.sensors) {
+    mmd.detIdToIndex[detId] = counter;
+    mmd.module_x[detId] = sensor.centerX;
+    mmd.module_y[detId] = sensor.centerY;
+    mmd.module_z[detId] = sensor.centerZ;
+    mmd.module_type[detId] = static_cast<unsigned int>(sensor.moduleType);
+    counter++;
+  }
+  mmd.detIdToIndex[kPixelModuleId] = counter;  //pixel module is the last module in the module list
+  counter++;
+  nModules = counter;
+
+  auto modulesBuffers = constructModuleCollection(pLStoLayer,
+                                                  mmd,
+                                                  nModules,
+                                                  nLowerModules,
+                                                  nPixels,
+                                                  pixelMapping,
+                                                  endcapGeometry,
+                                                  tiltedGeometry,
+                                                  moduleConnectionMap);
+
+  auto pixelMappingPtr = std::make_shared<PixelMap>(std::move(pixelMapping));
+  return std::make_unique<LSTESData<alpaka_common::DevHost>>(nModules,
+                                                             nLowerModules,
+                                                             nPixels,
+                                                             endcapGeometry.nEndCapMap,
+                                                             std::move(modulesBuffers),
+                                                             std::move(endcapGeometryDev),
+                                                             pixelMappingPtr);
+}