cleaned up surveyUtils

pylot/core/active/surveyUtils.py

@@ -1,48 +1,5 @@
 import numpy as np
 
-def generateSurvey(obsdir, shotlist):
-    from obspy.core import read
-    from pylot.core.active import seismicshot
-
-    shot_dict = {}
-    for shotnumber in shotlist:       # loop over data files
-        # generate filenames and read manual picks to a list
-        obsfile = obsdir + str(shotnumber) + '_pickle.dat'
-        #obsfile = obsdir + str(shotnumber) + '.dat'
-        
-        if not obsfile in shot_dict.keys():
-            shot_dict[shotnumber] = []
-        shot_dict[shotnumber] = seismicshot.SeismicShot(obsfile)
-        shot_dict[shotnumber].setParameters('shotnumber', shotnumber)
-        
-    return shot_dict
-
-def setParametersForShots(cutwindow, tmovwind, tsignal, tgap, receiverfile, sourcefile, shot_dict):
-    for shot in shot_dict.values():
-        shot.setCut(cutwindow)
-        shot.setTmovwind(tmovwind)
-        shot.setTsignal(tsignal)
-        shot.setTgap(tgap)
-        shot.setRecfile(receiverfile)
-        shot.setSourcefile(sourcefile)
-        shot.setOrder(order = 4)
-
-def removeEmptyTraces(shot_dict):
-    filename = 'removeEmptyTraces.out'
-    filename2 = 'updateTraces.out'
-    outfile = open(filename, 'w')
-    outfile2 = open(filename2, 'w')
-    for shot in shot_dict.values():
-        del_traceIDs = shot.updateTraceList()
-        removed = shot.removeEmptyTraces()
-        if removed is not None:
-            outfile.writelines('shot: %s, removed empty traces: %s\n' %(shot.getShotnumber(), removed))
-            outfile2.writelines('shot: %s, removed traceID(s) %s because they were not found in the corresponding stream\n' %(shot.getShotnumber(), del_traceIDs))
-    print '\nremoveEmptyTraces, updateTraces: Finished! See %s and %s for more information of removed traces.\n' %(filename, filename2)
-    outfile.close()
-    outfile2.close()
-
-
 def readParameters(parfile, parameter):
     from ConfigParser import ConfigParser
     parameterConfig = ConfigParser()
@@ -53,11 +10,6 @@ def readParameters(parfile, parameter):
 
     return value
 
-def setArtificialPick(shot_dict, traceID, pick):
-    for shot in shot_dict.values():
-        shot.setPick(traceID, pick)
-        shot.setPickwindow(traceID, shot.getCut())
-
 def fitSNR4dist(shot_dict, shiftdist = 5):
     dists = []
     picks = []
@@ -79,7 +31,6 @@ def fitSNR4dist(shot_dict, shiftdist = 5):
     plotFittedSNR(dists, snrthresholds, snrs)
     return fit_fn #### ZU VERBESSERN, sollte fertige funktion wiedergeben
 
-
 def plotFittedSNR(dists, snrthresholds, snrs):
     import matplotlib.pyplot as plt
     plt.interactive(True)
@@ -98,78 +49,8 @@ def setFittedSNR(shot_dict, shiftdist = 5, p1 = 0.004, p2 = -0.004):
             shot.setSNRthreshold(traceID, 1/(fit_fn(shot.getDistance(traceID) + shiftdist)**2)) ### s.o.
     print "\nsetFittedSNR: Finished setting of fitted SNR-threshold"
 
-#def linearInterp(dist_med, dist_start
-
-def exportFMTOMO(shot_dict, directory = 'FMTOMO_export', sourcefile = 'input_sf.in', ttFileExtension = '.tt'):
-    count = 0
-    fmtomo_factor = 1000 # transforming [m/s] -> [km/s]
-    LatAll = []; LonAll = []; DepthAll = []
-    srcfile = open(directory + '/' + sourcefile, 'w')
-    srcfile.writelines('%10s\n' %len(shot_dict)) # number of sources
-    for shotnumber in getShotlist(shot_dict):
-        shot = getShotForShotnumber(shot_dict, shotnumber)
-        ttfilename = str(shotnumber) + ttFileExtension
-        (x, y, z) = shot.getSrcLoc() # getSrcLoc returns (x, y, z)
-        srcfile.writelines('%10s %10s %10s\n' %(getAngle(y), getAngle(x), (-1)*z)) # lat, lon, depth
-        LatAll.append(getAngle(y)); LonAll.append(getAngle(x)); DepthAll.append((-1)*z)
-        srcfile.writelines('%10s\n' %1) # 
-        srcfile.writelines('%10s %10s %10s\n' %(1, 1, ttfilename))
-        ttfile = open(directory + '/' + ttfilename, 'w')
-        traceIDlist = shot.getTraceIDlist()
-        traceIDlist.sort()
-        ttfile.writelines(str(countPickedTraces(shot)) + '\n')
-        for traceID in traceIDlist:
-            if shot.getPick(traceID) is not None:
-                pick = shot.getPick(traceID) * fmtomo_factor
-                delta = shot.getPickError(traceID) * fmtomo_factor
-                (x, y, z) = shot.getRecLoc(traceID)
-                ttfile.writelines('%20s %20s %20s %10s %10s\n' %(getAngle(y), getAngle(x), (-1)*z, pick, delta))
-                LatAll.append(getAngle(y)); LonAll.append(getAngle(x)); DepthAll.append((-1)*z)
-                count += 1
-        ttfile.close()
-    srcfile.close()       
-    print 'Wrote output for %s traces' %count
-    print 'WARNING: output generated for FMTOMO-obsdata. Obsdata seems to take Lat, Lon, Depth and creates output for FMTOMO as Depth, Lat, Lon'
-    print 'Dimensions of the seismic Array, transformed for FMTOMO, are Depth(%s, %s), Lat(%s, %s), Lon(%s, %s)'%(
-        min(DepthAll), max(DepthAll), min(LatAll), max(LatAll), min(LonAll), max(LonAll))
-
-def getShotlist(shot_dict):
-    shotlist = []
-    for shot in shot_dict.values():
-        shotlist.append(shot.getShotnumber())
-    shotlist.sort()
-    return shotlist
-
-def getShotForShotnumber(shot_dict, shotnumber):
-    for shot in shot_dict.values():
-        if shot.getShotnumber() == shotnumber:
-            return shot
-
-def getAngle(distance):
-    '''
-    Function returns the angle on a Sphere of the radius R = 6371 [km] for a distance [km].
-    '''
-    PI = np.pi
-    R = 6371.
-    angle = distance * 180 / (PI * R)
-    return angle
-
-def countPickedTraces(shot):
-    numtraces = 0
-    for traceID in shot.getTraceIDlist():
-        if shot.getPick(traceID) is not None:
-            numtraces += 1
-    print "countPickedTraces: Found %s picked traces in shot number %s"  %(numtraces, shot.getShotnumber())
-    return numtraces
-
-def countAllPickedTraces(shot_dict):
-    traces = 0
-    for shot in shot_dict.values():
-        traces += countPickedTraces(shot)
-    return traces
-
 def findTracesInRanges(shot_dict, distancebin, pickbin):
-    '''                                                                                                                                                  
+    '''
     Returns traces corresponding to a certain area in a plot with all picks over the distances.
     
     :param: shot_dict, dictionary containing all shots that are used
@@ -180,7 +61,6 @@ def findTracesInRanges(shot_dict, distancebin, pickbin):
                                                                                      
     :param: pickbin
     :type: tuple, (t1[s], t2[s])
-                                                                                     
     '''
     shots_found = {}
     for shot in shot_dict.values():