Context¶
- class openmm.openmm.Context(*args)¶
A Context stores the complete state of a simulation. More specifically, it includes:
The current time
The position of each particle
The velocity of each particle
The values of configurable parameters defined by Force objects in the System
You can retrieve a snapshot of the current state at any time by calling getState(). This allows you to record the state of the simulation at various points, either for analysis or for checkpointing. getState() can also be used to retrieve the current forces on each particle and the current energy of the System.
- __init__(self, system, integrator) → Context¶
- __init__(self, system, integrator, platform) → Context
- __init__(self, system, integrator, platform, properties) → Context
- __init__(self, other) → Context
Construct a new Context in which to run a simulation, explicitly specifying what Platform should be used to perform calculations and the values of platform-specific properties.
- Parameters
system (System) – the System which will be simulated
integrator (Integrator) – the Integrator which will be used to simulate the System
platform (Platform) – the Platform to use for calculations
properties (map< std::string, std::string >) – a set of values for platform-specific properties. Keys are the property names.
Methods
__init__(-> Context -> Context -> Context)Construct a new Context in which to run a simulation, explicitly specifying what Platform should be used to perform calculations and the values of platform-specific properties.
applyConstraints(self, tol)Update the positions of particles so that all distance constraints are satisfied.
applyVelocityConstraints(self, tol)Update the velocities of particles so the net velocity of each constrained distance is zero.
computeVirtualSites(self)Recompute the locations of all virtual sites.
createCheckpoint(self)Create a checkpoint recording the current state of the Context.
getIntegrator()getMolecules(self)Get a description of how the particles in the system are grouped into molecules.
getParameter(self, name)Get the value of an adjustable parameter defined by a Force object in the System.
getParameters(self)Get all adjustable parameters that have been defined by Force objects in the System, along with their current values.
getPlatform(-> Platform)Get the Platform being used for calculations.
getState([getPositions, getVelocities, …])Get a State object recording the current state information stored in this context.
getStepCount(self)Get the current step count.
getSystem(self)Get System being simulated in this context.
getTime(self)Get the current time of the simulation (in picoseconds).
loadCheckpoint(self, checkpoint)Load a checkpoint that was written by createCheckpoint().
reinitialize(self[, preserveState])When a Context is created, it caches information about the System being simulated and the Force objects contained in it.
setParameter(self, name, value)Set the value of an adjustable parameter defined by a Force object in the System.
setPeriodicBoxVectors(self, a, b, c)Set the vectors defining the axes of the periodic box (measured in nm).
setPositions(self, positions)Set the positions of all particles in the System (measured in nm).
setState(self, state)Copy information from a State object into this Context.
setStepCount(self, count)Set the current step count.
setTime(self, time)Set the current time of the simulation (in picoseconds).
setVelocities(self, velocities)Set the velocities of all particles in the System (measured in nm/picosecond).
setVelocitiesToTemperature(self, temperature)Set the velocities of all particles in the System to random values chosen from a Boltzmann distribution at a given temperature.
Attributes
The membership flag
- property thisown¶
The membership flag
- getPlatform(self) → Platform¶
- getPlatform(self) → Platform
Get the Platform being used for calculations.
- setState(self, state)¶
Copy information from a State object into this Context. This restores the Context to approximately the same state it was in when the State was created. If the State does not include a piece of information (e.g. positions or velocities), that aspect of the Context is left unchanged.
Even when all possible information is included in the State, the effect of calling this method is still less complete than loadCheckpoint(). For example, it does not restore the internal states of random number generators. On the other hand, it has the advantage of not being hardware specific.
- getTime(self) → double¶
Get the current time of the simulation (in picoseconds).
- setTime(self, time)¶
Set the current time of the simulation (in picoseconds).
- getStepCount(self) → long long¶
Get the current step count.
- setStepCount(self, count)¶
Set the current step count.
- setPositions(self, positions)¶
Set the positions of all particles in the System (measured in nm). This method simply sets the positions without checking to see whether they satisfy distance constraints. If you want constraints to be enforced, call applyConstraints() after setting the positions.
- Parameters
positions (vector< Vec3 >) – a vector whose length equals the number of particles in the System. The i’th element contains the position of the i’th particle.
- setVelocities(self, velocities)¶
Set the velocities of all particles in the System (measured in nm/picosecond).
- Parameters
velocities (vector< Vec3 >) – a vector whose length equals the number of particles in the System. The i’th element contains the velocity of the i’th particle.
- setVelocitiesToTemperature(self, temperature, randomSeed=osrngseed())¶
Set the velocities of all particles in the System to random values chosen from a Boltzmann distribution at a given temperature.
- Parameters
temperature (double) – the temperature for which to select the velocities (measured in Kelvin)
randomSeed (int) – the random number seed to use when selecting velocities
- getParameters(self) → mapstringdouble¶
Get all adjustable parameters that have been defined by Force objects in the System, along with their current values.
- getParameter(self, name) → double¶
Get the value of an adjustable parameter defined by a Force object in the System.
- Parameters
name (string) – the name of the parameter to get
- setParameter(self, name, value)¶
Set the value of an adjustable parameter defined by a Force object in the System.
- Parameters
name (string) – the name of the parameter to set
value (double) – the value of the parameter
- setPeriodicBoxVectors(self, a, b, c)¶
Set the vectors defining the axes of the periodic box (measured in nm). They will affect any Force that uses periodic boundary conditions.
Triclinic boxes are supported, but the vectors must satisfy certain requirements. In particular, a must point in the x direction, b must point “mostly” in the y direction, and c must point “mostly” in the z direction. See the documentation for details.
- applyConstraints(self, tol)¶
Update the positions of particles so that all distance constraints are satisfied. This also recomputes the locations of all virtual sites.
- Parameters
tol (double) – the distance tolerance within which constraints must be satisfied.
- applyVelocityConstraints(self, tol)¶
Update the velocities of particles so the net velocity of each constrained distance is zero.
- Parameters
tol (double) – the velocity tolerance within which constraints must be satisfied.
- computeVirtualSites(self)¶
Recompute the locations of all virtual sites. There is rarely a reason to call this, since virtual sites are also updated by applyConstraints(). This is only for the rare situations when you want to enforce virtual sites but not constraints.
- reinitialize(self, preserveState=False)¶
When a Context is created, it caches information about the System being simulated and the Force objects contained in it. This means that, if the System or Forces are then modified, the Context does not see the changes. Call reinitialize() to force the Context to rebuild its internal representation of the System and pick up any changes that have been made.
This is an expensive operation, so you should try to avoid calling it too frequently. Most Force classes have an updateParametersInContext() method that provides a less expensive way of updating certain types of information. However, this method is the only way to make some types of changes, so it is sometimes necessary to call it.
By default, reinitializing a Context causes all state information (positions, velocities, etc.) to be discarded. You can optionally tell it to try to preserve state information. It does this by internally creating a checkpoint, then reinitializing the Context, then loading the checkpoint. Be aware that if the System has changed in a way that prevents the checkpoint from being loaded (such as changing the number of particles), this will throw an exception and the state information will be lost.
- getMolecules(self) → vectorii¶
Get a description of how the particles in the system are grouped into molecules. Two particles are in the same molecule if they are connected by constraints or bonds, where every Force object can define bonds in whatever way are appropriate to that force.
Each element lists the indices of all particles in a single molecule. Every particle is guaranteed to belong to exactly one molecule.
- getState(getPositions=False, getVelocities=False, getForces=False, getEnergy=False, getParameters=False, getParameterDerivatives=False, getIntegratorParameters=False, enforcePeriodicBox=False, groups=- 1)¶
Get a State object recording the current state information stored in this context.
- Parameters
getPositions (bool=False) – whether to store particle positions in the State
getVelocities (bool=False) – whether to store particle velocities in the State
getForces (bool=False) – whether to store the forces acting on particles in the State
getEnergy (bool=False) – whether to store potential and kinetic energy in the State
getParameters (bool=False) – whether to store context parameters in the State
getParameterDerivatives (bool=False) – whether to store parameter derivatives in the State
getIntegratorParameters (bool=False) – whether to store integrator parameters in the State
enforcePeriodicBox (bool=False) – if false, the position of each particle will be whatever position is stored in the Context, regardless of periodic boundary conditions. If true, particle positions will be translated so the center of every molecule lies in the same periodic box.
groups (set={0,1,2,...,31}) – a set of indices for which force groups to include when computing forces and energies. The default value includes all groups. groups can also be passed as an unsigned integer interpreted as a bitmask, in which case group i will be included if (groups&(1<<i)) != 0.
- createCheckpoint(self) → std::string¶
Create a checkpoint recording the current state of the Context. This should be treated as an opaque block of binary data. See loadCheckpoint() for more details.
Returns: a string containing the checkpoint data
- loadCheckpoint(self, checkpoint)¶
Load a checkpoint that was written by createCheckpoint().
A checkpoint contains not only publicly visible data such as the particle positions and velocities, but also internal data such as the states of random number generators. Ideally, loading a checkpoint should restore the Context to an identical state to when it was written, such that continuing the simulation will produce an identical trajectory. This is not strictly guaranteed to be true, however, and should not be relied on. For most purposes, however, the internal state should be close enough to be reasonably considered equivalent.
A checkpoint contains data that is highly specific to the Context from which it was created. It depends on the details of the System, the Platform being used, and the hardware and software of the computer it was created on. If you try to load it on a computer with different hardware, or for a System that is different in any way, loading is likely to fail. Checkpoints created with different versions of OpenMM are also often incompatible. If a checkpoint cannot be loaded, that is signaled by throwing an exception.
- Parameters:
checkpoint (string) the checkpoint data to load
