feat(imageIdSpecificStateManager): add replace imageIdToolState fun…

…ctions

There is currently no function to consume the same output from getImageIdToolState, which is the
state of a specific tool for the given imageId. addElementToolState requires a single item input,
and without replace, requires both a clear step and an iteration over all items to add into the
tool's id-specific state.

src/stateManagement/imageIdSpecificStateManager.js

@@ -94,6 +94,22 @@ function newImageIdSpecificToolStateManager() {
     return imageIdToolState[toolName];
   }
 
+  function replaceElementToolState(element, toolName, data) {
+    const enabledElement = external.cornerstone.getEnabledElement(element);
+
+    if (!enabledElement.image) {
+      return;
+    }
+    replaceImageIdToolState(enabledElement.image.imageId, toolName, data);
+  }
+
+  function replaceImageIdToolState(imageId, toolName, data) {
+    const imageIdToolState = toolState[imageId];
+
+    // Replace the toolState
+    imageIdToolState[toolName] = data;
+  }
+
   // Clears all tool data from this toolStateManager.
   function clearElementToolState(element) {
     const enabledElement = external.cornerstone.getEnabledElement(element);
@@ -115,9 +131,11 @@ function newImageIdSpecificToolStateManager() {
   return {
     get: getElementToolState,
     add: addElementToolState,
+    replace: replaceElementToolState,
     clear: clearElementToolState,
     getImageIdToolState,
     addImageIdToolState,
+    replaceImageIdToolState,
     clearImageIdToolState,
     saveImageIdToolState,
     restoreImageIdToolState,

src/stateManagement/imageIdSpecificStateManager.test.js

@@ -76,3 +76,36 @@ describe('imageIdSpecificStateManager.add', () => {
     });
   });
 });
+
+describe('imageIdSpecificStateManager.replace', () => {
+  it('replaces data in the existing toolState', () => {
+    const stateManager = newImageIdSpecificToolStateManager();
+    const toolName = 'TestTool';
+    const imageId = 'abc123';
+    const testElement = {
+      image: {
+        imageId,
+      },
+    };
+
+    externalModules.cornerstone.getEnabledElement.mockImplementationOnce(
+      () => testElement
+    );
+
+    // Setup with some intial data
+    stateManager.restoreImageIdToolState(imageId, {
+      [toolName]: { data: ['initialData'] },
+    });
+    // Add more data
+    stateManager.replace(testElement, toolName, {
+      data: ['replacedData1', 'replacedData2'],
+    });
+
+    // Check the results
+    const allToolState = stateManager.saveToolState();
+
+    expect(allToolState[imageId][toolName].data).toEqual(
+      expect.arrayContaining(['replacedData1', 'replacedData2'])
+    );
+  });
+});