marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[new] added attribute autopicks to and modified corresponding handling methods in class MainWindow

Browse Source
This commit is contained in:
Sebastian Wehling-Benatelli 2015-11-30 09:50:47 +01:00
parent ae21c9a149
commit 3f91fddd3e
1 changed files with 28 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
QtPyLoT.py
View File

@ -91,6 +91,7 @@ class MainWindow(QMainWindow):
self.filteroptions = {} self.filteroptions = {}
self.pickDlgs = {} self.pickDlgs = {}
self.picks = {} self.picks = {}
self.autopicks = {}
self.loc = False self.loc = False
# UI has to be set up before(!) children widgets are about to show up # UI has to be set up before(!) children widgets are about to show up
@ -476,12 +477,13 @@ class MainWindow(QMainWindow):
def getData(self): def getData(self):
return self.data return self.data
def getPicks(self): def getPicks(self, type='manual'):
return self.picks rdict = dict(auto=self.autopicks, manual=self.picks)
return rdict[type]
def getPicksOnStation(self, station): def getPicksOnStation(self, station, type='manual'):
try: try:
return self.getPicks()[station] return self.getPicks(type)[station]
except KeyError: except KeyError:
return None return None
@ -679,11 +681,8 @@ class MainWindow(QMainWindow):
self.thread.message.connect(self.addListItem) self.thread.message.connect(self.addListItem)
self.thread.start() self.thread.start()
self.drawPicks() def addPicks(self, station, picks, type='manual'):
stat_picks = self.getPicksOnStation(station, type)
def addPicks(self, station, picks):
stat_picks = self.getPicksOnStation(station)
rval = False rval = False
if not stat_picks: if not stat_picks:
stat_picks = picks stat_picks = picks
@ -707,7 +706,7 @@ class MainWindow(QMainWindow):
pass pass
else: else:
raise Exception('FATAL: Should never occur!') raise Exception('FATAL: Should never occur!')
self.getPicks()[station] = stat_picks self.getPicks(type=type)[station] = stat_picks
return rval return rval
def updatePicks(self): def updatePicks(self):
@ -734,24 +733,25 @@ class MainWindow(QMainWindow):
picks[station] = onsets.copy() picks[station] = onsets.copy()
self.picks.update(picks) self.picks.update(picks)
def drawPicks(self, station=None): def drawPicks(self, station=None, picktype='manual'):
# if picks to draw not specified, draw all picks available # if picks to draw not specified, draw all picks available
if not station: if not station:
for station in self.getPicks(): for station in self.getPicks(type=picktype):
self.drawPicks(station) self.drawPicks(station, picktype=picktype)
return return
# plotting picks # plotting picks
plotID = self.getStationID(station) plotID = self.getStationID(station)
ax = self.getPlotWidget().axes ax = self.getPlotWidget().axes
ylims = np.array([-.5, +.5]) + plotID ylims = np.array([-.5, +.5]) + plotID
phase_col = {'P': ('c', 'c--', 'b-'), phase_col = {'P': ('c', 'c--', 'b-', 'bv', 'b^'),
'S': ('m', 'm--', 'r-')} 'S': ('m', 'm--', 'r-', 'rv', 'r^')}
stat_picks = self.getPicks()[station] stat_picks = self.getPicks(type=picktype)[station]
for phase in stat_picks: for phase in stat_picks:
picks = stat_picks[phase] picks = stat_picks[phase]
if type(stat_picks[phase]) is not dict:
return
colors = phase_col[phase[0].upper()] colors = phase_col[phase[0].upper()]
stime = getGlobalTimes(self.getData().getWFData())[0] stime = getGlobalTimes(self.getData().getWFData())[0]
@ -761,11 +761,17 @@ class MainWindow(QMainWindow):
lpp = picks['lpp'] - stime lpp = picks['lpp'] - stime
spe = picks['spe'] spe = picks['spe']
if picktype == 'manual':
ax.fill_between([epp, lpp], ylims[0], ylims[1], ax.fill_between([epp, lpp], ylims[0], ylims[1],
alpha=.5, color=colors[0]) alpha=.5, color=colors[0])
ax.plot([mpp - spe, mpp - spe], ylims, colors[1], ax.plot([mpp - spe, mpp - spe], ylims, colors[1],
[mpp, mpp], ylims, colors[2], [mpp, mpp], ylims, colors[2],
[mpp + spe, mpp + spe], ylims, colors[1]) [mpp + spe, mpp + spe], ylims, colors[1])
elif picktype == 'auto':
ax.plot(mpp, ylims[1], colors[3],
mpp, ylims[0], colors[4])
else:
raise TypeError('Unknow picktype {0}'.format(picktype))
self.draw() self.draw()
def locateEvent(self): def locateEvent(self):