| 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 |
|
|---|
