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
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
|
# =================================================================================
# image.py
#
# A generic image base class. Defines a variety of global parameters and
# other stuff related to framebuffers.
# ================================================================================
from gifplot import *
# --------------------------------------------------------------------------------
# Default values (in case the user doesn't supply them
# --------------------------------------------------------------------------------
GRAPHICS_SIZE = 500
GRAPHICS_COLORMAP = "cmap"
# --------------------------------------------------------------------------
# Image base class
#
# All displays are consider to be "images" derived from this base class.
# The following member data and operations are available.
#
# self.frame : Framebuffer being used by this image
# self.colormap : Colormap being used with the image
# self.vxmin : Plotting region for this image
# self.vymin "
# self.vxmax "
# self.vymax "
# self.bgcolor : Default background color
# self.fgcolor : Default foreground color
#
# All images must support the following methods (images may of course use the
# definitions in the base class).
#
# self.show() : Show the current image
# self.draw() : Method for drawing the entire image on the frame buffer
# self.output() : Dump the current framebuffer out regardless of what state its in
# self.setview() : Method for setting the viewing region of this image
#
# The following methods are also encouraged for all images
#
# self.zoom() : Zoom in on the image
# self.left() : translate left
# self.right() : translate right
# self.up() : translate up
# self.down() : translate down
# self.rotr() : rotate right (3D only)
# self.rotl() : rotate left (3D only)
# self.rotd() : rotate down (3D only)
# self.rotu() : rotate up (3D only)
# self.rotc() : rotate around the center point (3D only)
# self.rotx() : rotate around the x axis (3D only)
# self.roty() : rotate around the y axis (3D only)
# self.rotz() : rotate around the z axis (3D only)
# self.center() : Set center point
#
# --------------------------------------------------------------------------
class Image:
Current = None # The currently active image
Default_framebuffer = None # Default framebuffer
Default_colormap = None # Default colormap
ImageNo = 0 # Current Image Number
bgcolor = BLACK
fgcolor = WHITE
def __init__(self):
init_framebuffer()
if Image.Default_colormap == None:
colormap(GRAPHICS_COLORMAP)
self.frame = Image.Default_framebuffer
self.filename = "img"+str(Image.ImageNo)+".gif"
self.imageno = 0
Image.ImageNo = Image.ImageNo + 1
self.colormap = Image.Default_colormap
self.stamp = ""
self.fgcolor = Image.fgcolor
self.bgcolor = Image.bgcolor
self.vxmin = 0
self.vymin = 0
self.vxmax = self.frame.width
self.vymax = self.frame.height
# Generate the image and dump it to the current output device
def show(self):
if self.frame == None:
print "Error. No frame buffer assigned to this image"
return
self.frame.noclip()
self.frame.clear(self.bgcolor)
self.frame.setclip(self.vxmin,self.vymin,self.vxmax,self.vymax)
self.draw()
self.output()
# Dump this image out over a socket or to a file
def output(self):
global Graphics_current, Graphics_buffer, Graphics_buffer_size
# Put a stamp on the image
self.frame.noclip()
self.frame.drawstring(1,1,self.fgcolor,self.bgcolor,self.stamp,HORIZONTAL)
# Make a GIF image
print "Writing " + self.filename
self.frame.writeGIF(self.colormap,self.filename)
Image.Current = self
def draw(self):
print "No drawing method specified for image " + self.filename
# Set the viewport for this image
def viewport(self,vx,vy,wx,wy):
self.vxmin = vx
self.vymin = vy
self.vxmax = wx
self.vymax = wy
self.frame.setclip(vx,vy,wx,wy)
# Image manipulation methods
def zoom(self,s):
print("zoom() unavailable in current image")
def left(self,s):
print("left() unavailable in current image")
def right(self,s):
print("right() unavailable in current image")
def up(self,s):
print("up() unavailable in current image")
def down(self,s):
print("down() unavailable in current image")
def rotr(self,s):
print("rotr() unavailable in current image")
def rotl(self,s):
print("rotl() unavailable in current image")
def rotu(self,s):
print("rotu() unavailable in current image")
def rotd(self,s):
print("rotd() unavailable in current image")
def rotc(self,s):
print("rotc() unavailable in current image")
def rotx(self,s):
print("rotx() unavailable in current image")
def roty(self,s):
print("roty() unavailable in current image")
def rotz(self,s):
print("rotz() unavailable in current image")
def center(self,x,y):
print("center() unavailable in current image")
# ---------------------------------------------------------------------------
# Image manipulation shortcuts
#
# Whenever an image is created or manipulated, 'image' gets set to
# that image. These commands operate on the last image used.
# ---------------------------------------------------------------------------
def show(*args):
if len(args) == 0:
Image.Current.show()
else :
for img in args:
img.show()
def zoom(s):
Image.Current.zoom(s)
Image.Current.show()
def left(s):
Image.Current.left(s)
Image.Current.show()
def right(s):
Image.Current.right(s)
Image.Current.show()
def up(s):
Image.Current.up(s)
Image.Current.show()
def center(x,y):
Image.Current.center(x,y)
Image.Current.show()
def down(s):
Image.Current.down(s)
Image.Current.show()
def rotr(deg):
Image.Current.rotr(deg)
Image.Current.show()
def rotl(deg):
Image.Current.rotl(deg)
Image.Current.show()
def rotd(deg):
Image.Current.rotd(deg)
Image.Current.show()
def rotu(deg):
Image.Current.rotu(deg)
Image.Current.show()
def rotc(deg):
Image.Current.rotc(deg)
Image.Current.show()
def rotx(deg):
Image.Current.rotx(deg)
Image.Current.show()
def roty(deg):
Image.Current.roty(deg)
Image.Current.show()
def rotz(deg):
Image.Current.rotz(deg)
Image.Current.show()
# ---------------------------------------------------------------------------
# These functions should be used to change the default values for new images
#
# ---------------------------------------------------------------------------
def set_default_framebuffer(f):
Image.Default_framebuffer = f
def colormap(filename):
c = ColorMap(filename)
Image.Default_colormap = c
def init_framebuffer():
if Image.Default_framebuffer == None:
Image.Default_framebuffer = FrameBuffer(GRAPHICS_SIZE,GRAPHICS_SIZE)
|
