scipy-type.hpp

Go to the documentation of this file.

/*
 * Copyright 2024 INRIA
 */
 
#ifndef __eigenpy_scipy_type_hpp__
#define __eigenpy_scipy_type_hpp__
 
#include "eigenpy/fwd.hpp"
#include "eigenpy/register.hpp"
#include "eigenpy/scalar-conversion.hpp"
#include "eigenpy/numpy-type.hpp"
 
namespace eigenpy {
 
struct EIGENPY_DLLAPI ScipyType {
  static ScipyType& getInstance();
 
  static void sharedMemory(const bool value);
 
  static bool sharedMemory();
 
  static bp::object getScipyType();
 
  static const PyTypeObject* getScipyCSRMatrixType();
  static const PyTypeObject* getScipyCSCMatrixType();
 
  template <typename SparseMatrix>
  static bp::object get_pytype_object(
      const Eigen::SparseMatrixBase<SparseMatrix>* ptr = nullptr) {
    EIGENPY_UNUSED_VARIABLE(ptr);
    return SparseMatrix::IsRowMajor ? getInstance().csr_matrix_obj
                                    : getInstance().csc_matrix_obj;
  }
 
  template <typename SparseMatrix>
  static PyTypeObject const* get_pytype(
      const Eigen::SparseMatrixBase<SparseMatrix>* ptr = nullptr) {
    EIGENPY_UNUSED_VARIABLE(ptr);
    return SparseMatrix::IsRowMajor ? getInstance().csr_matrix_type
                                    : getInstance().csc_matrix_type;
  }
 
  static int get_numpy_type_num(const bp::object& obj) {
    const PyTypeObject* type = Py_TYPE(obj.ptr());
    EIGENPY_USED_VARIABLE_ONLY_IN_DEBUG_MODE(type);
    assert(type == getInstance().csr_matrix_type ||
           type == getInstance().csc_matrix_type);
 
    bp::object dtype = obj.attr("dtype");
 
    const PyArray_Descr* npy_type =
        reinterpret_cast<PyArray_Descr*>(dtype.ptr());
    return npy_type->type_num;
  }
 
 protected:
  ScipyType();
 
  bp::object sparse_module;
 
  // SciPy types
  bp::object csr_matrix_obj, csc_matrix_obj;
  PyTypeObject *csr_matrix_type, *csc_matrix_type;
 
  bool shared_memory;
};
}  // namespace eigenpy
 
#endif  // ifndef __eigenpy_scipy_type_hpp__