Created a new interactor style to handle the Zoom camera

Thiago Franco de Moraes
1 parent 2ae6b471
Showing 2 changed files with 33 additions and 0 deletions Show diff stats
invesalius/data/styles.py
invesalius/data/viewer_slice.py
@@ -340,6 +340,32 @@ class SpinInteractorStyle(RightZoomInteractorStyle):
         iren.Render()
  
  
+class ZoomInteractorStyle(RightZoomInteractorStyle):
+    """
+    Interactor style responsible for zoom with movement of the mouse and the
+    left mouse button clicked.
+    """
+    def __init__(self, viewer):
+        RightZoomInteractorStyle.__init__(self)
+        
+        self.viewer = viewer
+
+        self.AddObserver("MouseMoveEvent", self.OnZoomMoveLeft)
+        self.viewer.interactor.Bind(wx.EVT_LEFT_DCLICK, self.OnUnZoom)
+
+    def OnZoomMoveLeft(self, obj, evt):
+        if self.left_pressed:
+            obj.Dolly()
+            obj.OnRightButtonDown()
+
+    def OnUnZoom(self, evt):
+        mouse_x, mouse_y = self.viewer.interactor.GetLastEventPosition()
+        ren = self.viewer.interactor.FindPokedRenderer(mouse_x, mouse_y)
+        #slice_data = self.get_slice_data(ren)
+        ren.ResetCamera()
+        ren.ResetCameraClippingRange()
+        #self.Reposition(slice_data)
+        self.viewer.interactor.Render()
  
  
 class ViewerStyle:
@@ -231,6 +231,13 @@ class Viewer(wx.Panel):
             self.interactor.SetInteractorStyle(style)
             self.interactor.Render()
  
+        elif state == const.STATE_ZOOM:
+            style = styles.ZoomInteractorStyle(self)
+
+            self.style = style
+            self.interactor.SetInteractorStyle(style)
+            self.interactor.Render()
+
         else:
             self.state = state
             action = {
...	...	@@ -340,6 +340,32 @@ class SpinInteractorStyle(RightZoomInteractorStyle):
340	340	iren.Render()
341	341
342	342
	343	+class ZoomInteractorStyle(RightZoomInteractorStyle):
	344	+ """
	345	+ Interactor style responsible for zoom with movement of the mouse and the
	346	+ left mouse button clicked.
	347	+ """
	348	+ def __init__(self, viewer):
	349	+ RightZoomInteractorStyle.__init__(self)
	350	+
	351	+ self.viewer = viewer
	352	+
	353	+ self.AddObserver("MouseMoveEvent", self.OnZoomMoveLeft)
	354	+ self.viewer.interactor.Bind(wx.EVT_LEFT_DCLICK, self.OnUnZoom)
	355	+
	356	+ def OnZoomMoveLeft(self, obj, evt):
	357	+ if self.left_pressed:
	358	+ obj.Dolly()
	359	+ obj.OnRightButtonDown()
	360	+
	361	+ def OnUnZoom(self, evt):
	362	+ mouse_x, mouse_y = self.viewer.interactor.GetLastEventPosition()
	363	+ ren = self.viewer.interactor.FindPokedRenderer(mouse_x, mouse_y)
	364	+ #slice_data = self.get_slice_data(ren)
	365	+ ren.ResetCamera()
	366	+ ren.ResetCameraClippingRange()
	367	+ #self.Reposition(slice_data)
	368	+ self.viewer.interactor.Render()
343	369
344	370
345	371	class ViewerStyle:
...	...
...	...	@@ -231,6 +231,13 @@ class Viewer(wx.Panel):
231	231	self.interactor.SetInteractorStyle(style)
232	232	self.interactor.Render()
233	233
	234	+ elif state == const.STATE_ZOOM:
	235	+ style = styles.ZoomInteractorStyle(self)
	236	+
	237	+ self.style = style
	238	+ self.interactor.SetInteractorStyle(style)
	239	+ self.interactor.Render()
	240	+
234	241	else:
235	242	self.state = state
236	243	action = {
...	...