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 | class CustomToolbarQT4Agg(CustomToolbarCommon, qt.QtGui.QToolBar): |
---|
12 | def __init__(self,parent): |
---|
13 | from asap.asapplotter import asapplotter |
---|
14 | if not isinstance(parent,asapplotter): |
---|
15 | return False |
---|
16 | if not parent._plotter: |
---|
17 | return False |
---|
18 | self._p = parent._plotter |
---|
19 | self.figmgr = self._p.figmgr |
---|
20 | self.canvas = self.figmgr.canvas |
---|
21 | self.mode = '' |
---|
22 | self.button = True |
---|
23 | self.pagecount = None |
---|
24 | CustomToolbarCommon.__init__(self,parent) |
---|
25 | # self.notewin = NotationWindowQT4Agg(master=self.canvas) |
---|
26 | self._add_custom_toolbar() |
---|
27 | |
---|
28 | def _add_custom_toolbar(self): |
---|
29 | qt.QtGui.QToolBar.__init__(self,parent=self.figmgr.window) |
---|
30 | self.figmgr.window.addToolBar(qt.QtCore.Qt.BottomToolBarArea,self) |
---|
31 | self.bNote = self._NewButton(master=self, |
---|
32 | text='notation', |
---|
33 | command=self.modify_note) |
---|
34 | self.bNote.setCheckable(True) |
---|
35 | |
---|
36 | self.bStat = self._NewButton(master=self, |
---|
37 | text='statistics', |
---|
38 | command=self.stat_cal) |
---|
39 | self.bStat.setCheckable(True) |
---|
40 | |
---|
41 | # page change oparations |
---|
42 | frPage = qt.QtGui.QWidget(parent=self,flags=qt.QtCore.Qt.Tool) |
---|
43 | loPage = qt.QtGui.QHBoxLayout(self) |
---|
44 | loPage.addStretch(1) |
---|
45 | self.lPagetitle = qt.QtGui.QLabel('Page:',parent=frPage) |
---|
46 | self.lPagetitle.setMargin(5) |
---|
47 | loPage.addWidget(self.lPagetitle) |
---|
48 | self.pagecount = qt.QtGui.QLabel(parent=frPage) |
---|
49 | self.pagecount.setStyleSheet("background-color: white") |
---|
50 | self.pagecount.setMargin(3) |
---|
51 | self.pagecount.setText(' 1') |
---|
52 | loPage.addWidget(self.pagecount) |
---|
53 | |
---|
54 | self.bNext = self._NewButton(master=frPage, |
---|
55 | text=' + ', |
---|
56 | command=self.next_page,addparent=False) |
---|
57 | loPage.addWidget(self.bNext) |
---|
58 | self.bPrev = self._NewButton(master=frPage, |
---|
59 | text=' - ', |
---|
60 | command=self.prev_page,addparent=False) |
---|
61 | loPage.addWidget(self.bPrev) |
---|
62 | frPage.setLayout(loPage) |
---|
63 | self.addWidget(frPage) |
---|
64 | |
---|
65 | self.bQuit = self._NewButton(master=self, |
---|
66 | text='Quit', |
---|
67 | command=self.quit) |
---|
68 | |
---|
69 | # if os.uname()[0] != 'Darwin': |
---|
70 | # self.bPrev.config(padx=5) |
---|
71 | # self.bNext.config(padx=5) |
---|
72 | |
---|
73 | self.pagecount.setText(' '*4) |
---|
74 | |
---|
75 | self.disable_button() |
---|
76 | return |
---|
77 | |
---|
78 | def _NewButton(self, master, text, command,addparent=True): |
---|
79 | b = qt.QtGui.QPushButton(text,parent=master) |
---|
80 | if addparent: master.addWidget(b) |
---|
81 | master.connect(b,qt.QtCore.SIGNAL('clicked()'),command) |
---|
82 | # if os.uname()[0] == 'Darwin': |
---|
83 | # b = Tk.Button(master=master, text=text, command=command) |
---|
84 | # else: |
---|
85 | # b = Tk.Button(master=master, text=text, padx=2, pady=2, |
---|
86 | # command=command) |
---|
87 | return b |
---|
88 | |
---|
89 | def show_pagenum(self,pagenum,formatstr): |
---|
90 | self.pagecount.setText(formatstr % (pagenum)) |
---|
91 | |
---|
92 | def spec_show(self): |
---|
93 | if not self.figmgr.toolbar.mode == '' or not self.button: return |
---|
94 | self.figmgr.toolbar.set_message('spec value: drag on a spec') |
---|
95 | if self.mode == 'spec': return |
---|
96 | self.mode = 'spec' |
---|
97 | # self.notewin.close_widgets() |
---|
98 | self.__disconnect_event() |
---|
99 | self._p.register('button_press',self._select_spectrum) |
---|
100 | |
---|
101 | def stat_cal(self): |
---|
102 | if not self.figmgr.toolbar.mode == '' or not self.button: |
---|
103 | # Get back button status BEFORE clicked |
---|
104 | self.bStat.setChecked(not self.bStat.isChecked()) |
---|
105 | return |
---|
106 | if self.mode == 'stat': |
---|
107 | # go back to spec mode |
---|
108 | self.bStat.setChecked(False) |
---|
109 | self.spec_show() |
---|
110 | return |
---|
111 | self.figmgr.toolbar.set_message('statistics: select a region') |
---|
112 | self.bStat.setChecked(True) |
---|
113 | self.bNote.setChecked(False) |
---|
114 | self.mode = 'stat' |
---|
115 | # self.notewin.close_widgets() |
---|
116 | self.__disconnect_event() |
---|
117 | self._p.register('button_press',self._single_mask) |
---|
118 | |
---|
119 | def modify_note(self): |
---|
120 | if not self.figmgr.toolbar.mode == '': |
---|
121 | # Get back button status BEFORE clicked |
---|
122 | self.bNote.setChecked(not self.bNote.isChecked()) |
---|
123 | return |
---|
124 | self.figmgr.toolbar.set_message('text: select a position/text') |
---|
125 | if self.mode == 'note': |
---|
126 | self.bNote.setChecked(False) |
---|
127 | self.mode = 'none' |
---|
128 | self.spec_show() |
---|
129 | return |
---|
130 | self.bStat.setChecked(False) |
---|
131 | self.bNote.setChecked(True) |
---|
132 | self.mode = 'note' |
---|
133 | self.__disconnect_event() |
---|
134 | self._p.register('button_press',self._mod_note) |
---|
135 | |
---|
136 | def quit(self): |
---|
137 | self.__disconnect_event() |
---|
138 | self.disable_button() |
---|
139 | self._p.quit() |
---|
140 | |
---|
141 | def enable_button(self): |
---|
142 | if self.button: return |
---|
143 | self.bStat.setEnabled(True) |
---|
144 | self.button = True |
---|
145 | self.spec_show() |
---|
146 | |
---|
147 | def disable_button(self): |
---|
148 | if not self.button: return |
---|
149 | self.bStat.setChecked(False) |
---|
150 | self.bStat.setDisabled(True) |
---|
151 | self.button = False |
---|
152 | self.mode = '' |
---|
153 | self.__disconnect_event() |
---|
154 | |
---|
155 | def enable_next(self): |
---|
156 | self.bNext.setEnabled(True) |
---|
157 | |
---|
158 | def disable_next(self): |
---|
159 | self.bNext.setDisabled(True) |
---|
160 | |
---|
161 | def enable_prev(self): |
---|
162 | self.bPrev.setEnabled(True) |
---|
163 | |
---|
164 | def disable_prev(self): |
---|
165 | self.bPrev.setDisabled(True) |
---|
166 | |
---|
167 | # pause buttons for slow operations |
---|
168 | def _pause_buttons(self,operation="end",msg=""): |
---|
169 | buttons = ["bStat","bNote","bQuit"] |
---|
170 | if operation == "start": |
---|
171 | enable = False |
---|
172 | else: |
---|
173 | enable = True |
---|
174 | for btn in buttons: |
---|
175 | getattr(self,btn).setEnabled(enable) |
---|
176 | self.figmgr.toolbar.set_message(msg) |
---|
177 | |
---|
178 | def delete_bar(self): |
---|
179 | self.__disconnect_event() |
---|
180 | self.destroy() |
---|
181 | |
---|
182 | def __disconnect_event(self): |
---|
183 | self._p.register('button_press',None) |
---|
184 | self._p.register('button_release',None) |
---|
185 | |
---|
186 | |
---|