core/randomize_rigid_body.py

This fragments shows how to either perturb or set the orientation of a rigid body randomly.

## \example core/randomize_rigid_body.py
# This fragments shows how to either perturb or set the orientation of a rigid
# body randomly.

import IMP.core
import IMP.algebra
import IMP
from IMP.algebra import ReferenceFrame3D
from IMP.algebra import Transformation3D

m = IMP.kernel.Model()
p = IMP.kernel.Particle(m)
rbd = IMP.core.RigidBody.setup_particle(p, ReferenceFrame3D())
translation = IMP.algebra.get_random_vector_in(
    IMP.algebra.get_unit_bounding_box_3d())

# we don't yet have python code to generate a nearby rotation
rotation = IMP.algebra.get_random_rotation_3d()
transformation = Transformation3D(rotation, translation)
# Option 1:
# note, this overwrites the existing position
# The True is to transform the members now rather than wait for a
# score state
rbd.set_reference_frame(ReferenceFrame3D(transformation))
# Option 2:
# perturb the existing transformation
composed_tr = IMP.algebra.compose \
    (rbd.get_reference_frame().get_transformation_to(),
     transformation)
rbd.set_reference_frame(ReferenceFrame3D(composed_tr))
# Alternative to Option 2:
IMP.core.transform(rbd, transformation)