CarmaSparseObjInterfacer Struct Reference

#include <sparse_obj.hpp>

Collaboration diagram for CarmaSparseObjInterfacer:

Static Public Member Functions
template<typename T >
static void	call (py::module &mod)

Detailed Description

Definition at line 56 of file sparse_obj.hpp.

Member Function Documentation

call()

template<typename T >

static void CarmaSparseObjInterfacer::call ( py::module &  mod )

static

Definition at line 57 of file sparse_obj.hpp.

                                                        {
    auto name = appendName<T>("sparse_obj_");
    using Class = CarmaSparseObj<T>;
 
    py::class_<Class>(mod, name.data(), py::buffer_protocol())
    .def("get_csr",[](Class &frame){
        py::object CSR = py::module::import("scipy.sparse").attr("csr_matrix");
        int dim1 = frame.get_dims(1);
        int dim2 = frame.get_dims(2);
        int nnz = frame.nz_elem;
 
        std::vector<int> rowind = std::vector<int>(dim1 + 1);
        std::vector<int> colind = std::vector<int>(nnz);
        std::vector<T> data = std::vector<T>(nnz);
 
        frame.sparse_to_host(rowind.data(), colind.data(), data.data());
        py::tuple csrMat = py::make_tuple(py::array(data.size(),data.data()), py::array(colind.size(),colind.data()),py::array(rowind.size(), rowind.data()));
        py::tuple shape = py::make_tuple(dim1,dim2);
 
 
        return CSR(csrMat,py::arg("shape")=shape);
        });
}

---

The documentation for this struct was generated from the following file:

• sparse_obj.hpp