From 274df86d989203823967bf34109801d30335a61d Mon Sep 17 00:00:00 2001
From: Jonas Gunz <himself@jonasgunz.de>
Date: Fri, 18 Oct 2024 02:04:08 +0200
Subject: add customization

---
 metchart/customization.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 metchart/customization.py

diff --git a/metchart/customization.py b/metchart/customization.py
new file mode 100644
index 0000000..84fc546
--- /dev/null
+++ b/metchart/customization.py
@@ -0,0 +1,32 @@
+from metpy.units import units
+
+import matplotlib as mpl
+from matplotlib.colors import LinearSegmentedColormap
+
+def register_colormaps():
+    # Define custom colormap
+    clcov_cmap = {
+        'red': (
+            (0.0, 0.0, 0.0),
+            (0.1, 0.9, 0.9),
+            (1.0, 0.3, 0.3),
+        ),
+        'green': (
+            (0.0, 0.5, 0.5),
+            (0.1, 0.9, 0.9),
+            (1.0, 0.3, 0.3),
+        ),
+        'blue': (
+            (0.0, 0.9, 0.9),
+            (0.1, 0.9, 0.9),
+            (1.0, 0.3, 0.3),
+        ),
+    }
+
+    mpl.colormaps.register(LinearSegmentedColormap('clcov', clcov_cmap))
+
+def register_units():
+    # Define custom gpm and gpdm units. The default gpm in metpy is aliased to meter.
+    # We need the correct definition
+    units.define('_gpm = 9.80665 * J/kg')
+    units.define('_gpdm = 10 * _gpm')
-- 
cgit v1.2.3