home
about
news
download
doc
source
systems
tests
bugs
contact
IMP Reference Guide
2.6.1
The Integrative Modeling Platform
IMP Manual
Reference Guide
Modules
Classes
Examples
version 2.6.1
display/basic_geometry.py
This example shows how to write basic geometric primitives to a file with color and name.
1
## \example display/basic_geometry.py
2
# This example shows how to write basic geometric primitives to a file
3
# with color and name.
4
5
from
__future__
import
print_function
6
import
IMP.display
7
import
sys
8
9
IMP.setup_from_argv
(sys.argv,
10
"Write basic geometric primitives to a file with color and name."
)
11
12
name =
IMP.create_temporary_file_name
(
"example"
,
".py"
)
13
print(
"File name is"
, name)
14
w =
IMP.display.PymolWriter
(name)
15
16
bb =
IMP.algebra.BoundingBox3D
(
IMP.algebra.Vector3D
(0, 0, 0),
17
IMP.algebra.Vector3D
(100, 100, 100))
18
# we could skip the outer loop if we only have one frame
19
for
f
in
range(0, 10):
20
w.set_frame(f)
21
g =
IMP.display.BoundingBoxGeometry
(bb)
22
g.set_name(
"bb"
)
23
w.add_geometry(g)
24
for
i
in
range(0, 10):
25
p =
IMP.algebra.get_random_vector_in
(bb)
26
g =
IMP.display.SphereGeometry
(
IMP.algebra.Sphere3D
(p, 10))
27
# give each a distinctive color
28
g.set_color(
IMP.display.get_display_color
(i))
29
g.set_name(str(i))
30
# add it to the file
31
w.add_geometry(g)