Context Navigation

← Previous Revision
Next Revision →
Blame
Revision Log

customgui_qt4agg.py

Visit:

Last change on this file was 2603, checked in by Kana Sugimoto, 12 years ago

New Development: No

JIRA Issue: No (bug fixes and a minor improvement)

Ready for Test: Yes

Interface Changes: No

What Interface Changed:

Test Programs: interactive test needed.

Put in Release Notes: No

Module(s): asapplotter, flagplotter, and sdplot (CASA)

Description:

Fixed bugs in asapplotter and flagplotter which prevented users from exiting
from notation mode, when the other buttons in the additional toolbar is not active.
Consintently use 'self._data' in asapplotter.plotgrid() instead of 'scan'.

File size: 29.7 KB

Line
1	import os
2	import matplotlib, numpy
3	from asap.logging import asaplog, asaplog_post_dec
4	from matplotlib.patches import Rectangle
5	from asap.parameters import rcParams
6	from asap._asap import stmath
7	from asap.customgui_base import *
8
9	import PyQt4 as qt
10
11	######################################
12	## Add CASA custom toolbar ##
13	######################################
14	class CustomToolbarQT4Agg(CustomToolbarCommon, qt.QtGui.QToolBar):
15	def __init__(self,parent):
16	from asap.asapplotter import asapplotter
17	if not isinstance(parent,asapplotter):
18	return False
19	if not parent._plotter:
20	return False
21	self._p = parent._plotter
22	self.figmgr = self._p.figmgr
23	self.canvas = self.figmgr.canvas
24	self.mode = ''
25	self.button = True
26	self.pagecount = None
27	CustomToolbarCommon.__init__(self,parent)
28	self.notewin = NotationWindowQT4Agg(master=self.canvas)
29	self._add_custom_toolbar()
30
31	def _add_custom_toolbar(self):
32	qt.QtGui.QToolBar.__init__(self,parent=self.figmgr.window)
33	self.figmgr.window.addToolBar(qt.QtCore.Qt.BottomToolBarArea,self)
34	self.bNote = self._NewButton(master=self,
35	text='notation',
36	command=self.modify_note,
37	balloon="add note on plot")
38	self.bNote.setCheckable(True)
39
40	self.bStat = self._NewButton(master=self,
41	text='statistics',
42	command=self.stat_cal,
43	balloon="calculate statistics")
44	self.bStat.setCheckable(True)
45
46	# page change oparations
47	frPage = qt.QtGui.QWidget(parent=self,flags=qt.QtCore.Qt.Tool)
48	loPage = qt.QtGui.QHBoxLayout(self)
49	loPage.addStretch(1)
50	self.lPagetitle = qt.QtGui.QLabel('Page:',parent=frPage)
51	self.lPagetitle.setMargin(5)
52	loPage.addWidget(self.lPagetitle)
53	self.pagecount = qt.QtGui.QLabel(parent=frPage)
54	self.pagecount.setStyleSheet("background-color: white")
55	self.pagecount.setMargin(3)
56	self.pagecount.setText(' 1')
57	loPage.addWidget(self.pagecount)
58
59	self.bNext = self._NewButton(master=frPage,
60	text=' + ',
61	command=self.next_page,
62	addit=False,
63	balloon="plot next page")
64	loPage.addWidget(self.bNext)
65	self.bPrev = self._NewButton(master=frPage,
66	text=' - ',
67	command=self.prev_page,addit=False,
68	balloon="plot previous page")
69	loPage.addWidget(self.bPrev)
70	frPage.setLayout(loPage)
71	self.addWidget(frPage)
72
73	self.bQuit = self._NewButton(master=self,
74	text='Quit',
75	command=self.quit,
76	balloon="Close window")
77
78	self.pagecount.setText(' '*4)
79
80	self.disable_button()
81	return
82
83	def _NewButton(self, master, text, command, balloon=None,addit=True):
84	b = qt.QtGui.QPushButton(text,parent=master)
85	if balloon: b.setToolTip(balloon)
86	if addit: master.addWidget(b)
87	master.connect(b,qt.QtCore.SIGNAL('clicked()'),command)
88	return b
89
90	def show_pagenum(self,pagenum,formatstr):
91	self.pagecount.setText(formatstr % (pagenum))
92
93	def spec_show(self):
94	if not self.figmgr.toolbar.mode == '' or not self.button: return
95	self.figmgr.toolbar.set_message("spec value: drag on a spec")
96	if self.mode == 'spec': return
97	self.mode = 'spec'
98	self.notewin.close_widgets()
99	self.__disconnect_event()
100	self._p.register('button_press',self._select_spectrum)
101
102	def stat_cal(self):
103	if not self.figmgr.toolbar.mode == '' or not self.button:
104	# Get back button status BEFORE clicked
105	self.bStat.setChecked(not self.bStat.isChecked())
106	return
107	if self.mode == 'stat':
108	# go back to spec mode
109	self.bStat.setChecked(False)
110	self.bStat.setToolTip("calculate statistics")
111	self.spec_show()
112	return
113	self.figmgr.toolbar.set_message("statistics: click at start and end channels")
114	self.bStat.setChecked(True)
115	self.bStat.setToolTip("Back to spec value mode")
116	self.bNote.setChecked(False)
117	self.bNote.setToolTip("add note on plot")
118	self.mode = 'stat'
119	self.notewin.close_widgets()
120	self.__disconnect_event()
121	self._p.register('button_press',self._single_mask)
122
123	def modify_note(self):
124	if not self.figmgr.toolbar.mode == '':
125	# Get back button status BEFORE clicked
126	self.bNote.setChecked(not self.bNote.isChecked())
127	return
128	if self.mode == 'note':
129	self.bNote.setChecked(False)
130	self.bNote.setToolTip("add note on plot")
131	self.mode = 'none'
132	self.spec_show()
133	if not self.button:
134	self.notewin.close_widgets()
135	self.__disconnect_event()
136	return
137	self.figmgr.toolbar.set_message("text: select a position/text")
138	self.bStat.setChecked(False)
139	self.bStat.setToolTip("calculate statistics")
140	self.bNote.setChecked(True)
141	self.bNote.setToolTip("Back to spec value mode")
142	self.mode = 'note'
143	self.__disconnect_event()
144	self._p.register('button_press',self._mod_note)
145
146	def quit(self):
147	self.__disconnect_event()
148	self.disable_button()
149	self._p.quit()
150
151	def enable_button(self):
152	if self.button: return
153	self.bStat.setEnabled(True)
154	self.button = True
155	self.spec_show()
156
157	def disable_button(self):
158	if not self.button: return
159	self.bStat.setChecked(False)
160	self.bStat.setDisabled(True)
161	self.button = False
162	self.mode = ''
163	self.__disconnect_event()
164
165	def enable_next(self):
166	self.bNext.setEnabled(True)
167
168	def disable_next(self):
169	self.bNext.setDisabled(True)
170
171	def enable_prev(self):
172	self.bPrev.setEnabled(True)
173
174	def disable_prev(self):
175	self.bPrev.setDisabled(True)
176
177	# pause buttons for slow operations
178	def _pause_buttons(self,operation="end",msg=""):
179	buttons = ["bStat","bNote","bQuit"]
180	if operation == "start":
181	enable = False
182	else:
183	enable = True
184	for btn in buttons:
185	getattr(self,btn).setEnabled(enable)
186	self.figmgr.toolbar.set_message(msg)
187
188	def delete_bar(self):
189	self.__disconnect_event()
190	self.destroy()
191
192	def __disconnect_event(self):
193	self._p.register('button_press',None)
194	self._p.register('button_release',None)
195
196	def _draw_span(self,axes,x0,x1,**kwargs):
197	height = self._p.figure.bbox.height
198	y1 = height - axes.bbox.y1
199	h = axes.bbox.height
200	w = abs(x1 - x0)
201	rect = [ int(val) for val in min(x0,x1), y1, w, h ]
202	self._p.canvas.drawRectangle( rect )
203	# nothing is returned by drawRectangle
204	return None
205
206	def _remove_span(self,span):
207	# Nothing to do with remove. just refresh (call only once)
208	self.canvas.draw()
209
210
211
212	######################################
213	## Notation box window ##
214	######################################
215	class NotationWindowQT4Agg(NotationWindowCommon):
216	"""
217	Backend based class to create widgets to add, modify, or delete
218	note on the plot.
219
220	Note:
221	Press LEFT-mouse button on the plot to ADD a note on the canvas.
222	A notation window will be loaded for specifying note string and
223	anchor. The note will be anchored on a position in whether figure-
224	(0-1 relative in a figure), panel- (0-1 relative in a plot axes),
225	or data-coordinate (data value in a plot axes).
226	Press RIGHT-mouse button on a note to MODIFY/DELETE it. A cascade
227	menu will be displayed and you can select an operation.
228	"""
229	def __init__(self,master=None):
230	self.parent = master
231	NotationWindowCommon.__init__(self,master=master)
232	self.anchval = None
233	self.textwin = self._create_textwindow(master=None)
234	self.menu = self._create_modmenu(master=self.parent)
235
236	### Notation window widget
237	def _create_textwindow(self,master=None):
238	"""Create notation window widget and iconfy it"""
239	#twin = qt.QtGui.QWidget(parent=master, flags=qt.QtCore.Qt.Popup)
240	twin = qt.QtGui.QWidget(parent=master, flags=qt.QtCore.Qt.Dialog)
241	twin.setWindowTitle("Notation")
242	self.textbox = self._NotationBox(parent=twin)
243	radiobox = self._AnchorRadio(parent=twin)
244	self.actionbs = self._ActionButtons(parent=twin)
245	vbox = qt.QtGui.QVBoxLayout(twin)
246	vbox.addWidget(self.textbox)
247	vbox.addWidget(radiobox)
248	vbox.addLayout(self.actionbs)
249	twin.setLayout(vbox)
250	#twin.setCentralWidget(self.textbox)
251	twin.hide()
252	return twin
253
254	def _NotationBox(self,parent=None):
255	textbox = qt.QtGui.QPlainTextEdit(parent=parent)
256	textbox.setStyleSheet("background-color: white")
257	fmetric = qt.QtGui.QFontMetrics(textbox.currentCharFormat().font())
258	textbox.resize(fmetric.width("A")20+fmetric.leading()2,
259	fmetric.height()*2+fmetric.ascent()+fmetric.descent())
260	del fmetric
261	textbox.setMinimumSize(textbox.size())
262	textbox.setUndoRedoEnabled(True)
263	textbox.setMidLineWidth(3)
264	textbox.setFrameShadow(qt.QtGui.QFrame.Sunken)
265	textbox.setCursor(qt.QtCore.Qt.IBeamCursor)
266	textbox.setFocus()
267	return textbox
268
269	def _AnchorRadio(self,parent=None):
270	# Returns a QGoupBox object which includes radio butons to
271	# select an anchor
272	anchbox = qt.QtGui.QGroupBox("anchor",parent=parent)
273	self.radio = qt.QtGui.QButtonGroup(parent=anchbox)
274	self.rFig = self._NewRadioButton(anchbox,"figure",\
275	bgr=self.radio,value=0,\
276	balloon="a fixed position in figure")
277	self.rAxis = self._NewRadioButton(anchbox,"panel",\
278	bgr=self.radio,value=1,\
279	balloon="a fixed realtive position in subplot")
280	self.rData = self._NewRadioButton(anchbox,"data",\
281	bgr=self.radio,value=2,\
282	balloon="a fixed data position in subplot")
283	hbox = qt.QtGui.QHBoxLayout(anchbox)
284	hbox.addWidget(self.rFig)
285	hbox.addWidget(self.rAxis)
286	hbox.addWidget(self.rData)
287	anchbox.setLayout(hbox)
288	# set initial selection "figure"
289	self.rFig.setChecked(True)
290	self.radio.setExclusive(True)
291	self.anchval = self.radio.checkedId()
292	return anchbox
293
294	def _NewRadioButton(self,parent,text,balloon=None,bgr=None,value=None):
295	rb= qt.QtGui.QRadioButton(text,parent=parent)
296	if bgr:
297	if value is not None:
298	bgr.addButton(rb,value)
299	else:
300	bgr.addButton(rb)
301	if balloon: rb.setToolTip(balloon)
302	return rb
303
304	def _enable_radio(self):
305	"""Enable 'panel' and 'data' radio button"""
306	self.rAxis.setEnabled(True)
307	self.rData.setEnabled(True)
308	# select Figure as the default value
309	self.rFig.setChecked(True)
310	self.anchval = self.radio.checkedId()
311
312	def _reset_radio(self):
313	"""Disable 'panel' and 'data' radio button"""
314	self.rAxis.setDisabled(True)
315	self.rData.setDisabled(True)
316	self.rFig.setEnabled(True)
317	# select Figure as the default value
318	self.rFig.setChecked(True)
319	self.anchval = self.radio.checkedId()
320
321	def _select_radio(self,selection):
322	"""Select a specified radio button"""
323	if not selection in self.anchors:
324	return
325	if selection == "data":
326	self.rData.setChecked(True)
327	elif selection == "axes":
328	self.rAxis.setChecked(True)
329	else:
330	self.rFig.setChecked(True)
331	self.anchval = self.radio.checkedId()
332
333	def _get_anchval(self):
334	"""Returns a integer of a selected radio button"""
335	self.anchval = self.radio.checkedId()
336	return self.anchval
337
338	def _get_note(self):
339	"""Returns a note string specified in the text box"""
340	return str(self.textbox.toPlainText())
341
342	def _clear_textbox(self):
343	"""Clear the text box"""
344	self.textbox.clear()
345
346	def _set_note(self,note=None):
347	"""Set a note string to the text box"""
348	self._clear_textbox()
349	if len(note) >0:
350	self.textbox.setPlainText(note)
351
352	def _ActionButtons(self,parent=None):
353	# Returns a layout object which includes "cancel" and "print" buttons
354	actbuts = qt.QtGui.QHBoxLayout()
355	bCancel = self._NewButton(parent,"cancel",self._cancel_text,\
356	addit=False,\
357	balloon="cancel printing/modifying")
358	bPrint = self._NewButton(parent,"print", self._print_text,\
359	addit=False,\
360	balloon="print text on plot")
361	actbuts.addWidget(bCancel)
362	actbuts.addWidget(bPrint)
363	return actbuts
364
365	def _NewButton(self, parent, text, command, balloon=None, addit=True):
366	b = qt.QtGui.QPushButton(text,parent=parent)
367	if balloon: b.setToolTip(balloon)
368	if addit: parent.addWidget(b)
369	parent.connect(b,qt.QtCore.SIGNAL('clicked()'),command)
370	return b
371
372	def _cancel_text(self):
373	"""
374	Cancel adding/modifying a note and close notaion window.
375	called when 'cancel' is selected.
376	"""
377	self.close_textwindow()
378
379	def _print_text(self):
380	"""
381	Add/Modify a note. Called when 'print' is selected on the
382	notation window.
383	"""
384	self.print_text()
385	self.close_textwindow()
386
387	def load_textwindow(self,event):
388	"""
389	Load text window at a event position to add a note on a plot.
390	Parameter:
391	event: an even object to specify the position to load
392	text window.
393	"""
394	self.close_modmenu()
395	if event.canvas != self.parent:
396	raise RuntimeError, "Got invalid event!"
397
398	self.event = event
399	is_ax = (event.inaxes != None)
400	(xpix, ypix) = self._disppix2screen(event.x, event.y)
401	offset = 5
402	self.show_textwindow(xpix+offset,ypix+offset,enableaxes=is_ax)
403
404	def show_textwindow(self,xpix,ypix,basetext=None,enableaxes=False):
405	"""
406	Load text window at a position of screen to add a note on a plot.
407	Parameters:
408	xpix, ypix: a pixel position from Upper-left corner
409	of the screen.
410	basetext: None (default) or any string.
411	A string to be printed on text box when loaded.
412	enableaxes: False (default) or True.
413	If True, 'panel' & 'data' radio button is enabled.
414	"""
415	if not self.textwin: return
416	self._reset_radio()
417	if enableaxes:
418	self._enable_radio()
419	self.textwin.activateWindow()
420	h = self.textwin.minimumHeight()
421	w = self.textwin.minimumWidth()
422	self.textwin.resize(w,h)
423	self.textwin.move(xpix,ypix)
424	self.textbox.setFocus()
425	self.textwin.raise_()
426	self.textwin.show()
427	if w*h <= 1: # Initial load
428	self.textwin.setMinimumSize(self.textwin.size())
429
430	def close_textwindow(self):
431	"""Close text window."""
432	self.seltext = {}
433	self._reset_radio()
434	self._clear_textbox()
435	self.textwin.hide()
436
437
438	### Modify/Delete menu widget
439	def _create_modmenu(self,master=None):
440	"""Create modify/delete menu widget"""
441	if master:
442	self.parent = master
443	if not self.parent:
444	return False
445	menu = qt.QtGui.QMenu(parent=self.parent)
446	menu.setTearOffEnabled(False)
447	menu.addAction("Modify",self._modify_note)
448	menu.addAction("Delete",self._delnote_dialog)
449	return menu
450
451	def load_modmenu(self,event):
452	"""
453	Load cascade menu at a event position to modify or delete
454	selected text.
455	Parameter:
456	event: an even object to specify the position to load
457	text window.
458	"""
459	self.close_textwindow()
460	self.seltext = self._get_selected_text(event)
461	if len(self.seltext) == 3:
462	canvas = event.canvas
463	corig = canvas.mapToGlobal(qt.QtCore.QPoint(0,0))
464	xpixs = corig.x() + int(event.x)
465	ypixs = corig.y() + canvas.height() - int(event.y)
466	self.menu.activateWindow()
467	self.menu.move(xpixs,ypixs)
468	self.menu.show()
469
470	def close_modmenu(self):
471	"""Close cascade menu."""
472	self.seltext = {}
473	self.menu.hide()
474
475	### load text window for modification
476	def _modify_note(self):
477	"""helper function to load text window to modify selected note"""
478	textobj = self.seltext['textobj']
479	(xtx, ytx) = textobj._get_xy_display()
480	is_ax = (self.seltext['anchor'] != 'figure')
481	if not is_ax:
482	# previous anchor is figure
483	pos = textobj.get_position()
484	is_ax = (self._get_axes_from_pos(pos,self.canvas) != None)
485
486	(xpix, ypix) = self._disppix2screen(xtx,ytx)
487	offset = int(textobj.get_size())*2
488	self.show_textwindow(xpix,ypix+offset,basetext=textobj.get_text(),\
489	enableaxes=is_ax)
490	self._select_radio(self.seltext['anchor'])
491	self._set_note(textobj.get_text())
492
493	### close all widgets
494	def close_widgets(self):
495	"""Close note window and menu"""
496	self.close_textwindow()
497	self.close_modmenu()
498
499	### dialog to confirm deleting note
500	def _delnote_dialog(self):
501	"""Load dialog to confirm deletion of the text"""
502	remind = "Delete text?\n '"+self.seltext['textobj'].get_text()+"'"
503	from PyQt4.QtGui import QMessageBox as mbox
504	answer = mbox.question(self.parent,"Delete?",remind,
505	buttons = mbox.Ok \| mbox.Cancel,
506	defaultButton=mbox.Cancel)
507	if answer == mbox.Ok:
508	self.delete_note()
509	else:
510	self.cancel_delete()
511
512	### helper functions
513	def _disppix2screen(self,xpixd,ypixd):
514	"""
515	helper function to calculate a pixel position form Upper-left
516	corner of the SCREEN from a pixel position (xpixd, ypixd)
517	from Lower-left of the CANVAS (which, e.g., event.x/y returns)
518
519	Returns:
520	(x, y): pixel position from Upper-left corner of the SCREEN.
521	"""
522	corig = self.parent.mapToGlobal(qt.QtCore.QPoint(0,0))
523	xpixs = corig.x() + xpixd
524	ypixs = corig.y() + self.parent.height() - ypixd
525	return (int(xpixs), int(ypixs))
526
527
528
529
530
531
532	###########################################
533	## Add CASA custom Flag toolbar ##
534	###########################################
535	class CustomFlagToolbarQT4Agg(CustomFlagToolbarCommon, qt.QtGui.QToolBar):
536	def __init__(self,parent):
537	from asap.asapplotter import asapplotter
538	if not isinstance(parent,asapplotter):
539	return False
540	if not parent._plotter:
541	return False
542	self._p = parent._plotter
543	self.figmgr = self._p.figmgr
544	self.canvas = self.figmgr.canvas
545	self.mode = ''
546	self.button = True
547	self.pagecount = None
548	CustomFlagToolbarCommon.__init__(self,parent)
549	self.notewin=NotationWindowQT4Agg(master=self.canvas)
550	self._add_custom_toolbar()
551
552	def _add_custom_toolbar(self):
553	qt.QtGui.QToolBar.__init__(self,parent=self.figmgr.window)
554	self.figmgr.window.addToolBar(qt.QtCore.Qt.BottomToolBarArea,self)
555	self.bRegion = self._NewButton(master=self,
556	text='region',
557	command=self.select_region,
558	balloon="select channel regions")
559	self.bRegion.setCheckable(True)
560
561	self.bPanel = self._NewButton(master=self,
562	text='panel',
563	command=self.select_panel,
564	balloon="select a whole spectrum")
565	self.bPanel.setCheckable(True)
566
567	self.bClear = self._NewButton(master=self,
568	text='clear',
569	command=self.cancel_select,
570	balloon="clear selections")
571
572	self.bFlag = self._NewButton(master=self,
573	text='flag',
574	command=self.flag,
575	balloon="flag selections")
576
577	self.bUnflag = self._NewButton(master=self,
578	text='unflag',
579	command=self.unflag,
580	balloon="unflag selections")
581
582	self.bStat = self._NewButton(master=self,
583	text='statistics',
584	command=self.stat_cal,
585	balloon="print statistics of selections")
586
587	self.bNote = self._NewButton(master=self,
588	text='notation',
589	command=self.modify_note,
590	balloon="add note on plot")
591	self.bNote.setCheckable(True)
592
593	# page change oparations
594	frPage = qt.QtGui.QWidget(parent=self,flags=qt.QtCore.Qt.Tool)
595	loPage = qt.QtGui.QHBoxLayout(self)
596	loPage.addStretch(1)
597	self.lPagetitle = qt.QtGui.QLabel('Page:',parent=frPage)
598	self.lPagetitle.setMargin(5)
599	loPage.addWidget(self.lPagetitle)
600	self.pagecount = qt.QtGui.QLabel(parent=frPage)
601	self.pagecount.setStyleSheet("background-color: white")
602	self.pagecount.setMargin(3)
603	self.pagecount.setText(' 1')
604	loPage.addWidget(self.pagecount)
605
606	self.bNext = self._NewButton(master=frPage,
607	text='+',
608	#imagename="hand.ppm",
609	command=self.next_page,
610	addit=False,
611	balloon="plot next page")
612	loPage.addWidget(self.bNext)
613	self.bPrev = self._NewButton(master=frPage,
614	text='-',
615	command=self.prev_page,
616	addit=False,
617	balloon="plot previous page")
618	loPage.addWidget(self.bPrev)
619	frPage.setLayout(loPage)
620	self.addWidget(frPage)
621
622	self.bQuit = self._NewButton(master=self,
623	text='Quit',
624	#imagename="stock_close.ppm",
625	command=self.quit,
626	balloon="Close window")
627
628	self.pagecount.setText(' '*4)
629	self.disable_button()
630	return
631
632	def _NewButton(self, master, text, command, balloon=None,addit=True,imagename=None):
633	img = None
634	if imagename:
635	imagename = os.path.join(matplotlib.rcParams['datapath'], 'images', imagename)
636	# img = Tk.PhotoImage(master=master, file=imagename)
637
638	b = qt.QtGui.QPushButton(text,parent=master)
639	if balloon: b.setToolTip(balloon)
640	if addit: master.addWidget(b)
641	master.connect(b,qt.QtCore.SIGNAL('clicked()'),command)
642	# if img: b.image = img
643	return b
644
645	def show_pagenum(self,pagenum,formatstr):
646	self.pagecount.setText(formatstr % (pagenum))
647
648	def spec_show(self):
649	if not self.figmgr.toolbar.mode == '' or not self.button: return
650	self.figmgr.toolbar.set_message('spec value: drag on a spec')
651	if self.mode == 'spec': return
652	self.mode = 'spec'
653	self.notewin.close_widgets()
654	self.__disconnect_event()
655	self._p.register('button_press',self._select_spectrum)
656
657	def modify_note(self):
658	if not self.figmgr.toolbar.mode == '':
659	# Get back button status BEFORE clicked
660	self.bNote.setChecked(not self.bNote.isChecked())
661	return
662	if self.mode == 'note':
663	self.bNote.setChecked(False)
664	self.bNote.setToolTip("add note on plot")
665	self.mode = 'none'
666	self.spec_show()
667	if not self.button:
668	self.notewin.close_widgets()
669	self.__disconnect_event()
670	return
671	self.figmgr.toolbar.set_message('text: select a position/text')
672	self.bNote.setChecked(True)
673	self.bNote.setToolTip("Back to spec value mode")
674	self.bRegion.setChecked(False)
675	self.bRegion.setToolTip("select channel regions")
676	self.bPanel.setChecked(False)
677	self.bPanel.setToolTip("select a whole spectrum")
678	self.mode = 'note'
679	self.__disconnect_event()
680	self._p.register('button_press',self._mod_note)
681
682	def select_region(self):
683	if not self.figmgr.toolbar.mode == '' or not self.button:
684	# Get back button status BEFORE clicked
685	self.bRegion.setChecked(not self.bRegion.isChecked())
686	if self.mode == 'region':
687	self.bRegion.setChecked(False)
688	self.bRegion.setToolTip("select channel regions")
689	self.mode = 'none'
690	self.spec_show()
691	return
692	self.figmgr.toolbar.set_message('select regions: click at start and end channels')
693	self.bNote.setChecked(False)
694	self.bNote.setToolTip("add note on plot")
695	self.bRegion.setChecked(True)
696	self.bRegion.setToolTip("Back to spec value mode")
697	self.bPanel.setChecked(False)
698	self.bPanel.setToolTip("select a whole spectrum")
699	self.mode = 'region'
700	self.notewin.close_widgets()
701	self.__disconnect_event()
702	self._p.register('button_press',self._add_region)
703
704	def select_panel(self):
705	if not self.figmgr.toolbar.mode == '' or not self.button:
706	# Get back button status BEFORE clicked
707	self.bPanel.setChecked(not self.bPanel.isChecked())
708	return
709	if self.mode == 'panel':
710	self.bPanel.setChecked(False)
711	self.bPanel.setToolTip("select subplots")
712	self.mode = 'none'
713	self.spec_show()
714	return
715	self.figmgr.toolbar.set_message('select spectra: click on subplots')
716	self.bNote.setChecked(False)
717	self.bNote.setToolTip("add note on plot")
718	self.bRegion.setChecked(False)
719	self.bRegion.setToolTip("select channel regions")
720	self.bPanel.setChecked(True)
721	self.bPanel.setToolTip("Back to spec value mode")
722	self.mode = 'panel'
723	self.notewin.close_widgets()
724	self.__disconnect_event()
725	self._p.register('button_press',self._add_panel)
726
727	def quit(self):
728	self.__disconnect_event()
729	self.disable_button()
730	self._p.unmap()
731
732	def enable_button(self):
733	if self.button: return
734	self.bRegion.setEnabled(True)
735	self.bPanel.setEnabled(True)
736	self.bClear.setEnabled(True)
737	self.bFlag.setEnabled(True)
738	self.bUnflag.setEnabled(True)
739	self.bStat.setEnabled(True)
740	self.button = True
741	self.spec_show()
742
743	def disable_button(self):
744	## disable buttons which don't work for plottp
745	if not self.button: return
746	self.bRegion.setChecked(False)
747	self.bRegion.setToolTip("select channel regions")
748	self.bPanel.setChecked(False)
749	self.bPanel.setToolTip("select subplots")
750
751	self.bRegion.setDisabled(True)
752	self.bPanel.setDisabled(True)
753	self.bClear.setDisabled(True)
754	self.bFlag.setDisabled(True)
755	self.bUnflag.setDisabled(True)
756	self.bStat.setDisabled(True)
757	self.bNext.setDisabled(True)
758	self.bPrev.setDisabled(True)
759	self.button = False
760	self.mode = ''
761	self.notewin.close_widgets()
762	self.__disconnect_event()
763
764	def enable_next(self):
765	self.bNext.setEnabled(True)
766
767	def disable_next(self):
768	self.bNext.setDisabled(True)
769
770	def enable_prev(self):
771	self.bPrev.setEnabled(True)
772
773	def disable_prev(self):
774	self.bPrev.setDisabled(True)
775
776	# pause buttons for slow operations
777	def _pause_buttons(self,operation="end",msg=""):
778	buttons = ["bRegion","bPanel","bClear","bFlag","bUnflag","bStat",
779	"bNote","bQuit"]
780	if operation == "start":
781	enable = False
782	else:
783	enable = True
784	for btn in buttons:
785	getattr(self,btn).setEnabled(enable)
786	self.figmgr.toolbar.set_message(msg)
787
788	def delete_bar(self):
789	self.__disconnect_event()
790	self.destroy()
791
792	def __disconnect_event(self):
793	self._p.register('button_press',None)
794	self._p.register('button_release',None)
795
796	def _draw_span(self,axes,x0,x1,**kwargs):
797	height = self._p.figure.bbox.height
798	y1 = height - axes.bbox.y1
799	h = axes.bbox.height
800	w = abs(x1 - x0)
801	rect = [ int(val) for val in min(x0,x1), y1, w, h ]
802	self._p.canvas.drawRectangle( rect )
803	# nothing is returned by drawRectangle
804	return None
805
806	def _remove_span(self,span):
807	# Nothing to do with remove.
808	pass

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: trunk/python/customgui_qt4agg.py

Download in other formats: