diff --git a/build_pop.py b/build_pop.py index 4b7126b8ef7c6ae249229ef8af25956cf3ee3d8d..4e8596a99f6c585d42dc6f3cfd1e483918ba9c97 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", ""]