because I can't let well enough alone

QIICR · Jan 30, 2024 · fe0ba32 · fe0ba32
1 parent 763023f
commit fe0ba32
Showing 1 changed file with 20 additions and 15 deletions.
diff --git a/libsrc/OverlapUtil.cpp b/libsrc/OverlapUtil.cpp
@@ -121,11 +121,11 @@ OFCondition OverlapUtil::getFramesForSegment(const Uint32 segmentNumber, OFVecto
             return EC_IllegalParameter;
         }
         Uint32 numFrames = static_cast<Uint32>(m_seg->getNumberOfFrames());
-         // use sets for performance and deduplication while adding frames to each set
-        OFVector<std::set<Uint32>> segmentFrameSet;
-        segmentFrameSet.resize(m_seg->getNumberOfSegments());
-        // Get Segmentation FG for each frame and remember the segment number for each frame
-        // in the vector m_segmentsForFrame
+        // use sets for performance and deduplication when adding frames to each segment index
+        OFVector<std::set<Uint32>> frameSetsForSegment;
+        frameSetsForSegment.resize(m_seg->getNumberOfSegments());
+        // Use the helper function getSegmentsForFrame() to pivot the data
+        // from segments on each frame, to the list of frames with contain each segment
         for (Uint32 f = 0; f < numFrames; f++)
         {
             OFCondition result;
@@ -137,7 +137,7 @@ OFCondition OverlapUtil::getFramesForSegment(const Uint32 segmentNumber, OFVecto
                      itr != segments.end();
                      itr++)
                 {
-                    segmentFrameSet[(*itr) - 1].insert(f);
+                    frameSetsForSegment[(*itr) - 1].insert(f);
                 }
             }
             else
@@ -146,12 +146,12 @@ OFCondition OverlapUtil::getFramesForSegment(const Uint32 segmentNumber, OFVecto
                 return result;
             }
         }
-        // copy the set back into the OFVector in m_framesForSegment
-        m_framesForSegment.resize(segmentFrameSet.size());
-        for (Uint32 s = 0; s < segmentFrameSet.size(); s++) {
-            m_framesForSegment[s].reserve(segmentFrameSet[s].size());
-            for (std::set<Uint32>::iterator itr = segmentFrameSet[s].begin();
-                 itr != segmentFrameSet[s].end();
+        // copy the sets back into the OFVectors in m_framesForSegment
+        m_framesForSegment.resize(frameSetsForSegment.size());
+        for (Uint32 s = 0; s < frameSetsForSegment.size(); s++) {
+            m_framesForSegment[s].reserve(frameSetsForSegment[s].size());
+            for (std::set<Uint32>::iterator itr = frameSetsForSegment[s].begin();
+                 itr != frameSetsForSegment[s].end();
                  itr++)
             {
                 m_framesForSegment[s].push_back(*itr);
@@ -352,7 +352,8 @@ OFCondition OverlapUtil::getSegmentsByPosition(SegmentsByPosition& result)
         {
             std::set<Uint32> segmentsInFrame;
             Uint32 frameNumber = m_logicalFramePositions[l][f];
-            if (getSegmentsForFrame(frameNumber, segmentsInFrame).good())
+            cond = getSegmentsForFrame(frameNumber, segmentsInFrame);
+            if (cond.good())
             {
                 for (std::set<Uint32>::iterator it = segmentsInFrame.begin();
                      it != segmentsInFrame.end();
@@ -364,9 +365,13 @@ OFCondition OverlapUtil::getSegmentsByPosition(SegmentsByPosition& result)
             else
             {
                 DCMSEG_ERROR("getSegmentsByPosition(): Cannot get segments for frame " << frameNumber);
-                return EC_IllegalCall;
+                break;
             }
         }
+        if (cond.bad())
+        {
+            break;
+        }
     }
     // print segments per logical frame  if debug log level is enabled
     if (cond.good() && DCM_dcmsegLogger.isEnabledFor(OFLogger::DEBUG_LOG_LEVEL))
@@ -375,7 +380,7 @@ OFCondition OverlapUtil::getSegmentsByPosition(SegmentsByPosition& result)
         printSegmentsByPosition(ss);
         DCMSEG_DEBUG(ss.str());
     }
-    DCMSEG_DEBUG("groupFramesByPosition(): Grouping segments by position took " << tm.getDiff() << " s");
+    DCMSEG_DEBUG("getSegmentsByPosition(): Getting segments by position took " << tm.getDiff() << " s");
     return cond;
 }