Fix non 3d annotations appearing in 3D viewport (#9)

* Fix non 3d annotations appearing in 3D viewport * Add toolContext in annotation metadata * Refactor code
RadicalImaging · Dec 21, 2023 · 925e880 · 925e880
1 parent c1dfbe4
commit 925e880
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/packages/tools/src/types/AnnotationTypes.ts b/packages/tools/src/types/AnnotationTypes.ts
@@ -47,6 +47,10 @@ type Annotation = {
      * VolumeId of the volume that the tool was configured to work on.
      */
     volumeId?: string;
+    /**
+     * The context of the annotation: 2D or 3D. If empty it will be considered as 2d context
+     */
+    toolContext?: string;
   };
   /**
    * Data for annotation, Derivatives need to define their own data types.

diff --git a/packages/tools/src/utilities/planar/filterAnnotationsForDisplay.ts b/packages/tools/src/utilities/planar/filterAnnotationsForDisplay.ts
@@ -71,7 +71,9 @@ export default function filterAnnotationsForDisplay(
       spacingInNormalDirection
     );
   } else if (viewport instanceof VolumeViewport3D) {
-    return annotations;
+    return annotations.filter((annotation) => {
+      return annotation.metadata?.toolContext?.toUpperCase() === '3D';
+    });
   } else {
     throw new Error(`Viewport Type ${viewport.type} not supported`);
   }