Added get_coordinate_cursor to CrossInteractorStyle

Thiago Franco de Moraes
1 parent 9cbf791e
Showing 1 changed file with 12 additions and 0 deletions Show diff stats
invesalius/data/styles.py
@@ -181,6 +181,18 @@ class CrossInteractorStyle(DefaultInteractorStyle):
             Publisher.sendMessage(('Set scroll position', 'SAGITAL'),
                                        coord[0])
  
+    def get_coordinate_cursor(self):
+        # Find position
+        x, y, z = self.picker.GetPickPosition()
+        bounds = self.viewer.slice_data.actor.GetBounds()
+        if bounds[0] == bounds[1]:
+            x = bounds[0]
+        elif bounds[2] == bounds[3]:
+            y = bounds[2]
+        elif bounds[4] == bounds[5]:
+            z = bounds[4]
+        return x, y, z
+
  
 class WWWLInteractorStyle(DefaultInteractorStyle):
     """
...	...	@@ -181,6 +181,18 @@ class CrossInteractorStyle(DefaultInteractorStyle):
181	181	Publisher.sendMessage(('Set scroll position', 'SAGITAL'),
182	182	coord[0])
183	183
	184	+ def get_coordinate_cursor(self):
	185	+ # Find position
	186	+ x, y, z = self.picker.GetPickPosition()
	187	+ bounds = self.viewer.slice_data.actor.GetBounds()
	188	+ if bounds[0] == bounds[1]:
	189	+ x = bounds[0]
	190	+ elif bounds[2] == bounds[3]:
	191	+ y = bounds[2]
	192	+ elif bounds[4] == bounds[5]:
	193	+ z = bounds[4]
	194	+ return x, y, z
	195	+
184	196
185	197	class WWWLInteractorStyle(DefaultInteractorStyle):
186	198	"""
...	...