palette picker multiflex

rowers/utils.py

@@ -5,15 +5,54 @@ import colorsys
 
 lbstoN = 4.44822
 
-def range_to_color_hex(groupcols):
-    rgb = [colorsys.hsv_to_rgb((207-4*x)/360.,
-                               0.06+0.89*x,
-                               1-0.38*x) for x in groupcols]
+def trcolors(r1,g1,b1,r2,g2,b2):
+    r1 = r1/255.
+    r2 = r2/255.
+    g1 = g1/255.
+    g2 = g2/255.
+    b2 = b2/255.
+    b1 = b1/255.
+    h1,s1,v1 = colorsys.rgb_to_hsv(r1,g1,b1)
+    h2,s2,v2 = colorsys.rgb_to_hsv(r2,g2,b2)
 
-    RGB = [(int(255.*r),int(255.*g),int(255.*b)) for (r, g, b) in rgb]
-    colors = ["#%02x%02x%02x" % (r, g, b) for (r, g, b) in RGB]
+    
+    return 360*h1,360*(h2-h1),s1,(s2-s1),v1,(v2-v1)
 
-    return colors
+palettes = {
+  'monochrome_blue':(207,-4,0.06,0.89,1.0,-0.38),
+  'gold_sunset':(47,-31,.26,-0.12,0.94,-0.5),
+  'blue_red':(207,-200,.85,0,.74,-.24),
+  'blue_green':(207,-120,.85,0,.75,.25),
+  'cyan_green':(192,-50,.08,.65,.98,-.34),
+  'cyan_purple':trcolors(237,248,251,136,65,157),
+  'green_blue':trcolors(240,249,232,8,104,172),
+  'orange_red':trcolors(254,240,217,179,0,0),
+  'cyan_blue':trcolors(241,238,246,4,90,141),
+  'cyan_green':trcolors(246,239,247,1,108,89),
+  'cyan_magenta':trcolors(241,238,246,152,0,67),
+  'beige_magenta':trcolors(254,235,226,122,1,119),
+  'yellow_green':trcolors(255,255,204,0,104,55),
+  'yellow_blue':trcolors(255,255,205,37,52,148),
+  'autumn':trcolors(255,255,212,153,52,4),
+  'yellow_red':trcolors(255,255,178,189,0,39)
+}
+
+
+def range_to_color_hex(groupcols,palette='monochrome_blue'):
+
+  try:
+    plt = palettes[palette]
+  except KeyErro:
+    plt = palettes['monochrome_blue']
+    
+  rgb = [colorsys.hsv_to_rgb((plt[0]+plt[1]*x)/360.,
+                             plt[2]+plt[3]*x,
+                             plt[4]+plt[5]*x) for x in groupcols]
+
+  RGB = [(int(255.*r),int(255.*g),int(255.*b)) for (r, g, b) in rgb]
+  colors = ["#%02x%02x%02x" % (r, g, b) for (r, g, b) in RGB]
+  
+  return colors
 
 def str2bool(v):
   return v.lower() in ("yes", "true", "t", "1")