All files / packages/tools/src/utilities/segmentation brushSizeForToolGroup.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
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
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { getToolGroup } from '../../store/ToolGroupManager';
import BrushTool from '../../tools/segmentation/BrushTool';
import triggerAnnotationRenderForViewportIds from '../triggerAnnotationRenderForViewportIds';
import { getRenderingEngine } from '@cornerstonejs/core';
import { getBrushToolInstances } from './utilities';
 
/**
 * Sets the brush size for all brush-based tools in a given tool group.
 * @param toolGroupId - The ID of the tool group to set the brush size for.
 * @param brushSize - The new brush size to set.
 * @param toolName - The name of the specific tool to set the brush size for (optional)
 * If not provided, all brush-based tools in the tool group will be affected.
 */
export function setBrushSizeForToolGroup(
  toolGroupId: string,
  brushSize: number,
  toolName?: string
): void {
  const toolGroup = getToolGroup(toolGroupId);
 
  if (toolGroup === undefined) {
    return;
  }
 
  const brushBasedToolInstances = getBrushToolInstances(toolGroupId, toolName);
 
  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);
}
 
/**
 * Gets the brush size for the first brush-based tool instance in a given tool group.
 * @param toolGroupId - The ID of the tool group to get the brush size for.
 * @param toolName - The name of the specific tool to get the brush size for (Optional)
 * If not provided, the first brush-based tool instance in the tool group will be used.
 * @returns The brush size of the selected tool instance, or undefined if no brush-based tool instance is found.
 */
export function getBrushSizeForToolGroup(
  toolGroupId: string,
  toolName?: string
): void {
  const toolGroup = getToolGroup(toolGroupId);
 
  if (toolGroup === undefined) {
    return;
  }
 
  const toolInstances = toolGroup._toolInstances;
 
  if (!Object.keys(toolInstances).length) {
    return;
  }
 
  const brushBasedToolInstances = getBrushToolInstances(toolGroupId, toolName);
 
  // 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;
}