Program Listing for File scipy-type.hpp

↰ Return to documentation for file (/tmp/ws/src/eigenpy/include/eigenpy/scipy-type.hpp)

/*
 * 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__