Function maxflow3d_wrapper

maxflow_python/maxflow.cpp:118–166

}

static PyObject *
maxflow3d_wrapper(PyObject *self, PyObject *args)
{
    PyObject *I=NULL, *P=NULL, *param=NULL;
    PyArrayObject *arr_I=NULL, *arr_P=NULL;
    if (!PyArg_ParseTuple(args, "OOO", &I, &P, &param)) return NULL;
    
    arr_I = (PyArrayObject*)PyArray_FROM_OTF(I, NPY_FLOAT32, NPY_IN_ARRAY);
    arr_P = (PyArrayObject*)PyArray_FROM_OTF(P, NPY_FLOAT32, NPY_IN_ARRAY);
    if (arr_I == NULL || arr_P == NULL) return NULL;
    float lambda = PyFloat_AsDouble(PyTuple_GET_ITEM(param, 0));
    float sigma  = PyFloat_AsDouble(PyTuple_GET_ITEM(param, 1));
    
    /*vv* code that makes use of arguments *vv*/
    int dimI = PyArray_NDIM(arr_I);             // number of dimensions
    int dimP = PyArray_NDIM(arr_P); 
    npy_intp * shapeI = PyArray_DIMS(arr_I);    // npy_intp array of length nd showing length in each dim
    npy_intp * shapeP = PyArray_DIMS(arr_P);

    if(dimI !=3 && dimI != 4){
        cout << "the input dimension can only be 3 or 4"<<endl;
        return NULL;
    }
    if(dimP != 4){
        cout << "dimension of probabilily map should be 4"<<endl;
        return NULL;
    }
    if(shapeI[0] != shapeP[0] || shapeI[1] != shapeP[1] || shapeI[2] != shapeP[2]){
        cout << "image and probability map have different sizes"<<endl;
        return NULL;
    }
    if(shapeP[3] != 2){
        cout << "probabilily map should have two channels"<<endl;
        return NULL;
    }
    
    int chns = 1;
    if(dimI == 4) chns = shapeI[3];
    npy_intp outshape[3] = {shapeI[0], shapeI[1], shapeI[2]};
    PyArrayObject * arr_L = (PyArrayObject*)  PyArray_SimpleNew(3, outshape, NPY_INT8);
    maxflow3d_inference((unsigned char *) arr_L->data, (const float *) arr_I->data, 
        (const float *) arr_P->data, NULL,
        shapeI[0], shapeI[1], shapeI[2], chns, 2, lambda, sigma);

    Py_DECREF(arr_I);
    Py_DECREF(arr_P);
    Py_INCREF(arr_L);
    return PyArray_Return(arr_L);
}

static PyObject *
interactive_maxflow3d_wrapper(PyObject *self, PyObject *args)