| 55 | | def generateColorscheme(masterColor, keys): |
| 56 | | """Generates a dictionary where the keys match the keys argument and |
| 57 | | the values are colors derivated from the masterColor. |
| 58 | | |
| 59 | | Each color is a lighter version of masterColor. This difference is |
| 60 | | computed based on the number of keys. |
| 61 | | |
| 62 | | The masterColor is given in a hex string format. |
| 63 | | """ |
| 64 | | r, g, b = hex2rgb(masterColor) |
| 65 | | light = 1.0 / (len(keys)*2) |
| 66 | | return dict([(key, lighten(r, g, b, light * i)) |
| 67 | | for i, key in enumerate(keys)]) |
| 68 | | |
| 69 | | |
| 70 | | def defaultColorscheme(keys): |
| 71 | | """Return the default color scheme (derived from a dark green)""" |
| 72 | | return generateColorscheme(DEFAULT_COLOR, keys) |
| 73 | | |
| 74 | | |
| 75 | | def getColorscheme(color, keys): |
| 76 | | """Get a color scheme from the six predefined ones or makes another |
| 77 | | one if the color is not found |
| 78 | | """ |
| 79 | | return generateColorscheme(colorSchemes.get(color, color), keys) |
| 80 | | |
| 81 | | |
| 82 | | # default colors for color schemes |
| 83 | | colorSchemes = dict( |
| | 56 | basicColors = dict( |
| | 64 | |
| | 65 | |
| | 66 | class ColorSchemeMetaclass(type): |
| | 67 | """This metaclass is used to autoregister all ColorScheme classes""" |
| | 68 | |
| | 69 | def __new__(mcs, name, bases, dict): |
| | 70 | klass = type.__new__(mcs, name, bases, dict) |
| | 71 | klass.registerColorScheme() |
| | 72 | return klass |
| | 73 | |
| | 74 | |
| | 75 | class ColorScheme(dict): |
| | 76 | """A color scheme is a dictionary where the keys match the keys |
| | 77 | constructor argument and the values are colors""" |
| | 78 | |
| | 79 | __metaclass__ = ColorSchemeMetaclass |
| | 80 | __registry__ = {} |
| | 81 | |
| | 82 | def __init__(self, keys): |
| | 83 | super(ColorScheme, self).__init__() |
| | 84 | |
| | 85 | @classmethod |
| | 86 | def registerColorScheme(cls): |
| | 87 | key = cls.__name__.replace('ColorScheme', '').lower() |
| | 88 | if key: |
| | 89 | cls.__registry__[key] = cls |
| | 90 | |
| | 91 | @classmethod |
| | 92 | def getColorScheme(cls, name, default=None): |
| | 93 | return cls.__registry__.get(name, default) |
| | 94 | |
| | 95 | |
| | 96 | class GradientColorScheme(ColorScheme): |
| | 97 | """In this color scheme each color is a lighter version of initialColor. |
| | 98 | |
| | 99 | This difference is computed based on the number of keys. |
| | 100 | |
| | 101 | The initialColor is given in a hex string format. |
| | 102 | """ |
| | 103 | |
| | 104 | def __init__(self, keys, initialColor=DEFAULT_COLOR): |
| | 105 | super(GradientColorScheme, self).__init__(keys) |
| | 106 | if initialColor in basicColors: |
| | 107 | initialColor = basicColors[initialColor] |
| | 108 | |
| | 109 | r, g, b = hex2rgb(initialColor) |
| | 110 | light = 1.0 / (len(keys) * 2) |
| | 111 | |
| | 112 | for i, key in enumerate(keys): |
| | 113 | self[key] = lighten(r, g, b, light * i) |
