pickle.hpp

//
// Copyright (c) 2023 INRIA
//
 
#ifndef __pinocchio_python_utils_pickle_hpp__
#define __pinocchio_python_utils_pickle_hpp__
 
#include <pinocchio/bindings/python/fwd.hpp>
 
namespace pinocchio
{
  namespace python
  {
    namespace bp = boost::python;
 
    template<typename Derived>
    struct PickleFromStringSerialization : bp::pickle_suite
    {
      static bp::tuple getinitargs(const Derived &)
      {
        return bp::make_tuple();
      }
 
      static bp::tuple getstate(const Derived & obj)
      {
        const std::string str(obj.saveToString());
        return bp::make_tuple(bp::str(str));
      }
 
      static void setstate(Derived & obj, bp::tuple tup)
      {
        if (bp::len(tup) == 0 || bp::len(tup) > 1)
        {
          throw eigenpy::Exception(
            "Pickle was not able to reconstruct the object from the loaded data.\n"
            "The pickle data structure contains too many elements.");
        }
 
        bp::object py_obj = tup[0];
        boost::python::extract<std::string> obj_as_string(py_obj.ptr());
        if (obj_as_string.check())
        {
          const std::string str = obj_as_string;
          obj.loadFromString(str);
        }
        else
        {
          throw eigenpy::Exception(
            "Pickle was not able to reconstruct the model from the loaded data.\n"
            "The entry is not a string.");
        }
      }
 
      static bool getstate_manages_dict()
      {
        return true;
      }
    };
  } // namespace python
} // namespace pinocchio
 
#endif // ifndef __pinocchio_python_utils_pickle_hpp__