From d2b950c215f660f9eaaa0d8d40ef2c3fbfcab5c3 Mon Sep 17 00:00:00 2001
From: Noe Brucy <noe.brucy@cea.fr>
Date: Mon, 5 Jul 2021 23:34:15 +0200
Subject: [PATCH] [plotter] fix norm

---
 plotter.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/plotter.py b/plotter.py
index 6e403c2..2250626 100644
--- a/plotter.py
+++ b/plotter.py
@@ -546,8 +546,9 @@ class Plotter(Aggregator, BaseProcessor):
         cmap="plasma",
         norm="log",
         put_cbar=True,
-        autoscale=True,
         transform=None,
+        vmin=None,
+        vmax=None,
         **kwargs,
     ):
         """
@@ -579,13 +580,15 @@ class Plotter(Aggregator, BaseProcessor):
         if transform is not None:
             dmap = transform(dmap)
 
-        if norm == "log":
-            norm = mpl.colors.LogNorm()
-        elif norm == "linear":
-            norm = mpl.colors.NoNorm()
+        if vmin is None:
+            vmin = np.min(dmap)
+        if vmax is None:
+            vmax = np.max(dmap)
 
-        if autoscale and norm is not None:
-            norm.autoscale(dmap)
+        if norm == "log":
+            norm = mpl.colors.LogNorm(vmin=vmin, vmax=vmax)
+        elif norm == "linear":
+            norm = mpl.colors.Normalize(vmin=vmin, vmax=vmax)
 
         im = plt.imshow(
             dmap, extent=im_extent, origin="lower", norm=norm, cmap=cmap, **kwargs