From cdca5a2141b77d544a1b5b2391d3a9115349a409 Mon Sep 17 00:00:00 2001
From: "Joachim Wuttke (o)" <j.wuttke@fz-juelich.de>
Date: Tue, 15 Dec 2020 19:53:04 +0100
Subject: [PATCH] colormap now from kwargs, environ, 'inferno'
---
Wrap/Python/plot_utils.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/Wrap/Python/plot_utils.py b/Wrap/Python/plot_utils.py
index 2cf3c73e9f0..0f770b92b25 100644
--- a/Wrap/Python/plot_utils.py
+++ b/Wrap/Python/plot_utils.py
@@ -23,6 +23,8 @@ except Exception as e:
label_fontsize = 16
+CMAP = "CMAP" in os.environ and os.environ["CMAP"] or 'inferno'
+
def get_axes_limits(result, units):
"""
@@ -100,10 +102,10 @@ def plot_array(array, axes_limits=None, **kwargs):
xlabel = kwargs.pop('xlabel', None)
ylabel = kwargs.pop('ylabel', None)
zlabel = kwargs.pop('zlabel', "Intensity")
-
title = kwargs.pop('title', None)
+ cmap = kwargs.pop('cmap', CMAP)
- im = plt.imshow(array, norm=norm, extent=axes_limits, **kwargs)
+ im = plt.imshow(array, cmap=cmap, norm=norm, extent=axes_limits, **kwargs)
cb = plt.colorbar(im, pad=0.025)
if xlabel:
--
GitLab