set(CMAKE_CXX_STANDARD 11)
find_package(PythonInterp 3.4 REQUIRED)
find_package(PythonLibs 3.4 REQUIRED)
find_package(SIP REQUIRED)
include(SIPMacros)

set(SIP_CONCAT_PARTS 4)
set(SIP_EXTRA_OPTIONS -e -o -y module.pyi)

set(SIP_SOURCES src/module.sip ...)

set(SIP_EXTRA_FILES_DEPEND ${SIP_SOURCES})
generate_sip_python_module_code(
    _module 
    src/module.sip
    py_cpp_files)
build_sip_python_module(
    _module
    src/module.sip
    "${py_cpp_files}")

list(GET py_cpp_files 0 PART0)
set_source_files_properties(${PART0}
    PROPERTIES COMPILE_DEFINITIONS SIP_IS_PART0)

%UnitPostIncludeCode
    #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
    #define PY_ARRAY_UNIQUE_SYMBOL module_ARRAY_API

    #ifndef SIP_IS_PART0
    #define NO_IMPORT_ARRAY
    #endif

    #include <numpy/arrayobject.h>
%End

%InitialisationCode
    import_array();
%End

Py_ssize_t documentCount = PyList_Size(docs);
std::unique_ptr<int[]> lengths { new int[documentCount] };
std::unique_ptr<int*[]> docPtrs { new int*[documentCount] };
PyObject *docsList = PyList_New(documentCount);

for (Py_ssize_t i = 0; i < documentCount; i++) {
    PyObject *item = PyList_GetItem(docs, i);
    PyObject *document = PyArray_FROMANY(
        item,
        NPY_UINT32,
        1, 1,
        NPY_ARRAY_C_CONTIGUOUS
    );
    if (document == nullptr) {
        sipIsErr = 1;
        break;
    }

    lengths[i] = PyArray_SHAPE((PyArrayObject*) document)[0];
    docPtrs[i] = (int*) PyArray_DATA((PyArrayObject*) document);

    PyList_SetItem(docsList, i, document);
}

sipKeepReference(sipSelf, (int) time(nullptr), docsList);

Py_BEGIN_ALLOW_THREADS
sipRes = sipCpp->init(
    documentCount,
    lengths.get(), docPtrs.get()
);
Py_END_ALLOW_THREADS

PyObject* wrapMatrix(
    double *matrix,
    int d1, int d2,
    PyObject *sipPySelf)
{
    npy_intp dims[] = { d1, d2 };

    return PyArray_SimpleNewFromData(2, dims, NPY_DOUBLE, matrix);
}