add if statement back

SegmentLinking · May 28, 2024 · 8dd1891 · 8dd1891
1 parent 7749db7
commit 8dd1891
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/code/core/write_sdl_ntuple.cc b/code/core/write_sdl_ntuple.cc
@@ -359,8 +359,12 @@ void setOccupancyBranches(SDL::Event<SDL::Acc>* event)
         moduleModule.push_back(modulesInGPU.modules[lowerIdx]);
         segmentOccupancy.push_back(segmentsInGPU.totOccupancySegments[lowerIdx]);
         mdOccupancy.push_back(mdsInGPU.totOccupancyMDs[lowerIdx]);
-        quintupletOccupancy.push_back(quintupletsInGPU.totOccupancyQuintuplets[lowerIdx]);
-        tripletOccupancy.push_back(tripletsInGPU.totOccupancyTriplets[lowerIdx]);
+
+        if(idx < *(modulesInGPU.nLowerModules))
+        {
+            quintupletOccupancy.push_back(quintupletsInGPU.totOccupancyQuintuplets[lowerIdx]);
+            tripletOccupancy.push_back(tripletsInGPU.totOccupancyTriplets[lowerIdx]);
+        }
     }
 
     ana.tx->setBranch<vector<int>>("module_layers", moduleLayer);