[addresses #167] started fixing the multiple phase saving issue

QtPyLoT.py

@@ -42,7 +42,8 @@ from pylot.core.read.data import Data
 from pylot.core.read.inputs import FilterOptions, AutoPickParameter
 from pylot.core.pick.autopick import autopickevent
 from pylot.core.util.defaults import FILTERDEFAULTS
-from pylot.core.util.errors import FormatError, DatastructureError
+from pylot.core.util.errors import FormatError, DatastructureError,\
+    OverwriteError
 from pylot.core.util.connection import checkurl
 from pylot.core.util.utils import fnConstructor, createEvent, getLogin,\
     createCreationInfo, getGlobalTimes
@@ -399,7 +400,18 @@ class MainWindow(QMainWindow):
     def saveData(self):
         settings = QSettings()
         exform = settings.value('data/exportFormat', 'QUAKEML')
+        try:
             self.getData().applyEVTData(self.getPicks())
+        except OverwriteError:
+            msgBox = QMessageBox()
+            msgBox.setText("Picks have been modified!")
+            msgBox.setInformativeText("Do you want to overwrite the picks and save?")
+            msgBox.setStandardButtons(QMessageBox.Save | QMessageBox.Discard |
+                                      QMessageBox.Cancel)
+            msgBox.setDefaultButton(QMessageBox.Save)
+            ret = msgBox.exec_()
+            if ret == QMessageBox.Save:
+                print('Overwrite and Save')
         try:
             self.getData().exportEvent(self.fname, exform)
         except FormatError:
@@ -407,7 +419,7 @@ class MainWindow(QMainWindow):
         except AttributeError, e:
             print 'warning: {0}'.format(e)
             directory = os.path.join(self.getRoot(), self.getEventFileName())
-            file_filter = "Seismic observation files (*.cnv *.obs *.xml)"
+            file_filter = "QuakeML file (*.xml);;VELEST observation file format (*.cnv);;NonLinLoc observation file (*.obs)"
             fname = QFileDialog.getSaveFileName(self, 'Save event data ...',
                                                 directory, file_filter)
             fbasename, exform = os.path.splitext(fname[0])

pylot/core/read/data.py

@@ -9,7 +9,7 @@ from obspy.core.event import Event, ResourceIdentifier, Pick, WaveformStreamID
 
 from pylot.core.read.io import readPILOTEvent
 from pylot.core.util.utils import fnConstructor, getGlobalTimes
-from pylot.core.util.errors import FormatError
+from pylot.core.util.errors import FormatError, OverwriteError
 
 
 class Data(object):
@@ -143,6 +143,9 @@ class Data(object):
         self.wfdata = self.getOriginalWFData().copy()
         self.dirty = False
 
+    def resetPicks(self):
+        self.getEvtData().picks = []
+
     def restituteWFData(self, fninventory):
         st = self.getWFData()
         inv = read_inventory(fninventory)
@@ -157,6 +160,8 @@ class Data(object):
 
         def applyPicks(picks):
             firstonset = None
+            if self.getEvtData().picks:
+                raise OverwriteError('Actual picks would be overwritten!')
             for station, onsets in picks.items():
                 print 'Reading picks on station %s' % station
                 for label, phase in onsets.items():

pylot/core/util/errors.py

@@ -16,3 +16,7 @@ class FormatError(Exception):
 
 class DatastructureError(Exception):
     pass
+
+
+class OverwriteError(IOError):
+    pass