registration.hpp

Go to the documentation of this file.

/*
 * Copyright 2014-2019, CNRS
 * Copyright 2018-2019, INRIA
 */
 
#ifndef __eigenpy_registration_hpp__
#define __eigenpy_registration_hpp__
 
#include "eigenpy/fwd.hpp"
#include "eigenpy/registration_class.hpp"
 
namespace eigenpy {
 
template <typename T>
inline bool check_registration() {
  const bp::type_info info = bp::type_id<T>();
  const bp::converter::registration* reg = bp::converter::registry::query(info);
  if (reg == NULL)
    return false;
  else if ((*reg).m_to_python == NULL)
    return false;
 
  return true;
}
 
template <typename T>
inline bool register_symbolic_link_to_registered_type() {
  if (eigenpy::check_registration<T>()) {
    const bp::type_info info = bp::type_id<T>();
    const bp::converter::registration* reg =
        bp::converter::registry::query(info);
    bp::handle<> class_obj(reg->get_class_object());
    bp::incref(class_obj.get());
    bp::scope().attr(reg->get_class_object()->tp_name) = bp::object(class_obj);
    return true;
  }
 
  return false;
}
 
template <typename T, typename Visitor>
inline bool register_symbolic_link_to_registered_type(const Visitor& visitor) {
  if (eigenpy::check_registration<T>()) {
    const bp::type_info info = bp::type_id<T>();
    const bp::converter::registration* reg =
        bp::converter::registry::query(info);
    bp::handle<> class_obj(reg->get_class_object());
    bp::incref(class_obj.get());
    bp::object object(class_obj);
    bp::scope().attr(reg->get_class_object()->tp_name) = object;
    registration_class<T> cl(object);
    cl.def(visitor);
    return true;
  }
 
  return false;
}
}  // namespace eigenpy
 
#endif  // ifndef __eigenpy_registration_hpp__