cleanup (worker)

This commit is contained in:
Marcel Paffrath 2016-06-01 14:12:49 +02:00
parent e341994825
commit 9996033ca5

View File

@ -1,14 +1,11 @@
# -*- coding: utf-8 -*-
import os
import sys
import numpy as np
from pylot.core.active import seismicshot
from pylot.core.active.surveyUtils import cleanUp
import copy_reg
import types
from pylot.core.util.utils import worker, _pickle_method
copy_reg.pickle(types.MethodType, _pickle_method)
def ppick(shot):
picks = []
for traceID in shot.getTraceIDlist():
@ -120,6 +117,15 @@ class Survey(object):
"cutwindow = %s, tMovingWindow = %f, tsignal = %f, tgap = %f"
% (cutwindow, tmovwind, tsignal, tgap))
def loadArray(self, path, receiverfile, sourcefile):
from pylot.core.active.seismicArrayPreparation import SeisArray
array = SeisArray(os.path.join(path, receiverfile))
array.addSourceLocations(os.path.join(path, sourcefile))
self.seisArray = array
def setManualPicksFromFiles(self, directory='picks'):
'''
Read manual picks from *.pck files in a directory.
@ -230,6 +236,7 @@ class Survey(object):
shotlist = []
print('pickAllShots: Setting pick parameters...')
for shot in self.data.values():
tstartpick = datetime.now()
shot.setVmin(vmin)
@ -238,7 +245,9 @@ class Survey(object):
shot.setPickParameters(folm = folm, method = HosAic, aicwindow = aicwindow)
shotlist.append(shot)
print('pickAllShots: Starting to pick...')
picks = worker(ppick, shotlist, cores)
print('Done!')
for shot in picks:
for item in shot: