Program Listing for File pickle-vector.hpp

↰ Return to documentation for file (include/pinocchio/bindings/python/utils/pickle-vector.hpp)

//
// Copyright (c) 2019-2020 CNRS INRIA
//

#ifndef __pinocchio_python_utils_pickle_vector_hpp__
#define __pinocchio_python_utils_pickle_vector_hpp__

#include <boost/python.hpp>
#include <boost/python/tuple.hpp>
#include <boost/python/stl_iterator.hpp>

namespace pinocchio
{
  namespace python
  {
    template<typename VecType>
    struct PickleVector : boost::python::pickle_suite
    {
      static boost::python::tuple getinitargs(const VecType&)
      { return boost::python::make_tuple(); }

      static boost::python::tuple getstate(boost::python::object op)
      {
        return boost::python::make_tuple(boost::python::list(boost::python::extract<const VecType&>(op)()));
      }

      static void setstate(boost::python::object op, boost::python::tuple tup)
      {
        if(boost::python::len(tup) > 0)
        {
          VecType & o = boost::python::extract<VecType&>(op)();
          boost::python::stl_input_iterator<typename VecType::value_type> begin(tup[0]), end;
          while(begin != end)
          {
             o.push_back(*begin);
             ++begin;
          }
        }
      }

      static bool getstate_manages_dict() { return true; }
    };
  }
}

#endif // ifndef __pinocchio_python_utils_pickle_vector_hpp__