import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.patches as mpatches
import matplotlib.path as mpath
import os
import scipy.ndimage.morphology
import matplotlib.cm as mcm

plt.style.use('classic')
# os.chdir('/tmp/b8ebfd6e1bfc304d39487750bc2a48be')
data = np.load('cp-data.npz')
path = mpath.Path(vertices=data['verts'], codes=data['codes'],
                  _interpolation_steps=data['steps'], readonly=data['ro'])
imshow_kwargs = {
    'vmin': None,
    'vmax': None,
    'resample': None,
    'filterrad': 4.0,
    'alpha': None,
    'filternorm': 1,
    'aspect': None,
    'interpolation': 'none',
    'cmap': None,
    'data': None,
    'url': None,
    'shape': None,
    'extent': [-20037508.342789244, 20037508.342789244,
               -8683259.716434667, 8683259.716434667],
    'norm': None,
    'imlim': None,
    'origin': 'lower'
}

for color in ['fc', 'red', 'green', 'blue']:
    mask = data[color + '_mask']
    # mask = scipy.ndimage.morphology.binary_erosion(mark, iterations=2)
    img = np.ma.MaskedArray(data[color + '_data'], mask)

    fig, ax = plt.subplots(figsize=(10, 5))

    background = mpatches.PathPatch(path, edgecolor='none', facecolor='red',
                                    zorder=-1, clip_on=False,
                                    transform=ax.transData)
    ax.add_patch(background)

    outline = mpatches.PathPatch(path, edgecolor='black', facecolor='none',
                                 zorder=2.5, clip_on=False,
                                 transform=ax.transData)
    ax.add_patch(outline)

    # Not necessary?
    ax.patch = background

    if color != 'fc':
        cmap = mcm.get_cmap(color.title() + 's')
        cmap.set_bad('pink')
        imshow_kwargs['cmap'] = cmap
    else:
        imshow_kwargs['cmap'] = None
    ax.imshow(img, **imshow_kwargs)

    ax.set_title('%s %s' % (color, mpl.__version__))

    fig.savefig('result-' + color + '.png')

plt.show()