import re
import matplotlib as mpl
import numpy

tudcolors = {
    'a': ('#5D85C3', '#009CDA', '#50B695', '#AFCC50', '#DDDF48', '#FFE05C',
          '#F8BA3C', '#EE7A34', '#E9503E', '#C9308E', '#804597'),
    'b': ('#005AA9', '#0083CC', '#009D81', '#99C000', '#C9D400', '#FDCA00',
          '#F5A300', '#EC6500', '#E6001A', '#A60084', '#721085'),
    'c': ('#004E8A', '#00689D', '#008877', '#7FAB16', '#B1BD00', '#D7AC00',
          '#D28700', '#CC4C03', '#B90F22', '#951169', '#611C73'),
    'd': ('#243572', '#004E73', '#00715E', '#6A8B22', '#99A604', '#AE8E00',
          '#BE6F00', '#A94913', '#961C26', '#732054', '#4C226A'),
}

# Store each color value in the dict as defined in the Style-Guide (e.g. tud9c)
tudcolors.update({'tud{}{}'.format(i + 1, s): col for s in tudcolors for i, col in enumerate(tudcolors[s])})

color_maps = {
    'blue-red': (tudcolors['tud1b'], tudcolors['tud9b']),
    'black-green': ('black', tudcolors['tud5c']),
    'violett-green': (tudcolors['tud11c'], tudcolors['tud5b'])
}


nominal_colors = {scheme: [tudcolors[scheme][i] for i in [1, 8, 3, 9, 6, 2]] for scheme in 'abcd'}


def full_colors(N, scheme='b'):
    cmap = mpl.colors.LinearSegmentedColormap.from_list('tud{}'.format(scheme), tudcolors[scheme])
    return ['#{:02x}{:02x}{:02x}'.format(*cmap(x, bytes=True)[:3]) for x in numpy.linspace(0, 1, N)]


def sequential_colors(N, cmap='blue-red', min=0, max=1):
    if cmap in color_maps:
        cmap = mpl.colors.LinearSegmentedColormap.from_list('tud_{}'.format(cmap), color_maps[cmap])
    elif '-' in cmap:
        cols = [tudcolors[k] if 'tud' in k else k for k in cmap.split('-')]
        cmap = mpl.colors.LinearSegmentedColormap.from_list(cmap, cols)
    else:
        cmap = mpl.pyplot.get_cmap(cmap)
    return ['#{:02x}{:02x}{:02x}'.format(*cmap(x, bytes=True)[:3]) for x in numpy.linspace(min, max, N)]