Merge branch 'rory/mplib-ax-warnings-2' of https://github.com/roryyorke/python-control into roryyorke-rory/mplib-ax-warnings-2

murrayrm · murrayrm · commit 19e57c1e29e8 · 2018-01-20T09:01:10.000-08:00
diff --git a/control/freqplot.py b/control/freqplot.py
@@ -135,6 +135,20 @@ def bode_plot(syslist, omega=None, dB=None, Hz=None, deg=None,
             else:
                 omega = sp.logspace(np.log10(omega_limits[0]), np.log10(omega_limits[1]), endpoint=True)
 
+    if Plot:
+        fig = plt.gcf()
+        ax_mag = None
+        ax_phase = None
+        for ax in fig.axes:
+            if ax.get_label() == 'pycontrol-bode-mag':
+                ax_mag = ax
+            elif ax.get_label() == 'pycontrol-bode-phs':
+                ax_phase = ax
+        if ax_mag is None or ax_phase is None:
+            plt.clf()
+            ax_mag = plt.subplot(211, label = 'pycontrol-bode-mag')
+            ax_phase = plt.subplot(212, label = 'pycontrol-bode-phs', sharex=ax_mag)
+
     mags, phases, omegas, nyquistfrqs = [], [], [], []
     for sys in syslist:
         if (sys.inputs > 1 or sys.outputs > 1):
@@ -177,8 +191,6 @@ def bode_plot(syslist, omega=None, dB=None, Hz=None, deg=None,
                 figlabel = str(random.randint(1, 1e6))
 
                 # Magnitude plot
-                ax_mag = plt.subplot(211, label=figlabel);
-
                 if dB:
                     pltline = ax_mag.semilogx(omega_plot, 20 * np.log10(mag),
                                               *args, **kwargs)
@@ -194,7 +206,6 @@ def bode_plot(syslist, omega=None, dB=None, Hz=None, deg=None,
                 ax_mag.set_ylabel("Magnitude (dB)" if dB else "Magnitude")
 
                 # Phase plot
-                ax_phase = plt.subplot(212, sharex=ax_mag);
                 if deg:
                     phase_plot = phase * 180. / math.pi
                 else:
@@ -362,32 +373,45 @@ def gangof4_plot(P, C, omega=None):
         S = feedback(1, L);
         T = L * S;
 
-        # Create a unique label to fix bug in matplotlib<=2.1
-        # See https://github.com/matplotlib/matplotlib/issues/9024
-        import random
-        figlabel = str(random.randint(1, 1e6))
+        plot_axes = {'t' : None, 's' : None, 'ps' : None, 'cs' : None}
+        for ax in plt.gcf().axes:
+            label = ax.get_label()
+            if label.startswith('pycontrol-gof-'):
+                key = label[len('pycontrol-gof-'):]
+                if key not in plot_axes:
+                    raise RuntimeError("unknown gof axis type '{}'".format(label))
+                plot_axes[key] = ax
+
+        # if any are missing, start from scratch
+        if any((ax is None for ax in plot_axes.values())):
+            plt.clf()
+            plot_axes = {'t' : plt.subplot(221,label='pycontrol-gof-t'),
+                         'ps' : plt.subplot(222,label='pycontrol-gof-ps'),
+                         'cs' : plt.subplot(223,label='pycontrol-gof-cs'),
+                         's' : plt.subplot(224,label='pycontrol-gof-s')}
+
+        # Plot the four sensitivity functions
 
-                # Plot the four sensitivity functions
         #! TODO: Need to add in the mag = 1 lines
         mag_tmp, phase_tmp, omega = T.freqresp(omega);
         mag = np.squeeze(mag_tmp)
         phase = np.squeeze(phase_tmp)
-        plt.subplot(221, label=figlabel); plt.loglog(omega, mag);
+        plot_axes['t'].loglog(omega, mag);
 
         mag_tmp, phase_tmp, omega = (P * S).freqresp(omega);
         mag = np.squeeze(mag_tmp)
         phase = np.squeeze(phase_tmp)
-        plt.subplot(222, label=figlabel); plt.loglog(omega, mag);
+        plot_axes['ps'].loglog(omega, mag);
 
         mag_tmp, phase_tmp, omega = (C * S).freqresp(omega);
         mag = np.squeeze(mag_tmp)
         phase = np.squeeze(phase_tmp)
-        plt.subplot(223, label=figlabel); plt.loglog(omega, mag);
+        plot_axes['cs'].loglog(omega, mag);
 
         mag_tmp, phase_tmp, omega = S.freqresp(omega);
         mag = np.squeeze(mag_tmp)
         phase = np.squeeze(phase_tmp)
-        plt.subplot(224, label=figlabel); plt.loglog(omega, mag);
+        plot_axes['s'].loglog(omega, mag);
 
 #
 # Utility functions
