4159 lines
170 KiB
Python
Executable File
4159 lines
170 KiB
Python
Executable File
#!/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 logging
|
|
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.event import Magnitude, Origin
|
|
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
|
|
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.event import Event
|
|
from pylot.core.io.location import create_creation_info, create_event
|
|
from pylot.core.util.widgets import FilterOptionsDialog, NewEventDlg, \
|
|
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):
|
|
__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 init_config_files(self, infile):
|
|
pylot_config_dir = os.path.join(os.path.expanduser('~'), '.pylot')
|
|
if not os.path.exists(pylot_config_dir):
|
|
os.mkdir(pylot_config_dir)
|
|
|
|
self._inputs = PylotParameter(infile)
|
|
if not infile:
|
|
self._inputs.reset_defaults()
|
|
# check for default pylot.in-file
|
|
infile = os.path.join(pylot_config_dir, '.pylot.in')
|
|
logging.warning('Using default input file {}'.format(infile))
|
|
self._inputs.export2File(infile)
|
|
self.infile = infile
|
|
|
|
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):
|
|
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 add_recentfile(self, event):
|
|
self.recentfiles.insert(0, event)
|
|
|
|
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):
|
|
'''
|
|
Creates and adds events by user selection of event folders to GUI.
|
|
'''
|
|
if not self.project:
|
|
self.createNewProject()
|
|
ed = GetExistingDirectories(self, 'Select event directories...')
|
|
if ed.exec_():
|
|
eventlist = [event for event in ed.selectedFiles() if not event.endswith('EVENTS-INFO')]
|
|
basepath = eventlist[0].split(os.path.basename(eventlist[0]))[0]
|
|
# small hack: if a file "eventlist.txt" is found in the basepath use only those events specified inside
|
|
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.split('\n')[0]) for filename in
|
|
infile.readlines()]
|
|
msg = 'Found file "eventlist.txt" in database path. WILL ONLY USE SELECTED EVENTS out of {} events ' \
|
|
'contained in this subset'
|
|
print(msg.format(len(eventlist_subset)))
|
|
eventlist = [eventname for eventname in eventlist if eventname in eventlist_subset]
|
|
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)
|
|
if not eventlist:
|
|
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')
|
|
return
|
|
else:
|
|
return
|
|
if not self.project:
|
|
print('No project found.')
|
|
return
|
|
|
|
# get path from first event in list and split them
|
|
path = eventlist[0]
|
|
try:
|
|
system_name = platform.system()
|
|
if system_name in ["Linux", "Darwin"]:
|
|
dirs = {
|
|
'database': path.split('/')[-2],
|
|
'datapath': os.path.split(path)[0], # path.split('/')[-3],
|
|
'rootpath': '/' + os.path.join(*path.split('/')[:-3])
|
|
}
|
|
elif system_name == "Windows":
|
|
rootpath = path.split('/')[:-3]
|
|
rootpath[0] += '/'
|
|
dirs = {
|
|
# TODO: Arrange path to meet Win standards
|
|
'database': path.split('/')[-2],
|
|
'datapath': path.split('/')[-3],
|
|
'rootpath': os.path.join(*rootpath)
|
|
}
|
|
except Exception as e:
|
|
dirs = {
|
|
'database': '',
|
|
'datapath': '',
|
|
'rootpath': ''
|
|
}
|
|
print('Warning: Could not automatically init folder structure. ({})'.format(e))
|
|
|
|
settings = QSettings()
|
|
settings.setValue("data/dataRoot", dirs['datapath']) # d irs['rootpath'])
|
|
settings.sync()
|
|
|
|
if not self.project.eventlist:
|
|
# init parameter object
|
|
self.setParameter(show=False)
|
|
# hide all parameter (show all needed parameter later)
|
|
self.paraBox.hide_parameter()
|
|
for directory in dirs.keys():
|
|
# set parameter
|
|
box = self.paraBox.boxes[directory]
|
|
self.paraBox.setValue(box, dirs[directory])
|
|
# show needed parameter in box
|
|
self.paraBox.show_parameter(directory)
|
|
dirs_box = self.paraBox.get_groupbox_dialog('Directories')
|
|
if not dirs_box.exec_():
|
|
return
|
|
self.project.rootpath = dirs['rootpath']
|
|
self.project.datapath = dirs['datapath']
|
|
else:
|
|
if hasattr(self.project, 'datapath'):
|
|
if not self.project.datapath == dirs['datapath']:
|
|
QMessageBox.warning(self, "PyLoT Warning",
|
|
'Datapath missmatch to current project!')
|
|
return
|
|
else:
|
|
self.project.rootpath = dirs['rootpath']
|
|
self.project.datapath = dirs['datapath']
|
|
|
|
self.project.add_eventlist(eventlist)
|
|
self.init_events()
|
|
self.setDirty(True)
|
|
|
|
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()
|
|
# change highlight color:
|
|
# palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Highlight,
|
|
# QtGui.QBrush(QtGui.QColor(0,0,127,127)))
|
|
# palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Highlight,
|
|
# QtGui.QBrush(QtGui.QColor(0,0,127,127)))
|
|
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
|
|
'''
|
|
|
|
# if pick widget is open, refresh tooltips as well
|
|
if self.apw:
|
|
self.apw.refresh_tooltips()
|
|
if hasattr(self, 'cmpw'):
|
|
self.cmpw.refresh_tooltips()
|
|
|
|
if not eventBox:
|
|
eventBox = self.eventBox
|
|
index = eventBox.currentIndex()
|
|
tv = QtWidgets.QTableView()
|
|
header = tv.horizontalHeader()
|
|
header.setSectionResizeMode(QtWidgets.QHeaderView.ResizeToContents)
|
|
header.setStretchLastSection(True)
|
|
header.hide()
|
|
tv.verticalHeader().hide()
|
|
tv.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
|
|
|
|
current_event = self.get_current_event()
|
|
|
|
eventBox.setView(tv)
|
|
eventBox.clear()
|
|
model = eventBox.model()
|
|
plmax = 0
|
|
# set maximum length of path string
|
|
for event in self.project.eventlist:
|
|
pl = len(event.path)
|
|
if pl > plmax:
|
|
plmax = pl
|
|
|
|
for id, event in enumerate(self.project.eventlist):
|
|
event_path = event.path
|
|
#phaseErrors = {'P': self._inputs['timeerrorsP'],
|
|
# 'S': self._inputs['timeerrorsS']}
|
|
|
|
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 in man_au_picks.keys():
|
|
if man_au_picks[ma]:
|
|
for picks in man_au_picks[ma].values():
|
|
for phasename, pick in picks.items():
|
|
if not type(pick) in [dict, AttribDict]:
|
|
continue
|
|
phase_ID = identifyPhaseID(phasename)
|
|
if not phase_ID in npicks[ma].keys():
|
|
continue
|
|
if pick.get('spe'):
|
|
npicks[ma][phase_ID] += 1
|
|
npicks_total[ma][phase_ID] += 1
|
|
|
|
event_ref = event.isRefEvent()
|
|
event_test = event.isTestEvent()
|
|
|
|
time = lat = lon = depth = localmag = None
|
|
if len(event.origins) == 1:
|
|
origin = event.origins[0]
|
|
time = origin.time + 0 # add 0 because there was an exception for time = 0s
|
|
lat = origin.latitude
|
|
lon = origin.longitude
|
|
depth = origin.depth
|
|
if len(event.magnitudes): # if magnitude information exists, i.e., event.magnitudes has at least 1 entry
|
|
moment_magnitude = event.magnitudes[0]
|
|
momentmag = '%4.1f' % moment_magnitude.mag
|
|
if len(event.magnitudes) > 1:
|
|
local_magnitude = event.magnitudes[1]
|
|
localmag = '%4.1f' % local_magnitude.mag
|
|
else:
|
|
localmag = ' '
|
|
|
|
else:
|
|
momentmag = ' '
|
|
localmag = ' '
|
|
|
|
# text = '{path:{plen}} | manual: [{p:3d}] | auto: [{a:3d}]'
|
|
# text = text.format(path=event_path,
|
|
# plen=plmax,
|
|
# p=event_npicks,
|
|
# a=event_nautopicks)
|
|
|
|
event_str = '{path:{plen}}'.format(path=event_path, plen=plmax)
|
|
if event.dirty:
|
|
event_str += '*'
|
|
item_path = QStandardItem(event_str)
|
|
item_time = QStandardItem('{}'.format(time.strftime("%Y-%m-%d %H:%M:%S") if time else ''))
|
|
item_lat = QStandardItem('{}'.format(lat))
|
|
item_lon = QStandardItem('{}'.format(lon))
|
|
item_depth = QStandardItem('{}'.format(depth))
|
|
item_localmag = QStandardItem('{}'.format(localmag))
|
|
item_momentmag = QStandardItem('{}'.format(momentmag))
|
|
|
|
item_nmp = QStandardItem()
|
|
item_nap = QStandardItem()
|
|
item_nmp.setIcon(self.manupicksicon_small)
|
|
item_nap.setIcon(self.autopicksicon_small)
|
|
|
|
for picktype, item_np in [('manual', item_nmp), ('auto', item_nap)]:
|
|
npicks_str = f"{npicks[picktype]['P']}|{npicks[picktype]['S']}"
|
|
#npicks_str += f"({npicks_total[picktype]['P']}/{npicks_total[picktype]['S']})"
|
|
item_np.setText(npicks_str)
|
|
|
|
item_ref = QStandardItem() # str(event_ref))
|
|
item_test = QStandardItem() # str(event_test))
|
|
if event_ref:
|
|
item_ref.setBackground(self._ref_test_colors['ref'])
|
|
if event_test:
|
|
item_test.setBackground(self._ref_test_colors['test'])
|
|
item_notes = QStandardItem(event.notes)
|
|
|
|
openIcon = self.style().standardIcon(QStyle.SP_DirOpenIcon)
|
|
item_path.setIcon(openIcon)
|
|
# if ref: set different color e.g.
|
|
# if event_ref:
|
|
# item.setBackground(self._colors['ref'])
|
|
# if event_test:
|
|
# itemt.setBackground(self._colors['test'])
|
|
# item.setForeground(QtGui.QColor('black'))
|
|
# font = item.font()
|
|
# font.setPointSize(10)
|
|
# item.setFont(font)
|
|
# item2.setForeground(QtGui.QColor('black'))
|
|
# item2.setFont(font)
|
|
itemlist = [item_path, item_time, item_lat, item_lon, item_depth,
|
|
item_localmag, item_momentmag, item_nmp, item_nap, item_ref, item_test, item_notes]
|
|
for item in itemlist:
|
|
item.setTextAlignment(Qt.AlignCenter)
|
|
if event_test and select_events == 'ref' or self.isEmpty(event_path):
|
|
for item in itemlist:
|
|
item.setEnabled(False)
|
|
|
|
# item color
|
|
self.setItemColor(itemlist, id, event, current_event)
|
|
|
|
model.appendRow(itemlist)
|
|
if not event.path == self.eventBox.itemText(id).split('*')[0].strip():
|
|
message = ('Path missmatch creating eventbox.\n'
|
|
'{} unequal {}.'
|
|
.format(event.path, self.eventBox.itemText(id)))
|
|
raise ValueError(message)
|
|
# not working with obspy events
|
|
# eventBox.setItemData(id, event)
|
|
eventBox.setCurrentIndex(index)
|
|
self.refreshRefTestButtons()
|
|
|
|
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 getEventFileName(self, type='manual'):
|
|
if self.get_fnames(type) is None:
|
|
self.set_fname(self.get_data().getEventFileName(), type)
|
|
return self.get_fnames(type)
|
|
|
|
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.
|
|
'''
|
|
# if self.fnames and self.okToContinue():
|
|
# self.setDirty(True)
|
|
# ans = self.data.setWFData(self.fnames)
|
|
# elif self.fnames is None and self.okToContinue():
|
|
# ans = self.data.setWFData(self.getWFFnames())
|
|
# else:
|
|
# ans = False
|
|
|
|
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 setWFstatus(self):
|
|
# '''
|
|
# show status of current data, can be either 'raw' or 'processed'
|
|
# :param status:
|
|
# :return:
|
|
# '''
|
|
# status = self.data.processed
|
|
# wf_stat_color = {True: 'green',
|
|
# False: 'black',
|
|
# None: None}
|
|
# wf_stat = {True: 'processed',
|
|
# False: 'raw',
|
|
# None: None}
|
|
# self.dataPlot.setQCboxItem(wf_stat[status])
|
|
|
|
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
|
|
# nth_sample_new = int(np.ceil(npts/npts_max))
|
|
# message = "You are about to plot a huge dataset with {npts} datapoints. With a current setting of " \
|
|
# "nth_sample = {nth_sample} a total of {npts2plot} points will be plotted which is more " \
|
|
# "than the maximum setting of {npts_max}. " \
|
|
# "PyLoT recommends to raise nth_sample from {nth_sample} to {nth_sample_new}. Do you want "\
|
|
# "to change nth_sample to {nth_sample_new} now?"
|
|
#
|
|
# ans = QMessageBox.question(self, self.tr("Optimize plot performance..."),
|
|
# self.tr(message.format(npts=npts,
|
|
# nth_sample=nth_sample,
|
|
# npts_max=npts_max,
|
|
# nth_sample_new=nth_sample_new,
|
|
# npts2plot=npts2plot)),
|
|
# QMessageBox.Yes | QMessageBox.No,
|
|
# QMessageBox.Yes)
|
|
# if ans == QMessageBox.Yes:
|
|
# settings.setValue("nth_sample", nth_sample_new)
|
|
# settings.sync()
|
|
|
|
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()
|
|
# self.drawPicks()
|
|
# self.draw()
|
|
|
|
def plotN(self):
|
|
self.setComponent('N')
|
|
self.plotWaveformDataThread()
|
|
# self.drawPicks()
|
|
# self.draw()
|
|
|
|
def plotE(self):
|
|
self.setComponent('E')
|
|
self.plotWaveformDataThread()
|
|
# self.drawPicks()
|
|
# self.draw()
|
|
|
|
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)
|
|
# self.drawPicks()
|
|
# self.draw()
|
|
|
|
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
|
|
# try:
|
|
# return self.filteroptions[self.getSeismicPhase()]
|
|
# except AttributeError as e:
|
|
# print(e)
|
|
# return FilterOptions(None, None, None)
|
|
|
|
def getFilters(self):
|
|
return self.filteroptions
|
|
|
|
def setFilterOptions(self, filterOptions): # , seismicPhase=None):
|
|
# if seismicPhase is None:
|
|
# self.getFilterOptions()[self.getSeismicPhase()] = filterOptions
|
|
# else:
|
|
# self.getFilterOptions()[seismicPhase] = 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 updateFilterOptions(self):
|
|
# try:
|
|
# settings = QSettings()
|
|
# if settings.value("filterdefaults",
|
|
# None) is None and not self.getFilters():
|
|
# for key, value in FILTERDEFAULTS.items():
|
|
# self.setFilterOptions(FilterOptions(**value), key)
|
|
# elif settings.value("filterdefaults", None) is not None:
|
|
# for key, value in settings.value("filterdefaults"):
|
|
# self.setFilterOptions(FilterOptions(**value), key)
|
|
# except Exception as e:
|
|
# self.update_status('Error ...')
|
|
# emsg = QErrorMessage(self)
|
|
# emsg.showMessage('Error: {0}'.format(e))
|
|
# else:
|
|
# self.update_status('Filter loaded ... '
|
|
# '[{0}: {1} Hz]'.format(
|
|
# self.getFilterOptions().getFilterType(),
|
|
# self.getFilterOptions().getFreq()))
|
|
# if self.filterActionP.isChecked() or self.filterActionS.isChecked():
|
|
# self.filterWaveformData()
|
|
|
|
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 setSeismicPhase(self, phase):
|
|
self.seismicPhase = self.seismicPhaseButtonGroup.getValue()
|
|
self.update_status('Seismic phase changed to '
|
|
'{0}'.format(self.getSeismicPhase()))
|
|
|
|
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 deletePicks(self, station, deleted_pick, type):
|
|
deleted_pick['station'] = station
|
|
self.addPicks(station, {}, type=type)
|
|
self.log_deleted_picks([deleted_pick])
|
|
|
|
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'])
|
|
|
|
# quality = getPickQuality(self.get_data().getWFData(),
|
|
# stat_picks, self._inputs, phaseID,
|
|
# compclass)
|
|
|
|
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 check4Comparison(self):
|
|
# mpicks = self.getPicks()
|
|
# apicks = self.getPicks('auto')
|
|
# for station, phases in mpicks.items():
|
|
# try:
|
|
# aphases = apicks[station]
|
|
# for phase in phases.keys():
|
|
# if phase in aphases.keys():
|
|
# return True
|
|
# except KeyError:
|
|
# continue
|
|
# return False
|
|
|
|
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 createNewEvent(self):
|
|
if self.okToContinue():
|
|
new = NewEventDlg()
|
|
if new.exec_() != QDialog.Rejected:
|
|
evtpar = new.getValues()
|
|
cinfo = create_creation_info(agency_id=self.agency)
|
|
event = create_event(evtpar['origintime'], cinfo)
|
|
self.data = Data(self, evtdata=event)
|
|
self.setDirty(True)
|
|
|
|
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 closeEvent(self, event):
|
|
if self.okToContinue():
|
|
if hasattr(self, 'logwidget'):
|
|
self.logwidget.close()
|
|
event.accept()
|
|
else:
|
|
event.ignore()
|
|
# self.closing.emit()
|
|
# QMainWindow.closeEvent(self, event)
|
|
|
|
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 Project(object):
|
|
'''
|
|
Pickable class containing information of a PyLoT project, like event lists and file locations.
|
|
'''
|
|
|
|
# TODO: remove rootpath
|
|
def __init__(self):
|
|
self.eventlist = []
|
|
self.location = None
|
|
self.rootpath = None
|
|
self.datapath = None
|
|
self.dirty = False
|
|
self.parameter = None
|
|
self._table = None
|
|
|
|
def add_eventlist(self, eventlist):
|
|
'''
|
|
Add events from an eventlist containing paths to event directories.
|
|
Will skip existing paths.
|
|
'''
|
|
if len(eventlist) == 0:
|
|
return
|
|
for item in eventlist:
|
|
event = Event(item)
|
|
event.rootpath = self.parameter['rootpath']
|
|
event.database = self.parameter['database']
|
|
event.datapath = self.parameter['datapath']
|
|
if not event.path in self.getPaths():
|
|
self.eventlist.append(event)
|
|
self.setDirty()
|
|
else:
|
|
print('Skipping event with path {}. Already part of project.'.format(event.path))
|
|
self.eventlist.sort(key=lambda x: x.pylot_id)
|
|
self.search_eventfile_info()
|
|
|
|
def remove_event(self, event):
|
|
self.eventlist.remove(event)
|
|
|
|
def remove_event_by_id(self, eventID):
|
|
for event in self.eventlist:
|
|
if eventID in str(event.resource_id):
|
|
self.remove_event(event)
|
|
break
|
|
|
|
def read_eventfile_info(self, filename, separator=','):
|
|
'''
|
|
Try to read event information from file (:param:filename) comparing specific event datetimes.
|
|
File structure (each row): event, date, time, magnitude, latitude, longitude, depth
|
|
separated by :param:separator each.
|
|
'''
|
|
with open(filename, 'r') as infile:
|
|
for line in infile.readlines():
|
|
eventID, date, time, mag, lat, lon, depth = line.split(separator)[:7]
|
|
# skip first line
|
|
try:
|
|
day, month, year = date.split('/')
|
|
except:
|
|
continue
|
|
year = int(year)
|
|
# hardcoded, if year only consists of 2 digits (e.g. 16 instead of 2016)
|
|
if year < 100:
|
|
year += 2000
|
|
datetime = '{}-{}-{}T{}'.format(year, month, day, time)
|
|
try:
|
|
datetime = UTCDateTime(datetime)
|
|
except Exception as e:
|
|
print(e, datetime, filename)
|
|
continue
|
|
for event in self.eventlist:
|
|
if eventID in str(event.resource_id) or eventID in event.origins:
|
|
if event.origins:
|
|
origin = event.origins[0] # should have only one origin
|
|
if origin.time == datetime:
|
|
origin.latitude = float(lat)
|
|
origin.longitude = float(lon)
|
|
origin.depth = float(depth)
|
|
else:
|
|
continue
|
|
elif not event.origins:
|
|
origin = Origin(resource_id=event.resource_id,
|
|
time=datetime, latitude=float(lat),
|
|
longitude=float(lon), depth=float(depth))
|
|
event.origins.append(origin)
|
|
event.magnitudes.append(Magnitude(resource_id=event.resource_id,
|
|
mag=float(mag),
|
|
mag_type='M'))
|
|
break
|
|
|
|
def search_eventfile_info(self):
|
|
'''
|
|
Search all datapaths in rootpath for filenames with given file extension fext
|
|
and try to read event info from it
|
|
'''
|
|
datapaths = []
|
|
fext = '.csv'
|
|
for event in self.eventlist:
|
|
if not event.datapath in datapaths:
|
|
datapaths.append(event.datapath)
|
|
for datapath in datapaths:
|
|
# datapath = os.path.join(self.rootpath, datapath)
|
|
if os.path.isdir(datapath):
|
|
for filename in os.listdir(datapath):
|
|
filename = os.path.join(datapath, filename)
|
|
if os.path.isfile(filename) and filename.endswith(fext):
|
|
try:
|
|
self.read_eventfile_info(filename)
|
|
except Exception as e:
|
|
print('Failed on reading eventfile info from file {}: {}'.format(filename, e))
|
|
else:
|
|
print("Directory %s does not exist!" % datapath)
|
|
|
|
def getPaths(self):
|
|
'''
|
|
Returns paths (eventlist) of all events saved in the project.
|
|
'''
|
|
paths = []
|
|
for event in self.eventlist:
|
|
paths.append(event.path)
|
|
return paths
|
|
|
|
def setDirty(self, value=True):
|
|
self.dirty = value
|
|
|
|
def getEventFromPath(self, path):
|
|
'''
|
|
Search for an event in the project by event path.
|
|
'''
|
|
for event in self.eventlist:
|
|
if event.path == path:
|
|
return event
|
|
|
|
def save(self, filename=None):
|
|
'''
|
|
Save PyLoT Project to a file.
|
|
Can be loaded by using project.load(filename).
|
|
'''
|
|
try:
|
|
import pickle
|
|
except ImportError:
|
|
import _pickle as pickle
|
|
|
|
if filename:
|
|
self.location = filename
|
|
else:
|
|
filename = self.location
|
|
|
|
table = self._table # MP: see below
|
|
try:
|
|
outfile = open(filename, 'wb')
|
|
self._table = [] # MP: Workaround as long as table cannot be saved as part of project
|
|
pickle.dump(self, outfile, protocol=pickle.HIGHEST_PROTOCOL)
|
|
self.setDirty(False)
|
|
self._table = table # MP: see above
|
|
return True
|
|
except Exception as e:
|
|
print('Could not pickle PyLoT project. Reason: {}'.format(e))
|
|
self.setDirty()
|
|
self._table = table # MP: see above
|
|
return False
|
|
|
|
@staticmethod
|
|
def load(filename):
|
|
'''
|
|
Load project from filename.
|
|
'''
|
|
import pickle
|
|
infile = open(filename, 'rb')
|
|
project = pickle.load(infile)
|
|
infile.close()
|
|
project.location = filename
|
|
print('Loaded %s' % filename)
|
|
return project
|
|
|
|
|
|
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()
|
|
# app.references.add(window)
|
|
# window.show()
|
|
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()
|
|
# pylot_app = QApplication(sys.argv)
|
|
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))
|