Enable querying hardmasked blocks

Author: dpryan79

py2bit.c

@@ -306,6 +306,75 @@ static PyObject *py2bitBases(pyTwoBit_t *self, PyObject *args, PyObject *kwds) {
     return NULL;
 }
 
+static PyObject *py2bitHardMaskedBlocks(pyTwoBit_t *self, PyObject *args, PyObject *kwds) {
+    PyObject *ret = NULL, *tup = NULL;
+    TwoBit *tb = self->tb;
+    char *chrom;
+    unsigned long startl = 0, endl = 0, totalBlocks = 0, tid;
+    uint32_t start, end, len, blockStart, blockEnd, i, j;
+    static char *kwd_list[] = {"chrom", "start", "end", NULL};
+
+    if(!tb) {
+        PyErr_SetString(PyExc_RuntimeError, "The 2bit file handle is not open!");
+        return NULL;
+    }
+
+    if(!PyArg_ParseTupleAndKeywords(args, kwds, "s|kk", kwd_list, &chrom, &startl, &endl)) {
+        PyErr_SetString(PyExc_RuntimeError, "You must supply at least a chromosome!");
+        return NULL;
+    }
+
+    //Get the chromosome ID
+    for(i=0; i<tb->hdr->nChroms; i++) {
+        if(strcmp(tb->cl->chrom[i], chrom) == 0) {
+            tid = i;
+            break;
+        }
+    }
+
+    len = twobitChromLen(tb, chrom);
+    if(len == 0) {
+        PyErr_SetString(PyExc_RuntimeError, "The specified chromosome doesn't exist in the 2bit file!");
+        return NULL;
+    }
+    if(endl == 0) endl = len;
+    if(endl > len) endl = len;
+    end = (uint32_t) endl;
+    if(startl >= endl && startl > 0) {
+        PyErr_SetString(PyExc_RuntimeError, "The start value must be less then the end value (and the end of the chromosome");
+        return NULL;
+    }
+    start = (uint32_t) startl;
+
+    // Count the total number of overlapping N-masked blocks
+    for(i=0; i<tb->idx->nBlockCount[tid]; i++) {
+        blockStart = tb->idx->nBlockStart[tid][i];
+        blockEnd = blockStart + tb->idx->nBlockSizes[tid][i];
+        if(blockStart < end && blockEnd > start) totalBlocks++;
+    }
+
+    // Form the output
+    ret = PyList_New(totalBlocks);
+    if(!ret) goto error;
+    for(i=0, j=0; i<tb->idx->nBlockCount[tid]; i++) {
+        blockStart = tb->idx->nBlockStart[tid][i];
+        blockEnd = blockStart + tb->idx->nBlockSizes[tid][i];
+        if(blockStart < end && blockEnd > start) {
+            tup = Py_BuildValue("(kk)", (unsigned long) blockStart, (unsigned long) blockEnd);
+            if(!tup) goto error;
+            if(PyList_SetItem(ret, j++, tup)) goto error;
+        }
+    }
+
+    return ret;
+
+error:
+    if(ret) Py_XDECREF(ret);
+    if(tup) Py_XDECREF(tup);
+    PyErr_SetString(PyExc_RuntimeError, "Received an error while constructing the output list and tuples!");
+    return NULL;
+}
+
 #if PY_MAJOR_VERSION >= 3
 PyMODINIT_FUNC PyInit_py2bit(void) {
     PyObject *res;

py2bit.h

@@ -1,7 +1,7 @@
 #include <Python.h>
 #include "2bit.h"
 
-#define pyTwoBitVersion "0.2.2"
+#define pyTwoBitVersion "0.3.0"
 
 typedef struct {
     PyObject_HEAD
@@ -16,6 +16,7 @@ static PyObject* py2bitClose(pyTwoBit_t *pybw, PyObject *args);
 static PyObject* py2bitChroms(pyTwoBit_t *pybw, PyObject *args);
 static PyObject *py2bitSequence(pyTwoBit_t *pybw, PyObject *args, PyObject *kwds);
 static PyObject *py2bitBases(pyTwoBit_t *pybw, PyObject *args, PyObject *kwds);
+static PyObject *py2bitHardMaskedBlocks(pyTwoBit_t *pybw, PyObject *args, PyObject *kwds);
 static void py2bitDealloc(pyTwoBit_t *pybw);
 
 static PyMethodDef tbMethods[] = {
@@ -145,6 +146,24 @@ bases. Counts may sum to less than the length of the region for the same reason.
 {'A': 0.12, 'C': 0.12, 'T': 0.12, 'G': 0.12}\n\
 >>> tb.bases(tb, \"chr1\", 24, 74, True)\n\
 {'A': 6, 'C': 6, 'T': 6, 'G': 6}\n\
+>>> tb.close()"},
+    {"hardMaskedBlocks", (PyCFunction)py2bitHardMaskedBlocks, METH_VARARGS|METH_KEYWORDS,
+"Retrieve a list of hard-masked blocks on a single-chromosome (or range on it).\n\
+\n\
+Positional arguments:\n\
+    chr:   Chromosome name\n\
+\n\
+Optional keyword arguments:\n\
+    start: Starting position (0-based)\n\
+    end:   Ending position (1-based)\n\
+\n\
+Returns:\n\
+    A list of tuples, with items start and end.\n\
+\n\
+>>> import py2bit\n\
+>>> tb = py2bit.open(\"test/test.2bit\")\n\
+>>> print(tb.hardMaskedBlocks(\"chr1\")\n\
+>>> \n\
 >>> tb.close()"},
     {"__enter__", (PyCFunction) py2bitEnter, METH_NOARGS, NULL},
     {"__exit__", (PyCFunction) py2bitClose, METH_VARARGS, NULL},

py2bitTest/test.py

@@ -46,3 +46,13 @@ def testSequence(self):
         tb = py2bit.open(self.fname, True)
         assert(tb.sequence("chr1", 1, 3) == "NN")
         assert(tb.sequence("chr1", 1, 2) == "N")
+        tb.close()
+
+    def testHardMaskedBlocks(self):
+        tb = py2bit.open(self.fname, True)
+        assert(tb.hardMaskedBlocks("chr1") == [(0, 50), (100, 150)])
+        assert(tb.hardMaskedBlocks("chr1", 25, 75) == [(0, 50)])
+        assert(tb.hardMaskedBlocks("chr1", 75, 100) == [])
+        assert(tb.hardMaskedBlocks("chr1", 75, 101) == [(100, 150)])
+        assert(tb.hardMaskedBlocks("chr2") == [(50, 100)])
+        tb.close()

setup.py

@@ -17,13 +17,13 @@
                     include_dirs = ['lib2bit', sysconfig.get_config_var("INCLUDEPY")])
 
 setup(name = 'py2bit',
-       version = '0.2.2',
+       version = '0.3.0',
        description = 'A package for accessing 2bit files using lib2bit',
        author = "Devon P. Ryan",
        author_email = "ryan@ie-freiburg.mpg.de",
-       url = "https://github.com/dpryan79/py2bit",
+       url = "https://github.com/deeptools/py2bit",
        license = "MIT",
-       download_url = "https://github.com/dpryan79/py2bit/tarball/0.2.2",
+       download_url = "https://github.com/deeptools/py2bit/tarball/0.3.0",
        keywords = ["bioinformatics", "2bit"],
        classifier = ["Development Status :: 5 - Production/Stable",
                      "Intended Audience :: Developers",