All files / packages/tools/src/utilities/segmentation brushThresholdForToolGroup.ts

0% Statements 0/25
0% Branches 0/13
0% Functions 0/3
0% Lines 0/25

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                                                                                                                                                 
import type { Types } from '@cornerstonejs/core';
import { getToolGroup } from '../../store/ToolGroupManager';
import triggerAnnotationRenderForViewportIds from '../triggerAnnotationRenderForViewportIds';
import { getRenderingEngine } from '@cornerstonejs/core';
import { getBrushToolInstances } from './utilities';
 
export function setBrushThresholdForToolGroup(
  toolGroupId: string,
  threshold: Types.Point2,
  otherArgs: Record<string, unknown> = { isDynamic: false }
) {
  const toolGroup = getToolGroup(toolGroupId);
 
  if (toolGroup === undefined) {
    return;
  }
 
  const brushBasedToolInstances = getBrushToolInstances(toolGroupId);
  const configuration = {
    ...otherArgs,
    ...(threshold !== undefined && { threshold }),
  };
 
  brushBasedToolInstances.forEach((tool) => {
    tool.configuration.strategySpecificConfiguration.THRESHOLD = {
      ...tool.configuration.strategySpecificConfiguration.THRESHOLD,
      ...configuration,
    };
  });
 
  // Trigger an annotation render for any viewports on the toolgroup
  const viewportsInfo = toolGroup.getViewportsInfo();
 
  if (!viewportsInfo.length) {
    return;
  }
 
  const { renderingEngineId } = viewportsInfo[0];
 
  // Use helper to get array of viewportIds, or we just end up doing this mapping
  // ourselves here.
  const viewportIds = toolGroup.getViewportIds();
 
  const renderingEngine = getRenderingEngine(renderingEngineId);
 
  triggerAnnotationRenderForViewportIds(renderingEngine, viewportIds);
}
 
export function getBrushThresholdForToolGroup(toolGroupId: string) {
  const toolGroup = getToolGroup(toolGroupId);
 
  if (toolGroup === undefined) {
    return;
  }
 
  const toolInstances = toolGroup._toolInstances;
 
  if (!Object.keys(toolInstances).length) {
    return;
  }
 
  const brushBasedToolInstances = getBrushToolInstances(toolGroupId);
  const brushToolInstance = brushBasedToolInstances[0];
 
  if (!brushToolInstance) {
    return;
  }
 
  // TODO -> Assumes the
  return brushToolInstance.configuration.strategySpecificConfiguration.THRESHOLD
    .threshold;
}