numpy-type.cpp

Go to the documentation of this file.

/*
 * Copyright 2018-2023 INRIA
 */
 
#include "eigenpy/numpy-type.hpp"
 
#include <patchlevel.h>  // For PY_MAJOR_VERSION
 
namespace eigenpy {
 
NumpyType& NumpyType::getInstance() {
  static NumpyType instance;
  return instance;
}
 
bp::object NumpyType::make(PyArrayObject* pyArray, bool copy) {
  return make((PyObject*)pyArray, copy);
}
 
bp::object NumpyType::make(PyObject* pyObj, bool /*copy*/) {
  bp::object m;
  m = bp::object(bp::handle<>(pyObj));  // nothing to do here
 
  Py_INCREF(m.ptr());
  return m;
}
 
void NumpyType::sharedMemory(const bool value) {
  getInstance().shared_memory = value;
}
 
bool NumpyType::sharedMemory() { return getInstance().shared_memory; }
 
const PyTypeObject* NumpyType::getNumpyArrayType() {
  return getInstance().NumpyArrayType;
}
 
NumpyType::NumpyType() {
  pyModule = bp::import("numpy");
 
#if PY_MAJOR_VERSION >= 3
  // TODO I don't know why this Py_INCREF is necessary.
  // Without it, the destructor of NumpyType SEGV sometimes.
  Py_INCREF(pyModule.ptr());
#endif
 
  NumpyArrayObject = pyModule.attr("ndarray");
  NumpyArrayType = reinterpret_cast<PyTypeObject*>(NumpyArrayObject.ptr());
 
  shared_memory = true;
}
}  // namespace eigenpy