source: trunk/python/flagtoolbar.py @ 2000

Last change on this file since 2000 was 2000, checked in by Kana Sugimoto, 13 years ago

New Development: No

JIRA Issue: Yes (CAS-1306/ASAP-205)

Ready for Test: Yes

Interface Changes: No

What Interface Changed: Please list interface changes

Test Programs: run flagplotter

Put in Release Notes: No

Module(s): flagtoolbar & flagplotter

Description:

Region selection and operation information will be printed on the logger
instead of the terminal.
Removed _statistics function in CustomFlagToolbarCommon? and merged it to stat_cal function.


File size: 29.3 KB
Line 
1import os
2import matplotlib, numpy
3from matplotlib.patches import Rectangle
4from asap.parameters import rcParams
5from asap.logging import asaplog, asaplog_post_dec
6from asap import scantable
7from asap._asap import stmath
8
9###########################################
10##    Add CASA custom Flag toolbar       ##
11###########################################
12class CustomFlagToolbarCommon:
13    def __init__(self,parent):
14        self.plotter=parent
15        #self.figmgr=self.plotter._plotter.figmgr
16        self._selregions = {}
17        self._selpanels = []
18        self._polygons = []
19        self._thisregion = None
20        self.xdataold=None
21       
22    ### select the nearest spectrum in pick radius
23    ###    and display spectral value on the toolbar.
24    def _select_spectrum(self,event):
25        # Do not fire event when in zooming/panning mode
26        mode = self.figmgr.toolbar.mode
27        if not mode == '':
28            return
29            # When selected point is out of panels
30        if event.inaxes == None:
31            return
32        # If not left button
33        if event.button != 1:
34            return
35
36        xclick=event.xdata
37        yclick=event.ydata
38        dist2=1000.
39        pickline=None
40        # If the pannel has picable objects
41        pflag=False
42        for lin in event.inaxes.lines:
43            if not lin.pickable():
44                continue
45            pflag=True
46            flag,pind = lin.contains(event)
47            if not flag:
48                continue
49            # Get nearest point
50            inds = pind['ind']
51            xlin = lin.get_xdata()
52            ylin = lin.get_ydata()
53            for i in inds:
54                d2=(xlin[i]-xclick)**2+(ylin[i]-yclick)**2
55                if dist2 >= d2:
56                    dist2 = d2
57                    pickline = lin
58        # No pickcable line in the pannel
59        if not pflag:
60            return
61        # Pickable but too far from mouse position
62        elif pickline is None:
63            picked='No line selected.'
64            self.figmgr.toolbar.set_message(picked)
65            return
66        del pind, inds, xlin, ylin
67        # Spectra are Picked
68        theplot = self.plotter._plotter
69        thetoolbar = self.figmgr.toolbar
70        thecanvas = self.figmgr.canvas
71        # Disconnect the default motion notify event
72        # Notice! the other buttons are also diabled!!!
73        thecanvas.mpl_disconnect(thetoolbar._idDrag)
74        # Get picked spectrum
75        xdata = pickline.get_xdata()
76        ydata = pickline.get_ydata()
77        titl=pickline.get_label()
78        titp=event.inaxes.title.get_text()
79        panel0=event.inaxes
80        picked="Selected: '"+titl+"' in panel '"+titp+"'."
81        thetoolbar.set_message(picked)
82        # Generate a navigation window
83        #naviwin=Navigationwindow(titp,titl)
84        #------------------------------------------------------#
85        # Show spectrum data at mouse position
86        def spec_data(event):
87            # Getting spectrum data of neiboring point
88            xclick=event.xdata
89            if event.inaxes != panel0:
90                return
91            ipoint=len(xdata)-1
92            for i in range(len(xdata)-1):
93                xl=xclick-xdata[i]
94                xr=xclick-xdata[i+1]
95                if xl*xr <= 0.:
96                    ipoint = i
97                    break
98            # Output spectral value on the navigation window
99            posi='[ %s, %s ]:  x = %.2f   value = %.2f'\
100                  %(titl,titp,xdata[ipoint],ydata[ipoint])
101            #naviwin.posi.set(posi)
102            thetoolbar.set_message(posi)
103        #------------------------------------------------------#
104        # Disconnect from mouse events
105        def discon(event):
106            #naviwin.window.destroy()
107            theplot.register('motion_notify',None)
108            # Re-activate the default motion_notify_event
109            thetoolbar._idDrag=thecanvas.mpl_connect('motion_notify_event',
110                                                     thetoolbar.mouse_move)
111            theplot.register('button_release',None)
112            return
113        #------------------------------------------------------#
114        # Show data value along with mouse movement
115        theplot.register('motion_notify',spec_data)
116        # Finish events when mouse button is released
117        theplot.register('button_release',discon)
118
119
120    ### Notation
121    def _mod_note(self,event):
122        # Do not fire event when in zooming/panning mode
123        if not self.figmgr.toolbar.mode == '':
124            return
125        if event.button ==1:
126            self.notewin.load_textwindow(event)
127        elif event.button == 3 and self._note_picked(event):
128            self.notewin.load_modmenu(event)
129        return
130
131    def _note_picked(self,event):
132        # just briefly check if any texts are picked
133        for textobj in self.canvas.figure.texts:
134            if textobj.contains(event)[0]:
135                return True
136        for ax in self.canvas.figure.axes:
137            for textobj in ax.texts:
138                if textobj.contains(event)[0]:
139                    return True
140        #print "No text picked"
141        return False
142
143    ### Region/Panel selection & oparations
144    ### add regions to selections
145    @asaplog_post_dec
146    def _add_region(self,event):
147        if not self.figmgr.toolbar.mode == '':
148            return
149        if event.button != 1 or event.inaxes == None:
150            return
151        # this row resolution assumes row panelling
152        irow = int(self._getrownum(event.inaxes))
153        if irow in self._selpanels:
154            msg = "The whole spectrum is already selected"
155            asaplog.post()
156            asaplog.push(msg)
157            asaplog.post('WARN')
158            return
159        #print "add the region to selections"
160        self._thisregion = {'axes': event.inaxes,'xs': event.xdata,
161                            'worldx': [event.xdata,event.xdata]}
162        self.plotter._plotter.register('button_press',None)
163        self.plotter._plotter.register('motion_notify', self._xspan_draw)
164        self.plotter._plotter.register('button_press', self._xspan_end)
165
166    def _xspan_draw(self,event):
167        if event.inaxes == self._thisregion['axes']:
168            xnow = event.xdata
169            self.xdataold = xnow
170        else:
171            xnow = self.xdataold
172        try: self.lastspan
173        except AttributeError: pass
174        else: self.lastspan.remove()
175
176        self.lastspan = self._thisregion['axes'].axvspan(self._thisregion['xs'],xnow,facecolor='0.7')
177        #self.plotter._plotter.show(False)
178        self.plotter._plotter.canvas.draw()
179        del xnow
180
181    def _xspan_end(self,event):
182        if not self.figmgr.toolbar.mode == '':
183            return
184        if event.button != 1:
185            return
186       
187        try: self.lastspan
188        except AttributeError: pass
189        else:
190            self.lastspan.remove()
191            del self.lastspan
192        if event.inaxes == self._thisregion['axes']:
193            xdataend = event.xdata
194        else:
195            xdataend=self.xdataold
196
197        self._thisregion['worldx'][1] = xdataend
198        lregion = self._thisregion['worldx']
199        pregion = self._thisregion['axes'].axvspan(lregion[0],lregion[1],
200                                                   facecolor='0.7')
201        #self.plotter._plotter.show(False)
202        self.plotter._plotter.canvas.draw()
203        self._polygons.append(pregion)
204        srow = self._getrownum(self._thisregion['axes'])
205        irow = int(srow)
206        if not self._selregions.has_key(srow):
207            self._selregions[srow] = []
208        self._selregions[srow].append(lregion)
209        del lregion, pregion, xdataend
210        sout = "selected region: "+str(self._thisregion['worldx'])+\
211              "(@row "+str(self._getrownum(self._thisregion['axes']))+")"
212        asaplog.push(sout)
213
214        # release event
215        self.plotter._plotter.register('button_press',None)
216        self.plotter._plotter.register('motion_notify',None)
217        # Clear up region selection
218        self._thisregion = None
219        self.xdataold = None
220        # finally recover region selection event
221        self.plotter._plotter.register('button_press',self._add_region)
222
223    ### add panels to selections
224    @asaplog_post_dec
225    def _add_panel(self,event):
226        if not self.figmgr.toolbar.mode == '':
227            return
228        if event.button != 1 or event.inaxes == None:
229            return
230        selax = event.inaxes
231        # this row resolution assumes row panelling
232        srow = self._getrownum(selax)
233        irow = int(srow)
234        if srow:
235            self._selpanels.append(irow)
236        shadow = Rectangle((0,0),1,1,facecolor='0.7',transform=selax.transAxes,visible=True)
237        self._polygons.append(selax.add_patch(shadow))
238        #self.plotter._plotter.show(False)
239        self.plotter._plotter.canvas.draw()
240        asaplog.push("row "+str(irow)+" is selected")
241        #print "selected rows =",self._selpanels
242        ## check for region selection of the spectra and overwrite it.
243        ##!!!! currently disabled for consistency with flag tools !!!!
244        #if self._selregions.has_key(srow):
245        #    self._selregions.pop(srow)
246        #    msg = "The whole spectrum is selected for row="+srow+". Region selection will be overwritten."
247        #    asaplog.push(msg)
248
249    def _getrownum(self,axis):
250        ### returns the row number of selected spectrum as a string ###
251        plabel = axis.get_title()
252        if plabel.startswith("row "):
253            return plabel.strip("row ")
254        return None
255
256    def _any_selection(self):
257        ### returns if users have selected any spectrum or region ###
258        if len(self._selpanels) or len(self._selregions):
259            return True
260        return False
261
262    def _plot_selections(self,regions=None,panels=None):
263        ### mark panels/spectra selections in the page
264        if not self._any_selection() and not (regions or panels):
265            return
266        regions = regions or self._selregions.copy() or {}
267        panels = panels or self._selpanels or []
268        if not isinstance(regions,dict):
269            asaplog.post()
270            asaplog.push("Invalid region specification")
271            asaplog.post('ERROR')
272        if not isinstance(panels,list):
273            asaplog.post()
274            asaplog.push("Invalid panel specification")
275            asaplog.post('ERROR')
276        strow = self._getrownum(self.plotter._plotter.subplots[0]['axes'])
277        enrow = self._getrownum(self.plotter._plotter.subplots[-1]['axes'])
278        for irow in range(int(strow),int(enrow)+1):
279            if regions.has_key(str(irow)):
280                ax = self.plotter._plotter.subplots[irow - int(strow)]['axes']
281                mlist = regions.pop(str(irow))
282                for i in range(len(mlist)):
283                    self._polygons.append(ax.axvspan(mlist[i][0],mlist[i][1],
284                                                     facecolor='0.7'))
285                del ax,mlist
286            if irow in panels:
287                ax = self.plotter._plotter.subplots[irow - int(strow)]['axes']
288                shadow = Rectangle((0,0),1,1,facecolor='0.7',
289                                   transform=ax.transAxes,visible=True)
290                self._polygons.append(ax.add_patch(shadow))
291                del ax,shadow
292        #self.plotter._plotter.show(False)
293        self.plotter._plotter.canvas.draw()
294        del regions,panels,strow,enrow
295
296    def _clear_selection_plot(self):
297        ### clear up polygons which mark selected spectra and regions ###
298        if len(self._polygons) > 0:
299            for shadow in self._polygons:
300                shadow.remove()
301            self.plotter._plotter.canvas.draw()
302        self._polygons = []
303
304    def _clearup_selections(self):
305        # clear-up selection and polygons
306        self._selpanels = []
307        self._selregions = {}
308        self._clear_selection_plot()
309
310    ### clear up selections
311    def cancel_select(self):
312        self.figmgr.toolbar.set_message('selections canceled')
313        # clear-up selection and polygons
314        self._clearup_selections()
315
316    ### flag selected spectra/regions
317    @asaplog_post_dec
318    def flag(self):
319        if not self._any_selection():
320            msg = "No selection to be Flagged"
321            asaplog.post()
322            asaplog.push(msg)
323            asaplog.post('WARN')
324            return
325        self._flag_operation(rows=self._selpanels,
326                             regions=self._selregions,unflag=False)
327        sout = "Flagged:\n"
328        sout += "  rows = "+str(self._selpanels)+"\n"
329        sout += "  regions: "+str(self._selregions)
330        asaplog.push(sout)
331        del sout
332        self._clearup_selections()
333        self._plot_page(pagemode="current")
334
335    ### unflag selected spectra/regions
336    @asaplog_post_dec
337    def unflag(self):
338        if not self._any_selection():
339            msg = "No selection to be Flagged"
340            asaplog.push(msg)
341            asaplog.post('WARN')
342            return
343        self._flag_operation(rows=self._selpanels,
344                             regions=self._selregions,unflag=True)
345        sout = "Unflagged:\n"
346        sout += "  rows = "+str(self._selpanels)+"\n"
347        sout += "  regions: "+str(self._selregions)
348        asaplog.push(sout)
349        del sout
350        self._clearup_selections()
351        self._plot_page(pagemode="current")
352
353    ### actual flag operation
354    @asaplog_post_dec
355    def _flag_operation(self,rows=None,regions=None,unflag=False):
356        scan = self.plotter._data
357        if not scan:
358            asaplog.post()
359            asaplog.push("Invalid scantable")
360            asaplog.post("ERROR")
361        if isinstance(rows,list) and len(rows) > 0:
362            scan.flag_row(rows=rows,unflag=unflag)
363        if isinstance(regions,dict) and len(regions) > 0:
364            for srow, masklist in regions.iteritems():
365                if not isinstance(masklist,list) or len(masklist) ==0:
366                    msg = "Ignoring invalid region selection for row = "+srow
367                    asaplog.post()
368                    asaplog.push(msg)
369                    asaplog.post("WARN")
370                    continue
371                irow = int(srow)
372                mask = scan.create_mask(masklist,invert=False,row=irow)
373                scan.flag(row=irow,mask=mask,unflag=unflag)
374                del irow, mask
375            del srow, masklist
376        del scan
377
378    ### show statistics of selected spectra/regions
379    @asaplog_post_dec
380    def stat_cal(self):
381        if not self._any_selection():
382            msg = "No selection to be calculated"
383            asaplog.push(msg)
384            asaplog.post('WARN')
385            return
386        #print "Calculate statistics of selected region and spectra"
387        self._selected_stats(rows=self._selpanels,regions=self._selregions)
388        #print "Statistics: "
389        #print "- rows: ",self._selpanels
390        #print "- regions:\n   ",self._selregions
391        self._clearup_selections()
392
393    @asaplog_post_dec
394    def _selected_stats(self,rows=None,regions=None):
395        scan = self.plotter._data
396        if not scan:
397            asaplog.post()
398            asaplog.push("Invalid scantable")
399            asaplog.post("ERROR")
400        mathobj = stmath( rcParams['insitu'] )
401        statval={}
402        statstr = ['max', 'min', 'mean', 'median', 'sum', 'stddev', 'rms']
403        if isinstance(rows,list) and len(rows) > 0:
404            for irow in rows:
405                for stat in statstr:
406                    #statval[stat] = scan.stats(stat=stat,row=irow)[0]
407                    statval[stat] = mathobj._statsrow(scan,[],stat,irow)[0]
408                self._print_stats(scan,irow,statval,statstr=statstr)
409            del irow
410        if isinstance(regions,dict) and len(regions) > 0:
411            for srow, masklist in regions.iteritems():
412                if not isinstance(masklist,list) or len(masklist) ==0:
413                    msg = "Ignoring invalid region selection for row = "+srow
414                    asaplog.post()
415                    asaplog.push(msg)
416                    asaplog.post("WARN")
417                    continue
418                irow = int(srow)
419                mask = scan.create_mask(masklist,invert=False,row=irow)
420                for stat in statstr:
421                    #statval[stat] = scan.stats(stat=stat,row=irow,mask = mask)[0]
422                    statval[stat] = mathobj._statsrow(scan,mask,stat,irow)[0]
423                self._print_stats(scan,irow,statval,statstr=statstr,mask=masklist)
424                del irow, mask
425            del srow, masklist
426        del scan, statval, mathobj
427
428    @asaplog_post_dec
429    def _print_stats(self,scan,row,stats,statstr=None,mask=None):
430        if not isinstance(scan, scantable):
431            asaplog.post()
432            asaplog.push("Invalid scantable")
433            asaplog.post("ERROR")
434        if row < 0 or row > scan.nrow():
435            asaplog.post()
436            asaplog.push("Invalid row number")
437            asaplog.post("ERROR")
438        if not isinstance(stats,dict) or len(stats) == 0:
439            asaplog.post()
440            asaplog.push("Invalid statistic value")
441            asaplog.post("ERROR")
442        maskstr = "All"
443        if mask:
444            maskstr = str(mask)
445        ssep = "-"*70+"\n"
446        sout = ssep
447        sout += ("Row=%d  Scan=%d  IF=%d  Pol=%d  Time=%s  mask=%s" % \
448                 (row, scan.getscan(row), scan.getif(row), scan.getpol(row), scan.get_time(row),maskstr))
449        sout += "\n"
450        statvals = []
451        if not len(statstr):
452            statstr = stats.keys()
453        for key in statstr:
454            sout += key.ljust(10)
455            statvals.append(stats.pop(key))
456        sout += "\n"
457        sout += ("%f "*len(statstr) % tuple(statvals))
458        sout += "\n"+ssep
459        asaplog.push(sout)
460        del sout, ssep, maskstr, statvals, key, scan, row, stats, statstr, mask
461
462    ### Page chages
463    ### go to the previous page
464    def prev_page(self):
465        self.figmgr.toolbar.set_message('plotting the previous page')
466        self._clear_selection_plot()
467        self._plot_page(pagemode="prev")
468        self._plot_selections()
469
470    ### go to the next page
471    def next_page(self):
472        self.figmgr.toolbar.set_message('plotting the next page')
473        self._clear_selection_plot()
474        self._plot_page(pagemode="next")
475        self._plot_selections()
476
477    ### actual plotting of the new page
478    def _plot_page(self,pagemode="next"):
479        if self.plotter._startrow <= 0:
480            msg = "The page counter is reset due to chages of plot settings. "
481            msg += "Plotting from the first page."
482            asaplog.post()
483            asaplog.push(msg)
484            asaplog.post('WARN')
485            goback = False
486           
487        self.plotter._plotter.hold()
488        self.plotter._plotter.legend(1)
489        #if goback:
490        #    self._set_prevpage_counter()
491        self._set_plot_counter(pagemode)
492        #self.plotter._plotter.clear()
493        self.plotter._plot(self.plotter._data)
494        self.set_pagecounter(self._get_pagenum())
495        self.plotter._plotter.release()
496        self.plotter._plotter.tidy()
497        self.plotter._plotter.show(hardrefresh=False)
498
499    ### calculate the panel ID and start row to plot a page
500    #def _set_prevpage_counter(self):
501    def _set_plot_counter(self, pagemode):
502        ## page operation should be either "previous", "current", or "next"
503        availpage = ["p","c","n"]
504        pageop = pagemode[0].lower()
505        if not (pageop in availpage):
506            asaplog.post()
507            asaplog.push("Invalid page operation")
508            asaplog.post("ERROR")
509        if pageop == "n":
510            # nothing necessary to plot the next page
511            return
512        # set row and panel counters to those of the 1st panel of previous page
513        maxpanel = 16
514        # the ID of the last panel in current plot
515        lastpanel = self.plotter._ipanel
516        # the number of current subplots
517        currpnum = len(self.plotter._plotter.subplots)
518       
519        # the nuber of previous subplots
520        start_ipanel = None
521        if pageop == "c":
522            start_ipanel = max(lastpanel-currpnum+1, 0)
523        else:
524            ## previous page
525            prevpnum = None
526            if self.plotter._rows and self.plotter._cols:
527                # when user set layout
528                prevpnum = self.plotter._rows*self.plotter._cols
529            else:
530                # no user specification
531                prevpnum = maxpanel
532            start_ipanel = max(lastpanel-currpnum-prevpnum+1, 0)
533            del prevpnum
534
535        # set the pannel ID of the last panel of the prev(-prev) page
536        self.plotter._ipanel = start_ipanel-1
537        if self.plotter._panelling == 'r':
538            self.plotter._startrow = start_ipanel
539        else:
540            # the start row number of the next panel
541            self.plotter._startrow = self.plotter._panelrows[start_ipanel]
542        del lastpanel,currpnum,start_ipanel
543
544    ### refresh the page counter
545    def set_pagecounter(self,page):
546        nwidth = int(numpy.ceil(numpy.log10(max(page,1))))+1
547        nwidth = max(nwidth,4)
548        formatstr = '%'+str(nwidth)+'d'
549        self.show_pagenum(page,formatstr)
550
551    def show_pagenum(self,pagenum,formatstr):
552        # passed to backend dependent class
553        pass       
554
555    def _get_pagenum(self):
556        maxpanel = 16
557        # get the ID of last panel in the current page
558        idlastpanel = self.plotter._ipanel
559        if self.plotter._rows and self.plotter._cols:
560            ppp = self.plotter._rows*self.plotter._cols
561        else:
562            ppp = maxpanel
563        return int(idlastpanel/ppp)+1
564
565#####################################
566##    Backend dependent Classes    ##
567#####################################
568### TkAgg
569if matplotlib.get_backend() == 'TkAgg':
570    import Tkinter as Tk
571    from notationwindow import NotationWindowTkAgg
572
573class CustomFlagToolbarTkAgg(CustomFlagToolbarCommon, Tk.Frame):
574    def __init__(self,parent):
575        from asap.asapplotter import asapplotter
576        if not isinstance(parent,asapplotter):
577            return False
578        if not parent._plotter:
579            return False
580        self._p = parent._plotter
581        self.figmgr = self._p.figmgr
582        self.canvas = self.figmgr.canvas
583        self.mode = ''
584        self.button = True
585        self.pagecount = None
586        CustomFlagToolbarCommon.__init__(self,parent)
587        self.notewin=NotationWindowTkAgg(master=self.canvas)
588        self._add_custom_toolbar()
589
590    def _add_custom_toolbar(self):
591        Tk.Frame.__init__(self,master=self.figmgr.window)
592        #self.bSpec=self._NewButton(master=self,
593        #                           text='spec value',
594        #                           command=self.spec_show)
595
596        self.bRegion=self._NewButton(master=self,
597                                   text='region',
598                                   command=self.select_region)
599        self.bPanel=self._NewButton(master=self,
600                                   text='panel',
601                                   command=self.select_panel)
602        self.bClear=self._NewButton(master=self,
603                                   text='clear',
604                                   command=self.cancel_select)
605        self.bFlag=self._NewButton(master=self,
606                                   text='flag',
607                                   command=self.flag)
608        self.bUnflag=self._NewButton(master=self,
609                                   text='unflag',
610                                   command=self.unflag)
611       
612        self.bStat=self._NewButton(master=self,
613                                   text='statistics',
614                                   command=self.stat_cal)
615
616        self.bNote=self._NewButton(master=self,
617                                   text='notation',
618                                   command=self.modify_note)
619
620        self.bQuit=self._NewButton(master=self,
621                                   text='Quit',
622                                   #file="stock_close.ppm",
623                                   command=self.quit,
624                                   side=Tk.RIGHT)
625       
626        # page change oparations
627        frPage = Tk.Frame(master=self,borderwidth=2,relief=Tk.GROOVE)
628        frPage.pack(ipadx=2,padx=10,side=Tk.RIGHT)
629        self.lPagetitle = Tk.Label(master=frPage,text='Page:',padx=5)
630                                   #width=8,anchor=Tk.E,padx=5)
631        self.lPagetitle.pack(side=Tk.LEFT)
632        self.pagecount = Tk.StringVar(master=frPage)
633        self.lPagecount = Tk.Label(master=frPage,
634                                   textvariable=self.pagecount,
635                                   padx=5,bg='white')
636        self.lPagecount.pack(side=Tk.LEFT,padx=3)
637       
638        self.bNext=self._NewButton(master=frPage,
639                                   text=' + ',
640                                   #file="hand.ppm",
641                                   command=self.next_page)
642        self.bPrev=self._NewButton(master=frPage,
643                                   text=' - ',
644                                   command=self.prev_page)
645
646        if os.uname()[0] != 'Darwin':
647            self.bPrev.config(padx=5)
648            self.bNext.config(padx=5)
649
650        self.pack(side=Tk.BOTTOM,fill=Tk.BOTH)
651        self.pagecount.set(' '*4)
652
653        self.disable_button()
654        return #self
655
656    def _NewButton(self, master, text, command, side=Tk.LEFT,file=None):
657        img = None
658        if file:
659            file = os.path.join(matplotlib.rcParams['datapath'], 'images', file)
660            img = Tk.PhotoImage(master=master, file=file)
661           
662        if os.uname()[0] == 'Darwin':
663            b = Tk.Button(master=master, text=text, image=img,
664                          command=command)
665            if img: b.image = img
666        else:
667            b = Tk.Button(master=master, text=text, image=img, padx=2, pady=2,
668                          command=command)
669            if img: b.image = img
670        b.pack(side=side)
671        return b
672
673    def show_pagenum(self,pagenum,formatstr):
674        self.pagecount.set(formatstr % (pagenum))
675
676    def spec_show(self):
677        if not self.figmgr.toolbar.mode == '' or not self.button: return
678        self.figmgr.toolbar.set_message('spec value: drag on a spec')
679        if self.mode == 'spec': return
680        self.mode='spec'
681        self.notewin.close_widgets()
682        self.__disconnect_event()
683        self._p.register('button_press',self._select_spectrum)
684
685    def modify_note(self):
686        if not self.figmgr.toolbar.mode == '': return
687        self.figmgr.toolbar.set_message('text: select a position/text')
688        if self.mode == 'note':
689            self.bNote.config(relief='raised')
690            self.mode='none'
691            self.spec_show()
692            return
693        self.bNote.config(relief='sunken')
694        self.bRegion.config(relief='raised')
695        self.bPanel.config(relief='raised')
696        self.mode='note'
697        self.__disconnect_event()
698        self._p.register('button_press',self._mod_note)
699
700    def select_region(self):
701        if not self.figmgr.toolbar.mode == '' or not self.button: return
702        self.figmgr.toolbar.set_message('select rectangle regions')
703        if self.mode == 'region':
704            self.bRegion.config(relief='raised')
705            self.mode='none'
706            self.spec_show()
707            return
708        self.bNote.config(relief='raised')
709        self.bRegion.config(relief='sunken')
710        self.bPanel.config(relief='raised')
711        self.mode='region'
712        self.notewin.close_widgets()
713        self.__disconnect_event()
714        self._p.register('button_press',self._add_region)
715
716    def select_panel(self):
717        if not self.figmgr.toolbar.mode == '' or not self.button: return
718        self.figmgr.toolbar.set_message('select panels')
719        if self.mode == 'panel':
720            self.bPanel.config(relief='raised')
721            self.mode='none'
722            self.spec_show()
723            return
724        self.bNote.config(relief='raised')
725        self.bRegion.config(relief='raised')
726        self.bPanel.config(relief='sunken')
727        self.mode='panel'
728        self.notewin.close_widgets()
729        self.__disconnect_event()
730        self._p.register('button_press',self._add_panel)
731
732    def quit(self):
733        self.__disconnect_event()
734        #self.delete_bar()
735        self.disable_button()
736        self.figmgr.window.wm_withdraw()
737
738    def enable_button(self):
739        if self.button: return
740        self.bRegion.config(state=Tk.NORMAL)
741        self.bPanel.config(state=Tk.NORMAL)
742        self.bClear.config(state=Tk.NORMAL)
743        self.bFlag.config(state=Tk.NORMAL)
744        self.bUnflag.config(state=Tk.NORMAL)
745        self.bStat.config(state=Tk.NORMAL)
746        self.button=True
747        self.spec_show()
748
749    def disable_button(self):
750        ## disable buttons which don't work for plottp
751        if not self.button: return
752        self.bRegion.config(relief='raised', state=Tk.DISABLED)
753        self.bPanel.config(relief='raised', state=Tk.DISABLED)
754        self.bClear.config(state=Tk.DISABLED)
755        self.bFlag.config(state=Tk.DISABLED)
756        self.bUnflag.config(state=Tk.DISABLED)
757        self.bStat.config(state=Tk.DISABLED)
758        self.bNext.config(state=Tk.DISABLED)
759        self.bPrev.config(state=Tk.DISABLED)
760        self.button=False
761        self.mode=''
762        self.notewin.close_widgets()
763        self.__disconnect_event()
764
765    def enable_next(self):
766        self.bNext.config(state=Tk.NORMAL)
767
768    def disable_next(self):
769        self.bNext.config(state=Tk.DISABLED)
770
771    def enable_prev(self):
772        self.bPrev.config(state=Tk.NORMAL)
773
774    def disable_prev(self):
775        self.bPrev.config(state=Tk.DISABLED)
776
777    def delete_bar(self):
778        self.__disconnect_event()
779        self.destroy()
780
781    def __disconnect_event(self):
782        self._p.register('button_press',None)
783        self._p.register('button_release',None)
Note: See TracBrowser for help on using the repository browser.