Function recover_map

ops/cpp_modules/src/cpp_modules.cpp:561–593

py::array_t<int> recover_map(py::array_t<int> contour_map, py::array_t<int> idx_sequence) {
    // input array
    py::buffer_info cm_buf = contour_map.request();
    int *cm_ptr = (int *) cm_buf.ptr;
    py::buffer_info is_buf = idx_sequence.request();
    int *is_ptr = (int *) is_buf.ptr;
    int h = cm_buf.shape[0];
    int w = cm_buf.shape[1];
    int l = is_buf.shape[0];

    // return array
    auto idx_map = py::array_t<int>(h * w);
    py::buffer_info im_buf = idx_map.request();
    int *im_ptr = (int *) im_buf.ptr;
    
    int pointer = 0;
    for (int i = 0; i < l; i++){
        int index = is_ptr[i];
        im_ptr[pointer] = index;
        pointer++;
        if (pointer >= h * w)
            break;
        while (cm_ptr[pointer] == 0){
            im_ptr[pointer] = index;
            pointer++;
            if (pointer >= h * w)
                break;
        }
    }

    idx_map.resize({h, w});
    return idx_map;
}


// feature extractor