pylot/PyLoT.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
PyLoT: Main program
===================
PyLoT is a seismic data processing software capable of picking seismic
phases (symmetric and asymmetric error assignment), exporting these to
several common phase data formats and post process the data, e.g. locating
events, via external localization software.
Additionally PyLoT is meant as an interface to autoPyLoT which can
automatically pick seismic phases, if the parameters have properly been
chosen for the particular data set.

Some icons are out of a free of charge icon set, which can be found here:
https://www.iconfinder.com/iconsets/flavour

:authors:
    Sebastian Wehling-Benatelli / Ludger Küperkoch / Marcel Paffrath
:copyright:
    The PyLoT Development Team (https://ariadne.geophysik.rub.de/trac/PyLoT)
:license:
    GNU Lesser General Public License, Version 3
    (http://www.gnu.org/copyleft/lesser.html)
"""

import argparse
import json
import os
import platform
import shutil
import sys
import traceback
from datetime import datetime

import matplotlib

matplotlib.use('Qt5Agg')

from PySide2 import QtGui, QtCore, QtWidgets
from PySide2.QtCore import QCoreApplication, QSettings, Signal, QFile, \
    QFileInfo, Qt, QSize
from PySide2.QtGui import QIcon, QKeySequence, QPixmap, QStandardItem
from PySide2.QtWidgets import QMainWindow, QInputDialog, QFileDialog, \
    QWidget, QHBoxLayout, QVBoxLayout, QStyle, QLabel, QFrame, QAction, \
    QDialog, QApplication, QMessageBox, QSplashScreen, \
    QActionGroup, QListWidget, QListView, QAbstractItemView, \
    QTreeView, QComboBox, QTabWidget, QPushButton, QGridLayout, QTableWidgetItem, QTableWidget
import numpy as np
from obspy import UTCDateTime, Stream
from obspy.core.util import AttribDict

from pylot.core.util.obspyDMT_interface import check_obspydmt_structure

import pyqtgraph as pg

try:
    from matplotlib.backends.backend_qt5agg import FigureCanvas
except ImportError:
    from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

from pylot.core.analysis.magnitude import LocalMagnitude, MomentMagnitude
from pylot.core.io.data import Data
from pylot.core.io.inputs import FilterOptions, PylotParameter
from autoPyLoT import autoPyLoT
from pylot.core.pick.compare import Comparison
from pylot.core.pick.utils import getQualityFromUncertainty
from pylot.core.io.phases import picksdict_from_picks
from pylot.core.io.project import Project
import pylot.core.loc.nll as nll
from pylot.core.util.errors import DatastructureError, \
    OverwriteError
from pylot.core.util.connection import checkurl
from pylot.core.util.dataprocessing import Metadata, restitute_data
from pylot.core.util.utils import fnConstructor, get_login, \
    full_range, readFilterInformation, pick_color_plt, \
    pick_linestyle_plt, identifyPhaseID, excludeQualityClasses, \
    transform_colors_mpl, transform_colors_mpl_str, getAutoFilteroptions, check_all_obspy, \
    check_all_pylot, get_bool, get_none
from pylot.core.util.gui import make_pen
from pylot.core.util.widgets import FilterOptionsDialog, PylotCanvas, WaveformWidgetPG, PropertiesDlg, HelpForm, createAction, PickDlg, \
    ComparisonWidget, TuneAutopicker, PylotParaBox, AutoPickDlg, CanvasWidget, AutoPickWidget, \
    CompareEventsWidget, ProgressBarWidget, AddMetadataWidget, SingleTextLineDialog, LogWidget, PickQualitiesFromXml, \
    SpectrogramTab, SearchFileByExtensionDialog
from pylot.core.util.array_map import Array_map
from pylot.core.util.structure import DATASTRUCTURE
from pylot.core.util.thread import Thread, Worker
from pylot.core.util.version import get_git_version as _getVersionString
from pylot.core.io.getEventListFromXML import geteventlistfromxml
from pylot.core.io.phases import getQualitiesfromxml

from pylot.styles import style_settings

if sys.version_info.major == 3:
    import icons_rc_3 as icons_rc
else:
    raise ImportError(f'Python version {sys.version_info.major} of current interpreter not supported.'
                      f'\nPlease use Python 3+.')

# workaround to prevent PyCharm from deleting icons_rc import when optimizing imports
icons_rc = icons_rc

locateTool = dict(nll=nll)


class MainWindow(QMainWindow):
    _metadata: Metadata
    __version__ = _getVersionString()
    closing = Signal()

    def __init__(self, parent=None, infile=None, reset_qsettings=False):
        super(MainWindow, self).__init__(parent)

        if infile and os.path.isfile(infile) is False:
            infile = QFileDialog().getOpenFileName(caption='Choose PyLoT-input file')[0]

            if not os.path.exists(infile):
                QMessageBox.warning(self, "PyLoT Warning",
                                    "No PyLoT-input file declared! Using default parameters!")
                infile = None

        self._inputs = PylotParameter(infile)
        if not infile:
            self._inputs.reset_defaults()

        self.infile = infile
        self._props = None

        self.gain = 1.

        self.dirty = False
        self.project = Project()
        self.project.parameter = self._inputs
        self.tap = None
        self.apw = None
        self.paraBox = None
        self.array_map = None
        self._metadata = Metadata(verbosity=0)
        self._eventChanged = [False, False]
        self.apd_local = None
        self.apd_sge = None
        self.stations_highlighted = []
        self.obspy_dmt = False
        self.nextStation = False

        self.poS_id = None
        self.ae_id = None
        self.scroll_id = None
        self._ctrl = False  # control key pressed
        self._shift = False  # shift key pressed

        self.drawnPicks = {'auto': {},
                           'manual': {}}

        # default factor for dataplot e.g. enabling/disabling scrollarea
        self.height_factor = 12
        self.plot_method = 'normal'

        # UI has to be set up before(!) children widgets are about to show up
        self.createAction = createAction
        # read settings
        settings = QSettings()
        if reset_qsettings:
            settings.clear()
        self.recentfiles = settings.value("data/recentEvents", [])
        self.dispComponent = str(settings.value("plotting/dispComponent", "Z"))

        # initialize event data
        if self.recentfiles:
            lastEvent = self.getLastEvent()
            self.data = Data(self, lastEvent)
        else:
            self.data = Data(self)
            self.data._new = False
        self.autodata = Data(self)

        self.fnames = None
        self.fnames_comp = None
        self._stime = None

        # track deleted picks for logging
        self.deleted_picks = {}

        # headers for event table
        self.table_headers = ['', 'Event', 'Time', 'Lat', 'Lon', 'Depth', 'Ml', 'Mw', '[N] MP', '[N] AP', 'Tuning Set',
                              'Test Set', 'Notes']

        # TODO: refactor rootpath to datapath
        while True:
            try:
                if settings.value("user/FullName", None) is None:
                    fulluser = QInputDialog.getText(self, "Enter Name:", "Full name")
                    settings.setValue("user/FullName", fulluser)
                    settings.setValue("user/Login", get_login())
                if settings.value("agency_id", None) is None:
                    agency = QInputDialog.getText(self,
                                                  "Enter authority/institution name:",
                                                  "Authority")
                    settings.setValue("agency_id", agency)
                structure_setting = settings.value("data/Structure", "PILOT")
                if not structure_setting:
                    structure_setting = 'PILOT'
                self.dataStructure = DATASTRUCTURE[structure_setting]()
                self.seismicPhase = str(settings.value("phase", "P"))
                if settings.value("data/dataRoot", None) is None:
                    dirname = QFileDialog().getExistingDirectory(
                        caption='Choose data root ...')
                    settings.setValue("data/dataRoot", dirname)
                if settings.value('useGuiFilter') is None:
                    settings.setValue('useGuiFilter', False)
                if settings.value('output/Format', None) is None:
                    outformat = QInputDialog.getText(self,
                                                     "Enter output format (*.xml, *.cnv, *.obs, *_focmec.in, *.pha):",
                                                     "Format")
                    settings.setValue("output/Format", outformat)
                if settings.value('autoFilter', None) is None:
                    settings.setValue('autoFilter', True)
                settings.sync()
                break
            except Exception as e:
                qmb = QMessageBox(self, icon=QMessageBox.Question,
                                  text='Could not init application settings: {}.'
                                       'Do you want to reset application settings?'.format(e),
                                  windowTitle='PyLoT - Init QSettings warning')
                qmb.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
                qmb.setDefaultButton(QMessageBox.No)
                ret = qmb.exec_()
                if ret == qmb.Yes:
                    settings.clear()
                else:
                    sys.exit()
                print('Settings cleared!')

        # setup UI
        self.setupUi()

        self.updateFilteroptions()

        self.loc = False

    def setupUi(self, use_logwidget=False):
        try:
            self.startTime = min(
                [tr.stats.starttime for tr in self.data.wfdata])
        except:
            self.startTime = UTCDateTime()

        self.init_styles()

        pylot_icon = QIcon()
        pylot_icon.addPixmap(QPixmap(':/icons/pylot.png'))

        self.setWindowTitle("PyLoT - do seismic processing the python way")
        self.setWindowIcon(pylot_icon)

        _widget = QWidget()
        self._main_layout = QVBoxLayout()

        quitIcon = self.style().standardIcon(QStyle.SP_MediaStop)
        helpIcon = self.style().standardIcon(QStyle.SP_DialogHelpButton)
        newFolderIcon = self.style().standardIcon(QStyle.SP_FileDialogNewFolder)

        # create resource icons
        newIcon = QIcon()
        newIcon.addPixmap(QPixmap(':/icons/newfile.png'))
        addIcon = QIcon()
        addIcon.addPixmap(QPixmap(':/icons/add.png'))
        saveIcon = QIcon()
        saveIcon.addPixmap(QPixmap(':/icons/save.png'))
        saveasIcon = QIcon()
        saveasIcon.addPixmap(QPixmap(':/icons/saveas.png'))
        saveProjectIcon = QIcon()
        saveProjectIcon.addPixmap(QPixmap(':/icons/saveproject.png'))
        saveProjectAsIcon = QIcon()
        saveProjectAsIcon.addPixmap(QPixmap(':/icons/saveprojectas.png'))
        openIcon = QIcon()
        openIcon.addPixmap(QPixmap(':/icons/openfile.png'))
        openProjectIcon = QIcon()
        openProjectIcon.addPixmap(QPixmap(':/icons/openproject.png'))
        openLocIcon = QIcon()
        openLocIcon.addPixmap(QPixmap(':/icons/openloc.png'))
        openEventIcon = QIcon()
        openEventIcon.addPixmap(QPixmap(':/icons/openpick.png'))
        openEventsIcon = QIcon()
        openEventsIcon.addPixmap(QPixmap(':/icons/openpicks.png'))
        saveEventsIcon = QIcon()
        saveEventsIcon.addPixmap(QPixmap(':/icons/savepicks.png'))
        prefIcon = QIcon()
        prefIcon.addPixmap(QPixmap(':/icons/preferences.png'))
        paraIcon = QIcon()
        paraIcon.addPixmap(QPixmap(':/icons/parameter.png'))
        deleteIcon = QIcon()
        deleteIcon.addPixmap(QPixmap(':/icons/delete.png'))
        self.inventoryIcon = QIcon()
        self.inventoryIcon.addPixmap(QPixmap(':/icons/inventory.png'))
        self.mapIcon = QIcon()
        self.mapIcon.addPixmap(QPixmap(':/icons/map.png'))
        self.autopicksicon_small = QIcon()
        self.autopicksicon_small.addPixmap(QPixmap(':/icons/autopicksicon_small.png'))
        self.manupicksicon_small = QIcon()
        self.manupicksicon_small.addPixmap(QPixmap(':/icons/manupicksicon_small.png'))
        loadpiloticon = QIcon()
        loadpiloticon.addPixmap(QPixmap(':/icons/Matlab_PILOT_icon.png'))
        p_icon = QIcon()
        p_icon.addPixmap(QPixmap(':/icons/key_P.png'))
        s_icon = QIcon()
        s_icon.addPixmap(QPixmap(':/icons/key_S.png'))
        print_icon = QIcon()
        print_icon.addPixmap(QPixmap(':/icons/printer.png'))
        self.filter_icon = QIcon()
        self.filter_icon.addPixmap(QPixmap(':/icons/filter.png'))
        self.filter_icon_p = QIcon()
        self.filter_icon_p.addPixmap(QPixmap(':/icons/filter_p.png'))
        self.filter_icon_s = QIcon()
        self.filter_icon_s.addPixmap(QPixmap(':/icons/filter_s.png'))
        z_icon = QIcon()
        z_icon.addPixmap(QPixmap(':/icons/key_Z.png'))
        n_icon = QIcon()
        n_icon.addPixmap(QPixmap(':/icons/key_N.png'))
        e_icon = QIcon()
        e_icon.addPixmap(QPixmap(':/icons/key_E.png'))
        autotune_icon = QIcon()
        autotune_icon.addPixmap(QPixmap(':/icons/tune.png'))
        autopylot_icon = QIcon()
        autopylot_icon.addPixmap(QPixmap(':/icons/autopylot_button'))
        locate_icon = QIcon()
        locate_icon.addPixmap(QPixmap(':/icons/locate_button.png'))
        compare_icon = QIcon()
        compare_icon.addPixmap(QPixmap(':/icons/compare_button.png'))
        qualities_icon = QIcon()
        qualities_icon.addPixmap(QPixmap(':/icons/pick_qualities_button.png'))
        eventlist_xml_icon = QIcon()
        eventlist_xml_icon.addPixmap(QPixmap(':/icons/eventlist_xml_button.png'))

        self.newProjectAction = self.createAction(self, "&New project ...",
                                                  self.createNewProject,
                                                  QKeySequence.New, newIcon,
                                                  "Create a new Project.")
        self.openProjectAction = self.createAction(self, "&Open project ...",
                                                   self.loadProject,
                                                   QKeySequence.Open,
                                                   openProjectIcon,
                                                   "Load project file")
        self.saveProjectAction = self.createAction(self, "&Save project ...",
                                                   self.saveProject,
                                                   QKeySequence.Save,
                                                   saveProjectIcon,
                                                   "Save project file")
        self.saveProjectAction.setEnabled(False)
        self.saveProjectAsAction = self.createAction(self, "Save project as ...",
                                                     self.saveProjectAs,
                                                     QKeySequence.SaveAs,
                                                     saveProjectAsIcon,
                                                     "Save project file as...")
        self.saveProjectAsAction.setEnabled(False)
        # newEventAction = self.createAction(self, "&New event ...",
        #                                    self.createNewEvent,
        #                                    QKeySequence.New, newIcon,
        #                                    "Create a new event.")
        self.openEventAction = self.createAction(self, "Load event information...",
                                                 self.load_data,
                                                 "Ctrl+M",
                                                 openEventIcon,
                                                 "Load event information for "
                                                 "the displayed event.")
        self.openEventAction.setEnabled(False)
        self.openEventAction.setData(None)

        self.openEventsAutoAction = self.createAction(self, "Load event information &automatically ... ",
                                                      self.load_multiple_data,
                                                      "Ctrl+A",
                                                      openEventsIcon,
                                                      "Load event data automatically "
                                                      "for all events.")
        self.openEventsAutoAction.setEnabled(False)
        self.openEventsAutoAction.setData(None)

        self.loadlocationaction = self.createAction(self, "Load &location ...",
                                                    self.load_loc, "Ctrl+L",
                                                    openLocIcon,
                                                    "Load location information on "
                                                    "the displayed event.")
        self.loadlocationaction.setEnabled(False)
        self.loadpilotevent = self.createAction(self, "Load PILOT &event ...",
                                                self.load_pilotevent, None,
                                                loadpiloticon,
                                                "Load PILOT event from information "
                                                "MatLab binary collections (created"
                                                " in former MatLab based version).")
        self.loadpilotevent.setEnabled(False)

        self.saveEventAction = self.createAction(self, "Save &event information ...",
                                                 self.saveData, "Ctrl+P",
                                                 saveEventsIcon, "Save event pick data,"
                                                                 " source origin and magnitude.")
        self.disableSaveEventAction()

        self.exportProjectTableAction = self.createAction(self, "Export Project table...",
                                                          self.exportProjectTableDialogs, None,
                                                          None, "Export Project table (csv)")

        self.addEventDataAction = self.createAction(self, "Add &events ...",
                                                    self.add_events,
                                                    "Ctrl+E", addIcon,
                                                    "Add event data")
        prefsEventAction = self.createAction(self, "Preferences",
                                             self.PyLoTprefs,
                                             QKeySequence.Preferences,
                                             prefIcon,
                                             "Edit PyLoT app preferences.")
        quitAction = self.createAction(self, "&Quit",
                                       QCoreApplication.instance().quit,
                                       QKeySequence.Close, quitIcon,
                                       "Close event and quit PyLoT")
        self.parameterAction = self.createAction(self, "Parameter",
                                                 self.setParameter,
                                                 None, paraIcon,
                                                 "Modify Parameter")
        self.deleteAutopicksAction = self.createAction(self, "Delete Autopicks",
                                                       self.deleteAllAutopicks,
                                                       None, deleteIcon,
                                                       "Delete all automatic picks from Project.")
        self.filterActionP = createAction(parent=self, text='Apply P Filter',
                                          slot=self.filterP,
                                          icon=self.filter_icon_p,
                                          tip='Toggle filtered/original'
                                              ' waveforms',
                                          checkable=True,
                                          shortcut='P')
        self.filterActionS = createAction(parent=self, text='Apply S Filter',
                                          slot=self.filterS,
                                          icon=self.filter_icon_s,
                                          tip='Toggle filtered/original'
                                              ' waveforms',
                                          checkable=True,
                                          shortcut='S')
        filterEditAction = self.createAction(self, "&Filter parameter ...",
                                             self.adjustFilterOptions,
                                             "Ctrl+F", self.filter_icon,
                                             """Adjust filter parameters.""")
        self.inventoryAction = self.createAction(self, "Manage &Inventories ...",
                                                 self.add_metadata,
                                                 "Ctrl+I", self.inventoryIcon,
                                                 """Manage metadata for current project""",
                                                 False)
        self.initMapAction = self.createAction(self, "Init array map ...",
                                               self.init_array_map,
                                               "Ctrl+M", self.mapIcon,
                                               """Initialize array map with current metadata""",
                                               False)
        self.initMapAction.setEnabled(False)
        self.selectPAction = self.createAction(self, "&P", self.alterPhase,
                                               "Alt+P",
                                               p_icon,
                                               "Toggle P phase.", True)
        self.selectSAction = self.createAction(self, "&S", self.alterPhase,
                                               "Alt+S",
                                               s_icon,
                                               "Toggle S phase", True)
        self.compare_action = self.createAction(self, "&Compare picks...",
                                                self.comparePicks, "Alt+C",
                                                compare_icon, "Comparison of "
                                                              "manual and "
                                                              "automatic pick "
                                                              "data.", False)
        self.compare_action.setEnabled(False)
        self.qualities_action = self.createAction(parent=self, text='Show pick qualities...',
                                                  slot=self.pickQualities, shortcut='Alt+Q',
                                                  icon=qualities_icon, tip='Histogram of pick qualities')
        self.qualities_action.setEnabled(False)
        # MP MP not yet implemented, therefore hide:
        # LK will be implemented soon, basic script has already (03/2021) been finished
        self.qualities_action.setVisible(True)

        self.eventlist_xml_action = self.createAction(parent=self, text='Create Eventlist from XML',
                                                      slot=self.eventlistXml, shortcut='Alt+X',
                                                      icon=eventlist_xml_icon,
                                                      tip='Create an Eventlist from a XML File')
        self.eventlist_xml_action.setEnabled(False)
        printAction = self.createAction(self, "&Print event ...",
                                        self.show_event_information, QKeySequence.Print,
                                        print_icon,
                                        "Print waveform section.")
        helpAction = self.createAction(self, "&Help ...", self.helpHelp,
                                       QKeySequence.HelpContents, helpIcon,
                                       """Show either the documentation
                                       homepage (internet connection available),
                                       or shipped documentation files.""")
        logAction = self.createAction(self, "&Show Log", self.showLogWidget,
                                      tip="""Display Log""")

        logAction.setEnabled(use_logwidget)

        # create button group for component selection

        componentGroup = QActionGroup(self)
        componentGroup.setExclusive(True)

        self.z_action = self.createAction(parent=componentGroup, text='Z',
                                          slot=self.plotZ, shortcut='Alt+Z',
                                          icon=z_icon, tip='Display the vertical (Z)'
                                                           ' component.',
                                          checkable=True)
        self.z_action.setChecked(True)
        self.z_action.setEnabled(False)

        self.n_action = self.createAction(parent=componentGroup, text='N',
                                          slot=self.plotN, shortcut='Alt+N',
                                          icon=n_icon,
                                          tip='Display the north-south (N) '
                                              'component.', checkable=True)
        self.n_action.setEnabled(False)

        self.e_action = self.createAction(parent=componentGroup, text='E',
                                          slot=self.plotE, shortcut='Alt+E',
                                          icon=e_icon,
                                          tip='Display the east-west (E) component.',
                                          checkable=True)
        self.e_action.setEnabled(False)

        componentActions = (self.z_action, self.n_action, self.e_action)

        self.auto_tune = self.createAction(parent=self, text='autoTune',
                                           slot=self.tune_autopicker, shortcut='Alt+Ctrl+A',
                                           icon=autotune_icon, tip='Tune autopicking algorithm.')

        self.auto_tune.setEnabled(False)

        self.auto_pick = self.createAction(parent=self, text='Current event',
                                           slot=self.autoPick, shortcut='Alt+Ctrl+A',
                                           icon=autopylot_icon, tip='Automatically pick'
                                                                    ' the displayed waveforms.')
        self.auto_pick.setEnabled(False)

        self.auto_pick_local = self.createAction(parent=self, text='Whole project (local machine)...',
                                                 slot=self.autoPickProject, shortcut=None,
                                                 icon=self.autopicksicon_small, tip='Automatically pick'
                                                                                    ' the complete project on local machine.')
        self.auto_pick_local.setEnabled(False)

        self.auto_pick_sge = self.createAction(parent=self, text='Whole project (grid engine)...',
                                               slot=self.autoPickProjectSGE, shortcut=None,
                                               icon=self.autopicksicon_small, tip='Automatically pick'
                                                                                  ' the complete project on grid engine.')
        self.auto_pick_sge.setEnabled(False)

        pickActions = (
            self.auto_tune, self.auto_pick, self.compare_action, self.qualities_action, self.eventlist_xml_action)

        # pickToolBar = self.addToolBar("PickTools")
        # pickToolActions = (selectStation, )
        # pickToolBar.setObjectName("PickTools")
        # self.addActions(pickToolBar, pickToolActions)
        self.locateEventAction = self.createAction(parent=self, text='locate the event',
                                                   slot=self.locate_event,
                                                   shortcut='Alt+Ctrl+L',
                                                   icon=locate_icon,
                                                   tip='Locate the event using '
                                                       'the displayed manual arrivals.')
        self.locateEventAction.setEnabled(False)

        locationToolActions = (self.locateEventAction,)

        # add top menu
        self.fileMenu = self.menuBar().addMenu('&File')
        self.fileMenuActions = (self.newProjectAction,
                                self.openProjectAction, self.saveProjectAction,
                                self.saveProjectAsAction, None,
                                self.addEventDataAction,
                                self.openEventAction, self.saveEventAction, None,
                                self.exportProjectTableAction, None,
                                quitAction)
        self.fileMenu.aboutToShow.connect(self.updateFileMenu)
        self.updateFileMenu()

        self.editMenu = self.menuBar().addMenu('&Edit')

        editActions = (self.filterActionP, self.filterActionS, filterEditAction, None,
                       # self.selectPAction, self.selectSAction, None,
                       self.inventoryAction, self.initMapAction, None,
                       prefsEventAction)
        # printAction) #TODO: print event?

        pickMenuActions = (self.parameterAction, self.deleteAutopicksAction)
        self.pickMenu = self.menuBar().addMenu('&Picking')
        self.autoPickMenu = self.pickMenu.addMenu(self.autopicksicon_small, 'Automatic picking')
        self.autoPickMenu.setEnabled(False)

        autoPickActions = (self.auto_pick, self.auto_pick_local, self.auto_pick_sge)

        self.helpMenu = self.menuBar().addMenu('&Help')
        helpActions = (helpAction, logAction)

        fileToolActions = (self.newProjectAction,
                           self.openProjectAction, self.saveProjectAction,
                           self.saveProjectAsAction)

        eventToolActions = (self.addEventDataAction,
                            self.openEventAction, self.openEventsAutoAction,
                            self.saveEventAction, self.loadlocationaction,
                            self.loadpilotevent)

        toolbars_keys = [
            "FileTools",
            "EventTools",
            "ComponentSelection",
            "autoPyLoT",
            "LocationTools"
        ]

        toolbars = {}

        for name in toolbars_keys:
            toolbars[name] = self.addToolBar(name)
            toolbars[name].setObjectName(name)
            toolbars[name].setMovable(False)

        self.addActions(self.editMenu, editActions)
        self.addActions(self.autoPickMenu, autoPickActions)
        self.addActions(self.pickMenu, pickMenuActions)
        self.addActions(self.helpMenu, helpActions)

        self.addActions(toolbars["FileTools"], fileToolActions)
        self.addActions(toolbars["EventTools"], eventToolActions)
        self.addActions(toolbars["ComponentSelection"], componentActions)
        self.addActions(toolbars["autoPyLoT"], pickActions)
        self.addActions(toolbars["LocationTools"], locationToolActions)

        # init pyqtgraph
        self.pg = pg

        # init style
        settings = QSettings()
        style = settings.value('style')
        self.set_style(style)

        # add event combo box, forward, backward and ref/test buttons
        self.eventBox = self.createEventBox()
        self.eventBox.setMaxVisibleItems(30)
        self.eventBox.setEnabled(False)
        self.previous_button = QPushButton('<')
        self.next_button = QPushButton('>')
        self.init_ref_test_buttons()
        self._event_layout = QHBoxLayout()
        self._event_layout.addWidget(QLabel('Event: '))
        self._event_layout.addWidget(self.eventBox)
        self._event_layout.addWidget(self.previous_button)
        self._event_layout.addWidget(self.next_button)
        self._event_layout.addWidget(self.ref_event_button)
        self._event_layout.addWidget(self.test_event_button)
        self._event_layout.setStretch(1, 1)  # set stretch of item 1 to 1
        self._main_layout.addLayout(self._event_layout)
        self.eventBox.activated.connect(self.refreshEvents)

        self.previous_button.clicked.connect(self.previous_event)
        self.next_button.clicked.connect(self.next_event)
        self.previous_button.setEnabled(False)
        self.next_button.setEnabled(False)

        # add main tab widget
        self.tabs = QTabWidget(self)
        self._main_layout.addWidget(self.tabs)
        self.tabs.currentChanged.connect(self.refreshTabs)

        # add progressbar
        self.mainProgressBarWidget = ProgressBarWidget(self)
        self._main_layout.addWidget(self.mainProgressBarWidget)

        # add scroll area used in case number of traces gets too high
        self.wf_scroll_area = QtWidgets.QScrollArea(self)
        self.wf_scroll_area.setVisible(False)
        self.no_data_label = QLabel('No Data')
        self.no_data_label.setStyleSheet('color: red')
        self.no_data_label.setAlignment(Qt.AlignCenter)

        # create central matplotlib figure canvas widget
        self.init_wfWidget()

        # init main widgets for main tabs
        wf_tab = QtWidgets.QWidget(self)
        array_tab = QtWidgets.QWidget(self)
        events_tab = QtWidgets.QWidget(self)
        spectro_tab = QtWidgets.QWidget(self)

        # init main widgets layouts
        self.wf_layout = QtWidgets.QVBoxLayout()
        self.array_layout = QtWidgets.QVBoxLayout()
        self.events_layout = QtWidgets.QVBoxLayout()
        self.spectro_layout = QtWidgets.QVBoxLayout()
        wf_tab.setLayout(self.wf_layout)
        array_tab.setLayout(self.array_layout)
        events_tab.setLayout(self.events_layout)
        spectro_tab.setLayout(self.spectro_layout)

        # tighten up layouts inside tabs
        for layout in [self.wf_layout, self.array_layout, self.events_layout]:
            layout.setSpacing(0)
            layout.setContentsMargins(0, 0, 0, 0)

        # add tabs to main tab widget
        self.tabs.addTab(wf_tab, 'Waveform Plot')
        self.tabs.addTab(array_tab, 'Array Map')
        self.tabs.addTab(events_tab, 'Eventlist')
        self.tabs.addTab(spectro_tab, 'Spectro')

        self.wf_layout.addWidget(self.no_data_label)
        self.wf_layout.addWidget(self.wf_scroll_area)
        self.wf_scroll_area.setWidgetResizable(True)
        self.init_array_tab()
        self.init_event_table()
        self.init_spectro_tab()
        self.tabs.setCurrentIndex(0)

        self.eventLabel = QLabel()
        self.eventLabel.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
        status = self.statusBar()
        status.setSizeGripEnabled(False)
        status.addPermanentWidget(self.eventLabel)
        status.showMessage("Ready", 500)

        _widget.setLayout(self._main_layout)
        _widget.showFullScreen()


        if use_logwidget:
            self.logwidget = LogWidget(parent=None)
            self.logwidget.show()
            self.stdout = self.logwidget.stdout
            self.stderr = self.logwidget.stderr

        self.setCentralWidget(_widget)

        # Need to store PickQualities Window somewhere so it doesnt disappear
        self.pickQualitiesWindow = None


    def init_wfWidget(self):
        xlab = self.startTime.strftime('seconds since %Y/%m/%d %H:%M:%S (%Z)')
        plottitle = None  # "Overview: {0} components ".format(self.getComponent())
        self.disconnectWFplotEvents()
        self.pg = pg
        self.dataPlot = WaveformWidgetPG(parent=self,
                                         title=plottitle)
        self.dataPlot.setCursor(Qt.CrossCursor)
        self.wf_scroll_area.setWidget(self.dataPlot)
        if self.get_current_event():
            self.plotWaveformDataThread()

    def init_ref_test_buttons(self):
        '''
        Initiate/create buttons for assigning events containing manual picks to reference or test set.
        '''
        self.ref_event_button = QtWidgets.QPushButton('Tune')
        self.test_event_button = QtWidgets.QPushButton('Test')
        self.ref_event_button.setMinimumWidth(100)
        self.test_event_button.setMinimumWidth(100)
        self.ref_event_button.setToolTip('Set manual picks of current ' +
                                         'event as reference picks for autopicker tuning.')
        self.test_event_button.setToolTip('Set manual picks of current ' +
                                          'event as test picks for autopicker testing.')
        self.ref_event_button.setCheckable(True)
        self.test_event_button.setCheckable(True)
        self.set_button_border_color(self.ref_event_button, self._style['ref']['rgba'])
        self.set_button_border_color(self.test_event_button, self._style['test']['rgba'])
        self.ref_event_button.clicked.connect(self.toggleRef)
        self.test_event_button.clicked.connect(self.toggleTest)
        self.ref_event_button.setEnabled(False)
        self.test_event_button.setEnabled(False)

    def showLogWidget(self):
        self.logwidget.show()
        self.logwidget.activateWindow()

    def keyPressEvent(self, event):
        if event.key() == QtCore.Qt.Key.Key_Control:
            self._ctrl = True
        if event.key() == QtCore.Qt.Key.Key_Shift:
            self._shift = True
        if event.key() == QtCore.Qt.Key.Key_Plus:
            self.increase_gain()
        if event.key() == QtCore.Qt.Key.Key_Minus:
            self.decrease_gain()
        if event.key() == QtCore.Qt.Key.Key_R:
            self.reset_gain()

    def keyReleaseEvent(self, event):
        if event.key() == QtCore.Qt.Key.Key_Control:
            self._ctrl = False
        if event.key() == QtCore.Qt.Key.Key_Shift:
            self._shift = False

    def increase_gain(self, factor=1.3):
        self.modify_gain('+', factor)

    def decrease_gain(self, factor=1.3):
        self.modify_gain('-', factor)

    def reset_gain(self):
        self.gain = 1
        self.plotWaveformDataThread(filter=False)

    def modify_gain(self, direction, factor):
        assert (direction in ['+', '-']), 'unknown direction'
        if self._ctrl:
            factor = factor ** 3
        if direction == '+':
            self.gain *= factor
        elif direction == '-':
            self.gain /= factor
        self.plotWaveformDataThread(filter=False)

    def init_styles(self):
        self._styles = {}
        styles = ['default', 'dark', 'bright']
        stylecolors = style_settings.stylecolors
        for style in styles:
            if style in stylecolors.keys():
                self._styles[style] = stylecolors[style]

        self._phasecolors = style_settings.phasecolors
        styles_dir = os.path.dirname(style_settings.__file__)

        for style, stylecolors in self._styles.items():
            stylesheet = stylecolors['stylesheet']['filename']
            if stylesheet:
                stylesheet_file = open(os.path.join(styles_dir, stylesheet), 'r')
                stylesheet = stylesheet_file.read()
                stylesheet_file.close()
            else:
                stylesheet = self.styleSheet()

            bg_color = stylecolors['background']['rgba']
            line_color = stylecolors['linecolor']['rgba']
            multcursor_color = stylecolors['multicursor']['rgba']

            # transform to 0-1 values for mpl and update dict
            stylecolors['background']['rgba_mpl'] = transform_colors_mpl(bg_color)
            stylecolors['linecolor']['rgba_mpl'] = transform_colors_mpl(line_color)
            multcursor_color = stylecolors['multicursor']['rgba_mpl'] = transform_colors_mpl(multcursor_color)

            stylecolors['stylesheet'] = stylesheet

    def set_style(self, stylename=None):
        if not stylename:
            stylename = 'default'
        if not stylename in self._styles:
            qmb = QMessageBox.warning(self, 'Could not find style',
                                      'Could not find style with name {}. Using default.'.format(stylename))
            self.set_style('default')
            return

        style = self._styles[stylename]
        self._style = style
        self._stylename = stylename
        self.setStyleSheet(style['stylesheet'])

        # colors for ref/test event
        self._ref_test_colors = {
            'ref': QtGui.QColor(*style['ref']['rgba']),
            'test': QtGui.QColor(*style['test']['rgba']),
        }

        # plot colors
        bg_color = style['background']['rgba']
        bg_color_mpl_na = transform_colors_mpl_str(bg_color, no_alpha=True)
        line_color = style['linecolor']['rgba']
        line_color_mpl_na = transform_colors_mpl_str(line_color, no_alpha=True)

        for param in matplotlib.rcParams:
            if 'color' in param and matplotlib.rcParams[param] in ['k', 'black']:
                matplotlib.rcParams[param] = line_color_mpl_na

        matplotlib.rc('axes',
                      edgecolor=line_color_mpl_na,
                      facecolor=bg_color_mpl_na,
                      labelcolor=line_color_mpl_na)
        matplotlib.rc('xtick',
                      color=line_color_mpl_na)
        matplotlib.rc('ytick',
                      color=line_color_mpl_na)
        matplotlib.rc('figure',
                      facecolor=bg_color_mpl_na)

        if self.pg:
            pg.setConfigOption('background', bg_color)
            pg.setConfigOption('foreground', line_color)

        settings = QSettings()
        settings.setValue('style', stylename)
        settings.sync()

    @property
    def metadata(self):
        return self._metadata

    @metadata.setter
    def metadata(self, value: Metadata):
        self._metadata = value

    def updateFilteroptions(self):
        filter_info = readFilterInformation(self._inputs)
        p_filter = filter_info['P']
        s_filter = filter_info['S']
        self.filteroptions = {'P': FilterOptions(p_filter['filtertype'],
                                                 p_filter['freq'],
                                                 p_filter['order']),
                              'S': FilterOptions(s_filter['filtertype'],
                                                 s_filter['freq'],
                                                 s_filter['order'])}

    def updateFileMenu(self):
        settings = QSettings()
        self.fileMenu.clear()
        self.recentProjectsMenu = self.fileMenu.addMenu('Recent Projects')
        for action in self.fileMenuActions[:-1]:
            if action is None:
                self.fileMenu.addSeparator()
            else:
                self.fileMenu.addAction(action)
        try:
            current = self.data.getID()
        except AttributeError:
            current = None
        recentEvents = []
        for eventID in self.recentfiles:
            fname = fnConstructor(eventID)
            if eventID != current and QFile.exists(fname):
                recentEvents.append(eventID)
                recentEvents.reverse()
                self.recentfiles = recentEvents[0:5]
                settings.setValue()
        if recentEvents:
            for i, eventID in enumerate(recentEvents):
                fname = fnConstructor(eventID)
                action = QAction(self.windowIcon(),
                                 "&{0} {1}".format(i + 1,
                                                   QFileInfo(fname).fileName()),
                                 self)
                action.setData(fname)
                self.connect(action, Signal("triggered()"),
                             self.load_data)
                self.fileMenu.addAction(action)
        self.fileMenu.addSeparator()
        self.fileMenu.addAction(self.fileMenuActions[-1])

        # add recent projects
        recentProjects = settings.value('recentProjects', [])
        if not type(recentProjects) == list:
            recentProjects = [recentProjects]
        for project in reversed(recentProjects):
            action = self.createAction(self, project,
                                       lambda fnm=project: self.loadProject(fnm),
                                       None, None)

            self.recentProjectsMenu.addAction(action)

    @property
    def inputs(self):
        return self._inputs

    @staticmethod
    def getRoot():
        settings = QSettings()
        return settings.value("data/dataRoot")

    def load_loc(self, fname=None):
        self.load_data(fname, loc=True)

    def load_pilotevent(self):
        filt = "PILOT location files (*LOC*.mat)"
        caption = "Select PILOT location file"
        fn_loc = QFileDialog().getOpenFileName(self, caption=caption,
                                               filter=filt, dir=self.getRoot())
        fn_loc = fn_loc[0]
        loc_dir = os.path.split(fn_loc)[0]
        filt = "PILOT phases files (*PHASES*.mat)"
        caption = "Select PILOT phases file"
        fn_phases = QFileDialog().getOpenFileName(self, caption=caption,
                                                  filter=filt, dir=loc_dir)
        fn_phases = fn_phases[0]

        fname_dict = dict(phasfn=fn_phases, locfn=fn_loc)
        self.load_data(fname_dict)

    def load_multiple_data(self):
        if not self.okToContinue():
            return
        refresh = False
        events = self.project.eventlist
        sld = SearchFileByExtensionDialog(label='Specify file extension: ', default_text='.xml',
                                          events=events)
        if not sld.exec_():
            return

        filenames = sld.getChecked()
        for event in events:
            for filename in filenames:
                if os.path.isfile(filename) and event.pylot_id in filename:
                    self.load_data(filename, draw=False, event=event, ask_user=True, merge_strategy=sld.merge_strategy)
                    refresh = True
        if not refresh:
            return
        if self.get_current_event().pylot_picks:
            self.refreshEvents()
        self.fill_eventbox()
        self.setDirty(True)

    def load_data(self, fname=None, loc=False, draw=True, event=None, ask_user=False, merge_strategy='Overwrite'):
        if not ask_user:
            if not self.okToContinue():
                return
        if fname is None:
            action = self.sender()
            if isinstance(action, QAction):
                fname = self.filename_from_action(action)
                if not fname:
                    return
        if not event:
            event = self.get_current_event()
        data = Data(self, event)
        try:
            data_new = Data(self, evtdata=str(fname))
            if merge_strategy == 'Overwrite':
                data = data_new
            elif merge_strategy == 'Merge':
                data += data_new
            else:
                raise NotImplementedError(f'Unknown merge strategy: {merge_strategy}')
        except ValueError:
            qmb = QMessageBox(self, icon=QMessageBox.Question,
                              text='Warning: Missmatch in event identifiers {} and {}. Continue?'.format(
                                  data_new.get_evt_data().resource_id,
                                  data.get_evt_data().resource_id),
                              windowTitle='PyLoT - Load data warning')
            qmb.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
            qmb.setDefaultButton(QMessageBox.No)
            ret = qmb.exec_()
            if ret == qmb.Yes:
                data_new.setNew()
                data += data_new
            else:
                return

        self.data = data
        message = 'Loading event info from file {}.'.format(fname)
        print(message)
        self.update_status(message)
        if not loc:
            self.updatePicks(event=event)
        if draw:
            if self.get_current_event().pylot_picks or self.get_current_event().pylot_autopicks:
                self.refreshEvents()
            self.setDirty(True)

    def set_button_border_color(self, button, color=None):
        '''
        Set background color of a button.
        button: type = QtGui.QAbstractButton
        color: type = QtGui.QColor or type = str (RGBA)
        '''
        if type(color) == QtGui.QColor:
            button.setStyleSheet({'QPushButton{background-color:transparent}'})
            palette = button.palette()
            role = button.backgroundRole()
            palette.setColor(role, color)
            button.setPalette(palette)
            button.setAutoFillBackground(True)
        elif type(color) == str:
            button.setStyleSheet('QPushButton{border-color: %s}'
                                 'QPushButton:checked{background-color: rgba%s}' % (color, color))
        elif type(color) == tuple:
            button.setStyleSheet('QPushButton{border-color: rgba%s}'
                                 'QPushButton:checked{background-color: rgba%s}' % (str(color), str(color)))
        elif not color:
            button.setStyleSheet(self.orig_parent._style['stylesheet'])

    def getWFFnames(self):
        try:
            evt = self.get_data().get_evt_data()
            if evt.pylot_picks:
                for pick in evt.pylot_picks:
                    try:
                        if pick.waveform_id is not None:
                            fname = pick.waveform_id.getSEEDstring()
                            if fname not in self.fnames:
                                self.fnames.append(fname)
                    except:
                        continue
            else:
                if self.dataStructure:
                    searchPath = self.dataStructure.expandDataPath()
                    fnames = QFileDialog.getOpenFileNames(self,
                                                          "Select waveform "
                                                          "files:",
                                                          dir=searchPath)
                    self.fnames = fnames[0]

                else:
                    raise DatastructureError('not specified')
            if not self.fnames:
                return None
            return self.fnames
        except DatastructureError as e:
            print(e)
            if not self._props:
                self._props = PropertiesDlg(self, infile=self.infile)

            if self._props.exec_() == QDialog.Accepted:
                return self.getWFFnames()
            else:
                return

    def getWFFnames_from_eventbox(self, eventbox: str = None, subpath: str = None) -> list:
        '''
        Return waveform filenames from event in eventbox.
        '''
        # TODO: add dataStructure class for obspyDMT here, this is just a workaround!
        eventpath = self.get_current_event_path(eventbox)
        if subpath:
            eventpath = os.path.join(eventpath, subpath)
        if not os.path.isdir(eventpath):
            return []
        if self.dataStructure:
            if not eventpath:
                return []
            fnames = [os.path.join(eventpath, f) for f in os.listdir(eventpath)]
        else:
            raise DatastructureError('not specified')
        return fnames

    @staticmethod
    def getPhaseID(phase):
        return identifyPhaseID(phase)

    def get_current_event(self, eventbox=None):
        '''
        Return event (type PyLoT.Event) currently selected in eventbox.
        '''
        if not eventbox:
            eventbox = self.eventBox
        path = eventbox.currentText().split('*')[0]
        return self.project.getEventFromPath(path)

    def get_current_event_path(self, eventbox=None):
        '''
        Return event path of event (type PyLoT.Event) currently selected in eventbox.
        '''
        event = self.get_current_event(eventbox)
        if event:
            return event.path

    def get_current_event_name(self, eventbox=None):
        '''
        Return event path of event (type PyLoT.Event) currently selected in eventbox.
        '''
        path = self.get_current_event_path(eventbox)
        if path:
            return path.split('/')[-1]

    def getLastEvent(self):
        return self.recentfiles[0]

    def add_events(self) -> None:
        '''
        Creates and adds events by user selection of event folders to GUI.
        '''
        if not self.project:
            self.createNewProject()

        eventlist, basepath = self.get_event_directories()
        if not eventlist:
            return

        eventlist = self.set_data_structure(basepath, eventlist)
        if not eventlist:
            self.show_no_events_message()
            return

        dirs = self.initialize_directory_structure(eventlist[0])
        if not dirs:
            return

        self.update_project_settings(dirs)

        self.project.add_eventlist(eventlist)
        self.init_events()
        self.setDirty(True)

    def get_event_directories(self):
        ed = GetExistingDirectories(self, 'Select event directories...')
        if not ed.exec_():
            return None, None

        eventlist = [event for event in ed.selectedFiles() if not event.endswith('EVENTS-INFO')]
        basepath = eventlist[0].split(os.path.basename(eventlist[0]))[0]
        eventlist = self.filter_eventlist_with_file(eventlist, basepath)
        return eventlist, basepath

    def filter_eventlist_with_file(self, eventlist, basepath):
        eventlist_file = os.path.join(basepath, 'eventlist.txt')
        if os.path.isfile(eventlist_file):
            with open(eventlist_file, 'r') as infile:
                eventlist_subset = [os.path.join(basepath, filename.strip()) for filename in infile]
                print(
                    f'Found file "eventlist.txt" in database path. WILL ONLY USE SELECTED EVENTS out of {len(eventlist_subset)} events contained in this subset')
                eventlist = [eventname for eventname in eventlist if eventname in eventlist_subset]
        return eventlist

    def set_data_structure(self, basepath, eventlist):
        if check_obspydmt_structure(basepath):
            print('Recognized obspyDMT structure in selected files. Setting Datastructure to ObspyDMT')
            self.dataStructure = DATASTRUCTURE['obspyDMT']()
            eventlist = check_all_obspy(eventlist)
        else:
            print('Setting Datastructure to PILOT')
            self.dataStructure = DATASTRUCTURE['PILOT']()
            eventlist = check_all_pylot(eventlist)
        return eventlist

    def show_no_events_message(self):
        print('No events found! Expected structure for event folders: [eEVID.DOY.YR],\n'
              ' e.g. eventID=1, doy=2, yr=2016: e0001.002.16 or obspyDMT database')

    def initialize_directory_structure(self, path):
        try:
            system_name = platform.system()
            if system_name in ["Linux", "Darwin"]:
                dirs = {
                    'database': path.split('/')[-2],
                    'datapath': os.path.split(path)[0],
                    'rootpath': '/' + os.path.join(*path.split('/')[:-3])
                }
            elif system_name == "Windows":
                rootpath = path.split('/')[:-3]
                rootpath[0] += '/'
                dirs = {
                    'database': path.split('/')[-2],
                    'datapath': path.split('/')[-3],
                    'rootpath': os.path.join(*rootpath)
                }
        except Exception as e:
            print(f'Warning: Could not automatically init folder structure. ({e})')
            return None

        self.save_directory_settings(dirs['datapath'])
        return dirs

    def save_directory_settings(self, datapath):
        settings = QSettings()
        settings.setValue("data/dataRoot", datapath)
        settings.sync()

    def update_project_settings(self, dirs):
        if not self.project.eventlist:
            self.set_parameter(show=False)
            self.paraBox.hide_parameter()
            self.update_paraBox(dirs)
            if not self.paraBox.get_groupbox_dialog('Directories').exec_():
                return
            self.project.rootpath = dirs['rootpath']
            self.project.datapath = dirs['datapath']
        else:
            if hasattr(self.project, 'datapath') and self.project.datapath != dirs['datapath']:
                QMessageBox.warning(self, "PyLoT Warning", 'Datapath mismatch to current project!')
                return
            self.project.rootpath = dirs['rootpath']
            self.project.datapath = dirs['datapath']

    def update_paraBox(self, dirs):
        for directory, path in dirs.items():
            box = self.paraBox.boxes[directory]
            self.paraBox.setValue(box, path)
            self.paraBox.show_parameter(directory)

    def remove_event(self, event):
        qmb = QMessageBox(self, icon=QMessageBox.Question,
                          text='Are you sure you want to delete event {}?'.format(event.pylot_id))
        qmb.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
        qmb.setDefaultButton(QMessageBox.Yes)
        ret = qmb.exec_()
        if not ret == qmb.Yes:
            return
        self.project.remove_event(event)
        self.init_events(True)

    @staticmethod
    def createEventBox():
        '''
        Eventbox generator.
        '''
        qcb = QComboBox()
        palette = qcb.palette()
        qcb.setPalette(palette)
        return qcb

    def init_events(self, new=False):
        '''
        Initiate GUI widgets in case of changed or newly added events.
        '''
        nitems = self.eventBox.count()
        is_data = self.data_check()
        if len(self.project.eventlist) == 0 or not is_data:
            print('No events to init.')
            self.clearWaveformDataPlot()
            if not is_data:
                new_path, modifypath = self.user_modify_path('Event folders not found! ')
                if modifypath:
                    self.modify_project_path(new_path)
                    self.init_events(True)
                    self.setDirty(True)
            return
        self.eventBox.setEnabled(True)
        self.fill_eventbox()
        if new:
            self.eventBox.setCurrentIndex(0)
        else:
            self.eventBox.setCurrentIndex(nitems)
        self.refreshEvents()
        tabindex = self.tabs.currentIndex()

    def user_modify_path(self, reason=''):
        dialog = QtWidgets.QInputDialog(parent=self)
        new_path, executed = dialog.getText(self, 'Change Project rootpath',
                                            '{}Rename project path {}:'.format(reason, self.project.rootpath))
        return new_path, executed

    def data_check(self):
        paths_exist = [os.path.exists(event.path) for event in self.project.eventlist]
        if all(paths_exist):
            return True
        elif not any(paths_exist):
            return False
        else:
            info_str = ''
            new_eventlist = []
            i = 0
            for event, path_exists in zip(self.project.eventlist, paths_exist):
                if not path_exists:
                    info_str += '\n{} exists: {}'.format(event.path, path_exists)
                else:
                    new_eventlist.append(self.project.eventlist[i])
                i += 1
            if len(new_eventlist) > 0:
                # adopt changes in event listings
                self.project.eventlist = new_eventlist
            print('Unable to find certain event paths:{}'.format(info_str))
            print("Removed these event paths from project eventlist!")
            return True

    def modify_project_path(self, new_rootpath):
        # TODO: change root to datapath
        self.project.rootpath = new_rootpath
        for event in self.project.eventlist:
            event.rootpath = new_rootpath
            event.path = os.path.join(event.rootpath, event.datapath, event.database, event.pylot_id)
            event.path = event.path.replace('\\', '/')
            event.path = event.path.replace('//', '/')

    def fill_eventbox(self, event=None, eventBox=None, select_events='all'):
        '''
        (Re)fill the selected eventBox (type = QtGui.QComboBox).

        :param: select_events, can be 'all', 'ref'
        :type: str
        '''

        self.refresh_tooltips()

        if not eventBox:
            eventBox = self.eventBox
        self.setup_eventBox(eventBox)

        current_event = self.get_current_event()
        eventBox.clear()
        model = eventBox.model()

        plmax = self.get_max_path_length()
        index = eventBox.currentIndex()

        for id, event in enumerate(self.project.eventlist):
            event_data = self.collect_event_data(event, plmax)
            itemlist = self.create_event_items(event_data)
            self.setItemColor(itemlist, id, event, current_event)

            if event.isTestEvent() and select_events == 'ref' or self.isEmpty(event.path):
                self.disable_items(itemlist)

            model.appendRow(itemlist)
            self.validate_event_path(event.path, id)

        eventBox.setCurrentIndex(index)
        self.refreshRefTestButtons()

    def refresh_tooltips(self):
        if self.apw:
            self.apw.refresh_tooltips()
        if hasattr(self, 'cmpw'):
            self.cmpw.refresh_tooltips()

    def setup_eventBox(self, eventBox):
        tv = QtWidgets.QTableView()
        header = tv.horizontalHeader()
        header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
        header.setStretchLastSection(True)
        header.hide()
        tv.verticalHeader().hide()
        tv.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
        eventBox.setView(tv)

    def get_max_path_length(self):
        if self.project.eventlist:
            return max(len(event.path) for event in self.project.eventlist)
        else:
            return None

    def collect_event_data(self, event, plmax):
        data = {
            'path': event.path,
            'picks': self.get_picks(event),
            'is_ref': event.isRefEvent(),
            'is_test': event.isTestEvent(),
            'time': None,
            'lat': None,
            'lon': None,
            'depth': None,
            'localmag': None,
            'momentmag': None,
            'notes': event.notes,
            'dirty': event.dirty,
            'plmax': plmax,
        }

        if event.origins:
            origin = event.origins[0]
            data['time'] = origin.time + 0  # add 0 to avoid exception for time = 0s
            data['lat'] = origin.latitude
            data['lon'] = origin.longitude
            data['depth'] = origin.depth

        if event.magnitudes:
            data['momentmag'] = '{:.1f}'.format(event.magnitudes[0].mag)
            data['localmag'] = '{:.1f}'.format(event.magnitudes[1].mag) if len(event.magnitudes) > 1 else ' '

        return data

    def get_picks(self, event):
        man_au_picks = {'manual': event.pylot_picks, 'auto': event.pylot_autopicks}
        npicks = {'manual': {'P': 0, 'S': 0}, 'auto': {'P': 0, 'S': 0}}
        npicks_total = {'manual': {'P': 0, 'S': 0}, 'auto': {'P': 0, 'S': 0}}

        for ma, picks in man_au_picks.items():
            if picks:
                for pick_values in picks.values():
                    for phasename, pick in pick_values.items():
                        if isinstance(pick, (dict, AttribDict)):
                            phase_ID = identifyPhaseID(phasename)
                            if phase_ID in npicks[ma]:
                                if pick.get('spe'):
                                    npicks[ma][phase_ID] += 1
                                npicks_total[ma][phase_ID] += 1

        return npicks, npicks_total

    def create_event_items(self, data):
        items = [
            self.create_standard_item(data['path'], data['plmax'],
                                      icon=self.style().standardIcon(QStyle.SP_DirOpenIcon)),
            self.create_standard_item(data['time'].strftime("%Y-%m-%d %H:%M:%S") if data['time'] else ''),
            self.create_standard_item(data['lat']),
            self.create_standard_item(data['lon']),
            self.create_standard_item(data['depth']),
            self.create_standard_item(data['localmag']),
            self.create_standard_item(data['momentmag']),
            self.create_pick_item(data['picks'][0]['manual']),
            self.create_pick_item(data['picks'][0]['auto']),
            self.create_ref_test_item(data['is_ref'], 'ref'),
            self.create_ref_test_item(data['is_test'], 'test'),
            self.create_standard_item(data['notes'])
        ]

        for item in items:
            item.setTextAlignment(Qt.AlignCenter)
            if data['dirty'] and item == items[0]:
                item.setText(f"{item.text()}*")

        return items

    def create_standard_item(self, text, max_length=None, icon=None):
        item = QStandardItem(f"{text:{max_length}}" if max_length else str(text))
        if icon:
            item.setIcon(icon)
        return item

    def create_pick_item(self, picks):
        item = QStandardItem()
        item.setText(f"{picks['P']}|{picks['S']}")
        return item

    def create_ref_test_item(self, condition, color_key):
        item = QStandardItem()
        if condition:
            item.setBackground(self._ref_test_colors[color_key])
        return item

    def disable_items(self, items):
        for item in items:
            item.setEnabled(False)

    def validate_event_path(self, path, id):
        if path != self.eventBox.itemText(id).split('*')[0].strip():
            raise ValueError(f"Path mismatch creating eventbox. {path} unequal {self.eventBox.itemText(id)}")

    def isEmpty(self, event_path):
        wf_stat = {True: 'processed',
                   False: 'raw',
                   None: None}

        # self.data.processed is only None for PyLoT datastructure, else True or False
        wf_dir = wf_stat[self.data.processed]
        if wf_dir is not None:
            wf_path = os.path.join(event_path, wf_dir)
            if wf_dir == 'processed' and not os.path.exists(wf_path):
                wf_path = os.path.join(event_path, 'raw')
        else:
            wf_path = event_path
        if not os.path.exists(wf_path):
            return True
        return not bool(os.listdir(wf_path))

    def filename_from_action(self, action):
        if action.data() is None:
            filt = "Supported file formats" \
                   " (*.mat *.qml *.xml *.kor *.evt)"
            caption = "Open an event file"
            fname = QFileDialog().getOpenFileName(self, caption=caption,
                                                  filter=filt,
                                                  dir=self.get_current_event_path())
            fname = fname[0]
        else:
            fname = str(action.data().toString())
        return fname

    def saveData(self, event=None, directory=None, outformats=['.xml', '.cnv', '.obs', '_focmec.in', '.pha']):
        '''
        Save event data to directory with specified output formats.
        :param event: PyLoT Event, if not set current event will be used
        :param directory: output directory, if not set default event path will be used
        :param outformats: str/list of output formats
        :return:
        '''
        if not event:
            event = self.get_current_event()
        if not type(outformats) == list:
            outformats = [outformats]

        def getSavePath(event, directory, outformats):
            if not directory:
                title = 'Save event data as {} to directory ...'.format(outformats)
                directory = QFileDialog.getExistingDirectory(self,
                                                             title,
                                                             event.path)
            else:
                directory = event.path

            filename = 'PyLoT_' + event.pylot_id
            eventfn = os.path.join(directory, filename)

            return eventfn

        fbasename = getSavePath(event, directory, outformats)

        uppererrorP = self._inputs['timeerrorsP']
        uppererrorS = self._inputs['timeerrorsS']
        # Inserted to prevent Bug in Eventlist
        self.get_data().setEvtData(event)
        try:
            self.get_data().applyEVTData(event, typ='event')  # getPicks())
        except OverwriteError:
            self.get_data().resetPicks()
            return self.saveData(event, directory, outformats)
        fcheck = ['auto', 'manual', 'origins', 'magnitude']

        saved_as = str()
        for outformat in outformats:
            try:
                self.get_data().exportEvent(fbasename, outformat, fcheck=fcheck,
                                            upperErrors=[uppererrorP[3], uppererrorS[3]])
                saved_as += str(outformat) + ' '
            except TypeError:
                print('WARNING: Format: {} not yet implemented'.format(outformat))
            self.get_data().setEvtData(event)

        msg = 'Event {} saved as {} in format(s) {}'.format(event.pylot_id, fbasename, saved_as.strip())
        self.update_status(msg)
        print(msg)
        event.dirty = False
        self.fill_eventbox()
        return True

    def exportEvents(self, outformats=['.xml'], events='all'):
        if events == 'all':
            events = self.project.eventlist
        assert type(events) == list, 'Wrong input type: {}'.format(type(events))
        for event in events:
            if not event.dirty:
                continue
            self.get_data().setEvtData(event)
            # save deleted picks to json file
            try:
                self.dump_deleted_picks(event.path)
            except Exception as e:
                print('WARNING! Could not save information on deleted picks {}. Reason: {}'.format(event.path, e))
                print(traceback.format_exc())
            try:
                self.saveData(event, event.path, outformats)
                event.dirty = False
            except Exception as e:
                print('WARNING! Could not save event {}. Reason: {}'.format(event.path, e))
        self.fill_eventbox()

    def enableSaveEventAction(self):
        self.saveEventAction.setEnabled(True)

    def disableSaveEventAction(self):
        self.saveEventAction.setEnabled(False)

    def getinfile(self):
        return self.infile

    def getComponent(self):
        return self.dispComponent

    def setComponent(self, component):
        self.dispComponent = component

    def get_data(self):
        return self.data

    def getPicks(self, type='manual'):
        if type == 'manual':
            return self.get_current_event().getPicks()
        if type == 'auto':
            return self.get_current_event().getAutopicks()

    def getPicksOnStation(self, station, type='manual'):
        try:
            return self.getPicks(type)[station]
        except KeyError:
            return None

    def comparePicks(self):
        comparisons = {}
        eventdict = {}
        for event in self.project.eventlist:
            if not self.comparable[event.pylot_id]:
                continue
            autopicks = excludeQualityClasses(event.getAutopicks(), [4],
                                              self._inputs['timeerrorsP'], self._inputs['timeerrorsS'])
            manupicks = excludeQualityClasses(event.getPicks(), [4],
                                              self._inputs['timeerrorsP'], self._inputs['timeerrorsS'])
            co = Comparison(auto=autopicks, manu=manupicks)
            comparisons[event.pylot_id] = co
            eventdict[event.pylot_id] = event
        if len(eventdict) < 1:
            return

        # init event selection options for autopick
        self.compareoptions = [('tune events', self.get_ref_events, self._style['ref']['rgba']),
                               ('test events', self.get_test_events, self._style['test']['rgba']),
                               ('all (picked) events', self.get_manu_picked_events, None)]

        self.cmpw = CompareEventsWidget(self, self.compareoptions, eventdict, comparisons)
        self.cmpw.start.connect(self.compareMulti)
        self.cmpw.refresh_tooltips()
        self.cmpw.show()

    def pickQualities(self):
        path = self.get_current_event_path()
        (_, plot) = getQualitiesfromxml(path, self._inputs.get('timeerrorsP'), self._inputs.get('timeerrorsS'),plotflag=1)
        self.pickQualitiesWindow = PickQualitiesFromXml(figure=plot, path=self.get_current_event_path(),inputVar=self._inputs)
        self.pickQualitiesWindow.showUI()
        return 

    # WIP JG
    def eventlistXml(self):
        path = self._inputs['rootpath'] + '/' + self._inputs['datapath'] + '/' + self._inputs['database']
        outpath = self.project.location[:self.project.location.rfind('/')]
        geteventlistfromxml(path, outpath)
        return

        # WIP JG
    def spectogramView(self):
        global test
        stations = []
        names = []
        traces = {}
        for tr in self.get_data().wfdata.traces:
            if not tr.stats.station in stations:
                stations.append(tr.stats.station)
                names.append(tr.stats.network + '.' + tr.stats.station)
        for station in stations:
            traces[station] = {}
        for ch in ['Z', 'N', 'E']:
            for tr in self.get_data().wfdata.select(component=ch).traces:
                traces[tr.stats.station][ch] = tr

        names.sort()
        a = self.get_current_event()

        print (self.get_data().wfdata.traces[0])

        test = SpectrogramTab(traces, self.get_data().wfdata)
        height = self.tabs.widget(0).height()
        width = self.tabs.widget(0).width()
        self.tabs.setCurrentIndex(3)
        figCanvas = test.makeSpecFig(direction=self.dispComponent, height = height, width = width, parent = self.tabs.widget)
        return figCanvas
        #self.spectro_layout.addWidget()
        # self.get_data().wfdata.spectrogram()
        # self.tabs.addTab(figCanvas, 'Spectrogram')
        # self.tabs[3] = figCanvas
        # self.refreshTabs()
        # test.show()



    def compareMulti(self):
        if not self.compareoptions:
            return
        for key, func, color in self.compareoptions:
            if self.cmpw.rb_dict[key].isChecked():
                # if radio button is checked break for loop and use func
                break
        eventlist = func()
        # use only events comparable
        eventlist_overlap = [event for event in eventlist if self.comparable[event.pylot_id]]
        compare_widget = self.buildMultiCompareWidget(eventlist_overlap)
        compare_widget.show()

    def buildMultiCompareWidget(self, eventlist):
        global_comparison = Comparison(eventlist=eventlist)
        compare_widget = ComparisonWidget(global_comparison, self)
        for events_name, rb in self.cmpw.rb_dict.items():
            if rb.isChecked():
                break
        compare_widget.setWindowTitle('Histograms for {}'.format(events_name))
        compare_widget.hideToolbar()
        compare_widget.setHistboxChecked(True)
        return compare_widget

    def getPlotWidget(self):
        return self.dataPlot

    @staticmethod
    def getWFID(ycoord):
        try:
            statID = int(round(ycoord))
        except TypeError as e:
            if 'a float is required' in e.message:
                return None
            else:
                raise e

        return statID

    def getStationIDs(self, station):
        wfIDs = []
        for wfID in self.getPlotWidget().getPlotDict().keys():
            actual_station = self.getPlotWidget().getPlotDict()[wfID].split('.')[1]
            if station == actual_station:
                wfIDs.append(wfID)
        return wfIDs

    def getStime(self):
        if self.get_data():
            return full_range(self.get_data().getWFData())[0]

    def addActions(self, target, actions):
        for action in actions:
            if action is None:
                target.addSeparator()
            else:
                target.addAction(action)

    def okToContinue(self):
        if self.dirty:
            qmb = QMessageBox(self, icon=QMessageBox.Question,
                              text='Do you wish to save changes in the current project?')
            qmb.setStandardButtons(QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel)
            qmb.setDefaultButton(QMessageBox.Save)
            ret = qmb.exec_()
            if ret == qmb.Save:
                return self.saveProject()
            elif ret == qmb.Cancel:
                return False
        return True

    def enableRefTestButtons(self, bool):
        '''
        Enable/disable reference and test event selection buttons.
        '''
        self.ref_event_button.setEnabled(bool)
        self.test_event_button.setEnabled(bool)

    def refreshRefTestButtons(self):
        '''
        Refresh state of reference and test event selection buttons depending on current event.
        '''
        event = self.get_current_event()
        if event:
            self.ref_event_button.setChecked(event.isRefEvent())
            self.test_event_button.setChecked(event.isTestEvent())
            self.enableRefTestButtons(bool(self.get_current_event().pylot_picks))
            return
        self.ref_event_button.setChecked(False)
        self.test_event_button.setChecked(False)
        self.enableRefTestButtons(False)

    def toggleRef(self):
        '''
        Toggle ref/test buttons when reference button gets clicked.
        '''
        ref = self.ref_event_button.isChecked()
        self.test_event_button.setChecked(False)
        self.get_current_event().setTestEvent(False)
        self.get_current_event().setRefEvent(ref)
        self.refreshTabs()
        if self.tap:
            self.tap.fill_eventbox()

    def toggleTest(self):
        '''
        Toggle ref/test buttons when test button gets clicked.
        '''
        test = self.test_event_button.isChecked()
        self.ref_event_button.setChecked(False)
        self.get_current_event().setRefEvent(False)
        self.get_current_event().setTestEvent(test)
        self.refreshTabs()
        if self.tap:
            self.tap.fill_eventbox()

    def checkEventButtons(self):
        if self.eventBox.currentIndex() == 0:
            prev_state = False
        else:
            prev_state = True
        if self.eventBox.currentIndex() == len(self.project.eventlist) - 1:
            next_state = False
        else:
            next_state = True
        self.previous_button.setEnabled(prev_state)
        self.next_button.setEnabled(next_state)

    def previous_event(self):
        self.eventBox.setCurrentIndex(self.eventBox.currentIndex() - 1)
        self.eventBox.activated.emit(-1)

    def next_event(self):
        self.eventBox.setCurrentIndex(self.eventBox.currentIndex() + 1)
        self.eventBox.activated.emit(+1)

    def refreshEvents(self):
        '''
        Refresh GUI when events get changed.
        '''
        # Attribute _eventChanged refers to change in first _eventChanged[0]
        # and second _eventChanged[1] main tabs.
        # E.g. plotting is not necessary when changing event in second tab and
        # array_map refresh is not necessary when changing event in waveform plot tab,
        # but gets necessary when switching from one to another after changing an event.
        self._eventChanged = [True, True]
        self.checkEventButtons()
        self.refreshTabs()

    def refreshTabs(self):
        '''
        Refresh main tabs depending on change of the current event in first/second tab.
        '''
        # Logical problem appearing somewhere when calling this function:
        # Loading an existing project from array_tab leads to two calls of newWF
        # which will read in data input twice. Therefore current tab is changed to 0
        # in loadProject before calling this function.
        self.fill_eventbox()
        #print(f'{self.get_current_event()=}')
        plotted = False
        if self.tabs.currentIndex() == 2:
            self.init_event_table()
        self.refreshRefTestButtons()

        # only refresh first/second tab when an event was changed.
        if self._eventChanged[0] or self._eventChanged[1]:
            event = self.get_current_event()
            if not event:
                return
        # if current tab is waveformPlot-tab and the data in this tab was not yet refreshed
        if self.tabs.currentIndex() == 0:
            if self._eventChanged[0]:
                if hasattr(self.project, 'eventlist'):
                    if len(self.project.eventlist) > 0:
                        self.newWF()
                        # keep track whether event was already plotted
                        plotted = True
        # if current tab is array_map-tab and the data in this tab was not yet refreshed
        if self.tabs.currentIndex() == 1:
            if self._eventChanged[1]:
                self.refresh_array_map()
                if not plotted and self._eventChanged[0]:
                    # newWF(plot=False) = load data without plotting
                    self.newWF(plot=False)
                self.update_obspy_dmt()
                self.refresh_array_map()
        if self.tabs.currentIndex() == 3:
            if self.spectroWidget != None:
                self.spectro_layout.removeWidget(self.spectroWidget)
            newSpectroWidget = self.spectogramView()
            self.spectro_layout.addWidget(newSpectroWidget)
            self.spectroWidget = newSpectroWidget

    def newWF(self, event=None, plot=True):
        '''
        Load new data and plot if necessary.
        '''
        self.loadWaveformDataThread(plot=plot)
        if plot:
            self._eventChanged[0] = False

    def loadWaveformDataThread(self, plot=True, load=True):
        '''
        Generates a modal thread to load waveform data and
        call modal plot thread method when finished.
        '''
        if load:
            self.prepareLoadWaveformData()
            self.wfd_thread = Thread(self, self.loadWaveformData,
                                     progressText='Reading data input...',
                                     pb_widget=self.mainProgressBarWidget)
        if load and plot:
            self.wfd_thread.finished.connect(self.plotWaveformDataThread)

        if load:
            self.wfd_thread.start()

        if plot and not load:
            self.plotWaveformDataThread()

    def prepareLoadWaveformData(self):
        self.fnames = self.getWFFnames_from_eventbox()
        self.fnames_comp = []
        fnames_comp = self.getWFFnames_from_eventbox(subpath='compare')
        self.dataPlot.activateCompareOptions(bool(fnames_comp))
        if fnames_comp:
            if self.dataPlot.comp_checkbox.isChecked():
                self.fnames_comp = fnames_comp

        eventpath = self.get_current_event_path()
        basepath = eventpath.split(os.path.basename(eventpath))[0]
        self.obspy_dmt = check_obspydmt_structure(basepath)
        self.dataPlot.activateObspyDMToptions(self.obspy_dmt)
        if self.obspy_dmt:
            self.prepareObspyDMT_data(eventpath)
            self.dataPlot.activateCompareOptions(True)

    def loadWaveformData(self):
        '''
        Load waveform data corresponding to current selected event.
        '''
        settings = QSettings()
        # process application events to wait for event items to appear in event box
        QApplication.processEvents()
        curr_event = self.get_current_event()
        if not curr_event:
            print('Could not find current event. Try reload?')
            return

        if len(curr_event.origins) > 0:
            origin_time = curr_event.origins[0].time
            tstart = settings.value('tstart') if get_none(settings.value('tstart')) else 0
            tstop = settings.value('tstop') if get_none(settings.value('tstop')) else 0
            tstart = origin_time + float(tstart)
            tstop = origin_time + float(tstop)
        else:
            tstart = None
            tstop = None

        self.data.setWFData(self.fnames,
                            self.fnames_comp,
                            checkRotated=True,
                            metadata=self.metadata,
                            tstart=tstart,
                            tstop=tstop, )

    def prepareObspyDMT_data(self, eventpath):
        qcbox_processed = self.dataPlot.qcombo_processed
        qcheckb_syn = self.dataPlot.comp_checkbox
        qcbox_processed.setEnabled(False)
        qcheckb_syn.setEnabled(False)
        for fpath in os.listdir(eventpath):
            fpath = fpath.split('/')[-1]
            if 'syngine' in fpath:
                eventpath_syn = os.path.join(eventpath, fpath)
                qcheckb_syn.setEnabled(True)
                if self.dataPlot.comp_checkbox.isChecked():
                    self.fnames_comp = [os.path.join(eventpath_syn, filename) for filename in os.listdir(eventpath_syn)]
            if 'processed' in fpath:
                qcbox_processed.setEnabled(True)
        if qcbox_processed.isEnabled():
            wftype = qcbox_processed.currentText()
        else:
            wftype = 'raw'
            qcbox_processed.setCurrentIndex(qcbox_processed.findText(wftype))
        eventpath_dmt = os.path.join(eventpath, wftype)
        self.fnames = [os.path.join(eventpath_dmt, filename) for filename in os.listdir(eventpath_dmt)]

    def check_plot_quantity(self):
        """
        Check the amount of samples to be plotted and ask user to reduce the amount if it is too large.
        :rtype: None
        """
        settings = QSettings()
        nth_sample = int(settings.value("nth_sample")) if settings.value("nth_sample") else 1
        npts_max = 1e7
        npts = self.get_npts_to_plot()
        if npts == 0:
            return False
        npts2plot = npts / nth_sample
        if npts2plot < npts_max:
            settings.setValue('large_dataset', False)
        else:
            settings.setValue('large_dataset', True)
            self.update_status('Dataset is very large. Using fast plotting method (MIN/MAX)', 10000)
        settings.sync()
        return True

    def get_npts_to_plot(self):
        if not hasattr(self.data, 'wfdata'):
            return 0
        return sum(trace.stats.npts for trace in self.data.getWFData())

    def connectWFplotEvents(self):
        '''
        Connect signals refering to WF-Dataplot (select station, tutor_user, scrolling)
        '''
        if self.pg:
            self.connect_pg()
        else:
            self.connect_mpl()

    def connect_pg(self):
        self.poS_id = self.dataPlot.plotWidget.scene().sigMouseClicked.connect(self.pickOnStation)

    def connect_mpl(self):
        if not self.poS_id:
            self.poS_id = self.dataPlot.mpl_connect('button_press_event',
                                                    self.pickOnStation)
        if not self.ae_id:
            self.ae_id = self.dataPlot.mpl_connect('axes_enter_event',
                                                   lambda event: self.tutor_user())

        if not self.scroll_id:
            self.scroll_id = self.dataPlot.mpl_connect('scroll_event',
                                                       self.scrollPlot)

    def disconnectWFplotEvents(self):
        '''
        Disconnect all signals refering to WF-Dataplot (select station, tutor_user, scrolling)
        '''
        if self.pg:
            self.disconnect_pg()
        else:
            self.disconnect_mpl()

    def disconnect_pg(self):
        if self.poS_id:
            try:
                self.dataPlot.plotWidget.scene().sigMouseClicked.disconnect()
            except:
                pass

    def disconnect_mpl(self):
        if self.poS_id:
            self.dataPlot.mpl_disconnect(self.poS_id)
        if self.ae_id:
            self.dataPlot.mpl_disconnect(self.ae_id)
        if self.scroll_id:
            self.dataPlot.mpl_disconnect(self.scroll_id)
        self.poS_id = None
        self.ae_id = None
        self.scroll_id = None

    def finish_pg_plot(self):
        self.getPlotWidget().updateWidget()
        plots = self.wfp_thread.data
        # do not show plot if no data are given
        self.wf_scroll_area.setVisible(len(plots) > 0)
        self.no_data_label.setVisible(not len(plots) > 0)
        for times, data, times_syn, data_syn in plots:
            self.dataPlot.plotWidget.getPlotItem().plot(times, data,
                                                        pen=self.dataPlot.pen_linecolor)
            if len(data_syn) > 0:
                self.dataPlot.plotWidget.getPlotItem().plot(times_syn, data_syn,
                                                            pen=self.dataPlot.pen_linecolor_syn)
        self.dataPlot.reinitMoveProxy()
        self.highlight_stations()

    def finishWaveformDataPlot(self):
        self.comparable = self.checkEvents4comparison()
        if self.pg:
            self.finish_pg_plot()
        else:
            self._max_xlims = self.dataPlot.getXLims(self.dataPlot.axes[0])
        plotWidget = self.getPlotWidget()
        plotDict = plotWidget.getPlotDict()
        pos = plotDict.keys()
        labels = ['{}.{}'.format(*plotDict[n].split('.')[:-2]) for n in pos]
        plotWidget.setYTickLabels(pos, labels)
        try:
            plotWidget.figure.tight_layout()
        except:
            pass
        self.connectWFplotEvents()
        self.loadlocationaction.setEnabled(True)
        self.auto_tune.setEnabled(True)
        self.auto_pick.setEnabled(True)
        self.auto_pick_local.setEnabled(True)
        self.auto_pick_sge.setEnabled(True)
        self.autoPickMenu.setEnabled(True)
        self.z_action.setEnabled(True)
        self.e_action.setEnabled(True)
        self.n_action.setEnabled(True)
        self.openEventAction.setEnabled(True)
        self.openEventsAutoAction.setEnabled(True)
        self.loadpilotevent.setEnabled(True)
        self.enableSaveEventAction()
        event = self.get_current_event()
        if event.pylot_picks:
            self.drawPicks(picktype='manual')
        if event.pylot_autopicks:
            self.drawPicks(picktype='auto')
        if event.pylot_picks or event.pylot_autopicks:
            self.locateEventAction.setEnabled(True)
            self.qualities_action.setEnabled(True)
            self.eventlist_xml_action.setEnabled(True)

        if True in self.comparable.values():
            self.compare_action.setEnabled(True)
        self.draw()
        # TODO: Quick and dirty, improve this later
        self.update_obspy_dmt()

    def update_obspy_dmt(self):
        self.metadata.clear_inventory()
        if self.obspy_dmt:
            invpath = os.path.join(self.get_current_event_path(), 'resp')
            if not invpath in self.metadata.inventories:
                self.metadata.add_inventory(invpath, obspy_dmt_inv=True)
            # check if directory is empty
            if os.listdir(invpath):
                self.init_map_button.setEnabled(True)
                self.initMapAction.setEnabled(True)

    @staticmethod
    def checkEvent4comparison(event):
        if event.pylot_picks and event.pylot_autopicks:
            for station in event.pylot_picks:
                if station in event.pylot_autopicks:
                    try:
                        autopick_p = event.pylot_autopicks[station]['P']['spe']
                    except KeyError:
                        autopick_p = None
                    try:
                        manupick_p = event.pylot_picks[station]['P']['spe']
                    except KeyError:
                        manupick_p = None
                    try:
                        autopick_s = event.pylot_autopicks[station]['S']['spe']
                    except KeyError:
                        autopick_s = None
                    try:
                        manupick_s = event.pylot_picks[station]['S']['spe']
                    except KeyError:
                        manupick_s = None
                    if autopick_p and manupick_p:
                        return True
                    elif autopick_s and manupick_s:
                        return True
        return False

    def checkEvents4comparison(self):
        # init dict to keep track whether event can be compared
        comparable = {}
        for event in self.project.eventlist:
            comparable[event.pylot_id] = self.checkEvent4comparison(event)
        return comparable

    def clearWaveformDataPlot(self, refresh_plot=False):
        self.disconnectWFplotEvents()
        if self.pg:
            self.dataPlot.plotWidget.getPlotItem().clear()
        else:
            for ax in self.dataPlot.axes:
                ax.cla()
        self.loadlocationaction.setEnabled(False)
        self.auto_tune.setEnabled(False)
        self.auto_pick.setEnabled(False)
        self.auto_pick_local.setEnabled(False)
        self.auto_pick_sge.setEnabled(False)
        self.autoPickMenu.setEnabled(False)
        self.z_action.setEnabled(False)
        self.e_action.setEnabled(False)
        self.n_action.setEnabled(False)
        self.openEventAction.setEnabled(False)
        self.openEventsAutoAction.setEnabled(False)
        self.loadpilotevent.setEnabled(False)
        self.compare_action.setEnabled(False)
        self.qualities_action.setEnabled(False)
        self.eventlist_xml_action.setEnabled(False)
        self.locateEventAction.setEnabled(False)
        if not refresh_plot:
            self.wf_scroll_area.setVisible(False)
            self.no_data_label.setVisible(True)
        self.disableSaveEventAction()
        self.draw()

    def plotWaveformDataThread(self, filter=True):
        '''
        Open a modal thread to plot current waveform data.
        '''
        if self.check_plot_quantity() == False:
            print('Nothing to plot.')
            return
        self.clearWaveformDataPlot(refresh_plot=True)
        self.wfp_thread = Thread(self, self.plotWaveformData,
                                 arg=filter,
                                 progressText='Plotting waveform data...',
                                 pb_widget=self.mainProgressBarWidget)
        self.wfp_thread.finished.connect(self.finishWaveformDataPlot)
        self.wfp_thread.start()

    def plotWaveformData(self, filter=True):
        '''
        Plot waveform data to current plotWidget.
        '''
        settings = QSettings()
        nth_sample = settings.value('nth_sample')
        if not nth_sample:
            nth_sample = 1
        zne_text = {'Z': 'vertical', 'N': 'north-south', 'E': 'east-west'}
        comp = self.getComponent()
        title = 'section: {0} components'.format(zne_text[comp])
        wfst = self.get_data().getWFData()
        wfsyn = self.get_data().getAltWFdata()
        if self.filterActionP.isChecked() and filter:
            self.filterWaveformData(plot=False, phase='P')
        elif self.filterActionS.isChecked() and filter:
            self.filterWaveformData(plot=False, phase='S')
        # wfst = self.get_data().getWFData().select(component=comp)
        # wfst += self.get_data().getWFData().select(component=alter_comp)
        plotWidget = self.getPlotWidget()
        self.adjustPlotHeight()
        if get_bool(settings.value('large_dataset')) == True:
            self.plot_method = 'fast'
        else:
            self.plot_method = 'normal'
        rval = plotWidget.plotWFData(wfdata=wfst, wfsyn=wfsyn, title=title, mapping=False, component=comp,
                                     nth_sample=int(nth_sample), method=self.plot_method, gain=self.gain)
        plots = rval if rval else []
        return plots

    def adjustPlotHeight(self):
        if self.pg:
            return
        height_need = len(self.data.getWFData()) * self.height_factor
        plotWidget = self.getPlotWidget()
        if self.tabs.widget(0).frameSize().height() < height_need:
            plotWidget.setMinimumHeight(height_need)
        else:
            plotWidget.setMinimumHeight(0)

    def plotZ(self):
        self.setComponent('Z')
        self.plotWaveformDataThread()

    def plotN(self):
        self.setComponent('N')
        self.plotWaveformDataThread()

    def plotE(self):
        self.setComponent('E')
        self.plotWaveformDataThread()

    def pushFilterWF(self, param_args):
        self.get_data().filterWFData(param_args)

    def filterP(self):
        self.filterActionS.setChecked(False)
        if self.filterActionP.isChecked():
            self.filterWaveformData(phase='P')
        else:
            self.resetWFData()

    def filterS(self):
        self.filterActionP.setChecked(False)
        if self.filterActionS.isChecked():
            self.filterWaveformData(phase='S')
        else:
            self.resetWFData()

    def resetWFData(self):
        self.get_data().resetWFData()
        self.plotWaveformDataThread()

    def filterWaveformData(self, plot=True, phase=None):
        if not self.get_current_event():
            return

        if self.get_data():
            if not phase:
                if self.filterActionP.isChecked():
                    phase = 'P'
                elif self.filterActionS.isChecked():
                    phase = 'S'
            if self.getFilterOptions():
                if (phase == 'P' and self.filterActionP.isChecked()) or (
                        phase == 'S' and self.filterActionS.isChecked()):
                    kwargs = self.getFilterOptions()[phase].parseFilterOptions()
                    self.pushFilterWF(kwargs)
                else:
                    self.get_data().resetWFData()
            elif self.filterActionP.isChecked() or self.filterActionS.isChecked():
                self.adjustFilterOptions()
            else:
                self.get_data().resetWFData()
            if plot:
                self.plotWaveformDataThread(filter=False)

    def getAutoFilteroptions(self, phase):
        return getAutoFilteroptions(phase, self._inputs)

    def adjustFilterOptions(self):
        fstring = "Filter Options"
        self.filterDlg = FilterOptionsDialog(titleString=fstring,
                                             parent=self)
        if self.filterDlg.exec_():
            filteroptions = self.filterDlg.getFilterOptions()
            self.setFilterOptions(filteroptions)
            if self.filterActionP.isChecked() or self.filterActionS.isChecked():
                kwargs = self.getFilterOptions()[self.getSeismicPhase()].parseFilterOptions()
                self.pushFilterWF(kwargs)
                self.plotWaveformDataThread()
            return True

    def checkFilterOptions(self):
        fstring = "Filter Options"
        self.filterDlg = FilterOptionsDialog(titleString=fstring,
                                             parent=self)
        filteroptions = self.filterDlg.getFilterOptions()
        self.setFilterOptions(filteroptions)
        filterP = filteroptions['P']
        filterS = filteroptions['S']
        minP, maxP = filterP.getFreq()
        minS, maxS = filterS.getFreq()
        self.paraBox.params_to_gui()

    def getFilterOptions(self):
        return self.filteroptions

    def setFilterOptions(self, filterOptions):
        self.filterOptions = filterOptions
        filterP = filterOptions['P']
        filterS = filterOptions['S']
        minP, maxP = filterP.getFreq()
        minS, maxS = filterS.getFreq()
        self._inputs.setParamKV('minfreq', (minP, minS))
        self._inputs.setParamKV('maxfreq', (maxP, maxS))
        self._inputs.setParamKV('filter_order', (filterP.getOrder(), filterS.getOrder()))
        self._inputs.setParamKV('filter_type', (filterP.getFilterType(), filterS.getFilterType()))

    def filterOptionsFromParameter(self):
        minP, minS = self._inputs['minfreq']
        maxP, maxS = self._inputs['maxfreq']
        orderP, orderS = self._inputs['filter_order']
        typeP, typeS = self._inputs['filter_type']

        filterP = self.getFilterOptions()['P']
        filterP.setFreq([minP, maxP])
        filterP.setOrder(orderP)
        filterP.setFilterType(typeP)

        filterS = self.getFilterOptions()['S']
        filterS.setFreq([minS, maxS])
        filterS.setOrder(orderS)
        filterS.setFilterType(typeS)

        self.checkFilterOptions()

    def getSeismicPhase(self):
        return self.seismicPhase

    def getTraceID(self, wfID):
        plot_dict = self.getPlotWidget().getPlotDict()
        return plot_dict.get(wfID)

    def getNetworkName(self, wfID):
        plot_dict = self.getPlotWidget().getPlotDict()
        if wfID in plot_dict.keys():
            return plot_dict[wfID].split('.')[0]

    def getStationName(self, wfID):
        plot_dict = self.getPlotWidget().getPlotDict()
        if wfID in plot_dict.keys():
            return plot_dict[wfID].split('.')[1]

    def getLocationName(self, wfID):
        plot_dict = self.getPlotWidget().getPlotDict()
        if wfID in plot_dict.keys():
            return plot_dict[wfID].split('.')[2]

    def alterPhase(self):
        pass

    def scrollPlot(self, gui_event):
        '''
        Function connected to mouse wheel scrolling inside WFdataPlot.
        Only used if amount of traces exceeds a certain limit creating
        a scroll area.
        '''
        button = gui_event.button
        x, y = gui_event.xdata, gui_event.ydata
        if not button == 'up' and not button == 'down':
            return
        if not self._ctrl and not self._shift:
            vbar = self.wf_scroll_area.verticalScrollBar()
            up_down = {'up': -60,
                       'down': 60}
            if vbar.maximum():
                vbar.setValue(vbar.value() + up_down[button])
        if self._ctrl:
            factor = {'up': 5. / 4.,
                      'down': 4. / 5.}
            self.height_factor *= factor[button]
            self.adjustPlotHeight()
        if self._shift:
            factor = {'up': 1. / 2.,
                      'down': 2.}
            xlims = self.dataPlot.getXLims(self.dataPlot.axes[0])
            xdiff = xlims[1] - xlims[0]
            xdiff *= factor[button]
            xl = x - 0.5 * xdiff
            xr = x + 0.5 * xdiff
            if xl < self._max_xlims[0]:
                xl = self._max_xlims[0]
            if xr > self._max_xlims[1]:
                xr = self._max_xlims[1]
            self.dataPlot.setXLims(self.dataPlot.axes[0], (xl, xr))
            self.dataPlot.draw()

    def pickOnStation(self, gui_event):
        if self.pg:
            button = gui_event.button()
            if not button in [1, 4]:
                return
        else:
            button = gui_event.button
            if not button == 1:
                return

        if self.pg:
            ycoord = self.dataPlot.plotWidget.getPlotItem().vb.mapSceneToView(gui_event.scenePos()).y()
        else:
            ycoord = gui_event.ydata

        wfID = self.getWFID(ycoord)

        if wfID is None: return

        network = self.getNetworkName(wfID)
        station = self.getStationName(wfID)
        location = self.getLocationName(wfID)
        seed_id = self.getTraceID(wfID)
        if button == 1:
            self.pickDialog(wfID, seed_id)
        elif button == 4:
            self.toggle_station_color(wfID, network, station, location)

    def toggle_station_color(self, wfID, network, station, location):
        black_pen = pg.mkPen((0, 0, 0))
        red_pen = pg.mkPen((200, 50, 50))
        line_item = self.dataPlot.plotWidget.getPlotItem().listDataItems()[wfID - 1]
        current_pen = line_item.opts['pen']
        nsl = '{}.{}.{}'.format(network, station, location)
        if current_pen == black_pen:
            line_item.setPen(red_pen)
            if not nsl in self.stations_highlighted:
                self.stations_highlighted.append(nsl)
        else:
            line_item.setPen(black_pen)
            if nsl in self.stations_highlighted:
                self.stations_highlighted.pop(self.stations_highlighted.index(nsl))

    def highlight_stations(self):
        for wfID, value in self.getPlotWidget().getPlotDict().items():
            station, channel, location, network = value.split('.')
            nsl = '{}.{}.{}'.format(network, station, location)
            if nsl in self.stations_highlighted:
                self.toggle_station_color(wfID, network, station, location)

    def pickDialog(self, wfID, seed_id=None):
        if not seed_id:
            seed_id = self.getTraceID(wfID)
        try:
            network, station, location = seed_id.split('.')[:3]
        except:
            print("Warning! No network, station, and location info available!")
            return
        self.update_status('picking on station {0}'.format(station))
        wfdata = self.get_data().getOriginalWFData().copy()
        wfdata_comp = self.get_data().getAltWFdata().copy()
        event = self.get_current_event()
        wftype = self.dataPlot.qcombo_processed.currentText() if self.obspy_dmt else None
        pickDlg = PickDlg(self, parameter=self._inputs,
                          data=wfdata.select(station=station),
                          data_compare=wfdata_comp.select(station=station),
                          station=station, network=network,
                          location=location,
                          picks=self.getPicksOnStation(station, 'manual'),
                          autopicks=self.getPicksOnStation(station, 'auto'),
                          metadata=self.metadata, event=event,
                          model=self.inputs.get('taup_model'),
                          filteroptions=self.filteroptions, wftype=wftype,
                          show_comp_data=self.dataPlot.comp_checkbox.isChecked())
        if self.filterActionP.isChecked():
            pickDlg.currentPhase = "P"
            pickDlg.filterWFData()
        elif self.filterActionS.isChecked():
            pickDlg.currentPhase = "S"
            pickDlg.filterWFData()
        pickDlg.nextStation.setChecked(self.nextStation)
        pickDlg.nextStation.stateChanged.connect(self.toggle_next_station)
        if pickDlg.exec_():
            if pickDlg._dirty:
                self.setDirty(True)
                self.update_status('picks accepted ({0})'.format(station))
                self.addPicks(station, pickDlg.getPicks(picktype='manual'), type='manual')
                self.addPicks(station, pickDlg.getPicks(picktype='auto'), type='auto')
                if pickDlg.removed_picks:
                    self.log_deleted_picks(pickDlg.removed_picks)
                self.enableSaveEventAction()
                self.comparable = self.checkEvents4comparison()
                if True in self.comparable.values():
                    self.compare_action.setEnabled(True)
                self.drawPicks(station)
                self.draw()
            if self.nextStation:
                if not self.get_loc_flag() and self.check4Loc():
                    self.locateEventAction.setEnabled(True)
                    self.set_loc_flag(True)
                elif self.get_loc_flag() and not self.check4Loc():
                    self.set_loc_flag(False)
                self.pickDialog(wfID - 1)

        else:
            self.update_status('picks discarded ({0})'.format(station))
        if not self.get_loc_flag() and self.check4Loc():
            self.locateEventAction.setEnabled(True)
            self.set_loc_flag(True)
        elif self.get_loc_flag() and not self.check4Loc():
            self.set_loc_flag(False)

    def toggle_next_station(self, signal):
        self.nextStation = bool(signal)

    def addListItem(self, text):
        textlist = text.split('\n')
        for text in textlist:
            self.listWidget.addItem(text)
        self.listWidget.scrollToBottom()

    def init_fig_dict(self):
        self.fig_dict = {}
        self.fig_keys = [
            'mainFig',
            'aicFig',
            'slength',
            'checkZ4s',
            'refPpick',
            'el_Ppick',
            'fm_picker',
            'el_S1pick',
            'el_S2pick',
            'refSpick',
            'aicARHfig',
            'plot_style'
        ]
        for key in self.fig_keys:
            if key == 'plot_style':
                fig = self._style
            else:
                fig = Figure()
            self.fig_dict[key] = fig

    def init_canvas_dict(self):
        self.canvas_dict = {}
        for key in self.fig_keys:
            if not key == 'plot_style':
                self.canvas_dict[key] = PylotCanvas(self.fig_dict[key], parent=self)

    def init_fig_dict_wadatijack(self, eventIDs):
        self.fig_dict_wadatijack = {}
        self.fig_keys_wadatijack = [
            'jackknife',
            'wadati',
            'plot_style'
        ]
        for eventID in eventIDs:
            self.fig_dict_wadatijack[eventID] = {}
            for key in self.fig_keys_wadatijack:
                if key == 'plot_style':
                    fig = self._style
                else:
                    fig = Figure()
                self.fig_dict_wadatijack[eventID][key] = fig

    def init_canvas_dict_wadatijack(self):
        self.canvas_dict_wadatijack = {}
        for eventID in self.fig_dict_wadatijack.keys():
            self.canvas_dict_wadatijack[eventID] = {}
            for key in self.fig_keys_wadatijack:
                if not key == 'plot_style':
                    self.canvas_dict_wadatijack[eventID][key] = PylotCanvas(self.fig_dict_wadatijack[eventID][key],
                                                                            parent=self)

    def tune_autopicker(self):
        '''
        Initiates TuneAutopicker widget used to interactively
        tune parameters for autopicking algorithm.
        '''
        # figures and canvas have to be iniated from the main GUI
        # thread to prevent handling of QPixmap objects outside of
        # the main thread
        self.init_fig_dict()
        # if not self.tap:
        # init TuneAutopicker object
        self.tap = TuneAutopicker(self, self.obspy_dmt)
        # first call of update to init tabs with empty canvas
        self.update_autopicker()
        # connect update signal of TuneAutopicker with update function
        # creating and filling figure canvas
        self.tap.update.connect(self.update_autopicker)
        self.tap.figure_tabs.setCurrentIndex(0)
        # else:
        #    self.update_autopicker()
        #    self.tap.fill_eventbox()
        self.tap.showMaximized()

    def update_autopicker(self):
        '''
        Create and fill TuneAutopicker tabs with figure canvas.
        '''
        self.init_canvas_dict()
        self.tap.fill_tabs(picked=True)
        for canvas in self.canvas_dict.values():
            canvas.setZoomBorders2content()
        if self.tap.pylot_picks:
            station = self.tap.get_current_station()
            p_pick = self.tap.pylot_picks[station].get('P')
            if p_pick:
                self.tap.pickDlg.autopicks['P_tuning'] = p_pick
                self.tap.pickDlg.drawPicks(phase='P_tuning', picktype='auto', picks=p_pick)
            s_pick = self.tap.pylot_picks[station].get('S')
            if s_pick:
                self.tap.pickDlg.autopicks['S_tuning'] = s_pick
                self.tap.pickDlg.drawPicks(phase='S_tuning', picktype='auto', picks=s_pick)

    def autoPick(self):
        autosave = self.get_current_event_path()
        if not os.path.exists(autosave):
            QMessageBox.warning(self, "PyLoT Warning",
                                "No autoPyLoT output declared!")
            return

        if not self.apw:
            # init event selection options for autopick
            self.pickoptions = [('current event', self.get_current_event, None),
                                ('tune events', self.get_ref_events, self._style['ref']['rgba']),
                                ('test events', self.get_test_events, self._style['test']['rgba']), ]
            # ('all (picked) events', self.get_manu_picked_events, None),
            # ('all events', self.get_all_events, None)]

            self.listWidget = QListWidget()
            self.setDirty(True)
            self.apw = AutoPickWidget(self, self.pickoptions)
            self.apw.insert_log_widget(self.listWidget)
            self.apw.refresh_tooltips()

            self.apw.start.connect(self.start_autopick)
        self.apw.show()

    def start_autopick(self):
        if not self.pickoptions:
            return
        for key, func, _ in self.pickoptions:
            if self.apw.rb_dict[key].isChecked():
                # if radio button is checked break for loop and use func
                break

        events = func()
        if not type(events) == list:
            events = [events]
        eventPaths = self.get_event_paths(events)
        eventIDs = self.get_event_ids(events)

        self.init_fig_dict_wadatijack(eventIDs)

        if not eventPaths:
            self.addListItem("No events found for '{}'".format(key))
            return
        else:
            self.addListItem("Picking the following events ({}):".format(key))
            for eventID in eventPaths:
                self.addListItem(str(eventID))

        self.apw.enable(False)

        # export current picks etc.
        self.exportEvents(['.xml'], events=events)

        wfpath = self.dataPlot.qcombo_processed.currentText() if self.obspy_dmt else ''
        # define arguments for picker
        args = {'parameter': self._inputs,
                'station': 'all',
                'fnames': 'None',
                'eventid': eventPaths,
                'iplot': 0,
                'fig_dict': None,
                'fig_dict_wadatijack': self.fig_dict_wadatijack,
                'savexml': False,
                'obspyDMT_wfpath': wfpath}
        # init pick thread
        self.mp_thread = QtCore.QThreadPool()
        self.mp_worker = Worker(autoPyLoT, args, redirect_stdout=True)

        self.addListItem(str(self._inputs))

        self.mp_worker.signals.message.connect(self.addListItem)
        self.mp_worker.signals.result.connect(self.finalizeAutoPick)
        self.mp_worker.signals.finished.connect(self.enableAutoPickWidget)

        self.mp_thread.start(self.mp_worker)

    def autoPickProject(self):
        if not self.apd_local:
            self.apd_local = AutoPickDlg(self, sge=False)
        self.apd_local.show()

    def autoPickProjectSGE(self):
        if not self.apd_sge:
            self.apd_sge = AutoPickDlg(self, sge=True)
        self.apd_sge.show()

    def finalizeAutoPick(self, result):
        if result:
            self.init_canvas_dict_wadatijack()
            for eventID in result.keys():
                event = self.get_event_from_id(eventID)
                if not event:
                    continue
                event.addAutopicks(result[eventID])
                jkw = CanvasWidget(self, self.canvas_dict_wadatijack[eventID]['jackknife'])
                wdw = CanvasWidget(self, self.canvas_dict_wadatijack[eventID]['wadati'])
                self.apw.add_plot_widget(jkw, 'Jackknife', eventID)
                self.apw.add_plot_widget(wdw, 'Wadati', eventID)
            self.apw.update_plots()
            self.drawPicks(picktype='auto')
            self.draw()

    def enableAutoPickWidget(self, event=None):
        self.apw.enable(True)

    def get_event_from_id(self, eventID):
        for event in self.project.eventlist:
            if event.pylot_id == eventID:
                return event

    @staticmethod
    def get_event_paths(eventlist):
        eventPaths = []
        for event in eventlist:
            eventPaths.append(event.path)
        return eventPaths

    @staticmethod
    def get_event_ids(eventlist):
        eventIDs = []
        for event in eventlist:
            eventIDs.append(event.pylot_id)
        return eventIDs

    def get_all_events(self):
        return self.project.eventlist

    def get_ref_events(self):
        events = []
        for event in self.project.eventlist:
            if event.isRefEvent():
                events.append(event)
        return events

    def get_test_events(self):
        events = []
        for event in self.project.eventlist:
            if event.isTestEvent():
                events.append(event)
        return events

    def get_manu_picked_events(self):
        events = []
        for event in self.project.eventlist:
            if len(event.pylot_picks) > 0:
                events.append(event)
        return events

    def addPicks(self, station, picks, type='manual'):
        stat_picks = self.getPicksOnStation(station, type)
        if not stat_picks:
            rval = False
        else:
            rval = True
        event = self.get_current_event()
        # create dictionary switch
        automanu = {'manual': event.setPick,
                    'auto': event.setAutopick}
        # dictionary consisting of set station only
        automanu[type](station=station, pick=picks)
        return rval

    def log_deleted_picks(self, deleted_picks, event_path=None):
        '''
        Log deleted picks to list self.deleted_picks
        '''
        if not event_path:
            event_path = self.get_current_event_path()
        for deleted_pick in deleted_picks:
            deleted_pick = dict(deleted_pick)
            if not event_path in self.deleted_picks.keys():
                self.deleted_picks[event_path] = []
            for key, value in deleted_pick.items():
                if type(value) == UTCDateTime:
                    deleted_pick[key] = str(value)
            deleted_pick['deletion_logtime'] = str(datetime.now())
            self.deleted_picks[event_path].append(deleted_pick)

    def dump_deleted_picks(self, event_path):
        '''
        Save deleted picks to json file for event in event_path. Load old file before and merge
        '''
        try:
            deleted_picks_from_file = self.load_deleted_picks(event_path)
        except Exception as e:
            msg = 'WARNING! Deleted picks file could not be loaded: {}' \
                  ' Check deleted picks file in path {} before continuing'
            print(msg.format(e, event_path))
            self.safetyCopy(event_path)
        deleted_picks_event = self.deleted_picks.get(event_path)
        if deleted_picks_event is None:
            return

        for pick in deleted_picks_from_file:
            if not pick in deleted_picks_event:
                deleted_picks_event.append(pick)

        # if there are no picks to save
        if not deleted_picks_event:
            return

        fpath = self.get_deleted_picks_fpath(event_path)
        with open(fpath, 'w') as outfile:
            json.dump(deleted_picks_event, outfile)

        # clear entry for this event
        self.deleted_picks[event_path] = []

    def safetyCopy(self, event_path):
        fpath = self.get_deleted_picks_fpath(event_path)
        fpath_new = fpath.split('.json')[0] + '_copy_{}.json'.format(datetime.now()).replace(' ', '_')
        shutil.move(fpath, fpath_new)

    def load_deleted_picks(self, event_path):
        ''' Load a list with deleted picks for event in event_path from file '''
        fpath = self.get_deleted_picks_fpath(event_path)
        if not os.path.isfile(fpath):
            return []
        with open(fpath, 'r') as infile:
            deleted_picks = json.load(infile)
        return deleted_picks

    def get_deleted_picks_fpath(self, event_path):
        return os.path.join(event_path, 'deleted_picks.json')

    def updatePicks(self, event=None):
        if not event:
            event = self.get_current_event()
        event.pylot_picks = {}
        event.pylot_autopicks = {}
        picksdict = picksdict_from_picks(evt=self.get_data().get_evt_data(), parameter=self.getParameter())
        event.addPicks(picksdict['manual'])
        event.addAutopicks(picksdict['auto'])

    def getParameter(self):
        if hasattr(self.project, 'parameter') and isinstance(self.project.parameter, PylotParameter):
            return self.project.parameter
        else:
            return self._inputs

    def drawPicks(self, station=None, picktype=None, stime=None):
        # if picktype not specified, draw both
        if not stime:
            stime = self.getStime()

        if not picktype:
            self.drawPicks(station, 'manual', stime)
            self.drawPicks(station, 'auto', stime)
            return

        if not picktype in ['manual', 'auto']:
            raise TypeError('Unknown picktype {0}'.format(picktype))

        # if picks to draw not specified, draw all picks available
        if not station:
            for station in self.getPicks(type=picktype):
                self.drawPicks(station, picktype=picktype, stime=stime)
            return

        if self.pg:
            pw = self.getPlotWidget().plotWidget
        else:
            ax = self.getPlotWidget().axes[0]

        if station in self.drawnPicks[picktype].keys():
            for item in self.drawnPicks[picktype][station]:
                pw.removeItem(item)
        self.drawnPicks[picktype][station] = []

        # check for station key in dictionary, else return
        if not station in self.getPicks(type=picktype):
            return

        # plotting picks
        plotIDs = self.getStationIDs(station)
        if not plotIDs:
            return

        for plotID in plotIDs:
            ylims = np.array([-.5, +.5]) + plotID

            stat_picks = self.getPicks(type=picktype)[station]
            settings = QSettings()

            for phase in stat_picks:
                if phase == 'SPt': continue  # wadati SP time
                picks = stat_picks[phase]
                if type(stat_picks[phase]) is not dict and type(stat_picks[phase]) is not AttribDict:
                    return

                phaseID = self.getPhaseID(phase)
                # get quality classes
                if phaseID == 'P':
                    quality = getQualityFromUncertainty(picks['spe'], self._inputs['timeerrorsP'])
                elif phaseID == 'S':
                    quality = getQualityFromUncertainty(picks['spe'], self._inputs['timeerrorsS'])

                if not picks['mpp']: continue

                mpp = picks['mpp'] - stime
                if picks['epp'] and picks['lpp']:
                    epp = picks['epp'] - stime
                    lpp = picks['lpp'] - stime
                else:
                    epp = None
                    lpp = None
                spe = picks['spe']

                if self.pg:
                    if spe:
                        if picks['epp'] and picks['lpp']:
                            pen = make_pen(picktype, phaseID, 'epp', quality)
                            self.drawnPicks[picktype][station].append(pw.plot([epp, epp], ylims,
                                                                              alpha=.25, pen=pen, name='EPP'))
                            pen = make_pen(picktype, phaseID, 'lpp', quality)
                            self.drawnPicks[picktype][station].append(pw.plot([lpp, lpp], ylims,
                                                                              alpha=.25, pen=pen, name='LPP'))
                            pen = make_pen(picktype, phaseID, 'spe', quality)
                            spe_l = pg.PlotDataItem([mpp - spe, mpp - spe], ylims, pen=pen,
                                                    name='{}-SPE'.format(phase))
                            spe_r = pg.PlotDataItem([mpp + spe, mpp + spe], ylims, pen=pen)
                            # Ugly fix: spe_l/r.curve.path is initialized to None and can't be accessed, causing the
                            # Warning: drawPicks: Could not create fill for symmetric pick error
                            # this generates them
                            spe_l.curve.shape()
                            spe_r.curve.shape()
                            try:
                                color = pen.color()
                                color.setAlpha(100.)
                                brush = pen.brush()
                                brush.setColor(color)
                                fill = pg.FillBetweenItem(spe_l, spe_r, brush=brush)
                                fb = pw.addItem(fill)
                                self.drawnPicks[picktype][station].append(fill)
                            except:
                                print('Warning: drawPicks: Could not create fill for symmetric pick error.')
                        pen = make_pen(picktype, phaseID, 'mpp', quality)
                        self.drawnPicks[picktype][station].append(
                            pw.plot([mpp, mpp], ylims, pen=pen, name='{}-Pick'.format(phase)))
                else:
                    if picktype == 'manual':
                        linestyle_mpp, width_mpp = pick_linestyle_plt(picktype, 'mpp')
                        color = pick_color_plt(picktype, self.getPhaseID(phase), quality)
                        if picks['epp'] and picks['lpp']:
                            ax.fill_between([epp, lpp], ylims[0], ylims[1],
                                            alpha=.25, color=color, label='EPP, LPP')
                        if spe:
                            linestyle_spe, width_spe = pick_linestyle_plt(picktype, 'spe')
                            ax.plot([mpp - spe, mpp - spe], ylims, color=color, linestyle=linestyle_spe,
                                    linewidth=width_spe, label='{}-SPE'.format(phase))
                            ax.plot([mpp + spe, mpp + spe], ylims, color=color, linestyle=linestyle_spe,
                                    linewidth=width_spe)
                            ax.plot([mpp, mpp], ylims, color=color, linestyle=linestyle_mpp, linewidth=width_mpp,
                                    label='{}-Pick (quality: {})'.format(phase, quality), picker=5)
                        else:
                            ax.plot([mpp, mpp], ylims, color=color, linestyle=linestyle_mpp, linewidth=width_mpp,
                                    label='{}-Pick (NO PICKERROR)'.format(phase), picker=5)
                    elif picktype == 'auto':
                        color = pick_color_plt(picktype, phase, quality)
                        linestyle_mpp, width_mpp = pick_linestyle_plt(picktype, 'mpp')
                        ax.plot(mpp, ylims[1], color=color, marker='v')
                        ax.plot(mpp, ylims[0], color=color, marker='^')
                        ax.vlines(mpp, ylims[0], ylims[1], color=color, linestyle=linestyle_mpp, linewidth=width_mpp,
                                  picker=5, label='{}-Autopick (quality: {})'.format(phase, quality))
                    else:
                        raise TypeError('Unknown picktype {0}'.format(picktype))

    def locate_event(self):
        """
        locate event using the manually picked phases
        :return:
        """
        if not self.okToContinue():
            return
        parameter = self._inputs
        loctool = 'nll'
        lt = locateTool[loctool]
        # get working directory
        locroot = parameter['nllocroot']
        #locroot = 'E:/NLL/src/Insheim'
        if locroot is None:
            self.PyLoTprefs()
            self.locate_event()

        ctrfile = os.path.join(locroot, 'run', parameter['ctrfile'])
        #ctrfile = 'E:/NLL/src/Insheim/run/Insheim_min1d032016.in'
        ttt = parameter['ttpatter']
        outfile = parameter['outpatter']
        eventname = self.get_current_event_name()
        obsdir = os.path.join(self._inputs['rootpath'], self._inputs['datapath'], self._inputs['database'], eventname)
        self.saveData(event=self.get_current_event(), directory=obsdir, outformats='.obs')
        filename = 'PyLoT_' + eventname
        locpath = os.path.join(locroot, 'loc', filename)
        phasefile = os.path.join(obsdir, filename + '.obs')
        lt.modify_inputs(ctrfile, locroot, filename, phasefile, ttt)
        try:
            lt.locate(ctrfile, self._inputs)
        except RuntimeError as e:
            print(e.message)
        # finally:
        #    os.remove(phasefile)
        self.get_data().applyEVTData(lt.read_location(locpath), typ='event')
        for event in self.calc_magnitude():
            self.get_data().applyEVTData(event, typ='event')

    def init_array_tab(self):
        '''
        Init second main tab if neither metadata nor
        array map are given. A button will be show calling self.get_metadata.
        '''
        if hasattr(self, 'metadata_widget'):
            if self.metadata_widget:
                self.metadata_widget.setParent(None)
                self.array_layout.removeWidget(self.metadata_widget)
        if hasattr(self, 'array_map'):
            if self.array_map:
                self.array_map.setParent(None)
                self.array_layout.removeWidget(self.array_map)
        self.metadata_widget = QWidget(self)
        grid_layout = QGridLayout()
        grid_layout.setColumnStretch(0, 1)
        grid_layout.setColumnStretch(2, 1)
        grid_layout.setRowStretch(0, 1)
        grid_layout.setRowStretch(4, 1)

        self.inventory_label = QLabel('No inventory set...')
        # init inventory button
        self.new_inv_button = QPushButton('Manage inventory files')
        self.new_inv_button.setIcon(self.inventoryIcon)
        self.new_inv_button.clicked.connect(self.inventoryAction.trigger)

        self.init_map_button = QPushButton('Init array map')
        self.init_map_button.setIcon(self.mapIcon)
        self.init_map_button.clicked.connect(self.initMapAction.trigger)
        self.init_map_button.setEnabled(False)

        grid_layout.addWidget(self.inventory_label, 1, 1)
        grid_layout.addWidget(self.new_inv_button, 2, 1)
        grid_layout.addWidget(self.init_map_button, 3, 1)

        self.metadata_widget.setLayout(grid_layout)
        self.array_layout.addWidget(self.metadata_widget)

    def init_array_map(self, checked=0, index=1):
        '''
        Try to init array map widget. If no metadata are given,
        self.get_metadata will be called.
        '''
        if checked: pass  # dummy argument for QAction trigger signal
        self.tabs.setCurrentIndex(1)
        # if there is no metadata (inventories is an empty list), just initialize the default empty tab
        if not self.metadata.inventories:
            self.init_array_tab()
            return
        if hasattr(self, 'metadata_widget'):
            if self.metadata_widget:
                self.metadata_widget.setParent(None)
                self.array_layout.removeWidget(self.metadata_widget)
        if hasattr(self, 'array_map'):
            if self.array_map:
                self.array_map.setParent(None)
                self.array_layout.removeWidget(self.array_map)
            elif not self.array_map:
                self.init_metadata()
                if not self.metadata:
                    return
        self.array_map = Array_map(self, self.metadata)
        # self.array_map_thread()
        self.array_layout.addWidget(self.array_map)
        self.tabs.setCurrentIndex(index)
        self.refresh_array_map()

    def init_spectro_tab(self):
        '''
        Init spectrogram tab with currently selected event.
        '''
        self.spectroWidget = None
        #self.spectro_layout.addWidget( self.spectogramView() )
        pass


    def array_map_thread(self):
        '''
        Start modal thread to init the array_map object.
        '''
        # Note: basemap generation freezes GUI but cannot be threaded as it generates a Pixmap.
        self.amt = Thread(self, self.array_map.init_map, arg=None, progressText='Generating map...',
                          pb_widget=self.mainProgressBarWidget)
        self.amt.finished.connect(self.finish_array_map)
        self.amt.start()

    def finish_array_map(self):
        '''
        Add array_map to GUI tab when array_map_thread has finished.
        '''
        self.array_map = self.amt.data
        self.array_layout.addWidget(self.array_map)
        # self.tabs.setCurrentIndex(index)
        # self.refresh_array_map()

    def refresh_array_map(self):
        '''
        Refresh array map when current event is changed.
        '''
        if not self.array_map:
            return
        # refresh with new picks here!!!
        event = self.get_current_event()
        if hasattr(event, 'origins'):
            if event.origins:
                lat = event.origins[0].latitude
                lon = event.origins[0].longitude
                self.array_map.eventLoc = (lat, lon)
        if self.get_current_event():
            self.array_map.refresh_drawings(self.get_current_event().getPicks(),
                                            self.get_current_event().getAutopicks(), )
        self._eventChanged[1] = False

    def init_event_table(self, tabindex=2):
        '''
        Build and initiate event table (3rd tab [index=2]) containing information of every event.
        '''

        def set_enabled(item, selectable=True, checkable=False):
            # modify item flags depending on case needed
            if selectable and not checkable:
                item.setFlags(QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable)
            elif selectable and checkable:
                item.setFlags(QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsSelectable)
            else:
                item.setFlags(QtCore.Qt.ItemIsEnabled)

        if self.project:
            eventlist = self.project.eventlist
        else:
            eventlist = []

        def cell_clicked(row=None, column=None):
            table = self.project._table
            event = self.project.getEventFromPath(table[row][1].text().split('*')[0])
            if column == 0:
                self.remove_event(event)

        def cell_changed(row=None, column=None):
            # connected to cell changes in event table
            # changes attributes of the corresponding event
            table = self.project._table
            event = self.project.getEventFromPath(table[row][1].text().split('*')[0])
            if column == 10 or column == 11:
                # toggle checked states (exclusive)
                item_ref = table[row][10]
                item_test = table[row][11]
                if column == 10 and item_ref.checkState():
                    item_test.setCheckState(QtCore.Qt.Unchecked)
                    event.setRefEvent(True)
                elif column == 10 and not item_ref.checkState():
                    event.setRefEvent(False)
                elif column == 11 and item_test.checkState():
                    item_ref.setCheckState(QtCore.Qt.Unchecked)
                    event.setTestEvent(True)
                elif column == 11 and not item_test.checkState():
                    event.setTestEvent(False)
                self.fill_eventbox()
            elif column == 12:
                # update event notes
                notes = table[row][12].text()
                event.addNotes(notes)
                self.fill_eventbox()

            event.dirty = True
            self.setDirty(True)

        current_event = self.get_current_event()
        scroll_item = None

        # generate delete icon
        del_icon = QIcon()
        del_icon.addPixmap(QPixmap(':/icons/delete.png'))

        # remove old table
        if hasattr(self, 'event_table'):
            self.event_table.setParent(None)
            self.events_layout.removeWidget(self.event_table)

        # init new qtable
        self.event_table = QTableWidget(self)
        self.event_table.setColumnCount(len(self.table_headers))
        self.event_table.setRowCount(len(eventlist))
        self.event_table.setHorizontalHeaderLabels(self.table_headers)

        # iterate through eventlist and generate items for table rows
        self.project._table = []
        for index, event in enumerate(eventlist):
            phaseErrors = {'P': self._inputs['timeerrorsP'],
                           'S': self._inputs['timeerrorsS']}

            ma_props = {'manual': event.pylot_picks,
                        'auto': event.pylot_autopicks}
            ma_count = {'manual': 0,
                        'auto': 0}

            for ma in ma_props.keys():
                if ma_props[ma]:
                    for picks in ma_props[ma].values():
                        for phasename, pick in picks.items():
                            if not type(pick) in [dict, AttribDict]:
                                continue
                            if pick.get('spe'):
                                ma_count[ma] += 1

            # init table items for current row
            item_delete = QTableWidgetItem()
            item_delete.setIcon(del_icon)
            item_path = QTableWidgetItem()
            item_time = QTableWidgetItem()
            item_lat = QTableWidgetItem()
            item_lon = QTableWidgetItem()
            item_depth = QTableWidgetItem()
            item_momentmag = QTableWidgetItem()
            item_localmag = QTableWidgetItem()
            item_nmp = QTableWidgetItem('{}'.format(ma_count['manual']))  # , ma_count_total['manual']))
            item_nmp.setIcon(self.manupicksicon_small)
            item_nap = QTableWidgetItem('{}'.format(ma_count['auto']))  # , ma_count_total['auto']))
            item_nap.setIcon(self.autopicksicon_small)
            item_ref = QTableWidgetItem()
            item_test = QTableWidgetItem()
            item_notes = QTableWidgetItem()

            event_str = event.path
            if event.dirty:
                event_str += '*'

            item_path.setText(event_str)
            if hasattr(event, 'origins'):
                if event.origins:
                    origin = event.origins[0]
                    item_time.setText(str(origin.time + 0).split('.')[0])  # +0 as workaround in case time=0s
                    item_lon.setText(str(origin.longitude))
                    item_lat.setText(str(origin.latitude))
                    item_depth.setText(str(origin.depth))
            if hasattr(event, 'magnitudes'):
                if event.magnitudes:
                    if len(event.magnitudes) > 1:
                        moment_magnitude = event.magnitudes[0]
                        moment_magnitude.mag = '%4.1f' % moment_magnitude.mag
                        item_momentmag.setText(str(moment_magnitude.mag))
                        local_magnitude = event.magnitudes[1]
                        local_magnitude.mag = '%4.1f' % local_magnitude.mag
                        item_localmag.setText(str(local_magnitude.mag))
                    else:
                        # check type of magnitude
                        if event.magnitudes[0].magnitude_type == 'Mw':
                            moment_magnitude = event.magnitudes[0]
                            moment_magnitude.mag = '%4.1f' % moment_magnitude.mag
                            item_momentmag.setText(str(moment_magnitude.mag))
                        elif event.magnitudes[0].magnitude_type == 'ML':
                            local_magnitude = event.magnitudes[0]
                            local_magnitude.mag = '%4.1f' % local_magnitude.mag
                            item_localmag.setText(str(local_magnitude.mag))

            item_notes.setText(event.notes)

            set_enabled(item_path, True, False)
            set_enabled(item_nmp, True, False)
            set_enabled(item_nap, True, False)
            set_enabled(item_delete, False, False)
            if event.pylot_picks:
                set_enabled(item_ref, True, True)
                set_enabled(item_test, True, True)
            else:
                set_enabled(item_ref, False, True)
                set_enabled(item_test, False, True)

            if event.isRefEvent():
                item_ref.setCheckState(QtCore.Qt.Checked)
            else:
                item_ref.setCheckState(QtCore.Qt.Unchecked)
            if event.isTestEvent():
                item_test.setCheckState(QtCore.Qt.Checked)
            else:
                item_test.setCheckState(QtCore.Qt.Unchecked)

            row = [item_delete, item_path, item_time, item_lat, item_lon, item_depth, item_localmag,
                   item_momentmag, item_nmp, item_nap, item_ref, item_test, item_notes]
            self.project._table.append(row)

            self.setItemColor(row, index, event, current_event)

            if event == current_event:
                scroll_item = item_path

            # manipulate items
            item_ref.setBackground(self._ref_test_colors['ref'])
            item_test.setBackground(self._ref_test_colors['test'])

        for r_index, row in enumerate(self.project._table):
            for c_index, item in enumerate(row):
                if type(item) == QTableWidgetItem:
                    self.event_table.setItem(r_index, c_index, item)
                elif type(item) in [QWidget, QPushButton]:
                    self.event_table.setCellWidget(r_index, c_index, item)

        header = self.event_table.horizontalHeader()
        header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
        header.setStretchLastSection(True)
        self.event_table.cellChanged[int, int].connect(cell_changed)
        self.event_table.cellClicked[int, int].connect(cell_clicked)

        self.events_layout.addWidget(self.event_table)
        self.tabs.setCurrentIndex(tabindex)

        self.event_table.scrollToItem(scroll_item)

    def exportProjectTableDialogs(self):
        if not self.project or not self.project._table:
            QMessageBox.warning(self, 'Nothing to export', 'No project or no project table to export yet!')
            return

        sld = SingleTextLineDialog(label='Specify separator (do not use within notes!): ', default_text=';')
        if not sld.exec_():
            return
        separator = sld.lineEdit.text()

        fd = QtWidgets.QFileDialog()
        fname = fd.getSaveFileName(self, 'Browse for file.',
                                   filter='Table (*.csv)')[0]
        if not fname: return

        if not fname.endswith('.csv'):
            fname += '.csv'

        try:
            self.exportProjectTable(fname, separator)
        except Exception as e:
            QMessageBox.warning(self, 'Could not save file',
                                'Could not save file {}.'.format(fname))
            return

    def exportProjectTable(self, filename, separator=';'):
        with open(filename, 'w') as outfile:
            for header in self.table_headers[1:13]:
                outfile.write('{}{}'.format(header, separator))
            outfile.write('\n')

            for row in self.project._table:
                row = row[1:13]
                event, time, lat, lon, depth, ml, mw, nmp, nap, tune, test, notes = row
                row_str = ''
                for index in range(len(row)):
                    row_str += '{}' + '{}'.format(separator)

                row_str = row_str.format(event.text(), time.text(), lat.text(), lon.text(), depth.text(), ml.text(),
                                         mw.text(), nmp.text(), nap.text(), bool(tune.checkState()),
                                         bool(test.checkState()), notes.text())
                outfile.write(row_str + '\n')

        message = 'Wrote table to file: {}'.format(filename)
        print(message)
        self.update_status(message)

    def setItemColor(self, item_list, index, event, current_event):
        def set_background_color(items, color):
            for item in items:
                item.setBackground(color)

        def set_foreground_color(items, color):
            for item in items:
                item.setForeground(color)

        if index % 2:
            set_background_color(item_list, QtGui.QColor(*(245, 245, 245, 255)))

        if self.isEmpty(event.path):
            set_foreground_color(item_list, QtGui.QColor(*(180, 180, 180, 255)))

        if event == current_event:
            set_background_color(item_list, QtGui.QColor(*(0, 143, 143, 255)))

    def set_metadata(self):
        self.project.inventories = self.metadata.inventories
        if self.metadata.inventories:
            self.init_map_button.setEnabled(True)
            self.initMapAction.setEnabled(True)
            self.inventory_label.setText('Inventory set!')
            self.setDirty(True)
        if not self.metadata.inventories:
            self.init_map_button.setEnabled(False)
            self.initMapAction.setEnabled(False)
            self.inventory_label.setText("No inventory set...")
            # self.setDirty(False)

    def add_metadata(self):
        self.add_metadata_widget = AddMetadataWidget(self, metadata=self.metadata)
        self.add_metadata_widget.accept_button.clicked.connect(self.set_metadata)

    def init_metadata(self, new=False, ask_default=True):
        if hasattr(self.project, 'inventories'):
            self.metadata = Metadata(verbosity=0)
            for inventory in self.project.inventories:
                self.metadata.add_inventory(inventory)

    def calc_magnitude(self):
        self.init_metadata()
        if not self.metadata:
            return None

        wf_copy = self.get_data().getWFData().copy()

        wf_select = Stream()
        # restitute only picked traces
        for station in np.unique(list(self.getPicks('manual').keys()) + list(self.getPicks('auto').keys())):
            wf_select += wf_copy.select(station=station)

        corr_wf = restitute_data(wf_select, self.metadata)
        # calculate moment magnitude
        moment_mag = MomentMagnitude(corr_wf, self.get_data().get_evt_data(), self.inputs.get('vp'),
                                     self.inputs.get('Qp'), self.inputs.get('rho'), verbosity=True)
        moment_mag.updated_event()
        event_moment_mag = moment_mag.net_magnitude()
        if event_moment_mag is not None:
            print("Network moment magnitude: %4.1f" % event_moment_mag.mag)

        # calculate local magnitude
        local_mag = LocalMagnitude(corr_wf, self.get_data().get_evt_data(), self.inputs.get('sstop'),
                                   self.inputs.get('WAscaling'), verbosity=True)

        magscaling = self.inputs.get('magscaling')
        event_loc_mag = local_mag.updated_event(magscaling)
        net_ml = local_mag.net_magnitude(magscaling)
        if net_ml:
            print("Network local magnitude: %4.1f" % net_ml.mag)
        if magscaling is None:
            scaling = False
        elif magscaling[0] == 0.0 and magscaling[1] == 0.0:
            scaling = False
        else:
            scaling = True
        if scaling:
            print("Network local magnitude scaled with:")
            print("%f * Ml + %f" % (magscaling[0], magscaling[1]))

        return event_loc_mag, event_moment_mag

    def check4Loc(self):
        return self.picksNum() >= 4

    def picksNum(self, type='manual'):
        num = 0
        for phases in self.getPicks(type).values():
            num += len(phases)
        return num

    def get_loc_flag(self):
        return self.loc

    def set_loc_flag(self, value):
        self.loc = value

    def check_loc_plt(self):
        evt = self.get_data().get_evt_data()
        if evt.origins and evt.magnitudes:
            return True
        return False

    def update_status(self, message, duration=10000):
        self.statusBar().showMessage(message, duration)
        if self.get_data() is not None:
            if not self.get_current_event() or not self.project.location:
                self.setWindowTitle("PyLoT - New project [*]")
            elif self.get_current_event():
                self.setWindowTitle("PyLoT - {} [*]".format(self.project.location))
            else:
                self.setWindowTitle(
                    "PyLoT - seismic processing the python way[*]")
        self.setWindowModified(self.dirty)

    def tutor_user(self):
        trace_pick = ' select trace to pick on station ...'
        strg_key = ' - [CTRL + mousewheel] vertical spacing'
        shift_key = ' - [SHIFT + mousewheel] horizontal zoom'
        message = trace_pick + strg_key + shift_key
        self.update_status(message, 10000)

    def show_event_information(self):
        pass

    def createNewProject(self):
        '''
        Create new project file.
        '''
        if not self.okToContinue():
            return
        self.project = Project()
        self.init_events(new=True)
        self.setDirty(False)
        self.project.parameter = self._inputs
        self.saveProjectAsAction.setEnabled(True)
        self.update_status('Created new project...', duration=1000)
        return True

    def loadProject(self, fnm=None):
        '''
        Load an existing project file.
        '''
        if not self.okToContinue():
            return
        if not fnm:
            settings = QSettings()
            dir = settings.value('current_project_path')
            dlg = QFileDialog(parent=self, directory=dir)
            fnm = dlg.getOpenFileName(self, 'Open project file...', filter='Pylot project (*.plp)')[0]
            if not fnm:
                return
            settings.setValue('current_project_path', os.path.split(fnm)[0])
        if not os.path.exists(fnm):
            QMessageBox.warning(self, 'Could not open file',
                                'Could not open project file {}. File does not exist.'.format(fnm))
            return
        if fnm:
            self.project = Project.load(fnm)
            if hasattr(self.project, 'parameter'):
                if self.project.parameter:
                    # do this step to update default parameter on older PyLoT projects
                    self.project.parameter.reinit_default_parameters()

                    self._inputs = self.project.parameter
                    self.updateFilteroptions()
            # added for backwards compatibility with older events not having a 'dirty' attribute
            for event in self.project.eventlist:
                if not hasattr(event, 'dirty'):
                    event.dirty = False
            self.tabs.setCurrentIndex(0)  # implemented to prevent double-loading of waveform data
            self.init_events(new=True)
            self.init_metadata()

            message = 'Opened project file {}.'.format(fnm)
            print(message)
            self.update_status(message)

            self.init_array_tab()
            self.set_metadata()
            self.add2recentProjects(fnm)
            self.setDirty(False)

    def add2recentProjects(self, fnm):
        settings = QtCore.QSettings()
        recent = settings.value('recentProjects', [])
        if not type(recent) == list:
            recent = [recent]
        if not fnm in recent:
            recent.append(fnm)
        new_recent = recent[-5:]
        settings.setValue('recentProjects', new_recent)
        settings.sync()

    def saveProjectAs(self, exists=False):
        '''
        Save back project to new pickle file.
        '''
        self.metadata.clear_inventory()
        dlg = QFileDialog(self)
        fnm = dlg.getSaveFileName(self, 'Create a new project file...', filter='Pylot project (*.plp)')
        filename = fnm[0]
        if not len(fnm[0]):
            return False
        if not filename.split('.')[-1] == 'plp':
            filename = fnm[0] + '.plp'
        self.project.parameter = self._inputs
        settings = QSettings()
        autosaveXML = get_bool(settings.value('autosaveXML', True))
        if autosaveXML:
            self.exportEvents()
        if not self.project.save(filename): return False
        self.setDirty(False)
        self.saveProjectAsAction.setEnabled(True)
        self.update_status('Saved new project to {}'.format(filename), duration=5000)
        self.add2recentProjects(filename)
        self.update_obspy_dmt()
        return True

    def saveProject(self, new=False):
        '''
        Save back project to pickle file.
        '''
        if self.project and not new:
            if not self.project.location:
                if not self.saveProjectAs(exists=True):
                    self.setDirty(True)
                    return False
            else:
                self.metadata.clear_inventory()
                self.project.parameter = self._inputs
                settings = QSettings()
                autosaveXML = get_bool(settings.value('autosaveXML', True))
                if autosaveXML:
                    self.exportEvents()
                if not self.project.save(): return False
                self.update_obspy_dmt()
            if not self.project.dirty:
                self.setDirty(False)
                self.update_status('Saved back project to file:\n{}'.format(self.project.location), duration=5000)
                return True
            else:
                # if still dirty because saving failed
                qmb = QMessageBox.warning(self, 'Could not save project',
                                          'Could not save back to original file.\nChoose new file')
                self.setDirty(True)
        return self.saveProjectAs(exists=True)

    def draw(self):
        self.fill_eventbox()
        self.getPlotWidget().draw()
        if self.plot_method == 'fast':
            self.dataPlot.setPermText(1, ' MIN/MAX plot ', color='red')
        else:
            self.dataPlot.setPermText(1)
        self.dataPlot.setPermText(0, '| Number of traces: {} | Gain: {}'.format(len(self.getPlotWidget().getPlotDict()),
                                                                                self.gain))

    def _setDirty(self):
        self.setDirty(True)

    def setDirty(self, value):
        self.saveProjectAction.setEnabled(value)
        self.saveProjectAsAction.setEnabled(True)
        self.project.setDirty(value)
        self.dirty = value
        self.fill_eventbox()

    def setParameter(self, checked=0, show=True):
        if checked: pass  # dummy argument to receive trigger signal (checked) if called by QAction
        if not self.paraBox:
            self.paraBox = PylotParaBox(self._inputs, parent=self, windowflag=Qt.Window)
            self.paraBox.accepted.connect(self._setDirty)
            self.paraBox.accepted.connect(self.filterOptionsFromParameter)
        if show:
            self.paraBox.params_to_gui()
            self.paraBox.show()

    def deleteAllAutopicks(self):
        qmb = QMessageBox(self, icon=QMessageBox.Question,
                          text='Are you sure you want to delete all automatic picks?')
        qmb.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
        qmb.setDefaultButton(QMessageBox.Yes)
        ret = qmb.exec_()
        if not ret == qmb.Yes:
            return
        for event in self.project.eventlist:
            event.pylot_autopicks = {}
            for index, pick in reversed(list(enumerate(event.picks))):
                if pick.method_id.id.endswith('auto'):
                    event.picks.pop(index)
        self.plotWaveformDataThread()
        self.refreshTabs()

    def PyLoTprefs(self):
        if not self._props:
            self._props = PropertiesDlg(self, infile=self.infile,
                                        inputs=self._inputs)
        else:
            self._props.setDirty(False)

        if self._props.exec_():
            if self._props.dirty == 1:
                if self.get_current_event():
                    self.plotWaveformDataThread()
            elif self._props.dirty == 2:
                self.refreshEvents()
            return

    def helpHelp(self):
        if checkurl():
            form = HelpForm(self,
                            'https://github.com/seismology-RUB/PyLoT')
        else:
            form = HelpForm(self, ':/help.html')
        form.show()

class GetExistingDirectories(QFileDialog):
    '''
    File dialog with possibility to select multiple folders.
    '''

    def __init__(self, *args):
        super(GetExistingDirectories, self).__init__(*args)
        self.setOption(self.DontUseNativeDialog, True)
        self.setOption(self.ReadOnly, True)
        self.setFileMode(self.Directory)
        self.setOption(self.ShowDirsOnly, True)
        self.findChildren(QListView)[0].setSelectionMode(QAbstractItemView.ExtendedSelection)
        self.findChildren(QTreeView)[0].setSelectionMode(QAbstractItemView.ExtendedSelection)


def create_window():
    app_created = False
    app = QCoreApplication.instance()
    # check for existing app (when using ipython)
    if app is None:
        app = QApplication(sys.argv)
        app_created = True
    # set aplication/organization name, domain (important to do this BEFORE setupUI is called for correct QSettings)
    app.setOrganizationName("Ruhr-University Bochum / BESTEC")
    app.setOrganizationDomain("rub.de")
    app.setApplicationName("PyLoT")
    app.references = set()
    return app, app_created


def main(project_filename=None, pylot_infile=None, reset_qsettings=False):

    # create the Qt application
    pylot_app, app_created = create_window()
    pixmap = QPixmap(":/splash/splash.png")
    splash = QSplashScreen(pixmap)
    splash.show()

    app_icon = QIcon()
    app_icon.addPixmap(QPixmap(':/icons/pylot.png'))

    # create the main window
    pylot_form = MainWindow(infile=pylot_infile, reset_qsettings=reset_qsettings)
    pylot_form.setWindowIcon(app_icon)
    pylot_form.setIconSize(QSize(60, 60))

    # set other App information
    pylot_app.setApplicationVersion(pylot_form.__version__)
    pylot_app.setWindowIcon(app_icon)
    pylot_app.processEvents()

    splash.showMessage('Loading. Please wait ...')
    pylot_app.processEvents()

    # Show main window and run the app
    pylot_form.showMaximized()
    pylot_app.processEvents()

    splash.finish(pylot_form)

    if project_filename:
        pylot_form.loadProject(args.project_filename)

    if app_created:
        pylot_app.exec_()
    else:
        return pylot_form


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Welcome to PyLoT.')
    parser.add_argument('-p', dest='project_filename', help='load project file',
                        default=None)
    parser.add_argument('-in', dest='input_filename', help='set pylot input file',
                        default=None)
    parser.add_argument('--reset_qsettings', default=False, action='store_true',
                        help='reset qsettings (debug option)')
    args = parser.parse_args()
    sys.exit(main(project_filename=args.project_filename, pylot_infile=args.input_filename,
                  reset_qsettings=args.reset_qsettings))