Function PYBIND11_MODULE

pointersect/pr/cpp/pr.cpp:961–980

namespace py = pybind11;
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

    // The following functions should not be called in python.
    // They are exposed for unit-testing.
    py::class_<idxmap>(m, "idxmap")
        .def(py::init())
        .def(py::init<size_t>())
        .def(py::init<const Table &>())
        .def("insert", &idxmap::insert)
        .def("get_pidx", &idxmap::get_pidx)
        .def("get_table", &idxmap::get_table);

    m.def("sub2ind", &sub2ind, "Sub index to linear index");
    m.def("ind2sub", &ind2sub, "linear index to sub index");
    m.def("gather_points", &gather_points, "Gather points in each grid cell");
    m.def("get_grid_idx", &get_grid_idx, "Compute the grid index a xyz point belongs to");
    m.def("grid_ray_intersection", &grid_ray_intersection, "Find out what are the grid cells a ray intersects with");
    m.def("grid_ray_intersection_v2", &grid_ray_intersection_v2, "Find out what are the grid cells a ray intersects with");
    m.def("find_neighbor_points_of_rays", &find_neighbor_points_of_rays, "Find the neighboring points of a ray");
}