Changeset 700 for mds-and-trees

Timestamp:

09/19/17 15:45:07 (7 years ago)

Author:

konrad

Message:

Colors support matplotlib cmaps now. 'color' property in config file can take two more key values: 'cmap' and 'normalize_cmap'. Cmap is defined as in ​https://matplotlib.org/devdocs/api/_as_gen/matplotlib.colors.LinearSegmentedColormap.html but uses arrays instead of tuples. If 'normalize_cmap" == false, step values (first value in each tuple) must be normalized. Otherwise, if 'normalize_cmap" == true, step values can be raw property values (from the range of values in the data).

File:

• mds-and-trees/tree-genealogy.py (modified) (6 diffs)

mds-and-trees/tree-genealogy.py

@@ -31 +31 @@
                 'color': {
                     'meaning': 'Lifespan',
+                    'normalize_cmap': False,
+                    'cmap': {},
                     'start': 'red',
                     'end': 'green',
@@ -51 +53 @@
                 'color': {
                     'meaning': 'adepth',
+                    'normalize_cmap': False,
+                    'cmap': {},
                     'start': 'black',
                     'end': 'red',
@@ -77 +81 @@
                 else:
                     destination[key] = value
-
             return destination
 
@@ -85 +88 @@
             self.settings = merge(c, self.settings)
             #print(json.dumps(self.settings, indent=4, sort_keys=True))
+
+        self.compile_cmaps()
+
+    def compile_cmaps(self):
+
+        # normalization of color breaking points
+        for part in ['dots', 'lines']:
+            if self.settings[part]['color']['cmap'] \
+                    and self.settings[part]['color']['normalize_cmap']:
+                cmap = self.settings[part]['color']['cmap']
+                min = self.design.props[self.settings[part]['color']['meaning'] + "_min"]
+                max = self.design.props[self.settings[part]['color']['meaning'] + "_max"]
+                for key in cmap:
+                    for arr in cmap[key]:
+                        arr[0] = (arr[0] - min) / (max-min)
+                    if cmap[key][0][0] != 0:
+                        cmap[key].insert(0, cmap[key][0][:])
+                        cmap[key][0][0] = 0
+                    if cmap[key][-1][0] != 1:
+                        cmap[key].append(cmap[key][-1][:])
+                        cmap[key][-1][0] = 1
+
+
+        for part in ['dots', 'lines']:
+            if self.settings[part]['color']['cmap']:
+                cmap = self.settings[part]['color']['cmap']
+                for key in cmap:
+                    new_arr = []
+                    for arr in cmap[key]:
+                        new_arr.append(tuple(arr))
+                    cmap[key] = tuple(new_arr)
+                #print(cmap)
+                self.settings[part]['color']['cmap'] = colors.LinearSegmentedColormap('Custom', cmap)
 
     def draw_dots(self, file, min_width, max_width, max_height):
@@ -137 +173 @@
         bias = self.settings[part][prop]['bias']
         if prop == "color":
-            return self.compute_color(start, end, value, bias)
+            if not self.settings[part][prop]['cmap']:
+                return self.compute_color(start, end, value, bias)
+            else:
+                return self.compute_color_from_cmap(self.settings[part][prop]['cmap'], value, bias)
         else:
             return self.compute_value(start, end, value, bias)
+
+    def compute_color_from_cmap(self, cmap, value, bias=1):
+        value = 1 - (1-value)**bias
+        rgba = cmap(value)
+        return (100*rgba[0], 100*rgba[1], 100*rgba[2])
+
 
     def compute_color(self, start, end, value, bias=1):
@@ -220 +265 @@
         offset = (min(fx-w, tx-w), min(fy-w, ty-w))
         size = (abs(fx-tx)+2*w, abs(fy-ty)+2*w)
+        if size[0] == 0 or size[1] == 0:
+            return
 
         c = style['color']