bugfixes (minSNR was 1, problem in for loop with break?)

This commit is contained in:
Marcel Paffrath 2015-12-07 11:21:11 +01:00
parent d5b277db99
commit 1a9f68f741

View File

@ -41,7 +41,7 @@ def plotFittedSNR(dists, snrthresholds, snrs):
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
plt.interactive(True) plt.interactive(True)
fig = plt.figure() fig = plt.figure()
plt.plot(dists, snrs, '.', markersize = 0.5, label = 'SNR values') plt.plot(dists, snrs, '.', markersize = 1.0, label = 'SNR values')
plt.plot(dists, snrthresholds, 'r.', markersize = 1, label = 'Fitted threshold') plt.plot(dists, snrthresholds, 'r.', markersize = 1, label = 'Fitted threshold')
plt.xlabel('Distance[m]') plt.xlabel('Distance[m]')
plt.ylabel('SNR') plt.ylabel('SNR')
@ -56,12 +56,12 @@ def setFittedSNR(shot_dict, shiftdist = 30, shiftSNR = 100, p1 = 0.004, p2 = -0.
for traceID in shot.getTraceIDlist(): ### IMPROVE for traceID in shot.getTraceIDlist(): ### IMPROVE
dist = shot.getDistance(traceID) + shiftdist dist = shot.getDistance(traceID) + shiftdist
snrthreshold = (1/(fit_fn(dist)**2)) - shiftSNR * np.exp(-0.05 * dist) snrthreshold = (1/(fit_fn(dist)**2)) - shiftSNR * np.exp(-0.05 * dist)
if snrthreshold <= 1: if snrthreshold < minSNR:
print('WARNING: SNR threshold %s lower %s. Set SNR threshold to %s.' print('WARNING: SNR threshold %s lower %s. Set SNR threshold to %s.'
%(snrthreshold, minSNR, minSNR)) %(snrthreshold, minSNR, minSNR))
shot.setSNRthreshold(traceID, minSNR) shot.setSNRthreshold(traceID, minSNR)
break else:
shot.setSNRthreshold(traceID, snrthreshold) shot.setSNRthreshold(traceID, snrthreshold)
print "setFittedSNR: Finished setting of fitted SNR-threshold" print "setFittedSNR: Finished setting of fitted SNR-threshold"