renaming
This commit is contained in:
parent
c9a75ca2d0
commit
ee16cef901
@ -70,6 +70,14 @@ class Survey(object):
|
|||||||
"cutwindow = %s, tMovingWindow = %f, tsignal = %f, tgap = %f"
|
"cutwindow = %s, tMovingWindow = %f, tsignal = %f, tgap = %f"
|
||||||
%(cutwindow, tmovwind, tsignal, tgap))
|
%(cutwindow, tmovwind, tsignal, tgap))
|
||||||
|
|
||||||
|
def setManualPicksFromFiles(self, directory = 'picks'):
|
||||||
|
'''
|
||||||
|
Read manual picks from *.pck files in a directory.
|
||||||
|
The * must be identical with the shotnumber.
|
||||||
|
'''
|
||||||
|
for shot in self.data.values():
|
||||||
|
shot.setManualPicksFromFile(directory)
|
||||||
|
|
||||||
def _removeAllEmptyTraces(self):
|
def _removeAllEmptyTraces(self):
|
||||||
filename = 'removeEmptyTraces.out'
|
filename = 'removeEmptyTraces.out'
|
||||||
count = 0
|
count = 0
|
||||||
@ -159,16 +167,16 @@ class Survey(object):
|
|||||||
def cleanBySPE(self, maxSPE):
|
def cleanBySPE(self, maxSPE):
|
||||||
for shot in self.data.values():
|
for shot in self.data.values():
|
||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
if shot.getFlag(traceID) == 1:
|
if shot.getPickFlag(traceID) == 1:
|
||||||
if shot.getSymmetricPickError(traceID) > maxSPE:
|
if shot.getSymmetricPickError(traceID) > maxSPE:
|
||||||
shot.setFlag(traceID, 0)
|
shot.setPickFlag(traceID, 0)
|
||||||
|
|
||||||
def plotSPE(self):
|
def plotSPE(self):
|
||||||
import matplotlib.pyplot as plt
|
import matplotlib.pyplot as plt
|
||||||
spe = []
|
spe = []
|
||||||
for shot in self.data.values():
|
for shot in self.data.values():
|
||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
if shot.getFlag(traceID) == 1:
|
if shot.getPickFlag(traceID) == 1:
|
||||||
spe.append(shot.getSymmetricPickError(traceID))
|
spe.append(shot.getSymmetricPickError(traceID))
|
||||||
spe.sort()
|
spe.sort()
|
||||||
plt.plot(spe, label = 'SPE')
|
plt.plot(spe, label = 'SPE')
|
||||||
@ -183,8 +191,8 @@ class Survey(object):
|
|||||||
numpicks = 0
|
numpicks = 0
|
||||||
for shot in self.data.values():
|
for shot in self.data.values():
|
||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
if shot.getFlag(traceID) == 0:
|
if shot.getPickFlag(traceID) == 0:
|
||||||
shot.setFlag(traceID, 1)
|
shot.setPickFlag(traceID, 1)
|
||||||
if shot.getSNR(traceID)[0] < shot.getSNRthreshold(traceID):
|
if shot.getSNR(traceID)[0] < shot.getSNRthreshold(traceID):
|
||||||
shot.removePick(traceID)
|
shot.removePick(traceID)
|
||||||
else:
|
else:
|
||||||
@ -248,7 +256,7 @@ class Survey(object):
|
|||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
snrlist.append(shot.getSNR(traceID)[0])
|
snrlist.append(shot.getSNR(traceID)[0])
|
||||||
dist.append(shot.getDistance(traceID))
|
dist.append(shot.getDistance(traceID))
|
||||||
if shot.getFlag(traceID) is not 0:
|
if shot.getPickFlag(traceID) is not 0:
|
||||||
pickedTraces += 1
|
pickedTraces += 1
|
||||||
info_dict[shot.getShotnumber()] = {'numtraces': numtraces,
|
info_dict[shot.getShotnumber()] = {'numtraces': numtraces,
|
||||||
'picked traces': [pickedTraces,
|
'picked traces': [pickedTraces,
|
||||||
@ -289,7 +297,7 @@ class Survey(object):
|
|||||||
traceIDlist.sort()
|
traceIDlist.sort()
|
||||||
ttfile.writelines(str(self.countPickedTraces(shot)) + '\n')
|
ttfile.writelines(str(self.countPickedTraces(shot)) + '\n')
|
||||||
for traceID in traceIDlist:
|
for traceID in traceIDlist:
|
||||||
if shot.getFlag(traceID) is not 0:
|
if shot.getPickFlag(traceID) is not 0:
|
||||||
pick = shot.getPick(traceID) * fmtomo_factor
|
pick = shot.getPick(traceID) * fmtomo_factor
|
||||||
delta = shot.getSymmetricPickError(traceID) * fmtomo_factor
|
delta = shot.getSymmetricPickError(traceID) * fmtomo_factor
|
||||||
(x, y, z) = shot.getRecLoc(traceID)
|
(x, y, z) = shot.getRecLoc(traceID)
|
||||||
@ -306,7 +314,7 @@ class Survey(object):
|
|||||||
def countPickedTraces(self, shot):
|
def countPickedTraces(self, shot):
|
||||||
count = 0
|
count = 0
|
||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
if shot.getFlag(traceID) is not 0:
|
if shot.getPickFlag(traceID) is not 0:
|
||||||
count += 1
|
count += 1
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@ -314,7 +322,7 @@ class Survey(object):
|
|||||||
count = 0
|
count = 0
|
||||||
for shot in self.data.values():
|
for shot in self.data.values():
|
||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
if shot.getFlag(traceID) is not 0:
|
if shot.getPickFlag(traceID) is not 0:
|
||||||
count += 1
|
count += 1
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@ -401,7 +409,7 @@ class Survey(object):
|
|||||||
for shot in self.data.values():
|
for shot in self.data.values():
|
||||||
for traceID in shot.getTraceIDlist():
|
for traceID in shot.getTraceIDlist():
|
||||||
if plotRemoved == False:
|
if plotRemoved == False:
|
||||||
if shot.getFlag(traceID) is not 0 or plotRemoved == True:
|
if shot.getPickFlag(traceID) is not 0 or plotRemoved == True:
|
||||||
dist.append(shot.getDistance(traceID))
|
dist.append(shot.getDistance(traceID))
|
||||||
pick.append(shot.getPick(traceID))
|
pick.append(shot.getPick(traceID))
|
||||||
snrlog.append(math.log10(shot.getSNR(traceID)[0]))
|
snrlog.append(math.log10(shot.getSNR(traceID)[0]))
|
||||||
|
Loading…
Reference in New Issue
Block a user