Merge pull request #58 from mirnylab/print_zoom_levels

Add function to print zoom resolutions

Author: nvictus

cooler/contrib/higlass.py

@@ -136,3 +136,30 @@ def get_info(file_path):
         }
 
     return info
+
+
+def get_quadtree_depth(chromsizes, binsize):
+    """
+    Depth of quad tree necessary to tesselate the concatenated genome with quad
+    tiles such that linear dimension of the tiles is a preset multiple of the 
+    genomic resolution.
+
+    """
+    tile_size_bp = TILE_SIZE * binsize
+    min_tile_cover = np.ceil(sum(chromsizes) / tile_size_bp)
+    return int(np.ceil(np.log2(min_tile_cover)))
+
+
+def get_zoom_resolutions(chromsizes, base_res):
+    return [base_res * 2**x for x in range(get_quadtree_depth(chromsizes, base_res)+1)]
+
+
+def print_zoom_resolutions(chromsizes_file, base_res):
+    """
+    Print comma-separated list of zoom resolutions for a given genome
+    and base resolution.
+
+    """
+    chromsizes = cooler.util.read_chromsizes(chromsizes_file, all_names=True)
+    resolutions = get_zoom_resolutions(chromsizes, base_res)
+    print(','.join(str(res) for res in resolutions))