From 50e7b8381222c1884cef03ae58bb6fdabb2c8697 Mon Sep 17 00:00:00 2001
From: marcel <marcel.paffrath@rub.de>
Date: Mon, 21 Aug 2017 15:11:07 +0200
Subject: [PATCH] [new] new function to exclude picks by quality

---
 QtPyLoT.py               |  2 +-
 pylot/core/util/utils.py | 31 +++++++++++++++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/QtPyLoT.py b/QtPyLoT.py
index 0d52073f..031214fd 100755
--- a/QtPyLoT.py
+++ b/QtPyLoT.py
@@ -74,7 +74,7 @@ from pylot.core.util.connection import checkurl
 from pylot.core.util.dataprocessing import read_metadata, restitute_data
 from pylot.core.util.utils import fnConstructor, getLogin, \
     full_range, readFilterInformation, trim_station_components, check4gaps, make_pen, pick_color_plt, \
-    pick_linestyle_plt, remove_underscores, check4doubled, identifyPhaseID
+    pick_linestyle_plt, remove_underscores, check4doubled, identifyPhaseID, excludeQualityClasses
 from pylot.core.util.event import Event
 from pylot.core.io.location import create_creation_info, create_event
 from pylot.core.util.widgets import FilterOptionsDialog, NewEventDlg, \
diff --git a/pylot/core/util/utils.py b/pylot/core/util/utils.py
index 426c966d..24ad261e 100644
--- a/pylot/core/util/utils.py
+++ b/pylot/core/util/utils.py
@@ -71,6 +71,37 @@ def gen_Pool(ncores=0):
     return pool
 
 
+def excludeQualityClasses(picks, qClasses, timeerrorsP, timeerrorsS):
+    '''
+    takes PyLoT picks dictionary and returns a new dictionary with certain classes excluded.
+    :param picks: PyLoT picks dictionary
+    :param qClasses: list (or int) of quality classes (0-4) to exclude
+    :param timeerrorsP: time errors for classes (0-4) for P
+    :param timeerrorsS: time errors for classes (0-4) for S
+    :return: new picks dictionary
+    '''
+    from pylot.core.pick.utils import getQualityFromUncertainty
+
+    if type(qClasses) in [int, float]:
+        qClasses = [qClasses]
+
+    picksdict_new = {}
+
+    phaseError = {'P': timeerrorsP,
+                  'S': timeerrorsS}
+
+    for station, phases in picks.items():
+        for phase, pick in phases.items():
+            pickerror = phaseError[identifyPhaseID(phase)]
+            quality = getQualityFromUncertainty(pick['spe'], pickerror)
+            if not quality in qClasses:
+                if not station in picksdict_new:
+                    picksdict_new[station] = {}
+                picksdict_new[station][phase] = pick
+
+    return picksdict_new
+
+
 def clims(lim1, lim2):
     """
     takes two pairs of limits and returns one pair of common limts