1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
## Automatically adapted for scipy Oct 31, 2005 by
# $Id: movie.py 1698 2006-03-14 23:12:10Z cookedm $
# ---------------------------------------------------------------------
#
# NAME: MOVIE.PY
#
# PURPOSE: Support functions for making animated sequences.
#
# CHANGES:
# 05/05/03 mdh time (in original yorick script) was changed to timing
# to avoid onflict with the time module, but two occurrences
# were missed.
#
# ---------------------------------------------------------------------
# Copyright (c) 1996, 1997, The Regents of the University of California.
# All rights reserved. See Legal.htm for full text and disclaimer.
from gist import *
from shapetest import *
from yorick import *
def movie (draw_frame, time_limit = 120., min_interframe = 0.0,
bracket_time = array ([2., 2.], Float ), lims = None, timing = 0) :
"""
movie (draw_frame, time_limit = 120., min_interframe = 0.0,
bracket_time = array ([2., 2.], Float ), lims = None, timing = 0)
runs a movie based on the given DRAW_FRAME function. The movie
stops after a total elapsed time of TIME_LIMIT seconds, which
defaults to 60 (one minute), or when the DRAW_FRAME function
returns zero.
note: All but the first argument are keyword arguments, with
defaults as shown.
def draw_frame(1) :
# Input argument i is the frame number.
# draw_frame should return non-zero if there are more
# frames in this movie. A zero return will stop the
# movie.
# draw_frame must NOT include any fma command if the
# making_movie variable is set (movie sets this variable
# before calling draw_frame)
If MIN_INTERFRAME is specified, a pause will be added as
necessary to slow down the movie. MIN_INTERFRAME is a time
necessary to slow down the movie. MIN_INTERFRAME is a time
in seconds (default 0).
The keyword bracket_time= (again a time in seconds) can be
used to adjust the duration of the pauses after the first
and last frames. It may also be a two element array [beg, end].
If the pause at the end is greater than five seconds, you will
be prompted to explain that hitting <RETURN> will abort the final
pause. Well, the Python version does not have this capability.
timing = 1 enables a timing printout for your movie.
If every frame of your movie has the same limits, use the
limits command to fix the limits before you call movie.
BUG: If you hit <RETURN> to start a movie early, it will not
pause at the end of the movie at all. You probably should
not use long initial pauses.
"""
global movie_timing, making_movie
if is_scalar (bracket_time) :
bracket_time = array ( [bracket_time, bracket_time], Float )
elif len (bracket_time) == 1 :
bracket_time = array ( [bracket_time [0], bracket_time [0]], Float )
elapsed = zeros (3)
this_frame = zeros (3)
window (wait = 1, style = "nobox.gs") # make sure window is ready to draw
fma ( )
animate (1)
making_movie = 1
i = 0
if timing == 1 :
elapsed = timer_ (elapsed)
elapsed0 = array(elapsed, copy = 1)
i = i + 1
more = draw_frame (i)
if lims != None:
limits (lims [0], lims [1], lims [2], lims [3])
else:
limits (square = 1)
fma ( )
if timing == 1 :
[elapsed, this_frame] = timer_ (elapsed, this_frame)
wait = bracket_time [0] - this_frame [2]
waited = waited0 = 0.0
if wait > 0 :
if wait > 5 :
print "Movie starts in", wait, "secs, or when you hit <RETURN>"
time.sleep (wait)
waited0 = waited0 + wait
else :
wait = min_interframe - this_frame [2]
if wait > 0 :
time.sleep (wait)
waited0 = waited0 + wait
while (more) :
this_frame = zeros (3)
i = i + 1
more = draw_frame (i)
if lims != None:
limits (lims [0], lims [1], lims [2], lims [3])
else :
limits (square = 1)
fma ( )
if not more :
break
if timing == 1 :
[elapsed, this_frame] = timer_ (elapsed, this_frame)
if (not more or (elapsed [2] - elapsed0 [2] > time_limit)) :
break
wait = min_interframe - this_frame [2]
if wait > 0 :
time.sleep (wait)
waited0 = waited0 + wait
if timing == 1 :
wait = bracket_time [1] - this_frame [2]
if wait > 0 :
if wait > 5 :
print "Holding last frame for", wait, "secs, or hit <RETURN>"
time.sleep (5) # wait)
waited0 = waited0 + wait
elapsed = timer_ (elapsed)
e = elapsed - elapsed0
movie_timing = [e [0], e [1], e [2], i, waited, waited0]
movie_stats ()
animate (0)
def movie_stats ( *timing ) :
"""
movie_stats ( ) or movie_stats ( timing )
prints statistics from the last movie command, or from the
command which produced TIMING. TIMING is the contents of the
movie_timing external variable after the movie command completes.
"""
if len (timing) > 0 :
timing = timing [0]
else :
timing = movie_timing
cpu = timing [0] + timing [1]
print "user", timing [0], "system", timing [1]
wall = timing [2]
nframes = timing [3]
waited = timing [4] + timing [5]
wait = timing [5]
print " Wall(sec) Wait(sec) CPU(sec)"
print " %9.3f %9.3f %8.3f %ld frames\n" % (wall, waited, cpu, nframes)
print " %9.3f %9.3f %8.3f per frame\n" % \
( (wall - waited) / nframes, wait / ( (nframes > 1) * (nframes - 1) + \
(nframes <= 1) * 1), cpu / nframes)
|
