aboutsummaryrefslogtreecommitdiff
path: root/metchart/customization.py
blob: 84fc5467f805e0215c69059cfdbaf832d7397c44 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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')