236 lines
8.4 KiB
Python
236 lines
8.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Wed Mar 19 11:27:35 2014
|
|
|
|
@author: sebastianw
|
|
"""
|
|
|
|
import matplotlib
|
|
|
|
matplotlib.use('Qt4Agg')
|
|
matplotlib.rcParams['backend.qt4'] = 'PySide'
|
|
|
|
from matplotlib.figure import Figure
|
|
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg
|
|
from PySide.QtGui import (QAction,
|
|
QApplication,
|
|
QCheckBox,
|
|
QComboBox,
|
|
QDialog,
|
|
QDialogButtonBox,
|
|
QDockWidget,
|
|
QDoubleSpinBox,
|
|
QGroupBox,
|
|
QGridLayout,
|
|
QHBoxLayout,
|
|
QIcon,
|
|
QKeySequence,
|
|
QLabel,
|
|
QLineEdit,
|
|
QMessageBox,
|
|
QSpinBox,
|
|
QTabWidget,
|
|
QTextBrowser,
|
|
QToolBar,
|
|
QVBoxLayout,
|
|
QWidget)
|
|
from PySide.QtCore import (Qt,
|
|
QUrl,
|
|
SIGNAL,
|
|
SLOT)
|
|
from pylot.core.util import OptionsError
|
|
from pylot.core.read import FilterOptions
|
|
|
|
|
|
class MPLWidget(FigureCanvasQTAgg):
|
|
|
|
def __init__(self, parent=None, xlabel='x', ylabel='y', title='Title'):
|
|
super(MPLWidget, self).__init__(Figure())
|
|
|
|
self.setParent(parent)
|
|
self.figure = Figure()
|
|
self.canvas = FigureCanvasQTAgg(self.figure)
|
|
self.axes = self.figure.add_subplot(111)
|
|
|
|
self.axes.set_xlabel(xlabel)
|
|
self.axes.set_ylabel(ylabel)
|
|
self.axes.set_title(title)
|
|
|
|
|
|
class PickDlg(QDialog):
|
|
|
|
def __init__(self, station=None, parent=None):
|
|
super(PickDlg, self).__init__(parent)
|
|
|
|
pass
|
|
|
|
|
|
class PropertiesDlg(QDialog):
|
|
|
|
def __init__(self, parent=None):
|
|
super(PropertiesDlg, self).__init__(parent)
|
|
|
|
self.tabBar = QTabBar()
|
|
self.buttonBox = QDialogButtonBox(QDialogButtonBox.Apply |
|
|
QDialogButtonBox.Close)
|
|
|
|
self.connect(self.buttonBox.button(QDialogButtonBox.Apply),
|
|
SIGNAL("clicked()"), self.apply)
|
|
self.connect(self.buttonBox, SIGNAL("rejected()"),
|
|
self, SLOT("reject()"))
|
|
pass
|
|
|
|
def apply(self):
|
|
pass
|
|
|
|
|
|
class FilterOptionsDock(QDialog):
|
|
|
|
def __init__(self, parent=None, titleString="Filter options",
|
|
filterOptions=None):
|
|
super(FilterOptionsDock, self).__init__()
|
|
|
|
if filterOptions and not isinstance(filterOptions, FilterOptions):
|
|
try:
|
|
fOptions = FilterOptions(**filterOptions)
|
|
filterOptions = fOptions
|
|
except Exception, e:
|
|
raise OptionsError('%s' % e)
|
|
else:
|
|
filterOptions = FilterOptions()
|
|
|
|
self.filterOptions = filterOptions
|
|
|
|
self.freqminLabel = QLabel()
|
|
self.freqminLabel.setText("minimum:")
|
|
self.freqminSpinBox = QDoubleSpinBox()
|
|
self.freqminSpinBox.setRange(5e-7, 1e6)
|
|
self.freqminSpinBox.setDecimals(2)
|
|
self.freqminSpinBox.setSuffix(' Hz')
|
|
self.freqminSpinBox.setValue(filterOptions.freq[0])
|
|
self.freqmaxLabel = QLabel()
|
|
self.freqmaxLabel.setText("maximum:")
|
|
self.freqmaxSpinBox = QDoubleSpinBox()
|
|
self.freqmaxSpinBox.setRange(5e-7, 1e6)
|
|
self.freqmaxSpinBox.setDecimals(2)
|
|
self.freqmaxSpinBox.setSuffix(' Hz')
|
|
if self.filterOptions.filterType in ['bandpass', 'bandstop']:
|
|
self.freqmaxSpinBox.setValue(self.filterOptions.freq[1])
|
|
|
|
typeOptions = ["bandpass", "bandstop", "lowpass", "highpass"]
|
|
|
|
self.orderLabel = QLabel()
|
|
self.orderLabel.setText("Order:")
|
|
self.orderSpinBox = QSpinBox()
|
|
self.orderSpinBox.setRange(2, 10)
|
|
self.selectTypeLabel = QLabel()
|
|
self.selectTypeLabel.setText("Select filter type:")
|
|
self.selectTypeCombo = QComboBox()
|
|
self.selectTypeCombo.addItems(typeOptions)
|
|
self.selectTypeLayout = QVBoxLayout()
|
|
self.selectTypeLayout.addWidget(self.orderLabel)
|
|
self.selectTypeLayout.addWidget(self.orderSpinBox)
|
|
self.selectTypeLayout.addWidget(self.selectTypeLabel)
|
|
self.selectTypeLayout.addWidget(self.selectTypeCombo)
|
|
|
|
self.freqGroupBox = QGroupBox("Frequency range")
|
|
self.freqGroupLayout = QGridLayout()
|
|
self.freqGroupLayout.addWidget(self.freqminLabel, 0, 0)
|
|
self.freqGroupLayout.addWidget(self.freqminSpinBox, 0, 1)
|
|
self.freqGroupLayout.addWidget(self.freqmaxLabel, 1, 0)
|
|
self.freqGroupLayout.addWidget(self.freqmaxSpinBox, 1, 1)
|
|
self.freqGroupBox.setLayout(self.freqGroupLayout)
|
|
|
|
self.layoutEditables = QHBoxLayout()
|
|
self.layoutEditables.addWidget(self.freqGroupBox)
|
|
self.layoutEditables.addLayout(self.selectTypeLayout)
|
|
|
|
self.setLayout(self.layoutEditables)
|
|
|
|
self.connect(self.freqminSpinBox, SIGNAL("valueChanged(double)"),
|
|
self.updateUi)
|
|
self.connect(self.freqmaxSpinBox, SIGNAL("valueChanged(double)"),
|
|
self.updateUi)
|
|
self.connect(self.orderSpinBox, SIGNAL("valueChanged(int)"),
|
|
self.updateUi)
|
|
self.connect(self.selectTypeCombo, SIGNAL("currentIndexChanged(int)"),
|
|
self.updateUi)
|
|
self.updateUi()
|
|
|
|
def updateUi(self):
|
|
if self.selectTypeCombo.currentText() not in ['bandpass', 'bandstop']:
|
|
self.freqminLabel.setText("cutoff:")
|
|
self.freqmaxLabel.setEnabled(False)
|
|
self.freqmaxSpinBox.setEnabled(False)
|
|
self.freqmaxSpinBox.setValue(self.freqminSpinBox.value())
|
|
else:
|
|
self.freqminLabel.setText("minimum:")
|
|
self.freqmaxLabel.setEnabled(True)
|
|
self.freqmaxSpinBox.setEnabled(True)
|
|
|
|
self.filterOptions.filterType = self.selectTypeCombo.currentText()
|
|
freq = []
|
|
freq.append(self.freqminSpinBox.value())
|
|
if self.filterOptions.filterType in ['bandpass', 'bandstop']:
|
|
if self.freqminSpinBox.value() > self.freqmaxSpinBox.value():
|
|
QMessageBox.warning(self, "Value error",
|
|
"Maximum frequency must be at least the "
|
|
"same value as minimum frequency (notch)!")
|
|
self.freqmaxSpinBox.setValue(self.freqminSpinBox.value())
|
|
self.freqmaxSpinBox.selectAll()
|
|
self.freqmaxSpinBox.setFocus()
|
|
return
|
|
freq.append(self.freqmaxSpinBox.value())
|
|
self.filterOptions.freq = freq
|
|
self.filterOptions.order = self.orderSpinBox.value()
|
|
return self.filterOptions
|
|
|
|
|
|
class LoadDataDlg(QDialog):
|
|
|
|
def __init__(self, parent=None):
|
|
super(LoadDataDlg, self).__init__(parent)
|
|
|
|
pass
|
|
|
|
|
|
class HelpForm(QDialog):
|
|
|
|
def __init__(self, page='https://ariadne.geophysik.rub.de/trac/PyLoT/wiki',
|
|
parent=None):
|
|
super(HelpForm, self).__init__(parent)
|
|
self.setAttribute(Qt.WA_DeleteOnClose)
|
|
self.setAttribute(Qt.WA_GroupLeader)
|
|
|
|
backAction = QAction(QIcon(":/back.png"), "&Back", self)
|
|
backAction.setShortcut(QKeySequence.Back)
|
|
homeAction = QAction(QIcon(":/home.png"), "&Home", self)
|
|
homeAction.setShortcut("Home")
|
|
self.pageLabel = QLabel()
|
|
|
|
toolBar = QToolBar()
|
|
toolBar.addAction(backAction)
|
|
toolBar.addAction(homeAction)
|
|
toolBar.addWidget(self.pageLabel)
|
|
self.textBrowser = QTextBrowser()
|
|
|
|
layout = QVBoxLayout()
|
|
layout.addWidget(toolBar)
|
|
layout.addWidget(self.textBrowser, 1)
|
|
self.setLayout(layout)
|
|
|
|
self.connect(backAction, SIGNAL("triggered()"),
|
|
self.textBrowser, SLOT("backward()"))
|
|
self.connect(homeAction, SIGNAL("triggered()"),
|
|
self.textBrowser, SLOT("home()"))
|
|
self.connect(self.textBrowser, SIGNAL("sourceChanged(QUrl)"),
|
|
self.updatePageTitle)
|
|
|
|
self.textBrowser.setSearchPaths([":/help"])
|
|
self.textBrowser.setSource(QUrl(page))
|
|
self.resize(400, 600)
|
|
self.setWindowTitle("{0} Help".format(QApplication.applicationName()))
|
|
|
|
def updatePageTitle(self):
|
|
self.pageLabel.setText(self.textBrowser.documentTitle())
|