1 | from os.path import expanduser, expandvars |
---|
2 | from asap._asap import Plotter2 |
---|
3 | |
---|
4 | class plotter2: |
---|
5 | def __init__(self): |
---|
6 | self._plotter = Plotter2() |
---|
7 | |
---|
8 | def set_output(self, filename=None, dev=None): |
---|
9 | if filename is None: |
---|
10 | filename = '' |
---|
11 | dev = 'xwindow' |
---|
12 | else: |
---|
13 | filename = filename.strip() |
---|
14 | if filename == '': |
---|
15 | dev = 'xwindow' |
---|
16 | else: |
---|
17 | if (dev is None) or (dev.strip() == ''): |
---|
18 | fnamelem = filename.lower().split('.') |
---|
19 | dev = fnamelem[len(fnamelem)-1].strip() |
---|
20 | if dev == 'ps': dev = 'cps' |
---|
21 | |
---|
22 | self._plotter.set_filename(expanduser(expandvars(filename))) |
---|
23 | self._plotter.set_device(dev.strip()) |
---|
24 | |
---|
25 | def set_vp(self, xmin, xmax, ymin, ymax, id=None): |
---|
26 | if id is None: |
---|
27 | if self._plotter.get_hasdefaultvp(): |
---|
28 | self._plotter.set_viewport(xmin, xmax, ymin, ymax, 0) |
---|
29 | else: |
---|
30 | return self._plotter.add_viewport(xmin, xmax, ymin, ymax) |
---|
31 | else: |
---|
32 | self._plotter.set_viewport(xmin, xmax, ymin, ymax, id) |
---|
33 | |
---|
34 | def show_vp(self, vpid=None): |
---|
35 | if vpid is None: vpid = -1 |
---|
36 | self._plotter.show_viewport(vpid) |
---|
37 | |
---|
38 | def hide_vp(self, vpid=None): |
---|
39 | if vpid is None: vpid = -1 |
---|
40 | self._plotter.hide_viewport(vpid) |
---|
41 | |
---|
42 | def set_range(self, xmin, xmax, ymin, ymax, vpid=None): |
---|
43 | if vpid is None: vpid = -1 |
---|
44 | self._plotter.set_range(xmin, xmax, ymin, ymax, vpid) |
---|
45 | |
---|
46 | def set_xrange(self, xmin, xmax, vpid=None): |
---|
47 | if vpid is None: vpid = -1 |
---|
48 | self._plotter.set_range_x(xmin, xmax, vpid) |
---|
49 | |
---|
50 | def set_yrange(self, ymin, ymax, vpid=None): |
---|
51 | if vpid is None: vpid = -1 |
---|
52 | self._plotter.set_range_y(ymin, ymax, vpid) |
---|
53 | |
---|
54 | def get_xrange(self, vpid=None): |
---|
55 | if vpid is None: vpid = -1 |
---|
56 | return self._plotter.get_range_x(vpid) |
---|
57 | |
---|
58 | def get_yrange(self, vpid=None): |
---|
59 | if vpid is None: vpid = -1 |
---|
60 | return self._plotter.get_range_y(vpid) |
---|
61 | |
---|
62 | def set_autorange(self, vpid=None): |
---|
63 | if vpid is None: vpid = -1 |
---|
64 | self._plotter.set_autorange(vpid) |
---|
65 | |
---|
66 | def set_xautorange(self, vpid=None): |
---|
67 | if vpid is None: vpid = -1 |
---|
68 | self._plotter.set_autorange_x(vpid) |
---|
69 | |
---|
70 | def set_yautorange(self, vpid=None): |
---|
71 | if vpid is None: vpid = -1 |
---|
72 | self._plotter.set_autorange_y(vpid) |
---|
73 | |
---|
74 | def set_fontsize(self, size, vpid=None): |
---|
75 | if vpid is None: vpid = -1 |
---|
76 | self._plotter.set_fontsize(size, vpid) |
---|
77 | |
---|
78 | def set_xtics(self, interval_major, num_minor=None, vpid=None): |
---|
79 | if vpid is None: vpid = -1 |
---|
80 | if num_minor is None: num_minor = 5 |
---|
81 | self._plotter.set_tics_x(interval_major, num_minor, vpid) |
---|
82 | |
---|
83 | def set_ytics(self, interval_major, num_minor=None, vpid=None): |
---|
84 | if vpid is None: vpid = -1 |
---|
85 | if num_minor is None: num_minor = 5 |
---|
86 | self._plotter.set_tics_y(interval_major, num_minor, vpid) |
---|
87 | |
---|
88 | def set_xautotics(self, vpid=None): |
---|
89 | if vpid is None: vpid = -1 |
---|
90 | self._plotter.set_autotics_x(vpid) |
---|
91 | |
---|
92 | def set_yautotics(self, vpid=None): |
---|
93 | if vpid is None: vpid = -1 |
---|
94 | self._plotter.set_autotics_y(vpid) |
---|
95 | |
---|
96 | def set_xnuminterval(self, interval, vpid=None): |
---|
97 | if vpid is None: vpid = -1 |
---|
98 | self._plotter.set_ninterval_x(interval, vpid) |
---|
99 | |
---|
100 | def set_ynuminterval(self, interval, vpid=None): |
---|
101 | if vpid is None: vpid = -1 |
---|
102 | self._plotter.set_ninterval_y(interval, vpid) |
---|
103 | |
---|
104 | def set_xnumlocation(self, location=None, vpid=None): |
---|
105 | if vpid is None: vpid = -1 |
---|
106 | if location is None: location = "l" |
---|
107 | self._plotter.set_nlocation_x(location.lower(), vpid) |
---|
108 | |
---|
109 | def set_ynumlocation(self, location=None, vpid=None): |
---|
110 | if vpid is None: vpid = -1 |
---|
111 | if location is None: location = "b" |
---|
112 | self._plotter.set_nlocation_y(location.lower(), vpid) |
---|
113 | |
---|
114 | def set_data(self, xdata, ydata, vpid=None, dataid=None): |
---|
115 | if dataid is None: |
---|
116 | if vpid is None: vpid = -1 |
---|
117 | dataid = -1 |
---|
118 | |
---|
119 | self._plotter.set_data(xdata, ydata, vpid, dataid) |
---|
120 | |
---|
121 | def set_line(self, color, width=None, style=None, vpid=None, dataid=None): |
---|
122 | if width is None: width = 1 |
---|
123 | if style is None: style = "solid" |
---|
124 | if vpid is None: vpid = -1 |
---|
125 | if dataid is None: dataid = -1 |
---|
126 | |
---|
127 | coloridx = self.get_colorindex(color) |
---|
128 | styleidx = self.get_linestyleindex(style) |
---|
129 | self._plotter.set_line(coloridx, width, styleidx, vpid, dataid) |
---|
130 | |
---|
131 | def show_line(self, vpid=None, dataid=None): |
---|
132 | if dataid is None: |
---|
133 | if vpid is None: vpid = -1 |
---|
134 | dataid = -1 |
---|
135 | |
---|
136 | self._plotter.show_line(vpid, dataid) |
---|
137 | |
---|
138 | def hide_line(self, vpid=None, dataid=None): |
---|
139 | if dataid is None: |
---|
140 | if vpid is None: vpid = -1 |
---|
141 | dataid = -1 |
---|
142 | |
---|
143 | self._plotter.hide_line(vpid, dataid) |
---|
144 | |
---|
145 | def set_point(self, type, size, color, vpid=None, dataid=None): |
---|
146 | if dataid is None: |
---|
147 | if vpid is None: vpid = -1 |
---|
148 | dataid = -1 |
---|
149 | coloridx = self.get_colorindex(color) |
---|
150 | self._plotter.set_point(type, size, coloridx, vpid, dataid) |
---|
151 | |
---|
152 | def show_point(self, vpid=None, dataid=None): |
---|
153 | if dataid is None: |
---|
154 | if vpid is None: vpid = -1 |
---|
155 | dataid = -1 |
---|
156 | |
---|
157 | self._plotter.show_point(vpid, dataid) |
---|
158 | |
---|
159 | def hide_point(self, vpid=None, dataid=None): |
---|
160 | if dataid is None: |
---|
161 | if vpid is None: vpid = -1 |
---|
162 | dataid = -1 |
---|
163 | |
---|
164 | self._plotter.hide_point(vpid, dataid) |
---|
165 | |
---|
166 | def set_xmask(self, xmin, xmax, color=None, fstyle=None, width=None, hsep=None, vpid=None): |
---|
167 | if color is None: color = "lightgray" |
---|
168 | if fstyle is None: fstyle = "solid" |
---|
169 | if width is None: width = 1 |
---|
170 | if hsep is None: hsep = 1.0 |
---|
171 | if vpid is None: vpid = -1 |
---|
172 | coloridx = self.get_colorindex(color) |
---|
173 | fstyleidx = self.get_fillstyleindex(fstyle) |
---|
174 | self._plotter.set_mask_x(xmin, xmax, coloridx, fstyleidx, width, hsep, vpid) |
---|
175 | |
---|
176 | def set_xlabel(self, label, style=None, size=None, posx=None, posy=None, vpid=None): |
---|
177 | if style is None: style = "" |
---|
178 | if size is None: size = 1.1 |
---|
179 | if posx is None: posx = -1.0 |
---|
180 | if posy is None: posy = -1.0 |
---|
181 | if vpid is None: vpid = -1 |
---|
182 | |
---|
183 | self._plotter.set_label_x(label, posx, posy, size, style, 1, 0, vpid) |
---|
184 | |
---|
185 | def set_ylabel(self, label, style=None, size=None, posx=None, posy=None, vpid=None): |
---|
186 | if style is None: style = "" |
---|
187 | if size is None: size = 1.1 |
---|
188 | if posx is None: posx = -1.0 |
---|
189 | if posy is None: posy = -1.0 |
---|
190 | if vpid is None: vpid = -1 |
---|
191 | |
---|
192 | self._plotter.set_label_y(label, posx, posy, size, style, 1, 0, vpid) |
---|
193 | |
---|
194 | def set_title(self, label, style=None, size=None, posx=None, posy=None, vpid=None): |
---|
195 | if style is None: style = "" |
---|
196 | if size is None: size = 1.5 |
---|
197 | if posx is None: posx = -1.0 |
---|
198 | if posy is None: posy = -1.0 |
---|
199 | if vpid is None: vpid = -1 |
---|
200 | |
---|
201 | self._plotter.set_title(label, posx, posy, size, style, 1, 0, vpid) |
---|
202 | |
---|
203 | def set_vpbgcolor(self, bgcolor, vpid=None): |
---|
204 | if vpid is None: vpid = -1 |
---|
205 | |
---|
206 | if bgcolor.strip() == "": |
---|
207 | coloridx = -1 |
---|
208 | else: |
---|
209 | coloridx = self.get_colorindex(bgcolor) |
---|
210 | |
---|
211 | self._plotter.set_vpbgcolor(coloridx, vpid) |
---|
212 | |
---|
213 | def plot(self): |
---|
214 | self._plotter.plot() |
---|
215 | |
---|
216 | def save(self, filename): |
---|
217 | prev_filename = self._plotter.get_filename() |
---|
218 | prev_dev = self._plotter.get_device() |
---|
219 | |
---|
220 | self.set_output(filename) |
---|
221 | self.plot() |
---|
222 | |
---|
223 | self.set_output(prev_filename, prev_dev) |
---|
224 | |
---|
225 | def get_vinfo(self): |
---|
226 | self._plotter.get_vinfo() |
---|
227 | |
---|
228 | def get_colorindex(self, colorname): |
---|
229 | name = colorname.strip().lower() |
---|
230 | |
---|
231 | if name == "white": idx = 0 # our definition of bgcolor |
---|
232 | if name == "black": idx = 1 # our definition of fgcolor |
---|
233 | if name == "red": idx = 2 |
---|
234 | if name == "green": idx = 3 |
---|
235 | if name == "blue": idx = 4 |
---|
236 | if name == "cyan": idx = 5 |
---|
237 | if name == "magenta": idx = 6 |
---|
238 | if name == "yellow": idx = 7 |
---|
239 | if name == "orange": idx = 8 |
---|
240 | if name == "yellowgreen": idx = 9 |
---|
241 | if name == "emerald": idx = 10 |
---|
242 | if name == "skyblue": idx = 11 |
---|
243 | if name == "purple": idx = 12 |
---|
244 | if name == "pink": idx = 13 |
---|
245 | if name == "gray": idx = 14 |
---|
246 | if name == "lightgray": idx = 15 |
---|
247 | |
---|
248 | return idx |
---|
249 | |
---|
250 | def list_colornames(self): |
---|
251 | print "plotter2: default color list ----" |
---|
252 | print " (0) white (background)" |
---|
253 | print " (1) black (foreground)" |
---|
254 | print " (2) red" |
---|
255 | print " (3) green" |
---|
256 | print " (4) blue" |
---|
257 | print " (5) cyan" |
---|
258 | print " (6) magenta" |
---|
259 | print " (7) yellow" |
---|
260 | print " (8) orange" |
---|
261 | print " (9) yellowgreen" |
---|
262 | print " (10) emerald" |
---|
263 | print " (11) skyblue" |
---|
264 | print " (12) purple" |
---|
265 | print " (13) pink" |
---|
266 | print " (14) gray" |
---|
267 | print " (15) lightgray" |
---|
268 | print "---------------------------------" |
---|
269 | |
---|
270 | def get_linestyleindex(self, fstyle): |
---|
271 | style = fstyle.strip().lower() |
---|
272 | if style == "solid": idx = 1 |
---|
273 | if style == "outline": idx = 2 |
---|
274 | if style == "hatched": idx = 3 |
---|
275 | if style == "crosshatched": idx = 4 |
---|
276 | |
---|
277 | return idx |
---|
278 | |
---|
279 | def list_linestyles(self): |
---|
280 | print "plotter2: fill style list ----" |
---|
281 | print " (1) solid" |
---|
282 | print " (2) dashed" |
---|
283 | print " (3) dash-dotted" |
---|
284 | print " (4) dotted" |
---|
285 | print " (5) dash-dot-dot-dotted" |
---|
286 | print "------------------------------" |
---|
287 | |
---|
288 | def get_fillstyleindex(self, fstyle): |
---|
289 | style = fstyle.strip().lower() |
---|
290 | if style == "solid": idx = 1 |
---|
291 | if style == "outline": idx = 2 |
---|
292 | if style == "hatched": idx = 3 |
---|
293 | if style == "crosshatched": idx = 4 |
---|
294 | |
---|
295 | return idx |
---|
296 | |
---|
297 | def list_fillstyles(self): |
---|
298 | print "plotter2: fill style list ----" |
---|
299 | print " (1) solid" |
---|
300 | print " (2) outline" |
---|
301 | print " (3) hatched" |
---|
302 | print " (4) crosshatched" |
---|
303 | print "------------------------------" |
---|
304 | |
---|
305 | """ |
---|
306 | def set_annotation(self, label, posx=None, posy=None, angle=None, fjust=None, size=None, style=None, color=None, bgcolor=None, vpid=None): |
---|
307 | if posx is None: posx = -1.0 |
---|
308 | if posy is None: posy = -1.0 |
---|
309 | if angle is None: angle = 0.0 |
---|
310 | if fjust is None: fjust = 0.5 |
---|
311 | if size is None: size = 2.0 |
---|
312 | if style is None: style = "" |
---|
313 | if color is None: color = 1 #default foreground colour (b) |
---|
314 | if bgcolor is None: bgcolor = 0 #default backgound colour (w) |
---|
315 | if vpid is None: vpid = -1 |
---|
316 | |
---|
317 | coloridx = self.get_colorindex(color) |
---|
318 | bgcoloridx = self.get_colorindex(bgcolor) |
---|
319 | self._plotter.set_annotation(label, posx, posy, angle, fjust, size, style, coloridx, bgcoloridx, vpid) |
---|
320 | """ |
---|