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

0% Statements 0/27
0% Branches 0/10
0% Functions 0/4
0% Lines 0/27

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 { getToolGroup } from '../../store/ToolGroupManager';
import BrushTool from '../../tools/segmentation/BrushTool';
import triggerAnnotationRenderForViewportIds from '../triggerAnnotationRenderForViewportIds';
import { getRenderingEngine } from '@cornerstonejs/core';
import getBrushToolInstances from './utilities';
 
export function setBrushSizeForToolGroup(
  toolGroupId: string,
  brushSize: number
): void {
  const toolGroup = getToolGroup(toolGroupId);
 
  if (toolGroup === undefined) {
    return;
  }
 
  const brushBasedToolInstances = getBrushToolInstances(toolGroupId);
 
  brushBasedToolInstances.forEach((tool: BrushTool) => {
    tool.configuration.brushSize = brushSize;
 
    // Invalidate the brush being rendered so it can update.
    tool.invalidateBrushCursor();
  });
 
  // Trigger an annotation render for any viewports on the toolgroup
  const viewportsInfo = toolGroup.getViewportsInfo();
 
  const viewportsInfoArray = Object.keys(viewportsInfo).map(
    (key) => viewportsInfo[key]
  );
 
  if (!viewportsInfoArray.length) {
    return;
  }
 
  const { renderingEngineId } = viewportsInfoArray[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 getBrushSizeForToolGroup(toolGroupId: string): void {
  const toolGroup = getToolGroup(toolGroupId);
 
  if (toolGroup === undefined) {
    return;
  }
 
  const toolInstances = toolGroup._toolInstances;
 
  if (!Object.keys(toolInstances).length) {
    return;
  }
 
  const brushBasedToolInstances = getBrushToolInstances(toolGroupId);
 
  // one is enough as they share the same brush size
  const brushToolInstance = brushBasedToolInstances[0];
 
  if (!brushToolInstance) {
    return;
  }
 
  // TODO -> Assumes the brush sizes are the same and set via these helpers.
  return brushToolInstance.configuration.brushSize;
}