diff --git a/src/geomtools/detector/draw.py b/src/geomtools/detector/draw.py index dbc78462ea57f4dc66979f6502960931355884be..c5f9c4b7f2f1b7d50e4861ff17804369fc707e82 100644 --- a/src/geomtools/detector/draw.py +++ b/src/geomtools/detector/draw.py @@ -1,6 +1,5 @@ -import numpy as np - import matplotlib.pyplot as plt +import numpy as np from mpl_toolkits.axes_grid1 import make_axes_locatable from .geom import assemble_data, get_pixel_positions @@ -32,8 +31,8 @@ def plot_detector_layout(panels, figsize=(16, 16), **kwargs): return fig, ax -def plot_data_on_detector(data, panels, figwidth=16, cmap=plt.cm.magma, - **kwargs): +def plot_data_on_detector(data, panels, colorbar=True, figwidth=16, + cmap=plt.cm.magma, **kwargs): """Plots data according to the detector geometry.""" if isinstance(panels, np.ndarray): @@ -51,15 +50,17 @@ def plot_data_on_detector(data, panels, figwidth=16, cmap=plt.cm.magma, extent = (nx - xc + 0.5, -xc - 0.5, -yc - 0.5, ny - yc + 0.5) im = ax.imshow(np.flip(img, axis=1), origin='lower', - extent=extent, vmin=0, vmax=1, cmap=cmap) + extent=extent, vmin=0, vmax=1, cmap=cmap, + interpolation="none") ax.plot([-50, 50], [0, 0], 'r') ax.plot([0, 0], [-50, 50], 'r') ax.set_aspect('equal') ax.axis(False) - ax_divider = make_axes_locatable(ax) - cax = ax_divider.append_axes("right", size="5%", pad="2%") + if colorbar: + ax_divider = make_axes_locatable(ax) + cax = ax_divider.append_axes("right", size="5%", pad="2%") + fig.colorbar(im, ax=ax, cax=cax) - fig.colorbar(im, ax=ax, cax=cax) return fig, ax