Factorize wheel handling and patching

printrun/gcview.py

@@ -154,14 +154,10 @@ class GcodeViewPanel(wxGLPanel):
         self.parent.setlayercb(new_layer)
         wx.CallAfter(self.Refresh)
 
-    def wheel(self, event):
-        """react to mouse wheel actions:
-            without shift: set max layer
-            with shift: zoom viewport
-        """
+    def handle_wheel(self, event):
         delta = event.GetWheelRotation()
         factor = 1.05
-        if event.ShiftDown():
+        if hasattr(self.parent, "model") and event.ShiftDown():
             if not self.parent.model:
                 return
             count = 1 if not event.ControlDown() else 10
@@ -176,6 +172,14 @@ class GcodeViewPanel(wxGLPanel):
         else:
             self.zoom(1 / factor, (x, y))
 
+    def wheel(self, event):
+        """react to mouse wheel actions:
+            without shift: set max layer
+            with shift: zoom viewport
+        """
+        self.handle_wheel(event)
+        wx.CallAfter(self.Refresh)
+
     def fit(self):
         if not self.parent.model or not self.parent.model.loaded:
             return

printrun/objectplater.py

@@ -103,6 +103,19 @@ class Plater(wx.Frame):
                     else:
                         orig_handler(event)
             patch_method(viewer, "handle_rotation", handle_rotation)
+        # Patch handle_wheel on the fly
+        if hasattr(viewer, "handle_wheel"):
+            def handle_wheel(self, event, orig_handler):
+                if not event.ShiftDown():
+                    delta = event.GetWheelRotation()
+                    angle = 10
+                    if delta > 0:
+                        self.parent.rotate_shape(angle / 2)
+                    else:
+                        self.parent.rotate_shape(-angle / 2)
+                else:
+                    orig_handler(event)
+            patch_method(viewer, "handle_wheel", handle_wheel)
         self.s = viewer
         self.mainsizer.Add(self.s, 1, wx.EXPAND)
 

printrun/stlview.py

@@ -170,26 +170,22 @@ class StlViewPanel(wxGLPanel):
         event.Skip()
         wx.CallAfter(self.Refresh)
 
+    def handle_wheel(self, event):
+        delta = event.GetWheelRotation()
+        factor = 1.05
+        x, y = event.GetPositionTuple()
+        x, y, _ = self.mouse_to_3d(x, y)
+        if delta > 0:
+            self.zoom(factor, (x, y))
+        else:
+            self.zoom(1 / factor, (x, y))
+
     def wheel(self, event):
         """react to mouse wheel actions:
         rotate object
             with shift zoom viewport
         """
-        delta = event.GetWheelRotation()
-        if not event.ShiftDown():
-            angle = 10
-            if delta > 0:
-                self.parent.rotate_shape(angle / 2)
-            else:
-                self.parent.rotate_shape(-angle / 2)
-        else:
-            factor = 1.05
-            x, y = event.GetPositionTuple()
-            x, y, _ = self.mouse_to_3d(x, y)
-            if delta > 0:
-                self.zoom(factor, (x, y))
-            else:
-                self.zoom(1 / factor, (x, y))
+        self.handle_wheel(event)
         wx.CallAfter(self.Refresh)
 
     def keypress(self, event):