feat(CPR): Add methods to switch CPR modes

Add a useStraightenedMode function and a useStretchedMode function

Author: bruyeret

Sources/Rendering/Core/ImageCPRMapper/example/index.js

@@ -355,26 +355,12 @@ angleEl.addEventListener('input', () =>
 );
 
 function useStraightenedMode() {
-  mapper.setCenterPoint(null);
-  mapper.setUseUniformOrientation(false);
-  mapper.getOrientedCenterline().setDistanceFunction(vec3.dist);
+  mapper.useStraightenedMode();
   updateDistanceAndDirection();
 }
 
 function useStretchedMode() {
-  mapper.setCenterPoint(centerline.getPoints().getPoint(0));
-  mapper.setUseUniformOrientation(true);
-  mapper.getOrientedCenterline().setDistanceFunction((a, b) => {
-    const worldTangent = vec3.transformQuat(
-      [],
-      mapper.getTangentDirection(),
-      mapper.getUniformOrientation()
-    );
-    const vec = vec3.subtract([], a, b);
-    const d2 = vec3.squaredLength(vec);
-    const x = vec3.dot(worldTangent, vec);
-    return Math.sqrt(d2 - x * x);
-  });
+  mapper.useStretchedMode();
   updateDistanceAndDirection();
 }
 

Sources/Rendering/Core/ImageCPRMapper/index.d.ts

@@ -206,6 +206,17 @@ export interface vtkImageCPRMapper extends vtkAbstractMapper3D {
 	 */
 	preRenderCheck(): boolean;
 
+	/**
+	 * Configure the mapper and the centerline to be in straightened CPR mode
+	 */
+	useStraightenedMode(): void;
+
+	/**
+	 * Configure the mapper and the centerline to be in strectched CPR mode
+	 * @param centerPoint The center point, optional, default to the first point of the centerline or [0, 0, 0]
+	 */
+	useStretchedMode(centerPoint?: vec3): void;
+
 	/**
 	 * Set the polydata used as a centerline
 	 * You can also use `publicAPI.setInputData(centerlineData, 1);`

Sources/Rendering/Core/ImageCPRMapper/index.js

@@ -271,6 +271,38 @@ function vtkImageCPRMapper(publicAPI, model) {
     return true;
   };
 
+  publicAPI.useStraightenedMode = () => {
+    publicAPI.setCenterPoint(null);
+    publicAPI.setUseUniformOrientation(false);
+    publicAPI.getOrientedCenterline().setDistanceFunction(vec3.dist);
+  };
+
+  publicAPI.useStretchedMode = (centerPoint) => {
+    const centerline = publicAPI.getOrientedCenterline();
+    // Set center point
+    if (centerPoint === undefined) {
+      // Get the first point of the centerline if there is one
+      const centerlinePoints = centerline.getPoints();
+      const newCenterPoint =
+        centerlinePoints.getNumberOfTuples() > 0
+          ? centerlinePoints.getPoint(0)
+          : [0, 0, 0];
+      publicAPI.setCenterPoint(newCenterPoint);
+    } else {
+      publicAPI.setCenterPoint(centerPoint);
+    }
+    // Enable uniform orientation
+    publicAPI.setUseUniformOrientation(true);
+    // Change distance function
+    centerline.setDistanceFunction((a, b) => {
+      const direction = publicAPI.getUniformDirection();
+      const vec = vec3.subtract([], a, b);
+      const d2 = vec3.squaredLength(vec);
+      const x = vec3.dot(direction, vec);
+      return Math.sqrt(d2 - x * x);
+    });
+  };
+
   publicAPI.setCenterlineData = (centerlineData) =>
     publicAPI.setInputData(centerlineData, 1);