From 8996df80e45c97ad5847ebbd19291cd772a2163d Mon Sep 17 00:00:00 2001
From: Joost van Griethuysen <JoostJM@users.noreply.github.com>
Date: Tue, 4 Feb 2025 15:05:55 +0100
Subject: [PATCH] fix: changing colormap for second volume in fusion viewport
 (#4746)

---
 extensions/cornerstone/src/commandsModule.ts | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/extensions/cornerstone/src/commandsModule.ts b/extensions/cornerstone/src/commandsModule.ts
index de3bf342f..5d2e34bc5 100644
--- a/extensions/cornerstone/src/commandsModule.ts
+++ b/extensions/cornerstone/src/commandsModule.ts
@@ -756,7 +756,12 @@ function commandsModule({
           displaySetInstanceUID = viewports.get(viewportId)?.displaySetInstanceUIDs[0];
         }
 
-        const volumeId = viewport.getVolumeId();
+        // ToDo: Find a better way of obtaining the volumeId that corresponds to the displaySetInstanceUID
+        const volumeId =
+          viewport
+            .getAllVolumeIds()
+            .find((_volumeId: string) => _volumeId.includes(displaySetInstanceUID)) ??
+          viewport.getVolumeId();
         viewport.setProperties({ colormap }, volumeId);
       }