Fixed Python wrapper for CustomIntegrator::getPerDofVariable()

d2a6f00c · Peter Eastman · 7b383c7e · d2a6f00c
Commit d2a6f00c authored Feb 25, 2012 by Peter Eastman
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

wrappers/python/src/swig_doxygen/swig_lib/python/extend.i wrappers/python/src/swig_doxygen/swig_lib/python/extend.i +8 -0

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
@@ -173,3 +173,11 @@
        self.this = system.this
  }
 }
+
+%extend OpenMM::CustomIntegrator {
+    PyObject* getPerDofVariable(int index) const {
+        std::vector<Vec3> values;
+        self->getPerDofVariable(index, values);
+        return copyVVec3ToList(values);
+    }
+}
\ No newline at end of file