repick button for plot_traces
This commit is contained in:
parent
d78b0f1cff
commit
4498c72c90
@ -7,6 +7,8 @@ from pylot.core.pick.CharFuns import HOScf
|
|||||||
from pylot.core.pick.CharFuns import AICcf
|
from pylot.core.pick.CharFuns import AICcf
|
||||||
from pylot.core.pick.utils import getSNR
|
from pylot.core.pick.utils import getSNR
|
||||||
from pylot.core.pick.utils import earllatepicker
|
from pylot.core.pick.utils import earllatepicker
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
plt.interactive('True')
|
||||||
|
|
||||||
class SeismicShot(object):
|
class SeismicShot(object):
|
||||||
'''
|
'''
|
||||||
@ -127,8 +129,15 @@ class SeismicShot(object):
|
|||||||
def getSourcefile(self):
|
def getSourcefile(self):
|
||||||
return self.paras['sourcefile']
|
return self.paras['sourcefile']
|
||||||
|
|
||||||
def getPick(self, traceID):
|
def getPick(self, traceID, returnRemoved = False):
|
||||||
|
if not self.getFlag(traceID) == 0:
|
||||||
return self.pick[traceID]['mpp']
|
return self.pick[traceID]['mpp']
|
||||||
|
if returnRemoved == True:
|
||||||
|
#print('getPick: Returned removed pick for shot %d, traceID %d' %(self.getShotnumber(), traceID))
|
||||||
|
return self.pick[traceID]['mpp']
|
||||||
|
|
||||||
|
def getPickIncludeRemoved(self, traceID):
|
||||||
|
return self.getPick(traceID, returnRemoved = True)
|
||||||
|
|
||||||
def getEarliest(self, traceID):
|
def getEarliest(self, traceID):
|
||||||
return self.pick[traceID]['epp']
|
return self.pick[traceID]['epp']
|
||||||
@ -136,6 +145,12 @@ class SeismicShot(object):
|
|||||||
def getLatest(self, traceID):
|
def getLatest(self, traceID):
|
||||||
return self.pick[traceID]['lpp']
|
return self.pick[traceID]['lpp']
|
||||||
|
|
||||||
|
def getSymmetricPickError(self, traceID):
|
||||||
|
pickerror = self.pick[traceID]['spe']
|
||||||
|
if np.isnan(pickerror) == True:
|
||||||
|
print "SPE is NaN for shot %s, traceID %s"%(self.getShotnumber(), traceID)
|
||||||
|
return pickerror
|
||||||
|
|
||||||
def getPickError(self, traceID):
|
def getPickError(self, traceID):
|
||||||
pickerror = abs(self.getEarliest(traceID) - self.getLatest(traceID))
|
pickerror = abs(self.getEarliest(traceID) - self.getLatest(traceID))
|
||||||
if np.isnan(pickerror) == True:
|
if np.isnan(pickerror) == True:
|
||||||
@ -209,7 +224,7 @@ class SeismicShot(object):
|
|||||||
:type: int
|
:type: int
|
||||||
'''
|
'''
|
||||||
return HOScf(self.getSingleStream(traceID), self.getCut(),
|
return HOScf(self.getSingleStream(traceID), self.getCut(),
|
||||||
self.getTmovwind(), self.getOrder())
|
self.getTmovwind(), self.getOrder(), stealthMode = True)
|
||||||
|
|
||||||
def getAICcf(self, traceID):
|
def getAICcf(self, traceID):
|
||||||
'''
|
'''
|
||||||
@ -232,7 +247,7 @@ class SeismicShot(object):
|
|||||||
tr_cf = Trace()
|
tr_cf = Trace()
|
||||||
tr_cf.data = self.getHOScf(traceID).getCF()
|
tr_cf.data = self.getHOScf(traceID).getCF()
|
||||||
st_cf += tr_cf
|
st_cf += tr_cf
|
||||||
return AICcf(st_cf, self.getCut(), self.getTmovwind())
|
return AICcf(st_cf, self.getCut(), self.getTmovwind(), stealthMode = True)
|
||||||
|
|
||||||
def getSingleStream(self, traceID): ########## SEG2 / SEGY ? ##########
|
def getSingleStream(self, traceID): ########## SEG2 / SEGY ? ##########
|
||||||
'''
|
'''
|
||||||
@ -291,11 +306,13 @@ class SeismicShot(object):
|
|||||||
def setEarllatepick(self, traceID, nfac = 1.5):
|
def setEarllatepick(self, traceID, nfac = 1.5):
|
||||||
tgap = self.getTgap()
|
tgap = self.getTgap()
|
||||||
tsignal = self.getTsignal()
|
tsignal = self.getTsignal()
|
||||||
tnoise = self.getPick(traceID) - tgap
|
tnoise = self.getPickIncludeRemoved(traceID) - tgap
|
||||||
|
|
||||||
(self.pick[traceID]['epp'], self.pick[traceID]['lpp'], tmp) = earllatepicker(self.getSingleStream(traceID),
|
(self.pick[traceID]['epp'], self.pick[traceID]['lpp'],
|
||||||
|
self.pick[traceID]['spe']) = earllatepicker(self.getSingleStream(traceID),
|
||||||
nfac, (tnoise, tgap, tsignal),
|
nfac, (tnoise, tgap, tsignal),
|
||||||
self.getPick(traceID))
|
self.getPickIncludeRemoved(traceID),
|
||||||
|
stealthMode = True)
|
||||||
|
|
||||||
def threshold(self, hoscf, aiccf, windowsize, pickwindow, folm = 0.6):
|
def threshold(self, hoscf, aiccf, windowsize, pickwindow, folm = 0.6):
|
||||||
'''
|
'''
|
||||||
@ -463,7 +480,7 @@ class SeismicShot(object):
|
|||||||
|
|
||||||
def setFlag(self, traceID, flag):
|
def setFlag(self, traceID, flag):
|
||||||
'Set flag = 0 if pick is invalid, else flag = 1'
|
'Set flag = 0 if pick is invalid, else flag = 1'
|
||||||
self.pick[traceID]['flag'] = 0
|
self.pick[traceID]['flag'] = flag
|
||||||
|
|
||||||
def getFlag(self, traceID):
|
def getFlag(self, traceID):
|
||||||
return self.pick[traceID]['flag']
|
return self.pick[traceID]['flag']
|
||||||
@ -570,7 +587,39 @@ class SeismicShot(object):
|
|||||||
# plt.plot(self.getDistArray4ttcPlot(), pickwindowarray_upperb, ':k')
|
# plt.plot(self.getDistArray4ttcPlot(), pickwindowarray_upperb, ':k')
|
||||||
|
|
||||||
def plot_traces(self, traceID, folm = 0.6): ########## 2D, muss noch mehr verbessert werden ##########
|
def plot_traces(self, traceID, folm = 0.6): ########## 2D, muss noch mehr verbessert werden ##########
|
||||||
import matplotlib.pyplot as plt
|
from matplotlib.widgets import Button
|
||||||
|
|
||||||
|
def onclick(event):
|
||||||
|
self.setPick(traceID, event.xdata)
|
||||||
|
self._drawStream(traceID, refresh = True)
|
||||||
|
self._drawCFs(traceID, folm, refresh = True)
|
||||||
|
fig.canvas.mpl_disconnect(self.traces4plot[traceID]['cid'])
|
||||||
|
plt.draw()
|
||||||
|
|
||||||
|
def connectButton(event = None):
|
||||||
|
cid = fig.canvas.mpl_connect('button_press_event', onclick)
|
||||||
|
self.traces4plot[traceID]['cid'] = cid
|
||||||
|
|
||||||
|
fig = plt.figure()
|
||||||
|
ax1 = fig.add_subplot(2,1,1)
|
||||||
|
ax2 = fig.add_subplot(2,1,2, sharex = ax1)
|
||||||
|
axb = fig.add_axes([0.15, 0.91, 0.05, 0.03])
|
||||||
|
button = Button(axb, 'repick', color = 'red', hovercolor = 'grey')
|
||||||
|
button.on_clicked(connectButton)
|
||||||
|
|
||||||
|
self.traces4plot = {}
|
||||||
|
if not traceID in self.traces4plot.keys():
|
||||||
|
self.traces4plot[traceID] = {'fig': fig,
|
||||||
|
'ax1': ax1,
|
||||||
|
'ax2': ax2,
|
||||||
|
'axb': axb,
|
||||||
|
'button': button,
|
||||||
|
'cid': None,}
|
||||||
|
|
||||||
|
self._drawStream(traceID)
|
||||||
|
self._drawCFs(traceID, folm)
|
||||||
|
|
||||||
|
def _drawStream(self, traceID, refresh = False):
|
||||||
from pylot.core.util.utils import getGlobalTimes
|
from pylot.core.util.utils import getGlobalTimes
|
||||||
from pylot.core.util.utils import prepTimeAxis
|
from pylot.core.util.utils import prepTimeAxis
|
||||||
|
|
||||||
@ -579,32 +628,47 @@ class SeismicShot(object):
|
|||||||
timeaxis = prepTimeAxis(stime, stream[0])
|
timeaxis = prepTimeAxis(stime, stream[0])
|
||||||
timeaxis -= stime
|
timeaxis -= stime
|
||||||
|
|
||||||
plt.interactive('True')
|
ax = self.traces4plot[traceID]['ax1']
|
||||||
|
|
||||||
hoscf = self.getHOScf(traceID)
|
if refresh == True:
|
||||||
aiccf = self.getAICcf(traceID)
|
xlim, ylim = ax.get_xlim(), ax.get_ylim()
|
||||||
|
ax.clear()
|
||||||
|
if refresh == True:
|
||||||
|
ax.set_xlim(xlim)
|
||||||
|
ax.set_ylim(ylim)
|
||||||
|
|
||||||
fig = plt.figure()
|
ax.set_title('Shot: %s, traceID: %s, pick: %s'
|
||||||
ax1 = plt.subplot(2,1,1)
|
%(self.getShotnumber(), traceID, self.getPick(traceID)))
|
||||||
plt.title('Shot: %s, traceID: %s, pick: %s' %(self.getShotnumber(), traceID, self.getPick(traceID)))
|
ax.plot(timeaxis, stream[0].data, 'k', label = 'trace')
|
||||||
ax1.plot(timeaxis, stream[0].data, 'k', label = 'trace')
|
ax.plot([self.getPick(traceID), self.getPick(traceID)],
|
||||||
ax1.plot([self.getPick(traceID), self.getPick(traceID)],
|
|
||||||
[min(stream[0].data),
|
[min(stream[0].data),
|
||||||
max(stream[0].data)],
|
max(stream[0].data)],
|
||||||
'r', label = 'mostlikely')
|
'r', label = 'mostlikely')
|
||||||
plt.legend()
|
ax.legend()
|
||||||
ax2 = plt.subplot(2,1,2, sharex = ax1)
|
|
||||||
ax2.plot(hoscf.getTimeArray(), hoscf.getCF(), 'b', label = 'HOS')
|
def _drawCFs(self, traceID, folm, refresh = False):
|
||||||
ax2.plot(hoscf.getTimeArray(), aiccf.getCF(), 'g', label = 'AIC')
|
hoscf = self.getHOScf(traceID)
|
||||||
ax2.plot([self.getPick(traceID), self.getPick(traceID)],
|
aiccf = self.getAICcf(traceID)
|
||||||
|
ax = self.traces4plot[traceID]['ax2']
|
||||||
|
|
||||||
|
if refresh == True:
|
||||||
|
xlim, ylim = ax.get_xlim(), ax.get_ylim()
|
||||||
|
ax.clear()
|
||||||
|
if refresh == True:
|
||||||
|
ax.set_xlim(xlim)
|
||||||
|
ax.set_ylim(ylim)
|
||||||
|
|
||||||
|
ax.plot(hoscf.getTimeArray(), hoscf.getCF(), 'b', label = 'HOS')
|
||||||
|
ax.plot(hoscf.getTimeArray(), aiccf.getCF(), 'g', label = 'AIC')
|
||||||
|
ax.plot([self.getPick(traceID), self.getPick(traceID)],
|
||||||
[min(np.minimum(hoscf.getCF(), aiccf.getCF())),
|
[min(np.minimum(hoscf.getCF(), aiccf.getCF())),
|
||||||
max(np.maximum(hoscf.getCF(), aiccf.getCF()))],
|
max(np.maximum(hoscf.getCF(), aiccf.getCF()))],
|
||||||
'r', label = 'mostlikely')
|
'r', label = 'mostlikely')
|
||||||
ax2.plot([0, self.getPick(traceID)],
|
ax.plot([0, self.getPick(traceID)],
|
||||||
[folm * max(hoscf.getCF()), folm * max(hoscf.getCF())],
|
[folm * max(hoscf.getCF()), folm * max(hoscf.getCF())],
|
||||||
'm:', label = 'folm = %s' %folm)
|
'm:', label = 'folm = %s' %folm)
|
||||||
plt.xlabel('Time [s]')
|
ax.set_xlabel('Time [s]')
|
||||||
plt.legend()
|
ax.legend()
|
||||||
|
|
||||||
def plot3dttc(self, step = 0.5, contour = False, plotpicks = False, method = 'linear', ax = None):
|
def plot3dttc(self, step = 0.5, contour = False, plotpicks = False, method = 'linear', ax = None):
|
||||||
'''
|
'''
|
||||||
@ -622,7 +686,6 @@ class SeismicShot(object):
|
|||||||
:param: method (optional), interpolation method; can be 'linear' (default) or 'cubic'
|
:param: method (optional), interpolation method; can be 'linear' (default) or 'cubic'
|
||||||
:type: 'string'
|
:type: 'string'
|
||||||
'''
|
'''
|
||||||
import matplotlib.pyplot as plt
|
|
||||||
from scipy.interpolate import griddata
|
from scipy.interpolate import griddata
|
||||||
from matplotlib import cm
|
from matplotlib import cm
|
||||||
from mpl_toolkits.mplot3d import Axes3D
|
from mpl_toolkits.mplot3d import Axes3D
|
||||||
@ -636,8 +699,8 @@ class SeismicShot(object):
|
|||||||
y.append(self.getRecLoc(traceID)[1])
|
y.append(self.getRecLoc(traceID)[1])
|
||||||
z.append(self.getPick(traceID))
|
z.append(self.getPick(traceID))
|
||||||
|
|
||||||
xaxis = np.arange(min(x)+1, max(x), step)
|
xaxis = np.arange(min(x), max(x), step)
|
||||||
yaxis = np.arange(min(y)+1, max(y), step)
|
yaxis = np.arange(min(y), max(y), step)
|
||||||
xgrid, ygrid = np.meshgrid(xaxis, yaxis)
|
xgrid, ygrid = np.meshgrid(xaxis, yaxis)
|
||||||
zgrid = griddata((x, y), z, (xgrid, ygrid), method = method)
|
zgrid = griddata((x, y), z, (xgrid, ygrid), method = method)
|
||||||
|
|
||||||
@ -662,7 +725,7 @@ class SeismicShot(object):
|
|||||||
|
|
||||||
plotmethod[method](*args)
|
plotmethod[method](*args)
|
||||||
|
|
||||||
def matshow(self, step = 0.5, method = 'linear', ax = None, plotRec = False, annotations = False):
|
def matshow(self, ax = None, step = 0.5, method = 'linear', plotRec = True, annotations = True, colorbar = True):
|
||||||
'''
|
'''
|
||||||
Plots a 2D matrix of the interpolated traveltimes. This needs less performance than plot3dttc
|
Plots a 2D matrix of the interpolated traveltimes. This needs less performance than plot3dttc
|
||||||
|
|
||||||
@ -672,27 +735,32 @@ class SeismicShot(object):
|
|||||||
:param: method (optional), interpolation method; can be 'linear' (default) or 'cubic'
|
:param: method (optional), interpolation method; can be 'linear' (default) or 'cubic'
|
||||||
:type: 'string'
|
:type: 'string'
|
||||||
|
|
||||||
:param: plotRec (optional), plot the receiver positions
|
:param: plotRec (optional), plot the receiver positions (colored scatter plot, should not be
|
||||||
|
deactivated because there might be receivers that are not inside the interpolated area)
|
||||||
:type: 'logical'
|
:type: 'logical'
|
||||||
|
|
||||||
:param: annotations (optional), displays traceIDs as annotations
|
:param: annotations (optional), displays traceIDs as annotations
|
||||||
:type: 'logical'
|
:type: 'logical'
|
||||||
'''
|
'''
|
||||||
import matplotlib.pyplot as plt
|
|
||||||
from scipy.interpolate import griddata
|
from scipy.interpolate import griddata
|
||||||
# plt.interactive('True')
|
|
||||||
|
|
||||||
x = []
|
x = []; xcut = []
|
||||||
y = []
|
y = []; ycut = []
|
||||||
z = []
|
z = []; zcut = []
|
||||||
|
tmin, tmax = self.getCut()
|
||||||
|
|
||||||
for traceID in self.pick.keys():
|
for traceID in self.pick.keys():
|
||||||
if self.getFlag(traceID) != 0:
|
if self.getFlag(traceID) != 0:
|
||||||
x.append(self.getRecLoc(traceID)[0])
|
x.append(self.getRecLoc(traceID)[0])
|
||||||
y.append(self.getRecLoc(traceID)[1])
|
y.append(self.getRecLoc(traceID)[1])
|
||||||
z.append(self.getPick(traceID))
|
z.append(self.getPick(traceID))
|
||||||
|
if self.getFlag(traceID) == 0 and self.getPickIncludeRemoved(traceID) is not None:
|
||||||
|
xcut.append(self.getRecLoc(traceID)[0])
|
||||||
|
ycut.append(self.getRecLoc(traceID)[1])
|
||||||
|
zcut.append(self.getPickIncludeRemoved(traceID))
|
||||||
|
|
||||||
xaxis = np.arange(min(x)+1, max(x), step)
|
xaxis = np.arange(min(x), max(x), step)
|
||||||
yaxis = np.arange(min(y)+1, max(y), step)
|
yaxis = np.arange(min(y), max(y), step)
|
||||||
xgrid, ygrid = np.meshgrid(xaxis, yaxis)
|
xgrid, ygrid = np.meshgrid(xaxis, yaxis)
|
||||||
zgrid = griddata((x, y), z, (xgrid, ygrid), method='linear')
|
zgrid = griddata((x, y), z, (xgrid, ygrid), method='linear')
|
||||||
|
|
||||||
@ -700,14 +768,28 @@ class SeismicShot(object):
|
|||||||
fig = plt.figure()
|
fig = plt.figure()
|
||||||
ax = plt.axes()
|
ax = plt.axes()
|
||||||
|
|
||||||
ax.imshow(zgrid, interpolation = 'none', extent = [min(x), max(x), min(y), max(y)])
|
ax.matshow(zgrid, extent = [min(x), max(x), min(y), max(y)], origin = 'lower')
|
||||||
|
plt.text(0.45, 0.9, 'shot: %s' %self.getShotnumber(), transform = ax.transAxes)
|
||||||
if annotations == True:
|
sc = ax.scatter(x, y, c = z, s = 30, label = 'picked shots', vmin = tmin, vmax = tmax, linewidths = 1.5)
|
||||||
for i, traceID in enumerate(self.pick.keys()):
|
sccut = ax.scatter(xcut, ycut, c = zcut, s = 30, edgecolor = 'm', label = 'cut out shots', vmin = tmin, vmax = tmax, linewidths = 1.5)
|
||||||
if shot.picks[traceID] != None:
|
if colorbar == True:
|
||||||
ax.annotate('%s' % traceID, xy=(x[i], y[i]), fontsize = 'x-small')
|
plt.colorbar(sc)
|
||||||
|
ax.set_xlabel('X')
|
||||||
|
ax.set_ylabel('Y')
|
||||||
|
ax.plot(self.getSrcLoc()[0], self.getSrcLoc()[1],'*k', markersize = 15) # plot source location
|
||||||
|
|
||||||
if plotRec == True:
|
if plotRec == True:
|
||||||
ax.plot(x, y, 'k.')
|
ax.scatter(x, y, c = z, s = 30)
|
||||||
|
|
||||||
|
if annotations == True:
|
||||||
|
for traceID in self.getTraceIDlist():
|
||||||
|
if self.getFlag(traceID) is not 0:
|
||||||
|
ax.annotate(' %s' %traceID , xy = (self.getRecLoc(traceID)[0], self.getRecLoc(traceID)[1]),
|
||||||
|
fontsize = 'x-small', color = 'k')
|
||||||
|
else:
|
||||||
|
ax.annotate(' %s' %traceID , xy = (self.getRecLoc(traceID)[0], self.getRecLoc(traceID)[1]),
|
||||||
|
fontsize = 'x-small', color = 'r')
|
||||||
|
|
||||||
plt.show()
|
plt.show()
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user