ENH: Simplify initialization of vtkQuaternion in EndoscopyLogic

jcfr · Nov 28, 2023 · cc0168c · cc0168c
1 parent b021205
commit cc0168c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/Modules/Scripted/Endoscopy/Endoscopy.py b/Modules/Scripted/Endoscopy/Endoscopy.py
@@ -902,32 +902,31 @@ def indexOfControlPointForDistanceAlongCurve(curve, distanceAlongInputCurve):
     @staticmethod
     def matrix3x3ToOrientation(matrix3x3):
         orientation = np.zeros((4,))
-        vtkQ = vtk.vtkQuaternion[np.float64]()
+        vtkQ = vtk.vtkQuaterniond()
         vtkQ.FromMatrix3x3(matrix3x3)
         orientation[0] = vtkQ.GetRotationAngleAndAxis(orientation[1:4])
         return orientation
 
     @staticmethod
     def orientationToMatrix3x3(orientation):
         matrix3x3 = np.zeros((3, 3))
-        vtkQ = vtk.vtkQuaternion[np.float64]()
+        vtkQ = vtk.vtkQuaterniond()
         vtkQ.SetRotationAngleAndAxis(*orientation)
         vtkQ.ToMatrix3x3(matrix3x3)
         return matrix3x3
 
     @staticmethod
     def orientationToQuaternion(orientation):
         quaternion = np.zeros((4,))
-        vtkQ = vtk.vtkQuaternion[np.float64]()
+        vtkQ = vtk.vtkQuaterniond()
         vtkQ.SetRotationAngleAndAxis(*orientation)
         vtkQ.Get(quaternion)
         return quaternion
 
     @staticmethod
     def quaternionToOrientation(quaternion):
         orientation = np.zeros((4,))
-        vtkQ = vtk.vtkQuaternion[np.float64]()
-        vtkQ.Set(*quaternion)
+        vtkQ = vtk.vtkQuaterniond(*quaternion)
         orientation[0] = vtkQ.GetRotationAngleAndAxis(orientation[1:4])
         return orientation