rmf/geometry.py

## \example rmf/geometry.py
# This example shows writing one piece of geometry to an RMF file and then
# reading it back.

from __future__ import print_function
import IMP.display
import IMP.rmf
import RMF
import sys

IMP.setup_from_argv(sys.argv, "geometry")

tfn = "rmf_geometry.rmf"

# open the RMF, clearing any existing contents
f = RMF.create_rmf_file(tfn)

# creating a box geometry
bb = IMP.algebra.BoundingBox3D(IMP.algebra.Vector3D(0, 0, 0),
                               IMP.algebra.Vector3D(10, 10, 10))
g = IMP.display.BoundingBoxGeometry(bb)

# add the geometry to the file
IMP.rmf.add_geometry(f, g)
IMP.rmf.save_frame(f, "zero")

bb = IMP.algebra.BoundingBox3D(IMP.algebra.Vector3D(1, 1, 1),
                               IMP.algebra.Vector3D(10, 10, 10))
g.set_geometry(bb)
# save a second frame with the bounding box
IMP.rmf.save_frame(f, "one")


del f
f = RMF.open_rmf_file_read_only(tfn)
# recreate the geometries from the file. The geometry will be the same
# but it will not be a IMP.display.BoundingBoxGeometry; it will be
# a set of cylinders instead.
gs = IMP.rmf.create_geometries(f)
IMP.rmf.load_frame(f, RMF.FrameID(0))
print(gs[0].get_name())
print("Try running rmf_display  on", tfn)

# load another frame
IMP.rmf.load_frame(f, RMF.FrameID(1))
# cast it to a BoundingBoxGeometry and print out the geometry
print(IMP.display.BoundingBoxGeometry.get_from(gs[0]).get_geometry())