From 02f04ff4f9d25d69ffba196f5a15af5b5512e9bf Mon Sep 17 00:00:00 2001
From: Antony Lee <anntzer.lee@gmail.com>
Date: Sat, 3 Jun 2017 20:47:04 -0700
Subject: [PATCH] Minor cleanups of the qt4 embedding examples.

`embedding_in_qt4_sgskip.py` would not run if the backend is set to
qt5agg in the rcfile, so we explicitly set it to qt4agg.  Importing
QtCore, QtGui from qt_compat is simpler than explicitly checking for
PySide.
---
 .../embedding_in_qt4_sgskip.py                | 16 +++++------
 .../embedding_in_qt4_wtoolbar_sgskip.py       | 28 ++++++-------------
 2 files changed, 16 insertions(+), 28 deletions(-)

diff --git a/examples/user_interfaces/embedding_in_qt4_sgskip.py b/examples/user_interfaces/embedding_in_qt4_sgskip.py
index cad3582f261d..46922d7078a4 100644
--- a/examples/user_interfaces/embedding_in_qt4_sgskip.py
+++ b/examples/user_interfaces/embedding_in_qt4_sgskip.py
@@ -14,22 +14,20 @@
 """
 
 from __future__ import unicode_literals
-import sys
 import os
 import random
-from matplotlib.backends import qt_compat
-use_pyside = qt_compat.QT_API == qt_compat.QT_API_PYSIDE
-if use_pyside:
-    from PySide import QtGui, QtCore
-else:
-    from PyQt4 import QtGui, QtCore
+import sys
 
 from numpy import arange, sin, pi
-from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
+
+import matplotlib
+matplotlib.use("Qt4Agg")
+from matplotlib.backends.backend_qt4agg import (
+    FigureCanvasQTAgg as FigureCanvas)
+from matplotlib.backends.qt_compat import QtCore, QtGui
 from matplotlib.figure import Figure
 
 progname = os.path.basename(sys.argv[0])
-progversion = "0.1"
 
 
 class MyMplCanvas(FigureCanvas):
diff --git a/examples/user_interfaces/embedding_in_qt4_wtoolbar_sgskip.py b/examples/user_interfaces/embedding_in_qt4_wtoolbar_sgskip.py
index 421a019a1452..c6367da9f72e 100644
--- a/examples/user_interfaces/embedding_in_qt4_wtoolbar_sgskip.py
+++ b/examples/user_interfaces/embedding_in_qt4_wtoolbar_sgskip.py
@@ -9,46 +9,38 @@
 import sys
 
 import numpy as np
+
 import matplotlib
 matplotlib.use("Qt4Agg")
-from matplotlib.figure import Figure
 from matplotlib.backend_bases import key_press_handler
 from matplotlib.backends.backend_qt4agg import (
     FigureCanvasQTAgg as FigureCanvas,
     NavigationToolbar2QT as NavigationToolbar)
-from matplotlib.backends import qt_compat
-use_pyside = qt_compat.QT_API == qt_compat.QT_API_PYSIDE
-
-if use_pyside:
-    from PySide.QtCore import *
-    from PySide.QtGui import *
-else:
-    from PyQt4.QtCore import *
-    from PyQt4.QtGui import *
+from matplotlib.backends.qt_compat import QtCore, QtGui
+from matplotlib.figure import Figure
 
 
-class AppForm(QMainWindow):
+class AppForm(QtGui.QMainWindow):
     def __init__(self, parent=None):
-        QMainWindow.__init__(self, parent)
-        #self.x, self.y = self.get_data()
+        QtGui.QMainWindow.__init__(self, parent)
         self.data = self.get_data2()
         self.create_main_frame()
         self.on_draw()
 
     def create_main_frame(self):
-        self.main_frame = QWidget()
+        self.main_frame = QtGui.QWidget()
 
         self.fig = Figure((5.0, 4.0), dpi=100)
         self.canvas = FigureCanvas(self.fig)
         self.canvas.setParent(self.main_frame)
-        self.canvas.setFocusPolicy(Qt.StrongFocus)
+        self.canvas.setFocusPolicy(QtCore.Qt.StrongFocus)
         self.canvas.setFocus()
 
         self.mpl_toolbar = NavigationToolbar(self.canvas, self.main_frame)
 
         self.canvas.mpl_connect('key_press_event', self.on_key_press)
 
-        vbox = QVBoxLayout()
+        vbox = QtGui.QVBoxLayout()
         vbox.addWidget(self.canvas)  # the matplotlib canvas
         vbox.addWidget(self.mpl_toolbar)
         self.main_frame.setLayout(vbox)
@@ -60,9 +52,7 @@ def get_data2(self):
     def on_draw(self):
         self.fig.clear()
         self.axes = self.fig.add_subplot(111)
-        #self.axes.plot(self.x, self.y, 'ro')
         self.axes.imshow(self.data, interpolation='nearest')
-        #self.axes.plot([1,2,3])
         self.canvas.draw()
 
     def on_key_press(self, event):
@@ -73,7 +63,7 @@ def on_key_press(self, event):
 
 
 def main():
-    app = QApplication(sys.argv)
+    app = QtGui.QApplication(sys.argv)
     form = AppForm()
     form.show()
     app.exec_()