All files / tools/src/stateManagement/segmentation/labelmapModel labelmapSegmentBindings.ts

30.95% Statements 13/42
26.66% Branches 8/30
63.63% Functions 7/11
30.95% Lines 13/42

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 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140                                  4715x                                                                 409x 409x       409x 409x 409x                 2497393x 2497393x       2497393x                               2943x 2943x       2943x             500x                                                                                
import type { Segmentation } from '../../../types/SegmentationStateTypes';
import type {
  LabelmapLayer,
  SegmentLabelmapBindingState,
} from '../../../types/LabelmapTypes';
import { ensureLabelmapState } from './normalizeLabelmapSegmentationData';
import {
  getLabelmap,
  getLabelmaps,
  removeLabelmap,
} from './labelmapLayerStore';
import { syncLegacyLabelmapData } from './labelmapLegacyAdapter';
 
function getSegmentBinding(
  segmentation: Segmentation,
  segmentIndex: number
): SegmentLabelmapBindingState | undefined {
  return ensureLabelmapState(segmentation)?.segmentBindings?.[segmentIndex];
}
 
function setSegmentBinding(
  segmentation: Segmentation,
  segmentIndex: number,
  binding: SegmentLabelmapBindingState
): void {
  const labelmapState = ensureLabelmapState(segmentation);
  if (!labelmapState) {
    return;
  }
 
  const previousBinding = labelmapState.segmentBindings[segmentIndex];
  if (previousBinding) {
    const previousLayer = labelmapState.labelmaps[previousBinding.labelmapId];
    if (previousLayer?.labelToSegmentIndex) {
      delete previousLayer.labelToSegmentIndex[previousBinding.labelValue];
    }
  }
 
  labelmapState.segmentBindings[segmentIndex] = binding;
  const layer = labelmapState.labelmaps[binding.labelmapId];
  if (layer) {
    layer.labelToSegmentIndex ||= {};
    layer.labelToSegmentIndex[binding.labelValue] = segmentIndex;
  }
}
 
function getSegmentsOnLabelmap(
  segmentation: Segmentation,
  labelmapId: string
): number[] {
  const labelmapState = ensureLabelmapState(segmentation);
  Iif (!labelmapState) {
    return [];
  }
 
  return Object.entries(labelmapState.segmentBindings)
    .filter(([, binding]) => binding.labelmapId === labelmapId)
    .map(([segmentIndex]) => Number(segmentIndex))
    .sort((a, b) => a - b);
}
 
function getSegmentIndexForLabelValue(
  segmentation: Segmentation,
  labelmapId: string,
  labelValue: number
): number | undefined {
  const layer = getLabelmap(segmentation, labelmapId);
  Iif (!layer || labelValue == null) {
    return;
  }
 
  return layer.labelToSegmentIndex?.[labelValue] ?? labelValue;
}
 
function getLabelValueForSegment(
  segmentation: Segmentation,
  segmentIndex: number
): number {
  return (
    getSegmentBinding(segmentation, segmentIndex)?.labelValue ?? segmentIndex
  );
}
 
function getLabelmapForSegment(
  segmentation: Segmentation,
  segmentIndex: number
): LabelmapLayer | undefined {
  const binding = getSegmentBinding(segmentation, segmentIndex);
  Iif (!binding) {
    return;
  }
 
  return getLabelmap(segmentation, binding.labelmapId);
}
 
function resolveLabelmapForSegment(
  segmentation: Segmentation,
  segmentIndex: number
): LabelmapLayer | undefined {
  return (
    getLabelmapForSegment(segmentation, segmentIndex) ??
    getLabelmaps(segmentation)[0]
  );
}
 
function removeSegmentBinding(
  segmentation: Segmentation,
  segmentIndex: number
): void {
  const labelmapState = ensureLabelmapState(segmentation);
  const binding = labelmapState?.segmentBindings?.[segmentIndex];
  if (!labelmapState || !binding) {
    return;
  }
 
  const layer = labelmapState.labelmaps[binding.labelmapId];
  if (layer?.labelToSegmentIndex) {
    delete layer.labelToSegmentIndex[binding.labelValue];
  }
 
  delete labelmapState.segmentBindings[segmentIndex];
 
  if (getSegmentsOnLabelmap(segmentation, binding.labelmapId).length === 0) {
    removeLabelmap(segmentation, binding.labelmapId);
  }
 
  syncLegacyLabelmapData(segmentation);
}
 
export {
  getLabelmapForSegment,
  getLabelValueForSegment,
  getSegmentBinding,
  getSegmentIndexForLabelValue,
  getSegmentsOnLabelmap,
  removeSegmentBinding,
  resolveLabelmapForSegment,
  setSegmentBinding,
};