Implement __copy__() and __deepcopy__() for Forces

Implement copy() and deepcopy() for Forces
681ab7ff · Peter Eastman · dea037c3 · 681ab7ff
Commit 681ab7ff authored Apr 04, 2012 by Peter Eastman
Show whitespace changes
Inline Side-by-side

Showing with 13 additions and 1 deletion

wrappers/python/src/swig_doxygen/swig_lib/python/extend.i wrappers/python/src/swig_doxygen/swig_lib/python/extend.i +13 -1

No files found.
--- a/wrappers/python/src/swig_doxygen/swig_lib/python/extend.i
+++ b/wrappers/python/src/swig_doxygen/swig_lib/python/extend.i
@@ -202,3 +202,15 @@
        return copyVVec3ToList(values);
    }
 }
+
+%extend OpenMM::Force {
+  %pythoncode {
+    def __copy__(self):
+        copy = self.__class__.__new__(self.__class__)
+        copy.__init__(self)
+        return copy
+
+    def __deepcopy__(self, memo):
+        return self.__copy__()
+  }
+}