From c7d7acd7e3770f9fe3d7e6cab8169b064c3b2d6b Mon Sep 17 00:00:00 2001
From: Sebastianw Wehling-Benatelli <sebastian.wehling@rub.de>
Date: Tue, 10 May 2016 09:54:21 +0200
Subject: [PATCH] [edit] implemented loop over PILOT database for reassessment

additionally the output of reassess_pilot_event is more verbose now
---
 pylot/core/io/phases.py              | 18 ++++++++++++++++--
 scripts/pylot-reasses-pilot-event.py |  2 +-
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/pylot/core/io/phases.py b/pylot/core/io/phases.py
index 13ad3eeb..a46c263b 100644
--- a/pylot/core/io/phases.py
+++ b/pylot/core/io/phases.py
@@ -267,7 +267,16 @@ def picks_from_picksdict(picks):
     return picks_list
 
 
-def reassess_pilot_event(root_dir, event_id, fn_param=None):
+def reassess_pilot_db(root_dir, out_dir=None, fn_param=None):
+    import glob
+
+    evt_list = glob.glob1(root_dir,'e????.???.??')
+
+    for evt in evt_list:
+        reassess_pilot_event(root_dir, evt, out_dir, fn_param)
+
+
+def reassess_pilot_event(root_dir, event_id, out_dir=None, fn_param=None):
     from obspy import read
 
     from pylot.core.io.inputs import AutoPickParameter
@@ -283,7 +292,9 @@ def reassess_pilot_event(root_dir, event_id, fn_param=None):
     phases_file = glob.glob(os.path.join(search_base, 'PHASES.mat'))
     if not phases_file:
         return
+    print('Opening PILOT phases file: {fn}'.format(fn=phases_file[0]))
     picks_dict = picks_from_pilot(phases_file[0])
+    print('Dictionary read from PHASES.mat:\n{0}'.format(picks_dict))
     for station in picks_dict.keys():
         fn_pattern = os.path.join(search_base, '{0}*'.format(station))
         try:
@@ -320,7 +331,10 @@ def reassess_pilot_event(root_dir, event_id, fn_param=None):
     evt = ope.Event(resource_id=event_id)
     evt.picks = picks_from_picksdict(picks_dict)
     # write phase information to file
-    fnout_prefix = os.path.join(root_dir, event_id, '{0}.'.format(event_id))
+    if not out_dir:
+        fnout_prefix = os.path.join(root_dir, event_id, '{0}.'.format(event_id))
+    else:
+        fnout_prefix = os.path.join(out_dir, event_id, '{0}.'.format(event_id))
     evt.write(fnout_prefix + 'xml', format='QUAKEML')
     #evt.write(fnout_prefix + 'cnv', format='VELEST')
 
diff --git a/scripts/pylot-reasses-pilot-event.py b/scripts/pylot-reasses-pilot-event.py
index f48a08f2..939bf860 100644
--- a/scripts/pylot-reasses-pilot-event.py
+++ b/scripts/pylot-reasses-pilot-event.py
@@ -21,4 +21,4 @@ if __name__ == '__main__':
     )
 
     args = parser.parse_args()
-    reassess_pilot_event(args.dir, args.id)
+    reassess_pilot_event(args.dir, args.id, None)