max num of proj layers added to channel assignment.

cms-L1TK · May 13, 2022 · 422361c · 422361c
1 parent a465123
commit 422361c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/L1Trigger/TrackFindingTracklet/interface/ChannelAssignment.h b/L1Trigger/TrackFindingTracklet/interface/ChannelAssignment.h
@@ -50,6 +50,8 @@ namespace trklet {
     int channelId(int seedType, int layerId) const;
     // max number of seeding layers
     int numSeedingLayers() const { return numSeedingLayers_; }
+    // max number of projection layers
+    int numProjectionLayers() const { return numProjectionLayers_; }
 
   private:
     // helper class to store configurations
@@ -76,6 +78,8 @@ namespace trklet {
     std::vector<int> offsetsStubs_;
     // max number of seeding layers
     int numSeedingLayers_;
+    // max number of projection layers
+    int numProjectionLayers_;
   };
 
 }  // namespace trklet

diff --git a/L1Trigger/TrackFindingTracklet/plugins/L1FPGATrackProducer.cc b/L1Trigger/TrackFindingTracklet/plugins/L1FPGATrackProducer.cc
@@ -617,11 +617,7 @@ void L1FPGATrackProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSe
   // number of stub channels
   const unsigned int numStreamsStub = N_SECTOR * channelAssignment_->numChannelsStub();
   // number of stub channels if all seed types streams padded to have same number of stub channels (for coding simplicity)
-  unsigned int maxNumProjectionLayers = 0;
-  for (int iSeedType = 0; iSeedType < channelAssignment_->numSeedTypes(); iSeedType++) {
-    maxNumProjectionLayers =
-        max(maxNumProjectionLayers, (unsigned int)channelAssignment_->numProjectionLayers(iSeedType));
-  }
+  const unsigned int maxNumProjectionLayers = channelAssignment_->numProjectionLayers();
   const unsigned int numStreamsStubRaw = numStreamsTrack * maxNumProjectionLayers;
 
   // Streams formatted to allow this code to run outside CMSSW.

diff --git a/L1Trigger/TrackFindingTracklet/src/ChannelAssignment.cc b/L1Trigger/TrackFindingTracklet/src/ChannelAssignment.cc
@@ -110,6 +110,8 @@ namespace trklet {
     }
     auto bigger = [](const vector<int>& lhs, const vector<int>& rhs) { return lhs.size() < rhs.size(); };
     numSeedingLayers_ = max_element(seedTypesSeedLayers_.begin(), seedTypesSeedLayers_.end(), bigger)->size();
+    numProjectionLayers_ =
+        max_element(seedTypesProjectionLayers_.begin(), seedTypesProjectionLayers_.end(), bigger)->size();
   }
 
   // sets channelId of given TTTrackRef, return false if track outside pt range