Remove interpolation by default for imshow

tracking/monitors/plotting.py

@@ -101,6 +101,8 @@ def plot_beamdata(filenames, dataset="mean", dimension="tau", stat_var="mean",
                 idx = -1
             else:
                 idx = np.where(time == int(turn))
+                if (idx.size == 0):
+                    raise ValueError("Turn is not valid.")
             
             if dataset == "current":
                 y = data[:,idx]*1e3
@@ -841,7 +843,7 @@ def streak_bunchspectrum(filename, bunch_number, dataset="incoherent",
     cmap = mpl.cm.inferno # sequential
     c = ax.imshow(data.T, cmap=cmap, origin='lower' , aspect='auto',
                   extent=[x_var.min(), x_var.max(), tmin, tmax],
-                  norm=option)
+                  norm=option, interpolation="none")
     cbar = fig.colorbar(c, ax=ax)
     cbar.ax.set_ylabel("FFT amplitude [a.u.]", rotation=270)
     ax.set_xlabel(xlabel)
@@ -1021,7 +1023,7 @@ def streak_beamspectrum(filename, dim="tau", f0=None, log_scale=True, fmin=None,
     cmap = mpl.cm.inferno # sequential
     c = ax.imshow(data.T, cmap=cmap, origin='lower' , aspect='auto',
                   extent=[x_var.min(), x_var.max(), tmin, tmax],
-                  norm=option)
+                  norm=option, interpolation="none")
     cbar = fig.colorbar(c, ax=ax)
     cbar.ax.set_ylabel("FFT amplitude [a.u.]", rotation=270)
     ax.set_xlabel(xlabel)
@@ -1099,6 +1101,8 @@ def plot_cavitydata(filename, cavity_name, phasor="cavity",
     if plot_type == "turn":
         
         index = np.array(time) == turn
+        if (index.size == 0):
+            raise ValueError("Turn is not valid.")
         ph = {"cavity":0, "beam":1}
         data = [np.array(cavity_data["cavity_phasor_record"])[:,index], 
                 np.array(cavity_data["beam_phasor_record"])[:,index]]