From b708b9cd230800fb3b6ec54fbff8d48c76c89453 Mon Sep 17 00:00:00 2001
From: Blaise Li <blaise.li__git@nsup.org>
Date: Mon, 20 Aug 2018 17:38:30 +0200
Subject: [PATCH] More generic colour-attributing code.

---
 small_RNA-seq/small_RNA-seq.snakefile | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/small_RNA-seq/small_RNA-seq.snakefile b/small_RNA-seq/small_RNA-seq.snakefile
index 960ac61..b45f3bb 100644
--- a/small_RNA-seq/small_RNA-seq.snakefile
+++ b/small_RNA-seq/small_RNA-seq.snakefile
@@ -3792,10 +3792,23 @@ def make_lib_colour_setter(colour_series_dict):
     return (colour_setter, lib2colour)
 
 
-small_type_series = {"pi" : ["pi"], "mi" : ["mi"], "gene" : ["prot_sisiu", "pseu_sisiu"], "repeat" : ["te_sisiu", "satel_sisiu", "simrep_sisiu"]}
+small_type_series = {
+    "pi" : ["pi"],
+    "mi" : ["mi"],
+    "gene" : ["prot_sisiu", "pseu_sisiu"],
+    "repeat" : ["te_sisiu", "satel_sisiu", "simrep_sisiu"]}
+
+
 def make_small_type_colour_setter(small_type_series):
-    base_colours = sns.color_palette("colorblind", 4)
-    series2base_colour = {"pi" : base_colours[0], "mi" : base_colours[1], "gene" : base_colours[2], "repeat" : base_colours[3]}
+    base_colours = sns.color_palette("colorblind", len(small_type_series))
+    # series2base_colour = {
+    #     "pi" : base_colours[0],
+    #     "mi" : base_colours[1],
+    #     "gene" : base_colours[2],
+    #     "repeat" : base_colours[3]}
+    series2base_colour = {
+        key : base_colours[i]
+        for (i, key) in enumerate(small_type_series.keys())}
     small_type2colour = {}
     for (type_series, small_types) in small_type_series.items():
         nb_colours = len(small_types)
@@ -3816,6 +3829,8 @@ def make_small_type_colour_setter(small_type_series):
 
 
 (small_type_colour_setter, small_type2colour) = make_small_type_colour_setter(small_type_series)
+
+
 (lib_colour_setter, lib2colour) = make_lib_colour_setter(genotype_series)
 #def make_sample_plotter(axis, libs=LIBS, treats=TREATS):
 def make_sample_plotter(axis):
-- 
GitLab