source: branches/casa-prerelease/pre-asap/python/casatoolbar.py@ 2052

Last change on this file since 2052 was 2052, checked in by Kana Sugimoto, 14 years ago

merged a bug fix in trunk(r2051)
File size: 15.2 KB
Line 
1import os
2import matplotlib, numpy
3from asap.logging import asaplog, asaplog_post_dec
4
5######################################
6## Add CASA custom toolbar ##
7######################################
8class CustomToolbarCommon:
9 def __init__(self,parent):
10 self.plotter=parent
11 #self.figmgr=self.plotter._plotter.figmgr
12
13 ### select the nearest spectrum in pick radius
14 ### and display spectral value on the toolbar.
15 def _select_spectrum(self,event):
16 # Do not fire event when in zooming/panning mode
17 mode = self.figmgr.toolbar.mode
18 if not mode == '':
19 return
20 # When selected point is out of panels
21 if event.inaxes == None:
22 return
23 # If not left button
24 if event.button != 1:
25 return
26
27 xclick=event.xdata
28 yclick=event.ydata
29 dist2=1000.
30 pickline=None
31 # If the pannel has picable objects
32 pflag=False
33 for lin in event.inaxes.lines:
34 if not lin.pickable():
35 continue
36 pflag=True
37 flag,pind = lin.contains(event)
38 if not flag:
39 continue
40 # Get nearest point
41 inds = pind['ind']
42 xlin = lin.get_xdata()
43 ylin = lin.get_ydata()
44 for i in inds:
45 d2=(xlin[i]-xclick)**2+(ylin[i]-yclick)**2
46 if dist2 >= d2:
47 dist2 = d2
48 pickline = lin
49 # No pickcable line in the pannel
50 if not pflag:
51 return
52 # Pickable but too far from mouse position
53 elif pickline is None:
54 picked='No line selected.'
55 self.figmgr.toolbar.set_message(picked)
56 return
57 del pind, inds, xlin, ylin
58 # Spectra are Picked
59 theplot = self.plotter._plotter
60 thetoolbar = self.figmgr.toolbar
61 thecanvas = self.figmgr.canvas
62 # Disconnect the default motion notify event
63 # Notice! the other buttons are also diabled!!!
64 thecanvas.mpl_disconnect(thetoolbar._idDrag)
65 # Get picked spectrum
66 xdata = pickline.get_xdata()
67 ydata = pickline.get_ydata()
68 titl=pickline.get_label()
69 titp=event.inaxes.title.get_text()
70 panel0=event.inaxes
71 picked="Selected: '"+titl+"' in panel '"+titp+"'."
72 thetoolbar.set_message(picked)
73 # Generate a navigation window
74 #naviwin=Navigationwindow(titp,titl)
75 #------------------------------------------------------#
76 # Show spectrum data at mouse position
77 def spec_data(event):
78 # Getting spectrum data of neiboring point
79 xclick=event.xdata
80 if event.inaxes != panel0:
81 return
82 ipoint=len(xdata)-1
83 for i in range(len(xdata)-1):
84 xl=xclick-xdata[i]
85 xr=xclick-xdata[i+1]
86 if xl*xr <= 0.:
87 ipoint = i
88 break
89 # Output spectral value on the navigation window
90 posi='[ %s, %s ]: x = %.2f value = %.2f'\
91 %(titl,titp,xdata[ipoint],ydata[ipoint])
92 #naviwin.posi.set(posi)
93 thetoolbar.set_message(posi)
94 #------------------------------------------------------#
95 # Disconnect from mouse events
96 def discon(event):
97 #naviwin.window.destroy()
98 theplot.register('motion_notify',None)
99 # Re-activate the default motion_notify_event
100 thetoolbar._idDrag=thecanvas.mpl_connect('motion_notify_event',
101 thetoolbar.mouse_move)
102 theplot.register('button_release',None)
103 return
104 #------------------------------------------------------#
105 # Show data value along with mouse movement
106 theplot.register('motion_notify',spec_data)
107 # Finish events when mouse button is released
108 theplot.register('button_release',discon)
109
110
111 ### Calculate statistics of the selected area.
112 def _single_mask(self,event):
113 # Do not fire event when in zooming/panning mode
114 if not self.figmgr.toolbar.mode == '':
115 return
116 # When selected point is out of panels
117 if event.inaxes == None:
118 return
119 if event.button ==1:
120 baseinv=True
121 elif event.button == 3:
122 baseinv=False
123 else:
124 return
125
126 def _calc_stats():
127 msk=mymask.get_mask()
128 mymask.scan.stats(stat='max',mask=msk)
129 mymask.scan.stats(stat='min',mask=msk)
130 mymask.scan.stats(stat='sum',mask=msk)
131 mymask.scan.stats(stat='mean',mask=msk)
132 mymask.scan.stats(stat='median',mask=msk)
133 mymask.scan.stats(stat='rms',mask=msk)
134 mymask.scan.stats(stat='stddev',mask=msk)
135
136 # Interactive mask definition
137 from asap.interactivemask import interactivemask
138 mymask=interactivemask(plotter=self.plotter,scan=self.plotter._data)
139 # Create initial mask
140 mymask.set_basemask(invert=baseinv)
141 # Inherit event
142 mymask.set_startevent(event)
143 # Set callback func
144 mymask.set_callback(_calc_stats)
145 # Selected mask
146 mymask.select_mask(once=True,showmask=False)
147
148 def _mod_note(self,event):
149 # Do not fire event when in zooming/panning mode
150 if not self.figmgr.toolbar.mode == '':
151 return
152 if event.button ==1:
153 self.notewin.load_textwindow(event)
154 elif event.button == 3 and self._note_picked(event):
155 self.notewin.load_modmenu(event)
156 return
157
158 def _note_picked(self,event):
159 # just briefly check if any texts are picked
160 for textobj in self.canvas.figure.texts:
161 if textobj.contains(event)[0]:
162 return True
163 for ax in self.canvas.figure.axes:
164 for textobj in ax.texts:
165 if textobj.contains(event)[0]:
166 return True
167 #print "No text picked"
168 return False
169
170 ### Page chages
171 ### go to the previous page
172 def prev_page(self):
173 self.figmgr.toolbar.set_message('plotting the previous page')
174 self._new_page(goback=True)
175
176 ### go to the next page
177 def next_page(self):
178 self.figmgr.toolbar.set_message('plotting the next page')
179 self._new_page(goback=False)
180
181 ### actual plotting of the new page
182 def _new_page(self,goback=False):
183 top = None
184 if self.plotter._startrow <= 0:
185 msg = "The page counter is reset due to chages of plot settings. "
186 msg += "Plotting from the first page."
187 asaplog.push(msg)
188 asaplog.post('WARN')
189 goback = False
190 else:
191 top = self.plotter._plotter.figure.subplotpars.top
192
193 self.plotter._plotter.hold()
194 if goback:
195 self._set_prevpage_counter()
196 #self.plotter._plotter.clear()
197 self.plotter._plot(self.plotter._data)
198 self.set_pagecounter(self._get_pagenum())
199 self.plotter._plotter.release()
200 self.plotter._plotter.tidy()
201 if self.plotter._headstring:
202 if top and top != self.plotter._margins[3]:
203 # work around for sdplot in CASA. complete checking in future?
204 self.plotter._plotter.figure.subplots_adjust(top=top)
205 self.plotter.print_header()
206 self.plotter._plotter.show(hardrefresh=False)
207 del top
208
209 ### calculate the panel ID and start row to plot the previous page
210 def _set_prevpage_counter(self):
211 # set row and panel counters to those of the 1st panel of previous page
212 maxpanel = 16
213 # the ID of the last panel in current plot
214 lastpanel = self.plotter._ipanel
215 # the number of current subplots
216 currpnum = len(self.plotter._plotter.subplots)
217 # the nuber of previous subplots
218 prevpnum = None
219 if self.plotter._rows and self.plotter._cols:
220 # when user set layout
221 prevpnum = self.plotter._rows*self.plotter._cols
222 else:
223 # no user specification
224 prevpnum = maxpanel
225
226 start_ipanel = max(lastpanel-currpnum-prevpnum+1, 0)
227 # set the pannel ID of the last panel of prev-prev page
228 self.plotter._ipanel = start_ipanel-1
229 if self.plotter._panelling == 'r':
230 self.plotter._startrow = start_ipanel
231 else:
232 # the start row number of the next panel
233 self.plotter._startrow = self.plotter._panelrows[start_ipanel]
234 del lastpanel,currpnum,prevpnum,start_ipanel
235
236 ### refresh the page counter
237 ### refresh the page counter
238 def set_pagecounter(self,page):
239 nwidth = int(numpy.ceil(numpy.log10(max(page,1))))+1
240 nwidth = max(nwidth,4)
241 formatstr = '%'+str(nwidth)+'d'
242 self.show_pagenum(page,formatstr)
243
244 def show_pagenum(self,pagenum,formatstr):
245 # passed to backend dependent class
246 pass
247
248 def _get_pagenum(self):
249 maxpanel = 16
250 # get the ID of last panel in the current page
251 idlastpanel = self.plotter._ipanel
252 if self.plotter._rows and self.plotter._cols:
253 ppp = self.plotter._rows*self.plotter._cols
254 else:
255 ppp = maxpanel
256 return int(idlastpanel/ppp)+1
257
258#####################################
259## Backend dependent Classes ##
260#####################################
261### TkAgg
262if matplotlib.get_backend() == 'TkAgg':
263 import Tkinter as Tk
264 from notationwindow import NotationWindowTkAgg
265
266class CustomToolbarTkAgg(CustomToolbarCommon, Tk.Frame):
267 def __init__(self,parent):
268 from asap.asapplotter import asapplotter
269 if not isinstance(parent,asapplotter):
270 return False
271 if not parent._plotter:
272 return False
273 self._p = parent._plotter
274 self.figmgr = self._p.figmgr
275 self.canvas = self.figmgr.canvas
276 self.mode = ''
277 self.button = True
278 self.pagecount = None
279 CustomToolbarCommon.__init__(self,parent)
280 self.notewin=NotationWindowTkAgg(master=self.canvas)
281 self._add_custom_toolbar()
282
283 def _add_custom_toolbar(self):
284 Tk.Frame.__init__(self,master=self.figmgr.window)
285 #self.bSpec=self._NewButton(master=self,
286 # text='spec value',
287 # command=self.spec_show)
288 self.bNote=self._NewButton(master=self,
289 text='notation',
290 command=self.modify_note)
291
292 self.bStat=self._NewButton(master=self,
293 text='statistics',
294 command=self.stat_cal)
295 self.bQuit=self._NewButton(master=self,
296 text='Quit',
297 command=self.quit,
298 side=Tk.RIGHT)
299
300 # page change oparations
301 frPage = Tk.Frame(master=self,borderwidth=2,relief=Tk.GROOVE)
302 frPage.pack(ipadx=2,padx=10,side=Tk.RIGHT)
303 self.lPagetitle = Tk.Label(master=frPage,text='Page:',padx=5)
304 self.lPagetitle.pack(side=Tk.LEFT)
305 self.pagecount = Tk.StringVar(master=frPage)
306 self.lPagecount = Tk.Label(master=frPage,
307 textvariable=self.pagecount,
308 padx=5,bg='white')
309 self.lPagecount.pack(side=Tk.LEFT,padx=3)
310
311 self.bNext=self._NewButton(master=frPage,
312 text=' + ',
313 command=self.next_page)
314 self.bPrev=self._NewButton(master=frPage,
315 text=' - ',
316 command=self.prev_page)
317
318 if os.uname()[0] != 'Darwin':
319 self.bPrev.config(padx=5)
320 self.bNext.config(padx=5)
321
322 self.pack(side=Tk.BOTTOM,fill=Tk.BOTH)
323 self.pagecount.set(' '*4)
324
325 self.disable_button()
326 return #self
327
328 def _NewButton(self, master, text, command, side=Tk.LEFT):
329 if os.uname()[0] == 'Darwin':
330 b = Tk.Button(master=master, text=text, command=command)
331 else:
332 b = Tk.Button(master=master, text=text, padx=2, pady=2,
333 command=command)
334 b.pack(side=side)
335 return b
336
337 def show_pagenum(self,pagenum,formatstr):
338 self.pagecount.set(formatstr % (pagenum))
339
340 def spec_show(self):
341 if not self.figmgr.toolbar.mode == '' or not self.button: return
342 self.figmgr.toolbar.set_message('spec value: drag on a spec')
343 if self.mode == 'spec': return
344 #self.bStat.config(relief='raised')
345 #self.bSpec.config(relief='sunken')
346 #self.bNote.config(relief='raised')
347 self.mode='spec'
348 self.notewin.close_widgets()
349 self.__disconnect_event()
350 self._p.register('button_press',self._select_spectrum)
351
352 def stat_cal(self):
353 if not self.figmgr.toolbar.mode == '' or not self.button: return
354 self.figmgr.toolbar.set_message('statistics: select a region')
355 if self.mode == 'stat':
356 # go back to spec mode
357 self.bStat.config(relief='raised')
358 self.spec_show()
359 return
360 #self.bSpec.config(relief='raised')
361 self.bStat.config(relief='sunken')
362 self.bNote.config(relief='raised')
363 self.mode='stat'
364 self.notewin.close_widgets()
365 self.__disconnect_event()
366 self._p.register('button_press',self._single_mask)
367
368 def modify_note(self):
369 if not self.figmgr.toolbar.mode == '': return
370 self.figmgr.toolbar.set_message('text: select a position/text')
371 if self.mode == 'note':
372 self.bNote.config(relief='raised')
373 self.mode='none'
374 self.spec_show()
375 return
376 #self.bSpec.config(relief='raised')
377 self.bStat.config(relief='raised')
378 self.bNote.config(relief='sunken')
379 self.mode='note'
380 self.__disconnect_event()
381 self._p.register('button_press',self._mod_note)
382
383 def quit(self):
384 self.__disconnect_event()
385 #self.delete_bar()
386 self.disable_button()
387 self.figmgr.window.wm_withdraw()
388
389 def enable_button(self):
390 if self.button: return
391 #self.bSpec.config(state=Tk.NORMAL)
392 self.bStat.config(state=Tk.NORMAL)
393 self.button=True
394 self.spec_show()
395
396 def disable_button(self):
397 if not self.button: return
398 #self.bSpec.config(relief='raised', state=Tk.DISABLED)
399 self.bStat.config(relief='raised', state=Tk.DISABLED)
400 #self.bNext.config(state=Tk.DISABLED)
401 #self.bPrev.config(state=Tk.DISABLED)
402 self.button=False
403 self.mode=''
404 self.__disconnect_event()
405
406 def enable_next(self):
407 self.bNext.config(state=Tk.NORMAL)
408
409 def disable_next(self):
410 self.bNext.config(state=Tk.DISABLED)
411
412 def enable_prev(self):
413 self.bPrev.config(state=Tk.NORMAL)
414
415 def disable_prev(self):
416 self.bPrev.config(state=Tk.DISABLED)
417
418 def delete_bar(self):
419 self.__disconnect_event()
420 self.destroy()
421
422 def __disconnect_event(self):
423 self._p.register('button_press',None)
424 self._p.register('button_release',None)
Note: See TracBrowser for help on using the repository browser.