IMP logo
IMP Reference Guide  2.12.0
The Integrative Modeling Platform
rmf/geometry.py
1 ## \example rmf/geometry.py
2 # This example shows writing one piece of geometry to an RMF file and then
3 # reading it back.
4 
5 from __future__ import print_function
6 import IMP.display
7 import IMP.rmf
8 import RMF
9 import sys
10 
11 IMP.setup_from_argv(sys.argv, "geometry")
12 
13 tfn = "rmf_geometry.rmf"
14 
15 # open the RMF, clearing any existing contents
16 f = RMF.create_rmf_file(tfn)
17 
18 # creating a box geometry
20  IMP.algebra.Vector3D(10, 10, 10))
22 
23 # add the geometry to the file
25 IMP.rmf.save_frame(f, "zero")
26 
28  IMP.algebra.Vector3D(10, 10, 10))
29 g.set_geometry(bb)
30 # save a second frame with the bounding box
31 IMP.rmf.save_frame(f, "one")
32 
33 
34 del f
35 f = RMF.open_rmf_file_read_only(tfn)
36 # recreate the geometries from the file. The geometry will be the same
37 # but it will not be a IMP.display.BoundingBoxGeometry; it will be
38 # a set of cylinders instead.
40 IMP.rmf.load_frame(f, RMF.FrameID(0))
41 print(gs[0].get_name())
42 print("Try running rmf_display on", tfn)
43 
44 # load another frame
45 IMP.rmf.load_frame(f, RMF.FrameID(1))
46 # cast it to a BoundingBoxGeometry and print out the geometry
47 print(IMP.display.BoundingBoxGeometry.get_from(gs[0]).get_geometry())
display::Geometries create_geometries(RMF::FileConstHandle parent)
Create geometry objects for the geometry nodes found in the file.
RMF::FrameID save_frame(RMF::FileHandle file, std::string name="")
Save the current state of the linked objects as a new RMF frame.
Strings setup_from_argv(const Strings &argv, std::string description, std::string positional_description, int num_positional)
void load_frame(RMF::FileConstHandle file, RMF::FrameID frame)
Load the given RMF frame into the state of the linked objects.
VectorD< 3 > Vector3D
Definition: VectorD.h:421
void add_geometry(RMF::FileHandle file, display::Geometry *r)
Add a single geometry to the file.
Output IMP model data in various file formats.
Support for the RMF file format for storing hierarchical molecular data and markup.