IMP  2.2.0
The Integrative Modeling Platform
rigid_brownian_dynamics.py
1 ## \example atom/rigid_brownian_dynamics.py
2 # This example shows how to run brownian dynamics with rigid bodies.
3 
4 import IMP.atom
5 import IMP.core
6 import IMP.algebra
7 import IMP.display
8 
9 
10 def create_rigid_body(m, name):
11  prb = IMP.core.RigidBody.setup_particle(
12  IMP.kernel.Particle(m), IMP.algebra.ReferenceFrame3D())
13  prb.set_coordinates_are_optimized(True)
14  prb.set_name(name + " rb")
15  ph = IMP.atom.Molecule.setup_particle(IMP.kernel.Particle(m))
16  ph.set_name(name)
17  for i in range(0, 2):
18  for j in range(0, 2):
19  for k in range(0, 2):
20  d = IMP.core.XYZR.setup_particle(IMP.kernel.Particle(m),
21  IMP.algebra.Sphere3D(
22  IMP.algebra.Vector3D(
23  i, j, k) * 10.0,
24  10))
25  ph.add_child(IMP.atom.Fragment.setup_particle(d))
26  IMP.atom.Mass.setup_particle(d, 1000)
27  prb.add_member(d)
28  d.set_name(name + str(i) + str(j) + str(k))
29  d = IMP.atom.RigidBodyDiffusion.setup_particle(prb)
30  d.set_rotational_diffusion_coefficient(
31  d.get_rotational_diffusion_coefficient() * 100)
32  return prb, ph
33 
34 
35 def display(i, w, hs):
36  w.set_frame(i)
37  for h in hs:
38  g = IMP.atom.HierarchyGeometry(h)
39  w.add_geometry(g)
40 
41 m = IMP.kernel.Model()
42 IMP.base.set_log_level(IMP.base.SILENT)
43 
44 rb0, h0 = create_rigid_body(m, "first")
45 rb1, h1 = create_rigid_body(m, "second")
46 rb1.set_reference_frame(IMP.algebra.ReferenceFrame3D(
47  IMP.algebra.Transformation3D(IMP.algebra.Vector3D(30, 0, 0))))
48 
49 ev = IMP.core.ExcludedVolumeRestraint(
50  IMP.container.ListSingletonContainer(
51  rb0.get_members() + rb1.get_members()),
52  1, 3)
53 m.add_restraint(ev)
54 
55 #h= IMP.core.Harmonic(0,1)
56 #s= IMP.core.DistanceToSingletonScore(h, IMP.algebra.Vector3D(0,0,0))
57 #r= IMP.core.SingletonRestraint(s, rb0.get_member(0))
58 # m.add_restraint(r)
59 cr = IMP.atom.create_distance_restraint(
60  IMP.atom.Selection(h0), IMP.atom.Selection(h1), 0, 1)
61 m.add_restraint(cr)
62 
63 bd = IMP.atom.BrownianDynamics(m)
64 bd.set_time_step(10000)
65 
66 nm = IMP.base.create_temporary_file_name("rigid_bd", ".pym")
67 nm = "rigid.pym"
68 w = IMP.display.PymolWriter(nm)
69 for i in range(0, 100):
70  display(i, w, [h0, h1])
71  bd.optimize(10)
IMP::algebra::Transformation3D
Simple 3D transformation class.
Definition: Transformation3D.h:30
IMP::core::RigidBody::setup_particle
static RigidBody setup_particle(kernel::Model *m, ParticleIndex pi, kernel::ParticleIndexesAdaptor ps)
Definition: rigid_bodies.h:168
IMP::base::set_log_level
void set_log_level(LogLevel l)
Set the current global log level.
IMP::base::create_temporary_file_name
std::string create_temporary_file_name(std::string prefix="imp_temp", std::string suffix="")
IMP::atom::create_distance_restraint
kernel::Restraint * create_distance_restraint(const Selection &n0, const Selection &n1, double x0, double k, std::string name="Distance%1%")
IMP::algebra::ReferenceFrame3D
A reference frame in 3D.
Definition: ReferenceFrame3D.h:20
IMP::core::XYZR::setup_particle
static XYZR setup_particle(kernel::Model *m, ParticleIndex pi)
Definition: XYZR.h:48
IMP::atom::BrownianDynamics
Simple Brownian dynamics simulator.
Definition: BrownianDynamics.h:67
IMP::atom::Molecule::setup_particle
static Molecule setup_particle(kernel::Model *m, ParticleIndex pi)
Definition: Molecule.h:37
IMP::container::ListSingletonContainer
Store a kernel::ParticleIndexes.
Definition: ListSingletonContainer.h:33
IMP::kernel::Particle
Class to handle individual model particles.
Definition: kernel/Particle.h:34
IMP::atom::Mass::setup_particle
static Mass setup_particle(kernel::Model *m, ParticleIndex pi, Float mass)
Definition: Mass.h:44
IMP::core
See IMP.core for more information.
Definition: AngleRestraint.h:19
IMP::algebra
See IMP.algebra for more information.
Definition: algebra_config.h:113
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::display::PymolWriter
Write a CGO file with the geometry.
Definition: PymolWriter.h:34
IMP::atom::Fragment::setup_particle
static Fragment setup_particle(kernel::Model *m, ParticleIndex pi)
Definition: Fragment.h:63
IMP::display
See IMP.display for more information.
Definition: ChimeraWriter.h:17
IMP::atom
See IMP.atom for more information.
Definition: angle_decorators.h:15
IMP::atom::create_rigid_body
IMP::core::RigidBody create_rigid_body(const Hierarchies &h, std::string name=std::string("created rigid body"))
Rigidify a molecule or collection of molecules.
IMP::atom::RigidBodyDiffusion::setup_particle
static RigidBodyDiffusion setup_particle(kernel::Model *m, ParticleIndex pi)
Definition: Diffusion.h:82
IMP::atom::HierarchyGeometry
Display an IMP::atom::Hierarchy particle as balls.
Definition: hierarchy_tools.h:187
IMP::kernel::Model
Class for storing model, its restraints, constraints, and particles.
Definition: kernel/Model.h:72