Update trackPlot.py

Added checks to test is 'chr' should be removed or added to chromosome name when plotting arcs or bedgraph files
deeptools · Oct 6, 2016 · 750bc0f · 750bc0f
1 parent b7f6d02
commit 750bc0f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/hicexplorer/trackPlot.py b/hicexplorer/trackPlot.py
@@ -455,6 +455,9 @@ def plot(self, ax, label_ax, chrom_region, start_region, end_region):
         score_list = []
         pos_list = []
 
+        if chrom_region not in self.interval_tree.keys():
+            chrom_region = change_chrom_names(chrom_region)
+
         for region in self.interval_tree[chrom_region].find(start_region - 10000,
                                                            end_region + 10000):
             score_list.append(float(region.value[0]))
@@ -1622,6 +1625,8 @@ def plot(self, ax, label_ax, chrom_region, region_start, region_end):
         max_diameter = 0
         count = 0
 
+        if chrom_region not in self.interval_tree.keys():
+            chrom_region = change_chrom_names(chrom_region)
         arcs_in_region = self.interval_tree[chrom_region].find(region_start, region_end)
 
         for idx, interval in enumerate(arcs_in_region):