From f94fbf99dc0bf9fd8d5d8b2181636d6fd2d007e4 Mon Sep 17 00:00:00 2001
From: Jonas Gunz <himself@jonasgunz.de>
Date: Thu, 19 Oct 2023 00:32:28 +0200
Subject: first wip meteogram

---
 run.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'run.py')

diff --git a/run.py b/run.py
index fa0f598..d0a004a 100755
--- a/run.py
+++ b/run.py
@@ -4,6 +4,8 @@ import sys
 import yaml
 import json
 import matplotlib.pyplot as plt
+import matplotlib as mpl
+from matplotlib.colors import LinearSegmentedColormap
 
 from metpy.units import units
 
@@ -12,6 +14,27 @@ from metpy.units import units
 units.define('_gpm = 9.80665 * J/kg')
 units.define('_gpdm = 10 * _gpm')
 
+# 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))
+
 FILE = 'config.yaml'
 if len(sys.argv) > 1:
     FILE = sys.argv[1]
-- 
cgit v1.2.3