IMP logo
IMP Reference Guide  2.6.1
The Integrative Modeling Platform
simulation.py
1 ## \example rmf/simulation.py
2 # This example shows writing a brownian dynamics simulation to a rmf file.
3 # It includes a bounding box, restraints and a hierarchy of particles.
4 
5 from __future__ import print_function, division
6 import IMP.atom
7 import IMP.rmf
8 import RMF
9 import IMP.container
10 import IMP.display
11 import sys
12 
13 IMP.setup_from_argv(sys.argv, "simulation")
14 
15 k = 10.0
16 np = 10
17 frames = 10
18 
19 m = IMP.Model()
20 bb = IMP.algebra.BoundingBox3D(IMP.algebra.Vector3D(0, 0, 0),
21  IMP.algebra.Vector3D(10, 10, 10))
22 
23 ps = [IMP.core.XYZR.setup_particle(IMP.Particle(m))
24  for i in range(0, np)]
25 h = IMP.atom.Hierarchy.setup_particle(IMP.Particle(m, "root"))
26 for p in ps:
27  p.set_radius(1)
28  p.set_coordinates_are_optimized(True)
29  IMP.atom.Mass.setup_particle(p, 30)
30  IMP.atom.Diffusion.setup_particle(p)
31  p.set_coordinates(IMP.algebra.get_random_vector_in(bb))
32 rs = []
33 link = IMP.core.HarmonicUpperBoundSphereDistancePairScore(0, k)
34 pl = []
35 for i in range(0, len(ps) // 2):
36  pp = (ps[2 * i], ps[2 * i + 1])
37  pl.append(pp)
38  # create OK staring position
39  pp[1].set_coordinates(IMP.algebra.get_random_vector_on(pp[0].get_sphere()))
40  IMP.display.Colored.setup_particle(pp[0], IMP.display.get_display_color(i))
41  IMP.display.Colored.setup_particle(pp[1], IMP.display.get_display_color(i))
42  hr = IMP.atom.Hierarchy.setup_particle(
43  IMP.Particle(m, "molecule " + str(i)))
44  hr.add_child(IMP.atom.Hierarchy.setup_particle(pp[0]))
45  hr.add_child(IMP.atom.Hierarchy.setup_particle(pp[1]))
46  h.add_child(hr)
47 r = IMP.container.PairsRestraint(link, pl, "Bonds")
48 rs.append(r)
49 
50 ev = IMP.core.ExcludedVolumeRestraint(ps, k, 1, "EV")
51 rs.append(ev)
52 
53 bbss = IMP.core.BoundingBox3DSingletonScore(
54  IMP.core.HarmonicUpperBound(0, k), bb)
55 bbr = IMP.container.SingletonsRestraint(bbss, ps)
56 rs.append(bbr)
57 
58 sf = IMP.core.RestraintsScoringFunction(rs, "SF")
59 
60 bd = IMP.atom.BrownianDynamics(m)
61 bd.set_log_level(IMP.SILENT)
62 bd.set_scoring_function(sf)
63 
64 bd.set_maximum_time_step(10)
65 
66 name = IMP.create_temporary_file_name("brownian", ".rmf")
67 rmf = RMF.create_rmf_file(name)
68 rmf.set_description("Brownian dyanmics trajectory with 10fs timestep.\n")
69 IMP.rmf.add_hierarchy(rmf, h)
70 IMP.rmf.add_restraints(rmf, rs)
71 IMP.rmf.add_geometry(rmf, IMP.display.BoundingBoxGeometry(bb))
72 
73 os = IMP.rmf.SaveOptimizerState(m, rmf)
74 os.update_always("initial conformation")
75 os.set_log_level(IMP.SILENT)
76 os.set_simulator(bd)
77 bd.add_optimizer_state(os)
78 
79 bd.optimize(frames)
80 
81 print("file is", rmf.get_name())
IMP::container::SingletonsRestraint
Applies a SingletonScore to each Singleton in a list.
Definition: SingletonsRestraint.h:34
IMP::core::HarmonicUpperBoundSphereDistancePairScore
A harmonic upper bound on the distance between two spheres.
Definition: SphereDistancePairScore.h:54
IMP::setup_from_argv
Strings setup_from_argv(const Strings &argv, std::string description, std::string positional_description, int num_positional)
IMP::container
Various classes to hold sets of particles.
IMP::core::HarmonicUpperBound
Upper bound harmonic function (non-zero when feature > mean)
Definition: core/HarmonicUpperBound.h:20
IMP::core::XYZR::setup_particle
static XYZR setup_particle(Model *m, ParticleIndex pi)
Definition: XYZR.h:48
IMP::core::GenericBoundingBox3DSingletonScore
Score particles based on how far outside a box they are.
Definition: BoundingBox3DSingletonScore.h:30
IMP::algebra::get_random_vector_in
Vector3D get_random_vector_in(const Cylinder3D &c)
Generate a random vector in a cylinder with uniform density.
IMP::Model
Class for storing model, its restraints, constraints, and particles.
Definition: Model.h:72
IMP::atom::Hierarchy::setup_particle
static Hierarchy setup_particle(Model *m, ParticleIndex pi, ParticleIndexesAdaptor children=ParticleIndexesAdaptor())
Create a Hierarchy of level t by adding the needed attributes.
Definition: atom/Hierarchy.h:255
IMP::rmf::add_hierarchy
void add_hierarchy(RMF::FileHandle fh, atom::Hierarchy hs)
IMP::atom::BrownianDynamics
Simple Brownian dynamics simulator.
Definition: BrownianDynamics.h:74
IMP::atom::Mass::setup_particle
static Mass setup_particle(Model *m, ParticleIndex pi, Float mass)
Definition: Mass.h:44
IMP::display::get_display_color
Color get_display_color(unsigned int i)
IMP::display::Colored::setup_particle
static Colored setup_particle(Model *m, ParticleIndex pi, Color color)
Definition: Colored.h:62
IMP::rmf::add_restraints
void add_restraints(RMF::NodeHandle fh, const Restraints &hs)
IMP::atom::Diffusion::setup_particle
static Diffusion setup_particle(Model *m, ParticleIndex pi, Float D)
Definition: Diffusion.h:50
IMP::algebra::Vector3D
VectorD< 3 > Vector3D
Definition: VectorD.h:395
IMP::core::ExcludedVolumeRestraint
Prevent a set of particles and rigid bodies from inter-penetrating.
Definition: ExcludedVolumeRestraint.h:40
IMP::Particle
Class to handle individual model particles.
Definition: Particle.h:37
IMP::rmf::add_geometry
void add_geometry(RMF::FileHandle file, display::Geometry *r)
Add a single geometry to the file.
IMP::algebra::get_random_vector_on
VectorD< D > get_random_vector_on(const SphereD< D > &s)
Generate a random vector on a sphere with uniform density.
Definition: vector_generators.h:100
IMP::display
Output IMP model data in various file formats.
IMP::atom
Functionality for loading, creating, manipulating and scoring atomic structures.
IMP::rmf
Support for the RMF file format for storing hierarchical molecular data and markup.
IMP::container::PairsRestraint
Applies a PairScore to each Pair in a list.
Definition: PairsRestraint.h:34
IMP::create_temporary_file_name
std::string create_temporary_file_name(std::string prefix="imp_temp", std::string suffix="")
Create a temporary file.