[plotter] fix norm

This commit is contained in:
Noe Brucy
2021-07-05 23:34:15 +02:00
parent a50fb60663
commit d2b950c215
+10 -7
View File
@@ -546,8 +546,9 @@ class Plotter(Aggregator, BaseProcessor):
cmap="plasma", cmap="plasma",
norm="log", norm="log",
put_cbar=True, put_cbar=True,
autoscale=True,
transform=None, transform=None,
vmin=None,
vmax=None,
**kwargs, **kwargs,
): ):
""" """
@@ -579,13 +580,15 @@ class Plotter(Aggregator, BaseProcessor):
if transform is not None: if transform is not None:
dmap = transform(dmap) dmap = transform(dmap)
if norm == "log": if vmin is None:
norm = mpl.colors.LogNorm() vmin = np.min(dmap)
elif norm == "linear": if vmax is None:
norm = mpl.colors.NoNorm() vmax = np.max(dmap)
if autoscale and norm is not None: if norm == "log":
norm.autoscale(dmap) norm = mpl.colors.LogNorm(vmin=vmin, vmax=vmax)
elif norm == "linear":
norm = mpl.colors.Normalize(vmin=vmin, vmax=vmax)
im = plt.imshow( im = plt.imshow(
dmap, extent=im_extent, origin="lower", norm=norm, cmap=cmap, **kwargs dmap, extent=im_extent, origin="lower", norm=norm, cmap=cmap, **kwargs