register.hpp
Go to the documentation of this file.
1 //
2 // Copyright (c) 2020 INRIA
3 //
4 
5 #ifndef __eigenpy_register_hpp__
6 #define __eigenpy_register_hpp__
7 
8 #include <algorithm>
9 #include <map>
10 #include <string>
11 #include <typeinfo>
12 
13 #include "eigenpy/exception.hpp"
14 #include "eigenpy/fwd.hpp"
15 #include "eigenpy/numpy.hpp"
16 
17 namespace eigenpy {
18 
20 struct EIGENPY_DLLAPI Register {
21  static PyArray_Descr *getPyArrayDescr(PyTypeObject *py_type_ptr);
22 
23  static PyArray_Descr *getPyArrayDescrFromTypeNum(const int type_num);
24 
25  template <typename Scalar>
26  static PyArray_Descr *getPyArrayDescrFromScalarType() {
27  if (!isNumpyNativeType<Scalar>()) {
28  const std::type_info &info = typeid(Scalar);
29  if (instance().type_to_py_type_bindings.find(&info) !=
30  instance().type_to_py_type_bindings.end()) {
31  PyTypeObject *py_type = instance().type_to_py_type_bindings[&info];
32  return instance().py_array_descr_bindings[py_type];
33  } else
34  return nullptr;
35  } else {
36  PyArray_Descr *new_descr =
38  return new_descr;
39  }
40  }
41 
42  template <typename Scalar>
43  static bool isRegistered() {
44  return isRegistered(Register::getPyType<Scalar>());
45  }
46 
47  static bool isRegistered(PyTypeObject *py_type_ptr);
48 
49  static int getTypeCode(PyTypeObject *py_type_ptr);
50 
51  template <typename Scalar>
52  static PyTypeObject *getPyType() {
53  if (!isNumpyNativeType<Scalar>()) {
54  const PyTypeObject *const_py_type_ptr =
55  bp::converter::registered_pytype<Scalar>::get_pytype();
56  if (const_py_type_ptr == NULL) {
57  std::stringstream ss;
58  ss << "The type " << typeid(Scalar).name()
59  << " does not have a registered converter inside Boot.Python."
60  << std::endl;
61  throw std::invalid_argument(ss.str());
62  }
63  PyTypeObject *py_type_ptr = const_cast<PyTypeObject *>(const_py_type_ptr);
64  return py_type_ptr;
65  } else {
66  PyArray_Descr *new_descr =
68  return new_descr->typeobj;
69  }
70  }
71 
72  template <typename Scalar>
73  static PyArray_Descr *getPyArrayDescr() {
74  if (!isNumpyNativeType<Scalar>()) {
75  return getPyArrayDescr(getPyType<Scalar>());
76  } else {
78  }
79  }
80 
81  template <typename Scalar>
82  static int getTypeCode() {
83  if (isNumpyNativeType<Scalar>())
85  else {
86  const std::type_info &info = typeid(Scalar);
87  if (instance().type_to_py_type_bindings.find(&info) !=
88  instance().type_to_py_type_bindings.end()) {
89  PyTypeObject *py_type = instance().type_to_py_type_bindings[&info];
90  int code = instance().py_array_code_bindings[py_type];
91 
92  return code;
93  } else
94  return -1; // type not registered
95  }
96  }
97 
98  static int registerNewType(
99  PyTypeObject *py_type_ptr, const std::type_info *type_info_ptr,
100  const int type_size, const int alignment, PyArray_GetItemFunc *getitem,
101  PyArray_SetItemFunc *setitem, PyArray_NonzeroFunc *nonzero,
102  PyArray_CopySwapFunc *copyswap, PyArray_CopySwapNFunc *copyswapn,
103  PyArray_DotFunc *dotfunc, PyArray_FillFunc *fill,
104  PyArray_FillWithScalarFunc *fillwithscalar);
105 
106  static Register &instance();
107 
108  private:
109  Register() {};
110 
112  bool operator()(const PyTypeObject *a, const PyTypeObject *b) const {
113  return std::string(a->tp_name) < std::string(b->tp_name);
114  }
115  };
116 
118  bool operator()(const std::type_info *a, const std::type_info *b) const {
119  return std::string(a->name()) < std::string(b->name());
120  }
121  };
122 
123  typedef std::map<const std::type_info *, PyTypeObject *, Compare_TypeInfo>
126 
127  typedef std::map<PyTypeObject *, PyArray_Descr *, Compare_PyTypeObject>
130 
131  typedef std::map<PyTypeObject *, int, Compare_PyTypeObject> MapCode;
133 };
134 
135 } // namespace eigenpy
136 
137 #endif // __eigenpy_register_hpp__
eigenpy::Register::getPyArrayDescrFromScalarType
static PyArray_Descr * getPyArrayDescrFromScalarType()
Definition: register.hpp:26
fill
void fill(Eigen::Ref< MatType > mat, const typename MatType::Scalar &value)
Definition: eigen_ref.cpp:62
eigenpy::Register::MapDescr
std::map< PyTypeObject *, PyArray_Descr *, Compare_PyTypeObject > MapDescr
Definition: register.hpp:128
eigenpy::call_PyArray_DescrFromType
PyArray_Descr * call_PyArray_DescrFromType(int typenum)
Definition: numpy.hpp:251
fwd.hpp
eigenpy::Register::isRegistered
static bool isRegistered()
Definition: register.hpp:43
eigenpy::Register::MapCode
std::map< PyTypeObject *, int, Compare_PyTypeObject > MapCode
Definition: register.hpp:131
eigenpy::Register::type_to_py_type_bindings
MapInfo type_to_py_type_bindings
Definition: register.hpp:125
eigenpy::Register::Register
Register()
Definition: register.hpp:109
eigenpy::Register
Structure collecting all the types registers in Numpy via EigenPy.
Definition: register.hpp:20
bench-switch.a
list a
Definition: bench-switch.py:12
eigenpy
Definition: alignment.hpp:14
eigenpy::Register::Compare_TypeInfo::operator()
bool operator()(const std::type_info *a, const std::type_info *b) const
Definition: register.hpp:118
eigenpy::Register::Compare_TypeInfo
Definition: register.hpp:117
eigenpy::Register::getPyArrayDescr
static PyArray_Descr * getPyArrayDescr()
Definition: register.hpp:73
eigenpy::NumpyEquivalentType
Definition: numpy.hpp:59
eigenpy::Register::Compare_PyTypeObject
Definition: register.hpp:111
setup.name
name
Definition: setup.in.py:179
eigenpy::Register::getTypeCode
static int getTypeCode()
Definition: register.hpp:82
numpy.hpp
eigenpy::Register::MapInfo
std::map< const std::type_info *, PyTypeObject *, Compare_TypeInfo > MapInfo
Definition: register.hpp:124
eigenpy::registerNewType
int registerNewType(PyTypeObject *py_type_ptr=NULL)
Definition: user-type.hpp:306
exception.hpp
eigenpy::Register::py_array_code_bindings
MapCode py_array_code_bindings
Definition: register.hpp:132
eigenpy::Register::getPyType
static PyTypeObject * getPyType()
Definition: register.hpp:52
eigenpy::Register::py_array_descr_bindings
MapDescr py_array_descr_bindings
Definition: register.hpp:129
eigenpy::Register::Compare_PyTypeObject::operator()
bool operator()(const PyTypeObject *a, const PyTypeObject *b) const
Definition: register.hpp:112


eigenpy
Author(s): Justin Carpentier, Nicolas Mansard
autogenerated on Fri Jun 14 2024 02:15:58