All files / polymorphic-segmentation/src/Contour/utils extractContourData.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import type { PolyDataClipCacheType } from '../../utilities/clipAndCacheSurfacesForViewport';
import type { RawContourData } from '../contourComputationStrategies';
 
/**
 * Extracts contour data from the given polyDataCache.
 * @param polyDataCache - The polyData cache containing intersection information.
 * @param segmentIndexMap - Optional map for mapping surface IDs to segment indices.
 * @returns A map of segment indices to an array of contour results.
 */
export function extractContourData(polyDataCache: PolyDataClipCacheType) {
  const rawResults = new Map() as RawContourData;
  if (!polyDataCache) {
    return rawResults;
  }
 
  for (const [segmentIndex, intersectionInfo] of polyDataCache) {
    const segmentIndexNumber = Number(segmentIndex);
 
    for (const [_, result] of intersectionInfo) {
      if (!result) {
        continue;
      }
 
      if (!rawResults.has(segmentIndexNumber)) {
        rawResults.set(segmentIndexNumber, []);
      }
 
      rawResults.get(segmentIndexNumber).push(result);
    }
  }
  return rawResults;
}