From 3710242bc3f3a8fc3e52a829af32d0bbbcda125a Mon Sep 17 00:00:00 2001
From: Gamelin Alexis <alexis.gamelin@synchrotron-soleil.fr>
Date: Thu, 6 Jan 2022 18:56:35 +0100
Subject: [PATCH] Remove interpolation by default for imshow
---
tracking/monitors/plotting.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/tracking/monitors/plotting.py b/tracking/monitors/plotting.py
index 3d57baf..e481411 100644
--- a/tracking/monitors/plotting.py
+++ b/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]]
--
GitLab