Compare commits
2 Commits
d7ee820de3
...
71d4269a4f
Author | SHA1 | Date | |
---|---|---|---|
71d4269a4f | |||
81e34875b9 |
6
PyLoT.py
Normal file → Executable file
6
PyLoT.py
Normal file → Executable file
@ -3590,7 +3590,7 @@ class MainWindow(QMainWindow):
|
|||||||
def calc_magnitude(self):
|
def calc_magnitude(self):
|
||||||
self.init_metadata()
|
self.init_metadata()
|
||||||
if not self.metadata:
|
if not self.metadata:
|
||||||
return None
|
return []
|
||||||
|
|
||||||
wf_copy = self.get_data().getWFData().copy()
|
wf_copy = self.get_data().getWFData().copy()
|
||||||
|
|
||||||
@ -3599,6 +3599,10 @@ class MainWindow(QMainWindow):
|
|||||||
for station in np.unique(list(self.getPicks('manual').keys()) + list(self.getPicks('auto').keys())):
|
for station in np.unique(list(self.getPicks('manual').keys()) + list(self.getPicks('auto').keys())):
|
||||||
wf_select += wf_copy.select(station=station)
|
wf_select += wf_copy.select(station=station)
|
||||||
|
|
||||||
|
if not wf_select:
|
||||||
|
logging.warning('Empty Stream in calc_magnitude. Return.')
|
||||||
|
return []
|
||||||
|
|
||||||
corr_wf = restitute_data(wf_select, self.metadata)
|
corr_wf = restitute_data(wf_select, self.metadata)
|
||||||
# calculate moment magnitude
|
# calculate moment magnitude
|
||||||
moment_mag = MomentMagnitude(corr_wf, self.get_data().get_evt_data(), self.inputs.get('vp'),
|
moment_mag = MomentMagnitude(corr_wf, self.get_data().get_evt_data(), self.inputs.get('vp'),
|
||||||
|
@ -268,9 +268,6 @@ class Metadata(object):
|
|||||||
if not fnames:
|
if not fnames:
|
||||||
# search for station name in filename
|
# search for station name in filename
|
||||||
fnames = glob.glob(os.path.join(path_to_inventory, '*' + station + '*'))
|
fnames = glob.glob(os.path.join(path_to_inventory, '*' + station + '*'))
|
||||||
if not fnames:
|
|
||||||
# search for network name in filename
|
|
||||||
fnames = glob.glob(os.path.join(path_to_inventory, '*' + network + '*'))
|
|
||||||
if not fnames:
|
if not fnames:
|
||||||
if self.verbosity:
|
if self.verbosity:
|
||||||
print('Could not find filenames matching station name, network name or seed id')
|
print('Could not find filenames matching station name, network name or seed id')
|
||||||
@ -282,7 +279,7 @@ class Metadata(object):
|
|||||||
continue
|
continue
|
||||||
invtype, robj = self._read_metadata_file(os.path.join(path_to_inventory, fname))
|
invtype, robj = self._read_metadata_file(os.path.join(path_to_inventory, fname))
|
||||||
try:
|
try:
|
||||||
# robj.get_coordinates(station_seed_id) # TODO: Commented out, failed with Parser, is this needed?
|
robj.get_coordinates(station_seed_id)
|
||||||
self.inventory_files[fname] = {'invtype': invtype,
|
self.inventory_files[fname] = {'invtype': invtype,
|
||||||
'data': robj}
|
'data': robj}
|
||||||
if station_seed_id in self.seed_ids.keys():
|
if station_seed_id in self.seed_ids.keys():
|
||||||
@ -290,6 +287,7 @@ class Metadata(object):
|
|||||||
self.seed_ids[station_seed_id] = fname
|
self.seed_ids[station_seed_id] = fname
|
||||||
return True
|
return True
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
logging.warning(e)
|
||||||
continue
|
continue
|
||||||
print('Could not find metadata for station_seed_id {} in path {}'.format(station_seed_id, path_to_inventory))
|
print('Could not find metadata for station_seed_id {} in path {}'.format(station_seed_id, path_to_inventory))
|
||||||
|
|
||||||
@ -654,6 +652,8 @@ def restitute_data(data, metadata, unit='VEL', force=False, ncores=0):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
# data = remove_underscores(data)
|
# data = remove_underscores(data)
|
||||||
|
if not data:
|
||||||
|
return
|
||||||
|
|
||||||
# loop over traces
|
# loop over traces
|
||||||
input_tuples = []
|
input_tuples = []
|
||||||
@ -661,9 +661,14 @@ def restitute_data(data, metadata, unit='VEL', force=False, ncores=0):
|
|||||||
input_tuples.append((tr, metadata, unit, force))
|
input_tuples.append((tr, metadata, unit, force))
|
||||||
data.remove(tr)
|
data.remove(tr)
|
||||||
|
|
||||||
pool = gen_Pool(ncores)
|
if ncores == 0:
|
||||||
result = pool.imap_unordered(restitute_trace, input_tuples)
|
result = []
|
||||||
pool.close()
|
for input_tuple in input_tuples:
|
||||||
|
result.append(restitute_trace(input_tuple))
|
||||||
|
else:
|
||||||
|
pool = gen_Pool(ncores)
|
||||||
|
result = pool.imap_unordered(restitute_trace, input_tuples)
|
||||||
|
pool.close()
|
||||||
|
|
||||||
for tr, remove_trace in result:
|
for tr, remove_trace in result:
|
||||||
if not remove_trace:
|
if not remove_trace:
|
||||||
|
Loading…
Reference in New Issue
Block a user