From 49f666d04918fe580b4327de473d5191f97db0c6 Mon Sep 17 00:00:00 2001
From: Floreal Cabanettes <floreal.cabanettes@inra.fr>
Date: Wed, 21 Jun 2017 16:17:48 +0200
Subject: [PATCH] Add chromosomes infos to confirmation task

---
 build_pop.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/build_pop.py b/build_pop.py
index 4b7126b..4e8596a 100755
--- a/build_pop.py
+++ b/build_pop.py
@@ -293,10 +293,13 @@ def confirm(deletions: dict, variants: dict):
     nb_dels = 0
     variants = sorted(variants["DEL"], key=lambda x: x["min"])
     variants_counts = OrderedDict()
+    variants_counts_chr = {}
     for variant in variants:
         variants_counts["{0}-{1}".format(variant["min"], variant["max"])] = 0
-    for deletes in deletions.values():
-        nb_dels += len(deletes)
+    for chrm, deletes in deletions.items():
+        nb_dels_chr = len(deletes)
+        nb_dels += nb_dels_chr
+        variants_counts_chr[chrm] = nb_dels_chr
         for delete in deletes:
             v_len = delete["length"]
             for variant in variants:
@@ -304,8 +307,12 @@ def confirm(deletions: dict, variants: dict):
                     variants_counts["{0}-{1}".format(variant["min"], variant["max"])] += 1
                     break
     print("We generate {0} deletion variants.".format(nb_dels))
+    print("Ranges:")
     for v_range, v_count in variants_counts.items():
         print("\t- Range {0}: {1}".format(v_range, v_count))
+    print("Chromosomes:")
+    for chrm in sorted(list(variants_counts_chr.keys())):
+        print("\t- {0}: {1}".format(chrm, variants_counts_chr[chrm]))
     print("")
     return input("Continue [Y/n]? ") in ["y", "Y", ""]
 
-- 
GitLab