check for mouse movement

Author: AdwaithBatchu

lib/matplotlib/backends/backend_qt.py

@@ -328,31 +328,10 @@ def leaveEvent(self, event):
     def mousePressEvent(self, event):
         button = self.buttond.get(event.button())
         if button is not None and self.figure is not None:
-            mpl_event = MouseEvent("button_press_event", self,
-                                        *self.mouseEventCoords(event), button,
-                                        modifiers=self._mpl_modifiers(),
-                                        guiEvent=event)
-            if button == MouseButton.RIGHT:
-                for ax in self.figure.get_axes():
-                    if ax.contains(mpl_event)[0] and ax.name == '3d':
-                        self.showContextMenu(event, ax)
-                        return
-
-            mpl_event._process()
-
-    def showContextMenu(self, event, ax):
-        menu = QtWidgets.QMenu(self)
-        menu.setTitle("3D View Options")
-
-        def set_view(elev, azim):
-            ax.view_init(elev=elev, azim=azim, roll=0)
-            self.draw_idle()
-
-        menu.addAction("Top View (XY)").triggered.connect(lambda: set_view(90, -90))
-        menu.addAction("Side View (YZ)").triggered.connect(lambda: set_view(0, 0))
-        menu.addAction("Front View (XZ)").triggered.connect(lambda: set_view(0, -90))
-
-        menu.exec(QtGui.QCursor.pos())
+            MouseEvent("button_press_event", self,
+                       *self.mouseEventCoords(event), button,
+                       modifiers=self._mpl_modifiers(),
+                       guiEvent=event)._process()
 
     def mouseDoubleClickEvent(self, event):
         button = self.buttond.get(event.button())
@@ -648,6 +627,14 @@ def full_screen_toggle(self):
         else:
             self.window.showFullScreen()
 
+    def context_menu(self, event, labels=None, actions=None):
+        if labels is None or actions is None:
+            return
+        menu = QtWidgets.QMenu(self.window)
+        for label, action in zip(labels, actions):
+            menu.addAction(label).triggered.connect(action)
+        menu.exec(QtGui.QCursor.pos())
+
     def _widgetclosed(self):
         CloseEvent("close_event", self.canvas)._process()
         if self.window._destroying:

lib/mpl_toolkits/mplot3d/axes3d.py

@@ -16,6 +16,7 @@
 import textwrap
 import warnings
 
+import functools
 import numpy as np
 
 import matplotlib as mpl
@@ -187,6 +188,8 @@ def __init__(
         pseudo_bbox = self.transLimits.inverted().transform([(0, 0), (1, 1)])
         self._pseudo_w, self._pseudo_h = pseudo_bbox[1] - pseudo_bbox[0]
 
+        self._right_click_moved = False
+
         # mplot3d currently manages its own spines and needs these turned off
         # for bounding box calculations
         self.spines[:].set_visible(False)
@@ -1357,6 +1360,7 @@ def clear(self):
 
     def _button_press(self, event):
         if event.inaxes == self:
+            self._right_click_moved = False
             self.button_pressed = event.button
             self._sx, self._sy = event.xdata, event.ydata
             toolbar = self.get_figure(root=True).canvas.toolbar
@@ -1367,6 +1371,19 @@ def _button_press(self, event):
 
     def _button_release(self, event):
         self.button_pressed = None
+
+        if event.button in self._zoom_btn and event.inaxes == self \
+            and not self._right_click_moved:
+            canvas = self.get_figure(root=True).canvas
+            canvas.manager.context_menu(
+                event,
+                labels=["XY", "YZ", "XZ"],
+                actions=[functools.partial(self.view_init, elev=90, azim=-90),
+                         functools.partial(self.view_init, elev=0, azim=0),
+                         functools.partial(self.view_init, elev=0, azim=-90)],
+            )
+            canvas.draw_idle()
+
         toolbar = self.get_figure(root=True).canvas.toolbar
         # backend_bases.release_zoom and backend_bases.release_pan call
         # push_current, so check the navigation mode so we don't call it twice
@@ -1621,6 +1638,7 @@ def _on_move(self, event):
         # Zoom
         elif self.button_pressed in self._zoom_btn:
             # zoom view (dragging down zooms in)
+            self._right_click_moved = True
             scale = h/(h - dy)
             self._scale_axis_limits(scale, scale, scale)