#!/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 os import platform import sys import matplotlib matplotlib.use('Qt4Agg') matplotlib.rcParams['backend.qt4'] = 'PySide' matplotlib.rcParams['savefig.dpi'] = 300 from PySide import QtGui, QtCore from PySide.QtCore import QCoreApplication, QSettings, Signal, QFile, \ QFileInfo, Qt, QSize from PySide.QtGui import QMainWindow, QInputDialog, QIcon, QFileDialog, \ QWidget, QHBoxLayout, QVBoxLayout, QStyle, QKeySequence, QLabel, QFrame, QAction, \ QDialog, QErrorMessage, QApplication, QPixmap, QMessageBox, QSplashScreen, \ QActionGroup, QListWidget, QLineEdit, QListView, QAbstractItemView, \ QTreeView, QComboBox, QTabWidget, QPushButton, QGridLayout import numpy as np from obspy import UTCDateTime 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_qt4agg import FigureCanvas except ImportError: from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas from matplotlib.backends.backend_qt4agg import NavigationToolbar2QT as NavigationToolbar 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 symmetrize_error, getQualityFromUncertainty, getPickQuality, get_quality_class from pylot.core.io.phases import picksdict_from_picks import pylot.core.loc.nll as nll from pylot.core.util.defaults import FILTERDEFAULTS 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, getLogin, \ full_range, readFilterInformation, trim_station_components, check4gaps, make_pen, pick_color_plt, \ pick_linestyle_plt, remove_underscores, check4doubled, identifyPhaseID, excludeQualityClasses, \ check4rotated, transform_colors_mpl, transform_colors_mpl_str, getAutoFilteroptions, check_all_obspy, \ check_all_pylot, real_Bool, SetChannelComponents 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, \ getDataType, ComparisonWidget, TuneAutopicker, PylotParaBox, AutoPickDlg, CanvasWidget, AutoPickWidget, \ CompareEventsWidget, ProgressBarWidget, AddMetadataWidget 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.styles import style_settings if sys.version_info.major == 3: import icons_rc_3 as icons_rc elif sys.version_info.major == 2: import icons_rc_2 as icons_rc else: raise ImportError('Could not determine python version.') locateTool = dict(nll=nll) class MainWindow(QMainWindow): __version__ = _getVersionString() closing = Signal() def __init__(self, parent=None, infile=None): super(MainWindow, self).__init__(parent) self.init_config_files(infile) self._props = None 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() 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() 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) 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", getLogin()) if settings.value("agency_id", None) is None: agency = QInputDialog.getText(self, "Enter authority/institution name:", "Authority") settings.setValue("agency_id", agency) 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!') self.fnames = None self._stime = None 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('compclass', None) is None: settings.setValue('compclass', SetChannelComponents()) 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):", "Format") settings.setValue("output/Format", outformat) if settings.value('autoFilter', None) is None: settings.setValue('autoFilter', True) settings.sync() # 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') print('Using default input file {}'.format(infile)) self._inputs.export2File(infile) self.infile = infile def setupUi(self): 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')) 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')) 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.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.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 qualitites...', 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: self.qualities_action.setVisible(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.""") # phaseToolBar = self.addToolBar("PhaseTools") # phaseToolActions = (self.selectPAction, self.selectSAction) # phaseToolBar.setObjectName("PhaseTools") # self.addActions(phaseToolBar, phaseToolActions) # 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) # 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, 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.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,) 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 = QtGui.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 = QtGui.QWidget(self) array_tab = QtGui.QWidget(self) events_tab = QtGui.QWidget(self) # init main widgets layouts self.wf_layout = QtGui.QVBoxLayout() self.array_layout = QtGui.QVBoxLayout() self.events_layout = QtGui.QVBoxLayout() wf_tab.setLayout(self.wf_layout) array_tab.setLayout(self.array_layout) events_tab.setLayout(self.events_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.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.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() self.setCentralWidget(_widget) 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 = QtGui.QPushButton('Tune') self.test_event_button = QtGui.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 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 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 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): self.fileMenu.clear() 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 = QSettings() 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]) @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 fext = '.xml' for event in events: path = event.path eventname = path.split('/')[-1] # or event.pylot_id filename = os.path.join(path, 'PyLoT_' + eventname + fext) if os.path.isfile(filename): self.load_data(filename, draw=False, event=event, overwrite=True) 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, overwrite=False): if not overwrite: 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)) data += data_new 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(type='auto', event=event) self.updatePicks(type='manual', 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=None): ''' 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) basepath = eventpath.split(os.path.basename(eventpath))[0] 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 = ed.selectedFiles() basepath = eventlist[0].split(os.path.basename(eventlist[0]))[0] if check_obspydmt_structure(basepath): print('Recognized obspyDMT structure in selected files. Settings 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': path.split('/')[-3], 'rootpath': '/' + os.path.join(*path.split('/')[:-3]) } elif system_name == "Windows": rootpath = path.split('/')[:-3] rootpath[0] += '/' dirs = { 'database': path.split('/')[-2], 'datapath': path.split('/')[-3], 'rootpath': os.path.join(*rootpath) } except Exception as e: dirs = { 'database': '', 'datapath': '', 'rootpath': '' } print('Warning: Could not automatically init folder structure. ({})'.format(e)) settings = QSettings() settings.setValue("data/dataRoot", dirs['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'] else: if hasattr(self.project, 'rootpath'): if not self.project.rootpath == dirs['rootpath']: QMessageBox.warning(self, "PyLoT Warning", 'Rootpath missmatch to current project!') return else: self.project.rootpath = dirs['rootpath'] 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 = QtGui.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 else: info_str = '' for event, path_exists in zip(self.project.eventlist, paths_exist): info_str += '\n{} exists: {}'.format(event.path, path_exists) print('Unable to find certain event paths:{}'.format(info_str)) return False def modify_project_path(self, new_rootpath): 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 = QtGui.QTableView() header = tv.horizontalHeader() header.setResizeMode(QtGui.QHeaderView.ResizeToContents) header.setStretchLastSection(True) header.hide() tv.verticalHeader().hide() tv.setSelectionBehavior(QtGui.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']} ma_props = {'manual': event.pylot_picks, 'auto': event.pylot_autopicks} ma_count = {'manual': 0, 'auto': 0} ma_count_total = {'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 ma_count_total[ma] += 1 event_ref = event.isRefEvent() event_test = event.isTestEvent() time = lat = lon = depth = mag = 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) == 1: magnitude = event.magnitudes[0] mag = magnitude.mag # 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 = QtGui.QStandardItem(event_str) item_time = QtGui.QStandardItem('{}'.format(time)) item_lat = QtGui.QStandardItem('{}'.format(lat)) item_lon = QtGui.QStandardItem('{}'.format(lon)) item_depth = QtGui.QStandardItem('{}'.format(depth)) item_mag = QtGui.QStandardItem('{}'.format(mag)) item_nmp = QtGui.QStandardItem('{}({})'.format(ma_count['manual'], ma_count_total['manual'])) item_nmp.setIcon(self.manupicksicon_small) item_nap = QtGui.QStandardItem('{}({})'.format(ma_count['auto'], ma_count_total['auto'])) item_nap.setIcon(self.autopicksicon_small) item_ref = QtGui.QStandardItem() # str(event_ref)) item_test = QtGui.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 = QtGui.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_mag, 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 is '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']): ''' 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'] 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) 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): return 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 getStationID(self, station): for wfID in self.getPlotWidget().getPlotDict().keys(): actual_station = self.getPlotWidget().getPlotDict()[wfID][0] if station == actual_station: return wfID return None 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.fill_eventbox() 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.fill_eventbox() 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. 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) 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_syn = [] 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) 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 self.data.setWFData(self.fnames, self.fnames_syn, checkRotated=True, metadata=self.metadata) def prepareObspyDMT_data(self, eventpath): qcbox_processed = self.dataPlot.qcombo_processed qcheckb_syn = self.dataPlot.syn_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.syn_checkbox.isChecked(): self.fnames_syn = [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() 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() # 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): 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, gaps = 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 = [plotDict[n][2] + '.' + plotDict[n][0] 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') self.locateEventAction.setEnabled(True) self.qualities_action.setEnabled(True) if event.pylot_autopicks: self.drawPicks(picktype='auto') self.qualities_action.setEnabled(True) if True in self.comparable.values(): self.compare_action.setEnabled(True) self.draw() # TODO: Quick and dirty, improve this on later iteration 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) # 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.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. ''' self.check_plot_quantity() 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().getSynWFData() 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 real_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) plots, gaps = rval if rval else ([], []) return plots, gaps 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 getStationName(self, wfID): plot_dict = self.getPlotWidget().getPlotDict() if wfID in plot_dict.keys(): return plot_dict[wfID][0] def getNetworkName(self, wfID): plot_dict = self.getPlotWidget().getPlotDict() if wfID in plot_dict.keys(): return plot_dict[wfID][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) if button == 1: self.pickDialog(wfID, network, station) elif button == 4: self.toggle_station_color(wfID, network, station) def toggle_station_color(self, wfID, network, station): 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'] nwst = '{}.{}'.format(network, station) if current_pen == black_pen: line_item.setPen(red_pen) if not nwst in self.stations_highlighted: self.stations_highlighted.append(nwst) else: line_item.setPen(black_pen) if nwst in self.stations_highlighted: self.stations_highlighted.pop(self.stations_highlighted.index(nwst)) def highlight_stations(self): for wfID, value in self.getPlotWidget().getPlotDict().items(): station, channel, network = value nwst = '{}.{}'.format(network, station) if nwst in self.stations_highlighted: self.toggle_station_color(wfID, network, station) def pickDialog(self, wfID, network=None, station=None): if not network: network = self.getNetworkName(wfID) if not station: station = self.getStationName(wfID) if not station or not network: return self.update_status('picking on station {0}'.format(station)) data = self.get_data().getOriginalWFData().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=data.select(station=station), station=station, network=network, picks=self.getPicksOnStation(station, 'manual'), autopicks=self.getPicksOnStation(station, 'auto'), metadata=self.metadata, event=event, filteroptions=self.filteroptions, wftype=wftype) if self.filterActionP.isChecked() or self.filterActionS.isChecked(): pickDlg.currentPhase = self.getSeismicPhase() 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') 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: 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 # if not stat_picks: # stat_picks = picks # else: # msgBox = QMessageBox(self) # msgBox.setText("The picks for station {0} have been " # "changed.".format(station)) # msgBox.setDetailedText("Old picks:\n" # "{old_picks}\n\n" # "New picks:\n" # "{new_picks}".format(old_picks=stat_picks, # new_picks=picks)) # msgBox.setInformativeText("Do you want to save your changes?") # msgBox.setStandardButtons(QMessageBox.Save | QMessageBox.Cancel) # msgBox.setDefaultButton(QMessageBox.Save) # ret = msgBox.exec_() # if ret == QMessageBox.Save: # stat_picks = picks # rval = True # elif ret == QMessageBox.Cancel: # pass # else: # raise Exception('FATAL: Should never occur!') # MP MP prompt redundant because new picks have to be accepted in the first place closing PickDlg def updatePicks(self, type='manual', event=None): if not event: event = self.get_current_event() picksdict = picksdict_from_picks(evt=self.get_data().get_evt_data()) if type == 'manual': event.addPicks(picksdict['manual']) # event.picks.update(picks) MP MP idea elif type == 'auto': event.addAutopicks(picksdict['auto']) 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 plotID = self.getStationID(station) if plotID is None: return 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) 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) 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'] if locroot is None: self.PyLoTprefs() self.locate_event() ctrfile = os.path.join(locroot, 'run', parameter['ctrfile']) 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) except RuntimeError as e: print(e.message) # finally: # os.remove(phasefile) self.get_data().applyEVTData(lt.read_location(locpath), typ='event') # self.get_data().applyEVTData(self.calc_magnitude(), 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, index=1): ''' Try to init array map widget. If no metadata are given, self.get_metadata will be called. ''' self.tabs.setCurrentIndex(1) 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.array_map_thread() self.array_layout.addWidget(self.array_map) self.tabs.setCurrentIndex(index) self.refresh_array_map() 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()) 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()) if column == 9 or column == 10: # toggle checked states (exclusive) item_ref = table[row][9] item_test = table[row][10] if column == 9 and item_ref.checkState(): item_test.setCheckState(QtCore.Qt.Unchecked) event.setRefEvent(True) elif column == 9 and not item_ref.checkState(): event.setRefEvent(False) elif column == 10 and item_test.checkState(): item_ref.setCheckState(QtCore.Qt.Unchecked) event.setTestEvent(True) elif column == 10 and not item_test.checkState(): event.setTestEvent(False) self.fill_eventbox() elif column == 11: # update event notes notes = table[row][11].text() event.addNotes(notes) self.fill_eventbox() current_event = self.get_current_event() # 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 = QtGui.QTableWidget(self) self.event_table.setColumnCount(12) self.event_table.setRowCount(len(eventlist)) self.event_table.setHorizontalHeaderLabels(['', 'Event', 'Time', 'Lat', 'Lon', 'Depth', 'Mag', '[N] MP', '[N] AP', 'Tuning Set', 'Test Set', 'Notes']) # 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} ma_count_total = {'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 ma_count_total[ma] += 1 # init table items for current row item_delete = QtGui.QTableWidgetItem() item_delete.setIcon(del_icon) item_path = QtGui.QTableWidgetItem() item_time = QtGui.QTableWidgetItem() item_lat = QtGui.QTableWidgetItem() item_lon = QtGui.QTableWidgetItem() item_depth = QtGui.QTableWidgetItem() item_mag = QtGui.QTableWidgetItem() item_nmp = QtGui.QTableWidgetItem('{}({})'.format(ma_count['manual'], ma_count_total['manual'])) item_nmp.setIcon(self.manupicksicon_small) item_nap = QtGui.QTableWidgetItem('{}({})'.format(ma_count['auto'], ma_count_total['auto'])) item_nap.setIcon(self.autopicksicon_small) item_ref = QtGui.QTableWidgetItem() item_test = QtGui.QTableWidgetItem() item_notes = QtGui.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: magnitude = event.magnitudes[0] item_mag.setText(str(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) column = [item_delete, item_path, item_time, item_lat, item_lon, item_depth, item_mag, item_nmp, item_nap, item_ref, item_test, item_notes] self.project._table.append(column) self.setItemColor(column, index, event, current_event) # 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) == QtGui.QTableWidgetItem: self.event_table.setItem(r_index, c_index, item) elif type(item) in [QtGui.QWidget, QtGui.QPushButton]: self.event_table.setCellWidget(r_index, c_index, item) header = self.event_table.horizontalHeader() header.setResizeMode(QtGui.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) 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) def add_metadata(self): self.add_metadata_widget = AddMetadataWidget(self, metadata=self.metadata) self.add_metadata_widget.close_button.clicked.connect(self.set_metadata) def init_metadata(self, new=False, ask_default=True): if hasattr(self.project, 'inventories'): self.metadata = Metadata() for inventory in self.project.inventories: self.metadata.add_inventory(inventory) def calc_magnitude(self, type='ML'): self.init_metadata() if not self.metadata: return None wf_copy = self.get_data().getWFData().copy() corr_wf = restitute_data(wf_copy, *self.metadata) # if not rest_flag: # raise ProcessingError('Restitution of waveform data failed!') if type == 'ML': local_mag = LocalMagnitude(corr_wf, self.get_data().get_evt_data(), self.inputs.get('sstop'), self.inputs.get('WAScaling'), verbosity=True) return local_mag.updated_event() elif type == 'Mw': 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) return moment_mag.updated_event() else: return None 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=5000): 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: dlg = QFileDialog(parent=self) fnm = dlg.getOpenFileName(self, 'Open project file...', filter='Pylot project (*.plp)') if not fnm: return fnm = fnm[0] if fnm: self.project = Project.load(fnm) if hasattr(self.project, 'parameter'): if self.project.parameter: 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.setDirty(False) self.init_metadata() self.init_array_tab() self.set_metadata() def saveProjectAs(self, exists=False): ''' Save back project to new pickle file. ''' 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 self.exportEvents() self.project.save(filename) self.setDirty(False) self.saveProjectAsAction.setEnabled(True) self.update_status('Saved new project to {}'.format(filename), duration=5000) 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.project.parameter = self._inputs self.exportEvents() self.project.save() 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: {} |'.format(len(self.getPlotWidget().getPlotDict()))) 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(): event.accept() else: event.ignore() # self.closing.emit() # QMainWindow.closeEvent(self, event) def setParameter(self, show=True): if not self.paraBox: self.paraBox = PylotParaBox(self._inputs, parent=self, windowflag=1) self.paraBox.accepted.connect(self._setDirty) self.paraBox.accepted.connect(self.filterOptionsFromParameter) if show: self.paraBox.params_to_gui() self.paraBox.show() def PyLoTprefs(self): if not self._props: self._props = PropertiesDlg(self, infile=self.infile, inputs=self._inputs) if self._props.exec_(): self.init_wfWidget() return def helpHelp(self): if checkurl(): form = HelpForm(self, 'https://ariadne.geophysik.ruhr-uni-bochum.de/trac/PyLoT/wiki') 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. ''' def __init__(self): self.eventlist = [] self.location = None self.rootpath = 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. ''' infile = open(filename, 'r') for line in infile.readlines(): eventID, date, time, mag, lat, lon, depth = line.split(separator)[:7] # skip first line try: month, day, 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 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) 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)) 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 cPickle except ImportError: import _pickle as cPickle if filename: self.location = filename else: filename = self.location try: outfile = open(filename, 'wb') cPickle.dump(self, outfile, -1) self.setDirty(False) except Exception as e: print('Could not pickle PyLoT project. Reason: {}'.format(e)) self.setDirty() @staticmethod def load(filename): ''' Load project from filename. ''' try: import cPickle except ImportError: import _pickle as cPickle infile = open(filename, 'rb') project = cPickle.load(infile) 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(args=None): project_filename = None # args.project_filename = 'C:/Shared/AlpArray/alparray_data/project_alparray_test.plp' pylot_infile = None if args: if args.project_filename: project_filename = args.project_filename if args.input_filename: pylot_infile = args.input_filename # 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) 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) args = parser.parse_args() sys.exit(main(args))