Function intra_predict

ops/cpp_modules/src/cpp_modules.cpp:248–285

}

py::array_t<float> intra_predict(py::array_t<int> seg_idx, py::array_t<float> model_param, py::array_t<float> transform_map) {
    // input array
    py::buffer_info si_buf = seg_idx.request();
    int *si_ptr = (int *) si_buf.ptr;
    py::buffer_info mp_buf = model_param.request();
    float *mp_ptr = (float *) mp_buf.ptr;
    py::buffer_info tm_buf = transform_map.request();
    float *tm_ptr = (float *) tm_buf.ptr;
    int h = si_buf.shape[0];
    int w = si_buf.shape[1];

    // return array
    auto pred_range_image = py::array_t<float>(si_buf.size);
    py::buffer_info pred_buf = pred_range_image.request();
    float *pred_ptr = (float *) pred_buf.ptr;

    for (int h_i = 0; h_i < h; h_i++) {
        for (int w_i = 0; w_i < w; w_i++){
            int model_idx = si_ptr[h_i * w + w_i];
            float param[4] = {mp_ptr[model_idx * 4],
                                       mp_ptr[model_idx * 4 + 1],
                                       mp_ptr[model_idx * 4 + 2],
                                       mp_ptr[model_idx * 4 + 3]};
            if (param[0] + param[1] + param[2] == 0){
                pred_ptr[h_i * w + w_i] = param[3];
            }
            else{
                float r_plane = -param[3] / (param[0] * tm_ptr[(h_i * w + w_i) * 3] +
                                             param[1] * tm_ptr[(h_i * w + w_i) * 3 + 1] +
                                             param[2] * tm_ptr[(h_i * w + w_i) * 3 + 2]);
                pred_ptr[h_i * w + w_i] = r_plane;
            }
        }
    }

    pred_range_image.resize({h, w, 1});
    return pred_range_image;
}


py::array_t<int> uniform_quantize(py::array_t<int> seg_idx, py::array_t<float> residual, float acc) {