updated autopick parameter writing routine

This commit is contained in:
Marcel Paffrath 2017-05-10 14:56:29 +02:00
parent 7d56e15ccf
commit d96bc3c621
3 changed files with 111 additions and 18 deletions

View File

@ -1 +1 @@
aeaa-dirty 7d56-dirty

View File

@ -198,9 +198,68 @@ class AutoPickParameter(object):
def export2File(self, fnout): def export2File(self, fnout):
fid_out = open(fnout, 'w') fid_out = open(fnout, 'w')
lines = [] lines = []
for key, value in self.iteritems(): # for key, value in self.iteritems():
lines.append('{key}\t{value}\n'.format(key=key, value=value)) # lines.append('{key}\t{value}\n'.format(key=key, value=value))
fid_out.writelines(lines) # fid_out.writelines(lines)
header = ('%This is a parameter input file for PyLoT/autoPyLoT.\n'+
'%All main and special settings regarding data handling\n'+
'%and picking are to be set here!\n'+
'%Parameters are optimized for local data sets!\n')
seperator = '%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%\n'
fid_out.write(header)
self.write_section(fid_out, self.get_main_para_names()['dirs'],
'main settings', seperator)
self.write_section(fid_out, self.get_main_para_names()['nlloc'],
'NLLoc settings', seperator)
self.write_section(fid_out, self.get_main_para_names()['smoment'],
'parameters for seismic moment estimation', seperator)
self.write_section(fid_out, self.get_main_para_names()['focmec'],
'focal mechanism', seperator)
self.write_section(fid_out, self.get_main_para_names()['pick'],
'common settings picker', seperator)
fid_out.write(('#special settings for calculating CF#\n'+
'%!!Edit the following only if you know what you are doing!!%\n'))
self.write_section(fid_out, self.get_special_para_names()['z'],
'Z-component', None)
self.write_section(fid_out, self.get_special_para_names()['h'],
'H-components', None)
self.write_section(fid_out, self.get_special_para_names()['fm'],
'first-motion picker', None)
self.write_section(fid_out, self.get_special_para_names()['quality'],
'quality assessment', None)
def write_section(self, fid, names, title, seperator):
if seperator:
fid.write(seperator)
fid.write('#{}#\n'.format(title))
l_val = 50
l_name = 15
l_ttip = 100
for name in names:
value = self[name]
if type(value) == list or type(value) == tuple:
value_tmp = ''
for vl in value:
value_tmp+= '{} '.format(vl)
value = value_tmp
tooltip = self.get_defaults()[name]['tooltip']
if not len(str(value)) > l_val:
value = '{:<{}} '.format(str(value), l_val)
else:
value = '{} '.format(str(value))
name += '#'
if not len(name) > l_name:
name = '#{:<{}} '.format(name, l_name)
else:
name = '#{} '.format(name)
if not len(tooltip) > l_ttip:
ttip = '%{:<{}}\n'.format(tooltip, l_ttip)
else:
ttip = '%{}\n'.format(tooltip)
line = value+name+ttip
fid.write(line)
class FilterOptions(object): class FilterOptions(object):

View File

@ -1590,11 +1590,39 @@ class AutoPickParaBox(QtGui.QWidget):
self.layout = QtGui.QHBoxLayout() self.layout = QtGui.QHBoxLayout()
self.layout.addWidget(self.tabs) self.layout.addWidget(self.tabs)
self.boxes = {} self.boxes = {}
self._init_sublayouts()
self.setLayout(self.layout) self.setLayout(self.layout)
self.add_main_parameters_tab() self.add_main_parameters_tab()
self.add_special_pick_parameters_tab() self.add_special_pick_parameters_tab()
self._toggle_advanced_settings()
def init_boxes(self, parameter_names, defaults=True): def _init_sublayouts(self):
self._main_layout = QtGui.QVBoxLayout()
self._advanced_layout = QtGui.QVBoxLayout()
self._create_advanced_cb()
def _create_advanced_cb(self):
self._advanced_cb = QtGui.QCheckBox('Enable Advanced Settings')
self._advanced_layout.addWidget(self._advanced_cb)
self._advanced_cb.toggled.connect(self._toggle_advanced_settings)
def _toggle_advanced_settings(self):
if self._advanced_cb.isChecked():
self._enable_advanced(True)
else:
self._enable_advanced(False)
def _enable_advanced(self, enable):
for lst in self.ap.get_special_para_names().values():
for param in lst:
box = self.boxes[param]
if type(box) is not list:
box.setEnabled(enable)
else:
for b in box:
b.setEnabled(enable)
def init_boxes(self, parameter_names, defaults=False):
grid = QtGui.QGridLayout() grid = QtGui.QGridLayout()
for index1, name in enumerate(parameter_names): for index1, name in enumerate(parameter_names):
@ -1606,6 +1634,8 @@ class AutoPickParaBox(QtGui.QWidget):
if not type(default_item['type']) == tuple: if not type(default_item['type']) == tuple:
if defaults: if defaults:
value = default_item['value'] value = default_item['value']
else:
value = self.ap[name]
typ = default_item['type'] typ = default_item['type']
box = self.create_box(value, typ, tooltip) box = self.create_box(value, typ, tooltip)
self.boxes[name] = box self.boxes[name] = box
@ -1613,6 +1643,8 @@ class AutoPickParaBox(QtGui.QWidget):
boxes = [] boxes = []
if defaults: if defaults:
values = default_item['value'] values = default_item['value']
else:
values = self.ap[name]
for index2, val in enumerate(values): for index2, val in enumerate(values):
typ = default_item['type'][index2] typ = default_item['type'][index2]
boxes.append(self.create_box(val, typ, tooltip)) boxes.append(self.create_box(val, typ, tooltip))
@ -1636,6 +1668,10 @@ class AutoPickParaBox(QtGui.QWidget):
box.setMaximum(100*value) box.setMaximum(100*value)
box.setValue(value) box.setValue(value)
elif typ == bool: elif typ == bool:
if value == 'True':
value = True
if value == 'False':
value = False
box = QtGui.QCheckBox() box = QtGui.QCheckBox()
box.setChecked(value) box.setChecked(value)
else: else:
@ -1661,32 +1697,30 @@ class AutoPickParaBox(QtGui.QWidget):
self.tabs.addTab(scrollA, name) self.tabs.addTab(scrollA, name)
def add_main_parameters_tab(self): def add_main_parameters_tab(self):
vb_layout = QtGui.QVBoxLayout() self.add_to_layout(self._main_layout, 'Directories',
self.add_to_layout(vb_layout, 'Directories',
self.ap.get_main_para_names()['dirs']) self.ap.get_main_para_names()['dirs'])
self.add_to_layout(vb_layout, 'NLLoc', self.add_to_layout(self._main_layout, 'NLLoc',
self.ap.get_main_para_names()['nlloc']) self.ap.get_main_para_names()['nlloc'])
self.add_to_layout(vb_layout, 'Seismic Moment', self.add_to_layout(self._main_layout, 'Seismic Moment',
self.ap.get_main_para_names()['smoment']) self.ap.get_main_para_names()['smoment'])
self.add_to_layout(vb_layout, 'Focal Mechanism', self.add_to_layout(self._main_layout, 'Focal Mechanism',
self.ap.get_main_para_names()['focmec']) self.ap.get_main_para_names()['focmec'])
self.add_to_layout(vb_layout, 'Pick Settings', self.add_to_layout(self._main_layout, 'Pick Settings',
self.ap.get_main_para_names()['pick'], self.ap.get_main_para_names()['pick'],
False) False)
self.add_tab(vb_layout, 'Main Settings') self.add_tab(self._main_layout, 'Main Settings')
def add_special_pick_parameters_tab(self): def add_special_pick_parameters_tab(self):
vb_layout = QtGui.QVBoxLayout() self.add_to_layout(self._advanced_layout, 'Z-component',
self.add_to_layout(vb_layout, 'Z-component',
self.ap.get_special_para_names()['z']) self.ap.get_special_para_names()['z'])
self.add_to_layout(vb_layout, 'H-components', self.add_to_layout(self._advanced_layout, 'H-components',
self.ap.get_special_para_names()['h']) self.ap.get_special_para_names()['h'])
self.add_to_layout(vb_layout, 'First-motion picker', self.add_to_layout(self._advanced_layout, 'First-motion picker',
self.ap.get_special_para_names()['fm']) self.ap.get_special_para_names()['fm'])
self.add_to_layout(vb_layout, 'Quality assessment', self.add_to_layout(self._advanced_layout, 'Quality assessment',
self.ap.get_special_para_names()['quality'], self.ap.get_special_para_names()['quality'],
False) False)
self.add_tab(vb_layout, 'Advanced Settings') self.add_tab(self._advanced_layout, 'Advanced Settings')
def gen_h_seperator(self): def gen_h_seperator(self):
seperator = QtGui.QFrame() seperator = QtGui.QFrame()