joints-variant.hpp

//
// Copyright (c) 2015-2021 CNRS INRIA
//
 
#ifndef __pinocchio_python_joints_variant_hpp__
#define __pinocchio_python_joints_variant_hpp__
 
#include <boost/algorithm/string/replace.hpp>
 
#include <boost/python.hpp>
 
#include "pinocchio/multibody/joint/joint-collection.hpp"
#include "pinocchio/bindings/python/multibody/joint/joints-models.hpp"
#include "pinocchio/bindings/python/multibody/joint/joints-datas.hpp"
#include "pinocchio/bindings/python/utils/printable.hpp"
 
namespace pinocchio
{
  namespace python
  {
    namespace bp = boost::python;
 
    template<typename T>
    std::string sanitizedClassname()
    {
      std::string className = boost::replace_all_copy(T::classname(), "<", "_");
      boost::replace_all(className, ">", "");
      return className;
    }
 
    template<typename VariantType>
    struct JointVariantVisitor : boost::static_visitor<PyObject *>
    {
      static result_type convert(VariantType const & jv)
      {
        return apply_visitor(JointVariantVisitor<VariantType>(), jv);
      }
 
      template<typename T>
      result_type operator()(T const & t) const
      {
        return boost::python::incref(boost::python::object(t).ptr());
      }
    };
 
    struct JointDataExposer
    {
      template<class T>
      void operator()(T)
      {
        expose_joint_data<T>(
          bp::class_<T>(
            sanitizedClassname<T>().c_str(), sanitizedClassname<T>().c_str(), bp::init<>())
            .def(JointDataBasePythonVisitor<T>())
            .def(PrintableVisitor<T>()));
        bp::implicitly_convertible<T, context::JointData>();
      }
    };
 
    struct JointModelExposer
    {
      template<class T>
      void operator()(T)
      {
        expose_joint_model<T>(
          bp::class_<T>(
            sanitizedClassname<T>().c_str(), sanitizedClassname<T>().c_str(), bp::no_init)
            .def(JointModelBasePythonVisitor<T>())
            .def(PrintableVisitor<T>()));
        bp::implicitly_convertible<T, context::JointModel>();
      }
    };
 
  } // namespace python
} // namespace pinocchio
 
#endif // ifndef __pinocchio_python_joints_variant_hpp__