Merge pull request #383 from cmastalli/topic/copyable

Extended copyable visitor
stack-of-tasks · Jul 31, 2023 · 7bddd6a · 7bddd6a
2 parents ab8c362 + f3edc2e
commit 7bddd6a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/include/eigenpy/copyable.hpp b/include/eigenpy/copyable.hpp
@@ -1,5 +1,6 @@
 //
-// Copyright (c) 2016-2021 CNRS INRIA
+// Copyright (c) 2016-2023 CNRS INRIA
+// Copyright (c) 2023 Heriot-Watt University
 //
 
 #ifndef __eigenpy_utils_copyable_hpp__
@@ -18,10 +19,14 @@ struct CopyableVisitor : public bp::def_visitor<CopyableVisitor<C> > {
   template <class PyClass>
   void visit(PyClass& cl) const {
     cl.def("copy", &copy, bp::arg("self"), "Returns a copy of *this.");
+    cl.def("__copy__", &copy, bp::arg("self"), "Returns a copy of *this.");
+    cl.def("__deepcopy__", &deepcopy, bp::args("self", "memo"),
+           "Returns a deep copy of *this.");
   }
 
  private:
   static C copy(const C& self) { return C(self); }
+  static C deepcopy(const C& self, bp::dict) { return C(self); }
 };
 }  // namespace eigenpy