All files / packages/tools/src/stateManagement/segmentation/polySeg/Surface convertLabelmapToSurface.ts

13.33% Statements 2/15
0% Branches 0/3
0% Functions 0/3
13.33% Lines 2/15

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82                                1x   1x                                                                                                                              
import {
  Types,
  cache,
  eventTarget,
  triggerEvent,
  Enums,
} from '@cornerstonejs/core';
import { getWebWorkerManager } from '@cornerstonejs/core';
import {
  LabelmapSegmentationData,
  LabelmapSegmentationDataStack,
  LabelmapSegmentationDataVolume,
} from '../../../../types/LabelmapTypes';
import { computeVolumeSegmentationFromStack } from '../../convertStackToVolumeSegmentation';
import { WorkerTypes } from '../../../../enums';
 
const workerManager = getWebWorkerManager();
 
const triggerWorkerProgress = (eventTarget, progress) => {
  triggerEvent(eventTarget, Enums.Events.WEB_WORKER_PROGRESS, {
    progress,
    type: WorkerTypes.POLYSEG_LABELMAP_TO_SURFACE,
  });
};
 
/**
 * Converts a labelmap representation to a surface representation.
 *
 * @param labelmapRepresentationData - The labelmap segmentation data.
 * @param segmentIndex - The index of the segment to convert.
 * @param isVolume - Optional flag indicating whether the labelmap is a volume or a stack. Default is true.
 * @returns A promise that resolves to the surface data.
 */
export async function convertLabelmapToSurface(
  labelmapRepresentationData: LabelmapSegmentationData,
  segmentIndex: number,
  isVolume = true
): Promise<Types.SurfaceData> {
  let volumeId;
  if (isVolume) {
    volumeId = (labelmapRepresentationData as LabelmapSegmentationDataVolume)
      .volumeId;
  } else {
    const { imageIdReferenceMap } =
      labelmapRepresentationData as LabelmapSegmentationDataStack;
    ({ volumeId } = await computeVolumeSegmentationFromStack({
      imageIdReferenceMap,
    }));
  }
 
  const volume = cache.getVolume(volumeId);
 
  const scalarData = volume.getScalarData();
  const { dimensions, spacing, origin, direction } = volume;
 
  triggerWorkerProgress(eventTarget, 0);
 
  const results = await workerManager.executeTask(
    'polySeg',
    'convertLabelmapToSurface',
    {
      scalarData,
      dimensions,
      spacing,
      origin,
      direction,
      segmentIndex,
    },
    {
      callbacks: [
        (progress) => {
          triggerWorkerProgress(eventTarget, progress);
        },
      ],
    }
  );
 
  triggerWorkerProgress(eventTarget, 1);
 
  return results;
}