Initial project version

pygrace/colors.py

@@ -0,0 +1,158 @@
+import matplotlib as mpl
+import matplotlib.pyplot as plt
+import numpy as np
+from collections import OrderedDict
+from itertools import cycle
+
+
+def make_cmap(colors, mapping=True):
+    if isinstance(colors, dict):
+        colors = np.array(colors.values())
+    bit_rgb = np.linspace(0, 1, 256)
+    position = np.linspace(0, 1, len(colors))
+    _cols = colors[:] + 0 
+    for i in range(len(colors)):
+        for j in xrange(3):
+            _cols[i][j] = bit_rgb[colors[i][j]]
+    if mapping:
+        cdict = {'red': [], 'green':[], 'blue':[]}
+        for pos, color in zip(position, _cols):
+            cdict['red'].append((pos, color[0], color[0]))
+            cdict['green'].append((pos, color[1], color[1]))
+            cdict['blue'].append((pos, color[2], color[2]))
+
+        cmap = mpl.colors.LinearSegmentedColormap('my_colormap', cdict, 256)
+    else:
+        cmap = mpl.colors.ListedColormap(_cols, 'my_colormap')
+    return cmap
+
+cm_bw = OrderedDict([('white', (255, 255, 255, 1)),
+                     ('black', (0, 0, 0, 1))])
+
+cm_grace = OrderedDict([
+    ('red',       (255, 0,   0,   255)),
+    ('green',     (0,   255, 0,   255)),
+    ('blue',      (0,   0,   255, 255)),
+    ('yellow',    (255, 255, 0,   255)),
+    ('brown',     (188, 143, 143, 255)),
+    ('grey',      (220, 220, 220, 255)),
+    ('violet',    (148, 0,   211, 255)),
+    ('cyan',      (0,   255, 255, 255)),
+    ('magenta',   (255, 0,   255, 255)),
+    ('orange',    (255, 165, 0,   255)),
+    ('indigo',    (114, 33,  188, 255)),
+    ('maroon',    (103, 7,   72,  255)),
+    ('turquoise', (64,  224, 208, 255)),
+    ('green4',    (0,   139, 0,   255))
+    ])
+
+
+cm_tuda = OrderedDict([
+    ('tud1a', (93,  133, 195, 255)),
+    ('tud2a', (0,   156, 218, 255)),
+    ('tud3a', (80,  182, 149, 255)),
+    ('tud4a', (175, 204, 80,  255)),
+    ('tud5a', (221, 223, 72,  255)),
+    ('tud6a', (255, 224, 92,  255)),
+    ('tud7a', (248, 186, 60,  255)),
+    ('tud8a', (238, 122, 52,  255)),
+    ('tud9a', (233, 80,  62,  255)),
+    ('tud10a', (201, 48,  142, 255)),
+    ('tud11a', (128, 69,  151, 255))
+    ])
+
+cm_tudb = OrderedDict([
+    ('tud1b', (0,   90,  169, 255)),
+    ('tud2b', (0,   131, 204, 255)),
+    ('tud3b', (0,   157, 129, 255)),
+    ('tud4b', (153, 192, 0,   255)),
+    ('tud5b', (201, 212, 0,   255)),
+    ('tud6b', (253, 202, 0,   255)),
+    ('tud7b', (245, 163, 0,   255)),
+    ('tud8b', (236, 101, 0,   255)),
+    ('tud9b', (230, 0,   26,  255)),
+    ('tud10b', (166, 0,   132, 255)),
+    ('tud11b', (114, 16,  133, 255))
+    ])
+
+cm_tudc = OrderedDict([
+    ('tud1c', (0,   78,  138, 255)),
+    ('tud2c', (0,   104, 157, 255)),
+    ('tud3c', (0,   136, 119, 255)),
+    ('tud4c', (127, 171, 22,  255)),
+    ('tud5c', (177, 189, 0,   255)),
+    ('tud6c', (215, 172, 0,   255)),
+    ('tud7c', (210, 135, 0,   255)),
+    ('tud8c', (204, 76,  3,   255)),
+    ('tud9c', (185, 15,  34,  255)),
+    ('tud10c', (149, 17,  105, 255)),
+    ('tud11c', (97,  28,  115, 255))
+    ])
+                
+                    
+cm_tudd = OrderedDict([
+    ('tud1d', (36,  53,  114, 255)),
+    ('tud2d', (0,   78,  115, 255)),
+    ('tud3d', (0,   113, 94,  255)),
+    ('tud4d', (106, 139, 55,  255)),
+    ('tud5d', (153, 166, 4,   255)),
+    ('tud6d', (174, 142, 0,   255)),
+    ('tud7d', (190, 111, 0,   255)),
+    ('tud8d', (169, 73,  19,  255)),
+    ('tud9d', (156, 28,  38,  255)),
+    ('tud10d', (115, 32,  84,  255)),
+    ('tud11d', (76,  34,  106, 255))
+    ])
+
+
+def combine_maps(*args):
+    color_map = OrderedDict()
+    for d in args:
+        for k, v in d.items():
+            color_map[k] = v
+    return color_map
+
+cm_tud = combine_maps(cm_tuda, cm_tudb, cm_tudc, cm_tudd)
+
+all_colors = combine_maps(cm_bw, cm_grace, cm_tud)
+
+colors = cm_tud.values()
+
+symbols = cycle(['o', 't', 'd', 's', '+'])
+
+sym_map = {None: '0', 'o': '1', 's': '2', 'd': '3', 't': '6', '+': '9'}
+
+
+class ColorDict(OrderedDict):
+    def __init__(self, *args, **kwargs):
+        super(ColorDict, self).__init__(*args, **kwargs)
+        for d in [cm_bw, cm_tuda, cm_tudb, cm_tudc, cm_tudd, cm_grace]:
+            old_length = len(self)
+            for k, c in enumerate(d.keys()):
+                rgb = d[c][:3]
+                self[rgb] = (k + old_length, c)
+        self._new = OrderedDict()
+
+    def __contains__(self, item):
+        if isinstance(item, tuple):
+            return item in self.keys()
+        elif isinstance(item. basestring):
+            return any([item == v[1] for v in self.values()])
+        else:
+            return False
+
+    def convert_color(self, color):
+        c = tuple(color[0:3])
+        if c not in self.keys():
+            self[c] = (str(self.__len__()), ''.join('{:02X}'.format(a) for a in c))
+            self._new[c] = (str(self.__len__()-1), ''.join('{:02X}'.format(a) for a in c))
+        return self[c][0]
+
+    def has_index(self, idx):
+        return any([idx == v[0] for v in self.values()])
+
+    def new_colors(self):
+        return self._new
+
+if __name__ == '__main__':
+    pass