Commit 0ef5ec85 authored by peastman's avatar peastman
Browse files

Context.getIntegrator() returns the original Python object

parent 18a7cac0
...@@ -90,6 +90,7 @@ SKIP_METHODS = [('State', 'getPositions'), ...@@ -90,6 +90,7 @@ SKIP_METHODS = [('State', 'getPositions'),
('UpdateTimeKernel',), ('UpdateTimeKernel',),
('VdwInfo',), ('VdwInfo',),
('WcaDispersionInfo',), ('WcaDispersionInfo',),
('Context', 'getIntegrator'),
('Context', 'createCheckpoint'), ('Context', 'createCheckpoint'),
('Context', 'loadCheckpoint'), ('Context', 'loadCheckpoint'),
('CudaPlatform',), ('CudaPlatform',),
......
...@@ -6,6 +6,9 @@ ...@@ -6,6 +6,9 @@
%extend OpenMM::Context { %extend OpenMM::Context {
%pythoncode %{ %pythoncode %{
def getIntegrator(self):
return self._integrator
def getState(self, getPositions=False, getVelocities=False, def getState(self, getPositions=False, getVelocities=False,
getForces=False, getEnergy=False, getParameters=False, getForces=False, getEnergy=False, getParameters=False,
getParameterDerivatives=False, enforcePeriodicBox=False, groups=-1): getParameterDerivatives=False, enforcePeriodicBox=False, groups=-1):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment