Source code for cameo.visualization.palette

# Copyright 2016 The Novo Nordisk Foundation Center for Biosustainability
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# See the License for the specific language governing permissions and
# limitations under the License.

import six

from palettable.colorbrewer.colorbrewer import _load_maps_by_type as load_colorbrewer_palette  # noqa

from palettable.cubehelix.cubehelix import _get_all_maps as load_cubehelix_palettes
from palettable.tableau.tableau import _get_all_maps as load_tableau_palettes
from palettable.wesanderson.wesanderson import _get_all_maps as load_wesanderson_palettes

from palettable.palette import Palette

[docs]class PaletteMapper(object): def __init__(self): self._palettes = {} @staticmethod def _split_palette_name(name): split = name.split("_") if len(split) == 1: raise ValueError("Invalid palette description '%s'" % name) name = split[0] reverse = False if len(split) == 2: size = int(split[1]) else: if split[-1] == "r": reverse = True size = int(split[-2]) else: size = int(split[-1]) return name, size, reverse
[docs] def add_palettalbe_palette(self, name, palette): assert isinstance(palette, Palette) name, size, reverse = self._split_palette_name(name) if name not in self._palettes: self._palettes[name] = {} if size not in self._palettes[name]: self._palettes[name][size] = {} self._palettes[name][size][reverse] = palette
[docs] def map_palette(self, name, n=3, reverse=False): try: name, size, reverse = self._split_palette_name(name) except ValueError: size = n palette = self._palettes[name] max_size = max(palette.keys()) min_size = min(palette.keys()) if size < min_size: size = min_size if size > max_size: size = max_size return palette[size][reverse]
mapper = PaletteMapper() for key, obj in six.iteritems(load_colorbrewer_palette("diverging")): mapper.add_palettalbe_palette(key, obj) for key, obj in six.iteritems(load_colorbrewer_palette("sequential")): mapper.add_palettalbe_palette(key, obj) for key, obj in six.iteritems(load_colorbrewer_palette("qualitative")): mapper.add_palettalbe_palette(key, obj) for key, obj in six.iteritems(load_cubehelix_palettes()): mapper.add_palettalbe_palette(key, obj) for key, obj in six.iteritems(load_tableau_palettes()): mapper.add_palettalbe_palette(key, obj) for key, obj in six.iteritems(load_wesanderson_palettes()): mapper.add_palettalbe_palette(key, obj)