marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[update] some bugfixes relating array_maps, removed some unneccessary files

Browse Source
This commit is contained in:
Marcel Paffrath
2019-10-23 11:33:37 +02:00
parent e94129ec36
commit 18aa67e5ba
11 changed files with 36 additions and 5184 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pylot/core/util/generate_array_maps.py
View File

@@ -9,12 +9,16 @@ from pylot.core.util.array_map import Array_map
import matplotlib.pyplot as plt
def main(project_file_path, manual=False, auto=True):
def main(project_file_path, manual=False, auto=True, file_format='png', f_ext=''):
project = Project.load(project_file_path)
nEvents = len(project.eventlist)
for index, event in enumerate(project.eventlist):
eventdir = event.path
# MP MP TESTING +++
#if not eventdir.endswith('20170908_044946.a'):
# continue
# MP MP ----
print('Working on event: {} ({}/{})'.format(eventdir, index + 1, nEvents))
# check for picks
manualpicks = event.getPicks()
@@ -30,14 +34,19 @@ def main(project_file_path, manual=False, auto=True):
# create figure to plot on
fig = plt.figure(figsize=(16,9))
# create array map object
map = Array_map(None, metadata, figure=fig, width=2.13e6, height=1.2e6)
map = Array_map(None, metadata, figure=fig, width=2.13e6, height=1.2e6, pointsize=15., linewidth=1.0)
# set combobox to auto/manual to plot correct pick type
map.comboBox_am.setCurrentIndex(map.comboBox_am.findText(pick_type))
# add picks to map and save file
map.refresh_drawings(manualpicks, autopicks)
fpath_out = os.path.join(eventdir, 'array_map_{}_{}.svg'.format(event.pylot_id, pick_type))
fpath_out = os.path.join(eventdir, 'array_map_{}_{}{}.{}'.format(event.pylot_id, pick_type, f_ext,
file_format))
fig.savefig(fpath_out, dpi=300.)
print('Wrote file: {}'.format(fpath_out))
if __name__ == '__main__':
main('/home/marcel/pylot_m7_mantle_correlated.plp')
#main('/home/marcel/pylot_m7_mantle_correlated_ORIG_AUTOPYLOT.plp')
#main('/home/marcel/test_project_obs_highf.plp', f_ext='_highf')
#main('/home/marcel/test_project_obs_lowf.plp', f_ext='_lowf')
#main('/home/marcel/test_project_obs_hybrid.plp', f_ext='_hybrid')
main('/home/marcel/test_project_obs_uhf.plp', f_ext='_uhf')