class HydroprintGUI(myqt.DialogWindow):
ConsoleSignal = QSignal(str)
def __init__(self, datamanager, parent=None):
super(HydroprintGUI, self).__init__(parent, maximize=True)
self.__updateUI = True
# Child widgets:
self.dmngr = datamanager
self.dmngr.wldsetChanged.connect(self.wldset_changed)
self.dmngr.wxdsetChanged.connect(self.wxdset_changed)
self.page_setup_win = PageSetupWin(self)
self.page_setup_win.newPageSetupSent.connect(self.layout_changed)
self.color_palette_win = ColorsSetupWin(self)
self.color_palette_win.newColorSetupSent.connect(self.update_colors)
# Memory path variable:
self.save_fig_dir = self.workdir
# Generate UI:
self.__initUI__()
def __initUI__(self):
# ---- Toolbar
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save the well hydrograph')
# btn_draw is usefull for debugging purposes
btn_draw = QToolButtonNormal(icons.get_icon('refresh'))
btn_draw.setToolTip('Force a refresh of the well hydrograph')
btn_draw.hide()
self.btn_load_layout = QToolButtonNormal(
icons.get_icon('load_graph_config'))
self.btn_load_layout.setToolTip(
"<p>Load graph layout for the current water level "
" datafile if it exists</p>")
self.btn_load_layout.clicked.connect(self.load_layout_isClicked)
self.btn_save_layout = QToolButtonNormal(
icons.get_icon('save_graph_config'))
self.btn_save_layout.setToolTip('Save current graph layout')
self.btn_save_layout.clicked.connect(self.save_layout_isClicked)
btn_bestfit_waterlvl = QToolButtonNormal(icons.get_icon('fit_y'))
btn_bestfit_waterlvl.setToolTip('Best fit the water level scale')
btn_bestfit_time = QToolButtonNormal(icons.get_icon('fit_x'))
btn_bestfit_time.setToolTip('Best fit the time scale')
self.btn_page_setup = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_page_setup.setToolTip('Show the page setup window')
self.btn_page_setup.clicked.connect(self.page_setup_win.show)
btn_color_pick = QToolButtonNormal(icons.get_icon('color_picker'))
btn_color_pick.setToolTip('<p>Show a window to setup the color palette'
' used to draw the hydrograph</p.')
btn_color_pick.clicked.connect(self.color_palette_win.show)
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.layout_changed)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Zoom Panel
btn_zoom_out = QToolButtonSmall(icons.get_icon('zoom_out'))
btn_zoom_out.setToolTip('Zoom out (ctrl + mouse-wheel-down)')
btn_zoom_out.clicked.connect(self.zoom_out)
btn_zoom_in = QToolButtonSmall(icons.get_icon('zoom_in'))
btn_zoom_in.setToolTip('Zoom in (ctrl + mouse-wheel-up)')
btn_zoom_in.clicked.connect(self.zoom_in)
self.zoom_disp = QSpinBox()
self.zoom_disp.setAlignment(Qt.AlignCenter)
self.zoom_disp.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.zoom_disp.setReadOnly(True)
self.zoom_disp.setSuffix(' %')
self.zoom_disp.setRange(0, 9999)
self.zoom_disp.setValue(100)
zoom_pan = myqt.QFrameLayout()
zoom_pan.setSpacing(3)
zoom_pan.addWidget(btn_zoom_out, 0, 0)
zoom_pan.addWidget(btn_zoom_in, 0, 1)
zoom_pan.addWidget(self.zoom_disp, 0, 2)
# LAYOUT :
btn_list = [btn_save, btn_draw,
self.btn_load_layout, self.btn_save_layout, VSep(),
btn_bestfit_waterlvl, btn_bestfit_time, VSep(),
self.btn_page_setup, btn_color_pick, self.btn_language,
VSep(), zoom_pan]
subgrid_toolbar = QGridLayout()
toolbar_widget = QWidget()
row = 0
for col, btn in enumerate(btn_list):
subgrid_toolbar.addWidget(btn, row, col)
subgrid_toolbar.setSpacing(5)
subgrid_toolbar.setContentsMargins(0, 0, 0, 0)
subgrid_toolbar.setColumnStretch(col + 1, 100)
toolbar_widget.setLayout(subgrid_toolbar)
# ---- LEFT PANEL
# SubGrid Hydrograph Frame :
self.hydrograph = hydrograph.Hydrograph()
self.hydrograph_scrollarea = mplFigViewer.ImageViewer()
self.hydrograph_scrollarea.zoomChanged.connect(self.zoom_disp.setValue)
grid_hydrograph = QGridLayout()
grid_hydrograph.addWidget(self.hydrograph_scrollarea, 0, 0)
grid_hydrograph.setRowStretch(0, 500)
grid_hydrograph.setColumnStretch(0, 500)
grid_hydrograph.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
# ASSEMBLING SubGrids :
grid_layout = QGridLayout()
self.grid_layout_widget = QFrame()
row = 0
grid_layout.addWidget(toolbar_widget, row, 0)
row += 1
grid_layout.addLayout(grid_hydrograph, row, 0)
grid_layout.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
grid_layout.setSpacing(5)
grid_layout.setColumnStretch(0, 500)
grid_layout.setRowStretch(1, 500)
self.grid_layout_widget.setLayout(grid_layout)
# ---- Right Panel
self.tabscales = self.__init_scalesTabWidget__()
self.right_panel = myqt.QFrameLayout()
self.right_panel.addWidget(self.dmngr, 0, 0)
self.right_panel.addWidget(self.tabscales, 1, 0)
self.right_panel.setRowStretch(2, 100)
self.right_panel.setSpacing(15)
# ---- MAIN GRID
mainGrid = QGridLayout()
mainGrid.addWidget(self.grid_layout_widget, 0, 0)
mainGrid.addWidget(VSep(), 0, 1)
mainGrid.addWidget(self.right_panel, 0, 2)
mainGrid.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
mainGrid.setSpacing(15)
mainGrid.setColumnStretch(0, 500)
mainGrid.setColumnMinimumWidth(2, 250)
self.setLayout(mainGrid)
# ---- EVENTS
# Toolbox Layout :
btn_bestfit_waterlvl.clicked.connect(self.best_fit_waterlvl)
btn_bestfit_time.clicked.connect(self.best_fit_time)
btn_draw.clicked.connect(self.draw_hydrograph)
btn_save.clicked.connect(self.select_save_path)
# Hydrograph Layout :
self.Ptot_scale.valueChanged.connect(self.layout_changed)
self.qweather_bin.currentIndexChanged.connect(self.layout_changed)
# ---- Init Image
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def __init_scalesTabWidget__(self):
class QRowLayout(QGridLayout):
def __init__(self, items, parent=None):
super(QRowLayout, self).__init__(parent)
for col, item in enumerate(items):
self.addWidget(item, 0, col)
self.setContentsMargins(0, 0, 0, 0)
self.setColumnStretch(0, 100)
# ---- Time axis properties
# Generate the widgets :
self.date_start_widget = QDateEdit()
self.date_start_widget.setDisplayFormat('01 / MM / yyyy')
self.date_start_widget.setAlignment(Qt.AlignCenter)
self.date_start_widget.dateChanged.connect(self.layout_changed)
self.date_end_widget = QDateEdit()
self.date_end_widget.setDisplayFormat('01 / MM / yyyy')
self.date_end_widget.setAlignment(Qt.AlignCenter)
self.date_end_widget.dateChanged.connect(self.layout_changed)
self.time_scale_label = QComboBox()
self.time_scale_label.setEditable(False)
self.time_scale_label.setInsertPolicy(QComboBox.NoInsert)
self.time_scale_label.addItems(['Month', 'Year'])
self.time_scale_label.setCurrentIndex(0)
self.time_scale_label.currentIndexChanged.connect(self.layout_changed)
self.dateDispFreq_spinBox = QSpinBox()
self.dateDispFreq_spinBox.setSingleStep(1)
self.dateDispFreq_spinBox.setMinimum(1)
self.dateDispFreq_spinBox.setMaximum(100)
self.dateDispFreq_spinBox.setValue(
self.hydrograph.date_labels_pattern)
self.dateDispFreq_spinBox.setAlignment(Qt.AlignCenter)
self.dateDispFreq_spinBox.setKeyboardTracking(False)
self.dateDispFreq_spinBox.valueChanged.connect(self.layout_changed)
# Setting up the layout :
widget_time_scale = QFrame()
widget_time_scale.setFrameStyle(0)
grid_time_scale = QGridLayout()
GRID = [[QLabel('From :'), self.date_start_widget],
[QLabel('To :'), self.date_end_widget],
[QLabel('Scale :'), self.time_scale_label],
[QLabel('Date Disp. Pattern:'),
self.dateDispFreq_spinBox]]
for i, ROW in enumerate(GRID):
grid_time_scale.addLayout(QRowLayout(ROW), i, 1)
grid_time_scale.setVerticalSpacing(5)
grid_time_scale.setContentsMargins(10, 10, 10, 10)
widget_time_scale.setLayout(grid_time_scale)
# ----- Water level axis properties
# Widget :
self.waterlvl_scale = QDoubleSpinBox()
self.waterlvl_scale.setSingleStep(0.05)
self.waterlvl_scale.setMinimum(0.05)
self.waterlvl_scale.setSuffix(' m')
self.waterlvl_scale.setAlignment(Qt.AlignCenter)
self.waterlvl_scale.setKeyboardTracking(False)
self.waterlvl_scale.valueChanged.connect(self.layout_changed)
self.waterlvl_scale.setFixedWidth(100)
self.waterlvl_max = QDoubleSpinBox()
self.waterlvl_max.setSingleStep(0.1)
self.waterlvl_max.setSuffix(' m')
self.waterlvl_max.setAlignment(Qt.AlignCenter)
self.waterlvl_max.setMinimum(-1000)
self.waterlvl_max.setMaximum(1000)
self.waterlvl_max.setKeyboardTracking(False)
self.waterlvl_max.valueChanged.connect(self.layout_changed)
self.waterlvl_max.setFixedWidth(100)
self.NZGridWL_spinBox = QSpinBox()
self.NZGridWL_spinBox.setSingleStep(1)
self.NZGridWL_spinBox.setMinimum(1)
self.NZGridWL_spinBox.setMaximum(50)
self.NZGridWL_spinBox.setValue(self.hydrograph.NZGrid)
self.NZGridWL_spinBox.setAlignment(Qt.AlignCenter)
self.NZGridWL_spinBox.setKeyboardTracking(False)
self.NZGridWL_spinBox.valueChanged.connect(self.layout_changed)
self.NZGridWL_spinBox.setFixedWidth(100)
self.datum_widget = QComboBox()
self.datum_widget.addItems(['Ground Surface', 'Sea Level'])
self.datum_widget.currentIndexChanged.connect(self.layout_changed)
# Layout :
subgrid_WLScale = QGridLayout()
GRID = [[QLabel('Minimum :'), self.waterlvl_max],
[QLabel('Scale :'), self.waterlvl_scale],
[QLabel('Grid Divisions :'), self.NZGridWL_spinBox],
[QLabel('Datum :'), self.datum_widget]]
for i, ROW in enumerate(GRID):
subgrid_WLScale.addLayout(QRowLayout(ROW), i, 1)
subgrid_WLScale.setVerticalSpacing(5)
subgrid_WLScale.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
WLScale_widget = QFrame()
WLScale_widget.setFrameStyle(0)
WLScale_widget.setLayout(subgrid_WLScale)
# ---- Weather Axis
# Widgets :
self.Ptot_scale = QSpinBox()
self.Ptot_scale.setSingleStep(5)
self.Ptot_scale.setMinimum(5)
self.Ptot_scale.setMaximum(500)
self.Ptot_scale.setValue(20)
self.Ptot_scale.setSuffix(' mm')
self.Ptot_scale.setAlignment(Qt.AlignCenter)
self.qweather_bin = QComboBox()
self.qweather_bin.setEditable(False)
self.qweather_bin.setInsertPolicy(QComboBox.NoInsert)
self.qweather_bin.addItems(['day', 'week', 'month'])
self.qweather_bin.setCurrentIndex(1)
# Layout :
layout = QGridLayout()
GRID = [[QLabel('Precip. Scale :'), self.Ptot_scale],
[QLabel('Resampling :'), self.qweather_bin]]
for i, row in enumerate(GRID):
layout.addLayout(QRowLayout(row), i, 1)
layout.setVerticalSpacing(5)
layout.setContentsMargins(10, 10, 10, 10) # (L,T,R,B)
layout.setRowStretch(i+1, 100)
widget_weather_scale = QFrame()
widget_weather_scale.setFrameStyle(0)
widget_weather_scale.setLayout(layout)
# ---- ASSEMBLING TABS
tabscales = QTabWidget()
tabscales.addTab(widget_time_scale, 'Time')
tabscales.addTab(WLScale_widget, 'Water Level')
tabscales.addTab(widget_weather_scale, 'Weather')
return tabscales
@property
def workdir(self):
return self.dmngr.workdir
# ---- Utilities
def zoom_in(self):
self.hydrograph_scrollarea.zoomIn()
def zoom_out(self):
self.hydrograph_scrollarea.zoomOut()
def update_colors(self):
self.hydrograph.update_colors()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
# ---- Datasets Handlers
@property
def wldset(self):
return self.dmngr.get_current_wldset()
@property
def wxdset(self):
return self.dmngr.get_current_wxdset()
def wldset_changed(self):
"""Handle when the water level dataset of the datamanager changes."""
if self.wldset is None:
self.clear_hydrograph()
return
else:
self.hydrograph.set_wldset(self.wldset)
self.hydrograph.gluedf = self.wldset.get_glue_at(-1)
# Load the manual measurements.
fname = os.path.join(
self.workdir, "Water Levels", 'waterlvl_manual_measurements')
tmeas, wlmeas = load_waterlvl_measures(fname, self.wldset['Well'])
self.wldset.set_wlmeas(tmeas, wlmeas)
# Setup the layout of the hydrograph.
layout = self.wldset.get_layout()
if layout is not None:
msg = ("Loading existing graph layout for well %s." %
self.wldset['Well'])
print(msg)
self.ConsoleSignal.emit('<font color=black>%s</font>' % msg)
self.load_graph_layout(layout)
else:
print('No graph layout exists for well %s.' % self.wldset['Well'])
# Fit Water Level in Layout :
self.__updateUI = False
self.best_fit_waterlvl()
self.best_fit_time()
if self.dmngr.set_closest_wxdset() is None:
self.draw_hydrograph()
self.__updateUI = True
def wxdset_changed(self):
"""Handle when the weather dataset of the datamanager changes."""
if self.wldset is None:
self.clear_hydrograph()
else:
self.hydrograph.set_wxdset(self.wxdset)
QCoreApplication.processEvents()
self.draw_hydrograph()
# ---- Draw Hydrograph Handlers
def best_fit_waterlvl(self):
wldset = self.dmngr.get_current_wldset()
if wldset is not None:
WLscale, WLmin = self.hydrograph.best_fit_waterlvl()
self.waterlvl_scale.setValue(WLscale)
self.waterlvl_max.setValue(WLmin)
def best_fit_time(self):
wldset = self.dmngr.get_current_wldset()
if wldset is not None:
date0, date1 = self.hydrograph.best_fit_time(wldset.xldates)
self.date_start_widget.setDate(QDate(date0[0], date0[1], date0[2]))
self.date_end_widget.setDate(QDate(date1[0], date1[1], date1[2]))
@QSlot()
def mrc_wl_changed(self):
"""
Force a redraw of the MRC water levels after the results have
changed for the dataset.
"""
self.hydrograph.draw_mrc_wl()
self.hydrograph.setup_legend()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
@QSlot(GLUEDataFrameBase)
def glue_wl_changed(self, gluedf):
"""
Force a redraw of the GLUE water levels after the results have
changed for the dataset.
"""
self.hydrograph.set_gluedf(gluedf)
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def layout_changed(self):
"""
When an element of the graph layout is changed in the UI.
"""
if self.__updateUI is False:
return
self.update_graph_layout_parameter()
if self.hydrograph.isHydrographExists is False:
return
sender = self.sender()
if sender == self.btn_language:
self.hydrograph.draw_ylabels()
self.hydrograph.setup_xticklabels()
self.hydrograph.setup_legend()
elif sender in [self.waterlvl_max, self.waterlvl_scale]:
self.hydrograph.setup_waterlvl_scale()
self.hydrograph.draw_ylabels()
elif sender == self.NZGridWL_spinBox:
self.hydrograph.setup_waterlvl_scale()
self.hydrograph.update_precip_scale()
self.hydrograph.draw_ylabels()
elif sender == self.Ptot_scale:
self.hydrograph.update_precip_scale()
self.hydrograph.draw_ylabels()
elif sender == self.datum_widget:
yoffset = int(self.wldset['Elevation']/self.hydrograph.WLscale)
yoffset *= self.hydrograph.WLscale
self.hydrograph.WLmin = (yoffset - self.hydrograph.WLmin)
self.waterlvl_max.blockSignals(True)
self.waterlvl_max.setValue(self.hydrograph.WLmin)
self.waterlvl_max.blockSignals(False)
# This is calculated so that trailing zeros in the altitude of the
# well is not carried to the y axis labels, so that they remain a
# int multiple of *WLscale*.
self.hydrograph.setup_waterlvl_scale()
self.hydrograph.draw_waterlvl()
self.hydrograph.draw_ylabels()
elif sender in [self.date_start_widget, self.date_end_widget]:
self.hydrograph.set_time_scale()
self.hydrograph.draw_weather()
self.hydrograph.draw_figure_title()
elif sender == self.dateDispFreq_spinBox:
self.hydrograph.set_time_scale()
self.hydrograph.setup_xticklabels()
elif sender == self.page_setup_win:
self.hydrograph.update_fig_size()
# Implicitly call : set_margins()
# draw_ylabels()
# set_time_scale()
# draw_figure_title
self.hydrograph.draw_waterlvl()
self.hydrograph.setup_legend()
elif sender == self.qweather_bin:
self.hydrograph.resample_bin()
self.hydrograph.draw_weather()
self.hydrograph.draw_ylabels()
elif sender == self.time_scale_label:
self.hydrograph.set_time_scale()
self.hydrograph.draw_weather()
else:
print('No action for this widget yet.')
# !!!! temporary fix until I can find a better solution !!!!
# sender.blockSignals(True)
if type(sender) in [QDoubleSpinBox, QSpinBox]:
sender.setReadOnly(True)
for i in range(10):
QCoreApplication.processEvents()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
for i in range(10):
QCoreApplication.processEvents()
if type(sender) in [QDoubleSpinBox, QSpinBox]:
sender.setReadOnly(False)
# sender.blockSignals(False)
def update_graph_layout_parameter(self):
# language :
self.hydrograph.language = self.btn_language.language
# Scales :
self.hydrograph.WLmin = self.waterlvl_max.value()
self.hydrograph.WLscale = self.waterlvl_scale.value()
self.hydrograph.RAINscale = self.Ptot_scale.value()
self.hydrograph.NZGrid = self.NZGridWL_spinBox.value()
# WL Datum :
self.hydrograph.WLdatum = self.datum_widget.currentIndex()
# Dates :
self.hydrograph.datemode = self.time_scale_label.currentText()
year = self.date_start_widget.date().year()
month = self.date_start_widget.date().month()
self.hydrograph.TIMEmin = xldate_from_date_tuple((year, month, 1), 0)
year = self.date_end_widget.date().year()
month = self.date_end_widget.date().month()
self.hydrograph.TIMEmax = xldate_from_date_tuple((year, month, 1), 0)
self.hydrograph.date_labels_pattern = self.dateDispFreq_spinBox.value()
# Page Setup :
self.hydrograph.fwidth = self.page_setup_win.pageSize[0]
self.hydrograph.fheight = self.page_setup_win.pageSize[1]
self.hydrograph.va_ratio = self.page_setup_win.va_ratio
self.hydrograph.trend_line = self.page_setup_win.isTrendLine
self.hydrograph.isLegend = self.page_setup_win.isLegend
self.hydrograph.isGraphTitle = self.page_setup_win.isGraphTitle
self.hydrograph.set_meteo_on(self.page_setup_win.is_meteo_on)
self.hydrograph.set_glue_wl_on(self.page_setup_win.is_glue_wl_on)
self.hydrograph.set_mrc_wl_on(self.page_setup_win.is_mrc_wl_on)
self.hydrograph.set_figframe_lw(self.page_setup_win.figframe_lw)
# Weather bins :
self.hydrograph.bwidth_indx = self.qweather_bin.currentIndex()
def clear_hydrograph(self):
"""Clear the hydrograph figure to show only a blank canvas."""
self.hydrograph.clf()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def draw_hydrograph(self):
if self.dmngr.wldataset_count() == 0:
msg = 'Please import a valid water level data file first.'
self.ConsoleSignal.emit('<font color=red>%s</font>' % msg)
self.emit_warning(msg)
return
self.update_graph_layout_parameter()
# Generate and Display Graph :
for i in range(5):
QCoreApplication.processEvents()
QApplication.setOverrideCursor(Qt.WaitCursor)
self.hydrograph.set_wldset(self.dmngr.get_current_wldset())
self.hydrograph.set_wxdset(self.dmngr.get_current_wxdset())
self.hydrograph.generate_hydrograph()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
QApplication.restoreOverrideCursor()
def select_save_path(self):
"""
Open a dialog where the user can select a file name to save the
hydrograph.
"""
if self.wldset is None:
return
ffmat = "*.pdf;;*.svg;;*.png"
fname = find_unique_filename(osp.join(
self.save_fig_dir, 'hydrograph_%s.pdf' % self.wldset['Well']))
fname, ftype = QFileDialog.getSaveFileName(
self, "Save Figure", fname, ffmat)
if fname:
ftype = ftype.replace('*', '')
fname = fname if fname.endswith(ftype) else fname + ftype
self.save_fig_dir = os.path.dirname(fname)
try:
self.save_figure(fname)
except PermissionError:
msg = "The file is in use by another application or user."
QMessageBox.warning(self, 'Warning', msg, QMessageBox.Ok)
self.select_save_path()
def save_figure(self, fname):
"""Save the hydrograph figure in a file."""
self.hydrograph.generate_hydrograph()
self.hydrograph.savefig(fname)
# ---- Graph Layout Handlers
def load_layout_isClicked(self):
"""Handle when the button to load the graph layout is clicked."""
if self.wldset is None:
self.emit_warning(
"Please import a valid water level data file first.")
return
layout = self.wldset.get_layout()
if layout is None:
self.emit_warning(
"No graph layout exists for well %s." % self.wldset['Well'])
else:
self.load_graph_layout(layout)
def load_graph_layout(self, layout):
"""Load the graph layout into the GUI."""
self.__updateUI = False
# Scales :
date = layout['TIMEmin']
date = xldate_as_tuple(date, 0)
self.date_start_widget.setDate(QDate(date[0], date[1], date[2]))
date = layout['TIMEmax']
date = xldate_as_tuple(date, 0)
self.date_end_widget.setDate(QDate(date[0], date[1], date[2]))
self.dateDispFreq_spinBox.setValue(layout['date_labels_pattern'])
self.waterlvl_scale.setValue(layout['WLscale'])
self.waterlvl_max.setValue(layout['WLmin'])
self.NZGridWL_spinBox.setValue(layout['NZGrid'])
self.Ptot_scale.setValue(layout['RAINscale'])
x = ['mbgs', 'masl'].index(layout['WLdatum'])
self.datum_widget.setCurrentIndex(x)
self.qweather_bin.setCurrentIndex(layout['bwidth_indx'])
self.time_scale_label.setCurrentIndex(
self.time_scale_label.findText(layout['datemode']))
# ---- Language and colors
self.btn_language.set_language(layout['language'])
self.color_palette_win.load_colors()
# ---- Page Setup
self.page_setup_win.pageSize = (layout['fwidth'], layout['fheight'])
self.page_setup_win.va_ratio = layout['va_ratio']
self.page_setup_win.isLegend = layout['legend_on']
self.page_setup_win.isGraphTitle = layout['title_on']
self.page_setup_win.isTrendLine = layout['trend_line']
self.page_setup_win.is_meteo_on = layout['meteo_on']
self.page_setup_win.is_glue_wl_on = layout['glue_wl_on']
self.page_setup_win.is_mrc_wl_on = layout['mrc_wl_on']
self.page_setup_win.figframe_lw = layout['figframe_lw']
self.page_setup_win.legend_on.set_value(layout['legend_on'])
self.page_setup_win.title_on.set_value(layout['title_on'])
self.page_setup_win.wltrend_on.set_value(layout['trend_line'])
self.page_setup_win.meteo_on.set_value(layout['meteo_on'])
self.page_setup_win.glue_wl_on.set_value(layout['glue_wl_on'])
self.page_setup_win.mrc_wl_on.set_value(layout['mrc_wl_on'])
self.page_setup_win.fframe_lw_widg.setValue(layout['figframe_lw'])
self.page_setup_win.fwidth.setValue(layout['fwidth'])
self.page_setup_win.fheight.setValue(layout['fheight'])
self.page_setup_win.va_ratio_spinBox.setValue(layout['va_ratio'])
# Check if Weather Dataset :
if layout['wxdset'] in self.dmngr.wxdsets:
self.dmngr.set_current_wxdset(layout['wxdset'])
else:
if self.dmngr.set_closest_wxdset() is None:
self.draw_hydrograph()
self.__updateUI = True
def save_layout_isClicked(self):
wldset = self.wldset
if wldset is None:
self.emit_warning(
"Please import a valid water level data file first.")
return
layout = wldset.get_layout()
if layout is not None:
msg = ('A graph layout already exists for well %s.Do you want to'
' you want to replace it?') % wldset['Well']
reply = QMessageBox.question(self, 'Save Graph Layout', msg,
QMessageBox.Yes | QMessageBox.No)
if reply == QMessageBox.Yes:
self.save_graph_layout()
elif reply == QMessageBox.No:
msg = "Graph layout not saved for well %s." % wldset['Well']
self.ConsoleSignal.emit('<font color=black>%s' % msg)
else:
self.save_graph_layout()
def save_graph_layout(self):
"""Save the graph layout in the project hdf5 file."""
print("Saving the graph layout for well %s..." % self.wldset['Well'],
end=" ")
layout = {'WLmin': self.waterlvl_max.value(),
'WLscale': self.waterlvl_scale.value(),
'RAINscale': self.Ptot_scale.value(),
'fwidth': self.page_setup_win.pageSize[0],
'fheight': self.page_setup_win.pageSize[1],
'va_ratio': self.page_setup_win.va_ratio,
'NZGrid': self.NZGridWL_spinBox.value(),
'bwidth_indx': self.qweather_bin.currentIndex(),
'date_labels_pattern': self.dateDispFreq_spinBox.value(),
'datemode': self.time_scale_label.currentText()}
layout['wxdset'] = None if self.wxdset is None else self.wxdset.name
year = self.date_start_widget.date().year()
month = self.date_start_widget.date().month()
layout['TIMEmin'] = xldate_from_date_tuple((year, month, 1), 0)
year = self.date_end_widget.date().year()
month = self.date_end_widget.date().month()
layout['TIMEmax'] = xldate_from_date_tuple((year, month, 1), 0)
if self.datum_widget.currentIndex() == 0:
layout['WLdatum'] = 'mbgs'
else:
layout['WLdatum'] = 'masl'
# ---- Page Setup
layout['title_on'] = bool(self.page_setup_win.isGraphTitle)
layout['legend_on'] = bool(self.page_setup_win.isLegend)
layout['language'] = self.btn_language.language
layout['trend_line'] = bool(self.page_setup_win.isTrendLine)
layout['meteo_on'] = bool(self.page_setup_win.is_meteo_on)
layout['glue_wl_on'] = bool(self.page_setup_win.is_glue_wl_on)
layout['mrc_wl_on'] = bool(self.page_setup_win.is_mrc_wl_on)
layout['figframe_lw'] = self.page_setup_win.figframe_lw
# Save the colors :
cdb = ColorsReader()
cdb.load_colors_db()
layout['colors'] = cdb.RGB
# Save the layout :
self.wldset.save_layout(layout)
msg = 'Layout saved successfully for well %s.' % self.wldset['Well']
self.ConsoleSignal.emit('<font color=black>%s</font>' % msg)
print("done")
def __initUI__(self):
# ---- Toolbar
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save the well hydrograph')
# btn_draw is usefull for debugging purposes
btn_draw = QToolButtonNormal(icons.get_icon('refresh'))
btn_draw.setToolTip('Force a refresh of the well hydrograph')
btn_draw.hide()
self.btn_load_layout = QToolButtonNormal(
icons.get_icon('load_graph_config'))
self.btn_load_layout.setToolTip(
"<p>Load graph layout for the current water level "
" datafile if it exists</p>")
self.btn_load_layout.clicked.connect(self.load_layout_isClicked)
self.btn_save_layout = QToolButtonNormal(
icons.get_icon('save_graph_config'))
self.btn_save_layout.setToolTip('Save current graph layout')
self.btn_save_layout.clicked.connect(self.save_layout_isClicked)
btn_bestfit_waterlvl = QToolButtonNormal(icons.get_icon('fit_y'))
btn_bestfit_waterlvl.setToolTip('Best fit the water level scale')
btn_bestfit_time = QToolButtonNormal(icons.get_icon('fit_x'))
btn_bestfit_time.setToolTip('Best fit the time scale')
self.btn_page_setup = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_page_setup.setToolTip('Show the page setup window')
self.btn_page_setup.clicked.connect(self.page_setup_win.show)
btn_color_pick = QToolButtonNormal(icons.get_icon('color_picker'))
btn_color_pick.setToolTip('<p>Show a window to setup the color palette'
' used to draw the hydrograph</p.')
btn_color_pick.clicked.connect(self.color_palette_win.show)
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.layout_changed)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Zoom Panel
btn_zoom_out = QToolButtonSmall(icons.get_icon('zoom_out'))
btn_zoom_out.setToolTip('Zoom out (ctrl + mouse-wheel-down)')
btn_zoom_out.clicked.connect(self.zoom_out)
btn_zoom_in = QToolButtonSmall(icons.get_icon('zoom_in'))
btn_zoom_in.setToolTip('Zoom in (ctrl + mouse-wheel-up)')
btn_zoom_in.clicked.connect(self.zoom_in)
self.zoom_disp = QSpinBox()
self.zoom_disp.setAlignment(Qt.AlignCenter)
self.zoom_disp.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.zoom_disp.setReadOnly(True)
self.zoom_disp.setSuffix(' %')
self.zoom_disp.setRange(0, 9999)
self.zoom_disp.setValue(100)
zoom_pan = myqt.QFrameLayout()
zoom_pan.setSpacing(3)
zoom_pan.addWidget(btn_zoom_out, 0, 0)
zoom_pan.addWidget(btn_zoom_in, 0, 1)
zoom_pan.addWidget(self.zoom_disp, 0, 2)
# LAYOUT :
btn_list = [
btn_save, btn_draw, self.btn_load_layout, self.btn_save_layout,
VSep(), btn_bestfit_waterlvl, btn_bestfit_time,
VSep(), self.btn_page_setup, btn_color_pick, self.btn_language,
VSep(), zoom_pan
]
subgrid_toolbar = QGridLayout()
toolbar_widget = QWidget()
row = 0
for col, btn in enumerate(btn_list):
subgrid_toolbar.addWidget(btn, row, col)
subgrid_toolbar.setSpacing(5)
subgrid_toolbar.setContentsMargins(0, 0, 0, 0)
subgrid_toolbar.setColumnStretch(col + 1, 100)
toolbar_widget.setLayout(subgrid_toolbar)
# ---- LEFT PANEL
# SubGrid Hydrograph Frame :
self.hydrograph = hydrograph.Hydrograph()
self.hydrograph_scrollarea = mplFigViewer.ImageViewer()
self.hydrograph_scrollarea.zoomChanged.connect(self.zoom_disp.setValue)
grid_hydrograph = QGridLayout()
grid_hydrograph.addWidget(self.hydrograph_scrollarea, 0, 0)
grid_hydrograph.setRowStretch(0, 500)
grid_hydrograph.setColumnStretch(0, 500)
grid_hydrograph.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
# ASSEMBLING SubGrids :
grid_layout = QGridLayout()
self.grid_layout_widget = QFrame()
row = 0
grid_layout.addWidget(toolbar_widget, row, 0)
row += 1
grid_layout.addLayout(grid_hydrograph, row, 0)
grid_layout.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
grid_layout.setSpacing(5)
grid_layout.setColumnStretch(0, 500)
grid_layout.setRowStretch(1, 500)
self.grid_layout_widget.setLayout(grid_layout)
# ---- Right Panel
self.tabscales = self.__init_scalesTabWidget__()
self.right_panel = myqt.QFrameLayout()
self.right_panel.addWidget(self.dmngr, 0, 0)
self.right_panel.addWidget(self.tabscales, 1, 0)
self.right_panel.setRowStretch(2, 100)
self.right_panel.setSpacing(15)
# ---- MAIN GRID
mainGrid = QGridLayout()
mainGrid.addWidget(self.grid_layout_widget, 0, 0)
mainGrid.addWidget(VSep(), 0, 1)
mainGrid.addWidget(self.right_panel, 0, 2)
mainGrid.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
mainGrid.setSpacing(15)
mainGrid.setColumnStretch(0, 500)
mainGrid.setColumnMinimumWidth(2, 250)
self.setLayout(mainGrid)
# ---- EVENTS
# Toolbox Layout :
btn_bestfit_waterlvl.clicked.connect(self.best_fit_waterlvl)
btn_bestfit_time.clicked.connect(self.best_fit_time)
btn_draw.clicked.connect(self.draw_hydrograph)
btn_save.clicked.connect(self.select_save_path)
# Hydrograph Layout :
self.Ptot_scale.valueChanged.connect(self.layout_changed)
self.qweather_bin.currentIndexChanged.connect(self.layout_changed)
# ---- Init Image
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
class HydroprintGUI(myqt.DialogWindow):
ConsoleSignal = QSignal(str)
def __init__(self, datamanager, parent=None):
super(HydroprintGUI, self).__init__(parent, maximize=True)
self.__updateUI = True
# Child widgets:
self.dmngr = datamanager
self.dmngr.wldsetChanged.connect(self.wldset_changed)
self.dmngr.wxdsetChanged.connect(self.wxdset_changed)
self.page_setup_win = PageSetupWin(self)
self.page_setup_win.newPageSetupSent.connect(self.layout_changed)
self.color_palette_win = ColorsSetupWin(self)
self.color_palette_win.newColorSetupSent.connect(self.update_colors)
# Memory path variable:
self.save_fig_dir = self.workdir
# Generate UI:
self.__initUI__()
def __initUI__(self):
# ---- Toolbar
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save the well hydrograph')
# btn_draw is usefull for debugging purposes
btn_draw = QToolButtonNormal(icons.get_icon('refresh'))
btn_draw.setToolTip('Force a refresh of the well hydrograph')
btn_draw.hide()
self.btn_load_layout = QToolButtonNormal(
icons.get_icon('load_graph_config'))
self.btn_load_layout.setToolTip(
"<p>Load graph layout for the current water level "
" datafile if it exists</p>")
self.btn_load_layout.clicked.connect(self.load_layout_isClicked)
self.btn_save_layout = QToolButtonNormal(
icons.get_icon('save_graph_config'))
self.btn_save_layout.setToolTip('Save current graph layout')
self.btn_save_layout.clicked.connect(self.save_layout_isClicked)
btn_bestfit_waterlvl = QToolButtonNormal(icons.get_icon('fit_y'))
btn_bestfit_waterlvl.setToolTip('Best fit the water level scale')
btn_bestfit_time = QToolButtonNormal(icons.get_icon('fit_x'))
btn_bestfit_time.setToolTip('Best fit the time scale')
self.btn_page_setup = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_page_setup.setToolTip('Show the page setup window')
self.btn_page_setup.clicked.connect(self.page_setup_win.show)
btn_color_pick = QToolButtonNormal(icons.get_icon('color_picker'))
btn_color_pick.setToolTip('<p>Show a window to setup the color palette'
' used to draw the hydrograph</p.')
btn_color_pick.clicked.connect(self.color_palette_win.show)
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.layout_changed)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Zoom Panel
btn_zoom_out = QToolButtonSmall(icons.get_icon('zoom_out'))
btn_zoom_out.setToolTip('Zoom out (ctrl + mouse-wheel-down)')
btn_zoom_out.clicked.connect(self.zoom_out)
btn_zoom_in = QToolButtonSmall(icons.get_icon('zoom_in'))
btn_zoom_in.setToolTip('Zoom in (ctrl + mouse-wheel-up)')
btn_zoom_in.clicked.connect(self.zoom_in)
self.zoom_disp = QSpinBox()
self.zoom_disp.setAlignment(Qt.AlignCenter)
self.zoom_disp.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.zoom_disp.setReadOnly(True)
self.zoom_disp.setSuffix(' %')
self.zoom_disp.setRange(0, 9999)
self.zoom_disp.setValue(100)
zoom_pan = myqt.QFrameLayout()
zoom_pan.setSpacing(3)
zoom_pan.addWidget(btn_zoom_out, 0, 0)
zoom_pan.addWidget(btn_zoom_in, 0, 1)
zoom_pan.addWidget(self.zoom_disp, 0, 2)
# LAYOUT :
btn_list = [
btn_save, btn_draw, self.btn_load_layout, self.btn_save_layout,
VSep(), btn_bestfit_waterlvl, btn_bestfit_time,
VSep(), self.btn_page_setup, btn_color_pick, self.btn_language,
VSep(), zoom_pan
]
subgrid_toolbar = QGridLayout()
toolbar_widget = QWidget()
row = 0
for col, btn in enumerate(btn_list):
subgrid_toolbar.addWidget(btn, row, col)
subgrid_toolbar.setSpacing(5)
subgrid_toolbar.setContentsMargins(0, 0, 0, 0)
subgrid_toolbar.setColumnStretch(col + 1, 100)
toolbar_widget.setLayout(subgrid_toolbar)
# ---- LEFT PANEL
# SubGrid Hydrograph Frame :
self.hydrograph = hydrograph.Hydrograph()
self.hydrograph_scrollarea = mplFigViewer.ImageViewer()
self.hydrograph_scrollarea.zoomChanged.connect(self.zoom_disp.setValue)
grid_hydrograph = QGridLayout()
grid_hydrograph.addWidget(self.hydrograph_scrollarea, 0, 0)
grid_hydrograph.setRowStretch(0, 500)
grid_hydrograph.setColumnStretch(0, 500)
grid_hydrograph.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
# ASSEMBLING SubGrids :
grid_layout = QGridLayout()
self.grid_layout_widget = QFrame()
row = 0
grid_layout.addWidget(toolbar_widget, row, 0)
row += 1
grid_layout.addLayout(grid_hydrograph, row, 0)
grid_layout.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
grid_layout.setSpacing(5)
grid_layout.setColumnStretch(0, 500)
grid_layout.setRowStretch(1, 500)
self.grid_layout_widget.setLayout(grid_layout)
# ---- Right Panel
self.tabscales = self.__init_scalesTabWidget__()
self.right_panel = myqt.QFrameLayout()
self.right_panel.addWidget(self.dmngr, 0, 0)
self.right_panel.addWidget(self.tabscales, 1, 0)
self.right_panel.setRowStretch(2, 100)
self.right_panel.setSpacing(15)
# ---- MAIN GRID
mainGrid = QGridLayout()
mainGrid.addWidget(self.grid_layout_widget, 0, 0)
mainGrid.addWidget(VSep(), 0, 1)
mainGrid.addWidget(self.right_panel, 0, 2)
mainGrid.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
mainGrid.setSpacing(15)
mainGrid.setColumnStretch(0, 500)
mainGrid.setColumnMinimumWidth(2, 250)
self.setLayout(mainGrid)
# ---- EVENTS
# Toolbox Layout :
btn_bestfit_waterlvl.clicked.connect(self.best_fit_waterlvl)
btn_bestfit_time.clicked.connect(self.best_fit_time)
btn_draw.clicked.connect(self.draw_hydrograph)
btn_save.clicked.connect(self.select_save_path)
# Hydrograph Layout :
self.Ptot_scale.valueChanged.connect(self.layout_changed)
self.qweather_bin.currentIndexChanged.connect(self.layout_changed)
# ---- Init Image
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def __init_scalesTabWidget__(self):
class QRowLayout(QGridLayout):
def __init__(self, items, parent=None):
super(QRowLayout, self).__init__(parent)
for col, item in enumerate(items):
self.addWidget(item, 0, col)
self.setContentsMargins(0, 0, 0, 0)
self.setColumnStretch(0, 100)
# ---- Time axis properties
# Generate the widgets :
self.date_start_widget = QDateEdit()
self.date_start_widget.setDisplayFormat('01 / MM / yyyy')
self.date_start_widget.setAlignment(Qt.AlignCenter)
self.date_start_widget.dateChanged.connect(self.layout_changed)
self.date_end_widget = QDateEdit()
self.date_end_widget.setDisplayFormat('01 / MM / yyyy')
self.date_end_widget.setAlignment(Qt.AlignCenter)
self.date_end_widget.dateChanged.connect(self.layout_changed)
self.time_scale_label = QComboBox()
self.time_scale_label.setEditable(False)
self.time_scale_label.setInsertPolicy(QComboBox.NoInsert)
self.time_scale_label.addItems(['Month', 'Year'])
self.time_scale_label.setCurrentIndex(0)
self.time_scale_label.currentIndexChanged.connect(self.layout_changed)
self.dateDispFreq_spinBox = QSpinBox()
self.dateDispFreq_spinBox.setSingleStep(1)
self.dateDispFreq_spinBox.setMinimum(1)
self.dateDispFreq_spinBox.setMaximum(100)
self.dateDispFreq_spinBox.setValue(self.hydrograph.date_labels_pattern)
self.dateDispFreq_spinBox.setAlignment(Qt.AlignCenter)
self.dateDispFreq_spinBox.setKeyboardTracking(False)
self.dateDispFreq_spinBox.valueChanged.connect(self.layout_changed)
# Setting up the layout :
widget_time_scale = QFrame()
widget_time_scale.setFrameStyle(0)
grid_time_scale = QGridLayout()
GRID = [[QLabel('From :'), self.date_start_widget],
[QLabel('To :'), self.date_end_widget],
[QLabel('Scale :'), self.time_scale_label],
[QLabel('Date Disp. Pattern:'), self.dateDispFreq_spinBox]]
for i, ROW in enumerate(GRID):
grid_time_scale.addLayout(QRowLayout(ROW), i, 1)
grid_time_scale.setVerticalSpacing(5)
grid_time_scale.setContentsMargins(10, 10, 10, 10)
widget_time_scale.setLayout(grid_time_scale)
# ----- Water level axis properties
# Widget :
self.waterlvl_scale = QDoubleSpinBox()
self.waterlvl_scale.setSingleStep(0.05)
self.waterlvl_scale.setMinimum(0.05)
self.waterlvl_scale.setSuffix(' m')
self.waterlvl_scale.setAlignment(Qt.AlignCenter)
self.waterlvl_scale.setKeyboardTracking(False)
self.waterlvl_scale.valueChanged.connect(self.layout_changed)
self.waterlvl_scale.setFixedWidth(100)
self.waterlvl_max = QDoubleSpinBox()
self.waterlvl_max.setSingleStep(0.1)
self.waterlvl_max.setSuffix(' m')
self.waterlvl_max.setAlignment(Qt.AlignCenter)
self.waterlvl_max.setMinimum(-1000)
self.waterlvl_max.setMaximum(1000)
self.waterlvl_max.setKeyboardTracking(False)
self.waterlvl_max.valueChanged.connect(self.layout_changed)
self.waterlvl_max.setFixedWidth(100)
self.NZGridWL_spinBox = QSpinBox()
self.NZGridWL_spinBox.setSingleStep(1)
self.NZGridWL_spinBox.setMinimum(1)
self.NZGridWL_spinBox.setMaximum(50)
self.NZGridWL_spinBox.setValue(self.hydrograph.NZGrid)
self.NZGridWL_spinBox.setAlignment(Qt.AlignCenter)
self.NZGridWL_spinBox.setKeyboardTracking(False)
self.NZGridWL_spinBox.valueChanged.connect(self.layout_changed)
self.NZGridWL_spinBox.setFixedWidth(100)
self.datum_widget = QComboBox()
self.datum_widget.addItems(['Ground Surface', 'Sea Level'])
self.datum_widget.currentIndexChanged.connect(self.layout_changed)
# Layout :
subgrid_WLScale = QGridLayout()
GRID = [[QLabel('Minimum :'), self.waterlvl_max],
[QLabel('Scale :'), self.waterlvl_scale],
[QLabel('Grid Divisions :'), self.NZGridWL_spinBox],
[QLabel('Datum :'), self.datum_widget]]
for i, ROW in enumerate(GRID):
subgrid_WLScale.addLayout(QRowLayout(ROW), i, 1)
subgrid_WLScale.setVerticalSpacing(5)
subgrid_WLScale.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
WLScale_widget = QFrame()
WLScale_widget.setFrameStyle(0)
WLScale_widget.setLayout(subgrid_WLScale)
# ---- Weather Axis
# Widgets :
self.Ptot_scale = QSpinBox()
self.Ptot_scale.setSingleStep(5)
self.Ptot_scale.setMinimum(5)
self.Ptot_scale.setMaximum(500)
self.Ptot_scale.setValue(20)
self.Ptot_scale.setSuffix(' mm')
self.Ptot_scale.setAlignment(Qt.AlignCenter)
self.qweather_bin = QComboBox()
self.qweather_bin.setEditable(False)
self.qweather_bin.setInsertPolicy(QComboBox.NoInsert)
self.qweather_bin.addItems(['day', 'week', 'month'])
self.qweather_bin.setCurrentIndex(1)
# Layout :
layout = QGridLayout()
GRID = [[QLabel('Precip. Scale :'), self.Ptot_scale],
[QLabel('Resampling :'), self.qweather_bin]]
for i, row in enumerate(GRID):
layout.addLayout(QRowLayout(row), i, 1)
layout.setVerticalSpacing(5)
layout.setContentsMargins(10, 10, 10, 10) # (L,T,R,B)
layout.setRowStretch(i + 1, 100)
widget_weather_scale = QFrame()
widget_weather_scale.setFrameStyle(0)
widget_weather_scale.setLayout(layout)
# ---- ASSEMBLING TABS
tabscales = QTabWidget()
tabscales.addTab(widget_time_scale, 'Time')
tabscales.addTab(WLScale_widget, 'Water Level')
tabscales.addTab(widget_weather_scale, 'Weather')
return tabscales
@property
def workdir(self):
return self.dmngr.workdir
# ---- Utilities
def zoom_in(self):
self.hydrograph_scrollarea.zoomIn()
def zoom_out(self):
self.hydrograph_scrollarea.zoomOut()
def update_colors(self):
self.hydrograph.update_colors()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
# ---- Datasets Handlers
@property
def wldset(self):
return self.dmngr.get_current_wldset()
@property
def wxdset(self):
return self.dmngr.get_current_wxdset()
def wldset_changed(self):
"""Handle when the water level dataset of the datamanager changes."""
if self.wldset is None:
self.clear_hydrograph()
return
else:
wldset = self.wldset
self.hydrograph.set_wldset(wldset)
self.hydrograph.gluedf = self.wldset.get_glue_at(-1)
# Load the manual measurements.
fname = os.path.join(self.workdir, "Water Levels",
'waterlvl_manual_measurements')
tmeas, wlmeas = load_waterlvl_measures(fname, wldset['Well'])
wldset.set_wlmeas(tmeas, wlmeas)
# Setup the layout of the hydrograph.
layout = wldset.get_layout()
if layout is not None:
msg = 'Loading existing graph layout for well %s.' % wldset['Well']
print(msg)
self.ConsoleSignal.emit('<font color=black>%s</font>' % msg)
self.load_graph_layout(layout)
else:
print('No graph layout exists for well %s.' % wldset['Well'])
# Fit Water Level in Layout :
self.__updateUI = False
self.best_fit_waterlvl()
self.best_fit_time()
self.dmngr.set_closest_wxdset()
self.__updateUI = True
def wxdset_changed(self):
"""Handle when the weather dataset of the datamanager changes."""
if self.wldset is None:
self.clear_hydrograph()
else:
self.hydrograph.set_wxdset(self.wxdset)
QCoreApplication.processEvents()
self.draw_hydrograph()
# ---- Draw Hydrograph Handlers
def best_fit_waterlvl(self):
wldset = self.dmngr.get_current_wldset()
if wldset is not None:
WLscale, WLmin = self.hydrograph.best_fit_waterlvl()
self.waterlvl_scale.setValue(WLscale)
self.waterlvl_max.setValue(WLmin)
def best_fit_time(self):
wldset = self.dmngr.get_current_wldset()
if wldset is not None:
date0, date1 = self.hydrograph.best_fit_time(wldset['Time'])
self.date_start_widget.setDate(QDate(date0[0], date0[1], date0[2]))
self.date_end_widget.setDate(QDate(date1[0], date1[1], date1[2]))
@QSlot()
def mrc_wl_changed(self):
"""
Force a redraw of the MRC water levels after the results have
changed for the dataset.
"""
self.hydrograph.draw_mrc_wl()
self.hydrograph.setup_legend()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
@QSlot(GLUEDataFrameBase)
def glue_wl_changed(self, gluedf):
"""
Force a redraw of the GLUE water levels after the results have
changed for the dataset.
"""
self.hydrograph.set_gluedf(gluedf)
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def layout_changed(self):
"""
When an element of the graph layout is changed in the UI.
"""
if self.__updateUI is False:
return
self.update_graph_layout_parameter()
if self.hydrograph.isHydrographExists is False:
return
sender = self.sender()
if sender == self.btn_language:
self.hydrograph.draw_ylabels()
self.hydrograph.setup_xticklabels()
self.hydrograph.setup_legend()
elif sender in [self.waterlvl_max, self.waterlvl_scale]:
self.hydrograph.setup_waterlvl_scale()
self.hydrograph.draw_ylabels()
elif sender == self.NZGridWL_spinBox:
self.hydrograph.setup_waterlvl_scale()
self.hydrograph.update_precip_scale()
self.hydrograph.draw_ylabels()
elif sender == self.Ptot_scale:
self.hydrograph.update_precip_scale()
self.hydrograph.draw_ylabels()
elif sender == self.datum_widget:
yoffset = int(self.wldset['Elevation'] / self.hydrograph.WLscale)
yoffset *= self.hydrograph.WLscale
self.hydrograph.WLmin = (yoffset - self.hydrograph.WLmin)
self.waterlvl_max.blockSignals(True)
self.waterlvl_max.setValue(self.hydrograph.WLmin)
self.waterlvl_max.blockSignals(False)
# This is calculated so that trailing zeros in the altitude of the
# well is not carried to the y axis labels, so that they remain a
# int multiple of *WLscale*.
self.hydrograph.setup_waterlvl_scale()
self.hydrograph.draw_waterlvl()
self.hydrograph.draw_ylabels()
elif sender in [self.date_start_widget, self.date_end_widget]:
self.hydrograph.set_time_scale()
self.hydrograph.draw_weather()
self.hydrograph.draw_figure_title()
elif sender == self.dateDispFreq_spinBox:
self.hydrograph.set_time_scale()
self.hydrograph.setup_xticklabels()
elif sender == self.page_setup_win:
self.hydrograph.update_fig_size()
# Implicitly call : set_margins()
# draw_ylabels()
# set_time_scale()
# draw_figure_title
self.hydrograph.draw_waterlvl()
self.hydrograph.setup_legend()
elif sender == self.qweather_bin:
self.hydrograph.resample_bin()
self.hydrograph.draw_weather()
self.hydrograph.draw_ylabels()
elif sender == self.time_scale_label:
self.hydrograph.set_time_scale()
self.hydrograph.draw_weather()
else:
print('No action for this widget yet.')
# !!!! temporary fix until I can find a better solution !!!!
# sender.blockSignals(True)
if type(sender) in [QDoubleSpinBox, QSpinBox]:
sender.setReadOnly(True)
for i in range(10):
QCoreApplication.processEvents()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
for i in range(10):
QCoreApplication.processEvents()
if type(sender) in [QDoubleSpinBox, QSpinBox]:
sender.setReadOnly(False)
# sender.blockSignals(False)
def update_graph_layout_parameter(self):
# language :
self.hydrograph.language = self.btn_language.language
# Scales :
self.hydrograph.WLmin = self.waterlvl_max.value()
self.hydrograph.WLscale = self.waterlvl_scale.value()
self.hydrograph.RAINscale = self.Ptot_scale.value()
self.hydrograph.NZGrid = self.NZGridWL_spinBox.value()
# WL Datum :
self.hydrograph.WLdatum = self.datum_widget.currentIndex()
# Dates :
self.hydrograph.datemode = self.time_scale_label.currentText()
year = self.date_start_widget.date().year()
month = self.date_start_widget.date().month()
self.hydrograph.TIMEmin = xldate_from_date_tuple((year, month, 1), 0)
year = self.date_end_widget.date().year()
month = self.date_end_widget.date().month()
self.hydrograph.TIMEmax = xldate_from_date_tuple((year, month, 1), 0)
self.hydrograph.date_labels_pattern = self.dateDispFreq_spinBox.value()
# Page Setup :
self.hydrograph.fwidth = self.page_setup_win.pageSize[0]
self.hydrograph.fheight = self.page_setup_win.pageSize[1]
self.hydrograph.va_ratio = self.page_setup_win.va_ratio
self.hydrograph.trend_line = self.page_setup_win.isTrendLine
self.hydrograph.isLegend = self.page_setup_win.isLegend
self.hydrograph.isGraphTitle = self.page_setup_win.isGraphTitle
self.hydrograph.set_meteo_on(self.page_setup_win.is_meteo_on)
self.hydrograph.set_glue_wl_on(self.page_setup_win.is_glue_wl_on)
self.hydrograph.set_mrc_wl_on(self.page_setup_win.is_mrc_wl_on)
self.hydrograph.set_figframe_lw(self.page_setup_win.figframe_lw)
# Weather bins :
self.hydrograph.bwidth_indx = self.qweather_bin.currentIndex()
def clear_hydrograph(self):
"""Clear the hydrograph figure to show only a blank canvas."""
self.hydrograph.clf()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def draw_hydrograph(self):
if self.dmngr.wldataset_count() == 0:
msg = 'Please import a valid water level data file first.'
self.ConsoleSignal.emit('<font color=red>%s</font>' % msg)
self.emit_warning(msg)
return
self.update_graph_layout_parameter()
# Generate and Display Graph :
for i in range(5):
QCoreApplication.processEvents()
QApplication.setOverrideCursor(Qt.WaitCursor)
self.hydrograph.set_wldset(self.dmngr.get_current_wldset())
self.hydrograph.set_wxdset(self.dmngr.get_current_wxdset())
self.hydrograph.generate_hydrograph()
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
QApplication.restoreOverrideCursor()
def select_save_path(self):
"""
Open a dialog where the user can select a file name to save the
hydrograph.
"""
if self.wldset is None:
return
ffmat = "*.pdf;;*.svg;;*.png"
fname = find_unique_filename(
osp.join(self.save_fig_dir,
'hydrograph_%s.pdf' % self.wldset['Well']))
fname, ftype = QFileDialog.getSaveFileName(self, "Save Figure", fname,
ffmat)
if fname:
ftype = ftype.replace('*', '')
fname = fname if fname.endswith(ftype) else fname + ftype
self.save_fig_dir = os.path.dirname(fname)
try:
self.save_figure(fname)
except PermissionError:
msg = "The file is in use by another application or user."
QMessageBox.warning(self, 'Warning', msg, QMessageBox.Ok)
self.select_save_path()
def save_figure(self, fname):
"""Save the hydrograph figure in a file."""
self.hydrograph.generate_hydrograph()
self.hydrograph.savefig(fname)
# ---- Graph Layout Handlers
def load_layout_isClicked(self):
"""Handle when the button to load the graph layout is clicked."""
if self.wldset is None:
self.emit_warning(
"Please import a valid water level data file first.")
return
layout = self.wldset.get_layout()
if layout is None:
self.emit_warning("No graph layout exists for well %s." %
self.wldset['Well'])
else:
self.load_graph_layout(layout)
def load_graph_layout(self, layout):
"""Load the graph layout into the GUI."""
self.__updateUI = False
# Scales :
date = layout['TIMEmin']
date = xldate_as_tuple(date, 0)
self.date_start_widget.setDate(QDate(date[0], date[1], date[2]))
date = layout['TIMEmax']
date = xldate_as_tuple(date, 0)
self.date_end_widget.setDate(QDate(date[0], date[1], date[2]))
self.dateDispFreq_spinBox.setValue(layout['date_labels_pattern'])
self.waterlvl_scale.setValue(layout['WLscale'])
self.waterlvl_max.setValue(layout['WLmin'])
self.NZGridWL_spinBox.setValue(layout['NZGrid'])
self.Ptot_scale.setValue(layout['RAINscale'])
x = ['mbgs', 'masl'].index(layout['WLdatum'])
self.datum_widget.setCurrentIndex(x)
self.qweather_bin.setCurrentIndex(layout['bwidth_indx'])
self.time_scale_label.setCurrentIndex(
self.time_scale_label.findText(layout['datemode']))
# ---- Language and colors
self.btn_language.set_language(layout['language'])
self.color_palette_win.load_colors()
# ---- Page Setup
self.page_setup_win.pageSize = (layout['fwidth'], layout['fheight'])
self.page_setup_win.va_ratio = layout['va_ratio']
self.page_setup_win.isLegend = layout['legend_on']
self.page_setup_win.isGraphTitle = layout['title_on']
self.page_setup_win.isTrendLine = layout['trend_line']
self.page_setup_win.is_meteo_on = layout['meteo_on']
self.page_setup_win.is_glue_wl_on = layout['glue_wl_on']
self.page_setup_win.is_mrc_wl_on = layout['mrc_wl_on']
self.page_setup_win.figframe_lw = layout['figframe_lw']
self.page_setup_win.legend_on.set_value(layout['legend_on'])
self.page_setup_win.title_on.set_value(layout['title_on'])
self.page_setup_win.wltrend_on.set_value(layout['trend_line'])
self.page_setup_win.meteo_on.set_value(layout['meteo_on'])
self.page_setup_win.glue_wl_on.set_value(layout['glue_wl_on'])
self.page_setup_win.mrc_wl_on.set_value(layout['mrc_wl_on'])
self.page_setup_win.fframe_lw_widg.setValue(layout['figframe_lw'])
self.page_setup_win.fwidth.setValue(layout['fwidth'])
self.page_setup_win.fheight.setValue(layout['fheight'])
self.page_setup_win.va_ratio_spinBox.setValue(layout['va_ratio'])
# Check if Weather Dataset :
if layout['wxdset'] in self.dmngr.wxdsets:
self.dmngr.set_current_wxdset(layout['wxdset'])
else:
self.dmngr.set_closest_wxdset()
self.__updateUI = True
def save_layout_isClicked(self):
wldset = self.wldset
if wldset is None:
self.emit_warning(
"Please import a valid water level data file first.")
return
layout = wldset.get_layout()
if layout is not None:
msg = ('A graph layout already exists for well %s.Do you want to'
' you want to replace it?') % wldset['Well']
reply = QMessageBox.question(self, 'Save Graph Layout', msg,
QMessageBox.Yes | QMessageBox.No)
if reply == QMessageBox.Yes:
self.save_graph_layout()
elif reply == QMessageBox.No:
msg = "Graph layout not saved for well %s." % wldset['Well']
self.ConsoleSignal.emit('<font color=black>%s' % msg)
else:
self.save_graph_layout()
def save_graph_layout(self):
"""Save the graph layout in the project hdf5 file."""
print("Saving the graph layout for well %s..." % self.wldset['Well'],
end=" ")
layout = {
'WLmin': self.waterlvl_max.value(),
'WLscale': self.waterlvl_scale.value(),
'RAINscale': self.Ptot_scale.value(),
'fwidth': self.page_setup_win.pageSize[0],
'fheight': self.page_setup_win.pageSize[1],
'va_ratio': self.page_setup_win.va_ratio,
'NZGrid': self.NZGridWL_spinBox.value(),
'bwidth_indx': self.qweather_bin.currentIndex(),
'date_labels_pattern': self.dateDispFreq_spinBox.value(),
'datemode': self.time_scale_label.currentText()
}
layout['wxdset'] = None if self.wxdset is None else self.wxdset.name
year = self.date_start_widget.date().year()
month = self.date_start_widget.date().month()
layout['TIMEmin'] = xldate_from_date_tuple((year, month, 1), 0)
year = self.date_end_widget.date().year()
month = self.date_end_widget.date().month()
layout['TIMEmax'] = xldate_from_date_tuple((year, month, 1), 0)
if self.datum_widget.currentIndex() == 0:
layout['WLdatum'] = 'mbgs'
else:
layout['WLdatum'] = 'masl'
# ---- Page Setup
layout['title_on'] = bool(self.page_setup_win.isGraphTitle)
layout['legend_on'] = bool(self.page_setup_win.isLegend)
layout['language'] = self.btn_language.language
layout['trend_line'] = bool(self.page_setup_win.isTrendLine)
layout['meteo_on'] = bool(self.page_setup_win.is_meteo_on)
layout['glue_wl_on'] = bool(self.page_setup_win.is_glue_wl_on)
layout['mrc_wl_on'] = bool(self.page_setup_win.is_mrc_wl_on)
layout['figframe_lw'] = self.page_setup_win.figframe_lw
# Save the colors :
cdb = ColorsReader()
cdb.load_colors_db()
layout['colors'] = cdb.RGB
# Save the layout :
self.wldset.save_layout(layout)
msg = 'Layout saved successfully for well %s.' % self.wldset['Well']
self.ConsoleSignal.emit('<font color=black>%s</font>' % msg)
print("done")
def __initUI__(self):
# ---- Toolbar
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save the well hydrograph')
# btn_draw is usefull for debugging purposes
btn_draw = QToolButtonNormal(icons.get_icon('refresh'))
btn_draw.setToolTip('Force a refresh of the well hydrograph')
btn_draw.hide()
self.btn_load_layout = QToolButtonNormal(
icons.get_icon('load_graph_config'))
self.btn_load_layout.setToolTip(
"<p>Load graph layout for the current water level "
" datafile if it exists</p>")
self.btn_load_layout.clicked.connect(self.load_layout_isClicked)
self.btn_save_layout = QToolButtonNormal(
icons.get_icon('save_graph_config'))
self.btn_save_layout.setToolTip('Save current graph layout')
self.btn_save_layout.clicked.connect(self.save_layout_isClicked)
btn_bestfit_waterlvl = QToolButtonNormal(icons.get_icon('fit_y'))
btn_bestfit_waterlvl.setToolTip('Best fit the water level scale')
btn_bestfit_time = QToolButtonNormal(icons.get_icon('fit_x'))
btn_bestfit_time.setToolTip('Best fit the time scale')
self.btn_page_setup = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_page_setup.setToolTip('Show the page setup window')
self.btn_page_setup.clicked.connect(self.page_setup_win.show)
btn_color_pick = QToolButtonNormal(icons.get_icon('color_picker'))
btn_color_pick.setToolTip('<p>Show a window to setup the color palette'
' used to draw the hydrograph</p.')
btn_color_pick.clicked.connect(self.color_palette_win.show)
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.layout_changed)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Zoom Panel
btn_zoom_out = QToolButtonSmall(icons.get_icon('zoom_out'))
btn_zoom_out.setToolTip('Zoom out (ctrl + mouse-wheel-down)')
btn_zoom_out.clicked.connect(self.zoom_out)
btn_zoom_in = QToolButtonSmall(icons.get_icon('zoom_in'))
btn_zoom_in.setToolTip('Zoom in (ctrl + mouse-wheel-up)')
btn_zoom_in.clicked.connect(self.zoom_in)
self.zoom_disp = QSpinBox()
self.zoom_disp.setAlignment(Qt.AlignCenter)
self.zoom_disp.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.zoom_disp.setReadOnly(True)
self.zoom_disp.setSuffix(' %')
self.zoom_disp.setRange(0, 9999)
self.zoom_disp.setValue(100)
zoom_pan = myqt.QFrameLayout()
zoom_pan.setSpacing(3)
zoom_pan.addWidget(btn_zoom_out, 0, 0)
zoom_pan.addWidget(btn_zoom_in, 0, 1)
zoom_pan.addWidget(self.zoom_disp, 0, 2)
# LAYOUT :
btn_list = [btn_save, btn_draw,
self.btn_load_layout, self.btn_save_layout, VSep(),
btn_bestfit_waterlvl, btn_bestfit_time, VSep(),
self.btn_page_setup, btn_color_pick, self.btn_language,
VSep(), zoom_pan]
subgrid_toolbar = QGridLayout()
toolbar_widget = QWidget()
row = 0
for col, btn in enumerate(btn_list):
subgrid_toolbar.addWidget(btn, row, col)
subgrid_toolbar.setSpacing(5)
subgrid_toolbar.setContentsMargins(0, 0, 0, 0)
subgrid_toolbar.setColumnStretch(col + 1, 100)
toolbar_widget.setLayout(subgrid_toolbar)
# ---- LEFT PANEL
# SubGrid Hydrograph Frame :
self.hydrograph = hydrograph.Hydrograph()
self.hydrograph_scrollarea = mplFigViewer.ImageViewer()
self.hydrograph_scrollarea.zoomChanged.connect(self.zoom_disp.setValue)
grid_hydrograph = QGridLayout()
grid_hydrograph.addWidget(self.hydrograph_scrollarea, 0, 0)
grid_hydrograph.setRowStretch(0, 500)
grid_hydrograph.setColumnStretch(0, 500)
grid_hydrograph.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
# ASSEMBLING SubGrids :
grid_layout = QGridLayout()
self.grid_layout_widget = QFrame()
row = 0
grid_layout.addWidget(toolbar_widget, row, 0)
row += 1
grid_layout.addLayout(grid_hydrograph, row, 0)
grid_layout.setContentsMargins(0, 0, 0, 0) # (L, T, R, B)
grid_layout.setSpacing(5)
grid_layout.setColumnStretch(0, 500)
grid_layout.setRowStretch(1, 500)
self.grid_layout_widget.setLayout(grid_layout)
# ---- Right Panel
self.tabscales = self.__init_scalesTabWidget__()
self.right_panel = myqt.QFrameLayout()
self.right_panel.addWidget(self.dmngr, 0, 0)
self.right_panel.addWidget(self.tabscales, 1, 0)
self.right_panel.setRowStretch(2, 100)
self.right_panel.setSpacing(15)
# ---- MAIN GRID
mainGrid = QGridLayout()
mainGrid.addWidget(self.grid_layout_widget, 0, 0)
mainGrid.addWidget(VSep(), 0, 1)
mainGrid.addWidget(self.right_panel, 0, 2)
mainGrid.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
mainGrid.setSpacing(15)
mainGrid.setColumnStretch(0, 500)
mainGrid.setColumnMinimumWidth(2, 250)
self.setLayout(mainGrid)
# ---- EVENTS
# Toolbox Layout :
btn_bestfit_waterlvl.clicked.connect(self.best_fit_waterlvl)
btn_bestfit_time.clicked.connect(self.best_fit_time)
btn_draw.clicked.connect(self.draw_hydrograph)
btn_save.clicked.connect(self.select_save_path)
# Hydrograph Layout :
self.Ptot_scale.valueChanged.connect(self.layout_changed)
self.qweather_bin.currentIndexChanged.connect(self.layout_changed)
# ---- Init Image
self.hydrograph_scrollarea.load_mpl_figure(self.hydrograph)
def __initGUI__(self):
# ---- Navigator
self.btn_prev = QToolButtonNormal(icons.get_icon('go_previous'))
self.btn_prev.clicked.connect(self.navigate_brf)
self.btn_next = QToolButtonNormal(icons.get_icon('go_next'))
self.btn_next.clicked.connect(self.navigate_brf)
self.current_brf = QSpinBox()
self.current_brf.setRange(0, 99)
self.current_brf.setAlignment(Qt.AlignCenter)
self.current_brf.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.current_brf.setCorrectionMode(
QAbstractSpinBox.CorrectToNearestValue)
self.current_brf.valueChanged.connect(self.navigate_brf)
self.current_brf.setValue(0)
self.total_brf = QLabel('/ 0')
# ---- Language button
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.plot_brf)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Toolbar
# Generate the buttons :
self.btn_del = QToolButtonNormal(icons.get_icon('clear_search'))
self.btn_del.setToolTip('Delete current BRF results')
self.btn_del.clicked.connect(self.del_brf)
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save current BRF graph...')
btn_save.clicked.connect(self.select_savefig_path)
self.btn_setp = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_setp.setToolTip('Show graph layout parameters...')
self.btn_setp.clicked.connect(self.toggle_graphpannel)
# Generate the layout :
self.tbar = myqt.QFrameLayout()
buttons = [
btn_save, self.btn_del,
VSep(), self.btn_prev, self.current_brf, self.total_brf,
self.btn_next,
VSep(), self.btn_setp, self.btn_language
]
for btn in buttons:
if isinstance(btn, QLayout):
self.tbar.addLayout(btn, 1, self.tbar.columnCount())
else:
self.tbar.addWidget(btn, 1, self.tbar.columnCount())
row = self.tbar.columnCount()
self.tbar.addWidget(HSep(), 0, 0, 1, row + 1)
self.tbar.setColumnStretch(row, 100)
self.tbar.setContentsMargins(10, 0, 10, 10) # (l, t, r, b)
# ---- Graph Canvas
self.fig_frame = QFrame()
self.fig_frame.setFrameStyle(StyleDB().frame)
self.fig_frame.setObjectName("figframe")
self.fig_frame.setStyleSheet("#figframe {background-color:white;}")
self.brf_canvas = FigureCanvasQTAgg(BRFFigure())
fflay = QGridLayout(self.fig_frame)
fflay.setContentsMargins(0, 0, 0, 0) # (left, top, right, bottom)
fflay.addWidget(self.tbar, 1, 0)
fflay.addWidget(self.brf_canvas, 0, 0)
# ---- Graph Options Panel
self.graph_opt_panel = BRFOptionsPanel()
self.graph_opt_panel.sig_graphconf_changed.connect(self.plot_brf)
# ---- Main Layout
ml = QGridLayout(self)
ml.addWidget(self.fig_frame, 0, 2)
ml.addWidget(self.graph_opt_panel, 0, 3)
ml.setColumnStretch(1, 100)
class BRFViewer(QWidget):
"""
Window that is used to show all the results produced with for the
currently selected water level dataset.
"""
def __init__(self, wldset=None, parent=None):
super(BRFViewer, self).__init__(parent)
self.__save_ddir = osp.dirname(__rootdir__)
self.setWindowTitle('BRF Results Viewer')
self.setWindowIcon(icons.get_icon('master'))
self.setWindowFlags(Qt.Window | Qt.CustomizeWindowHint
| Qt.WindowMinimizeButtonHint
| Qt.WindowCloseButtonHint)
self.__initGUI__()
self.set_wldset(wldset)
def __initGUI__(self):
# ---- Navigator
self.btn_prev = QToolButtonNormal(icons.get_icon('go_previous'))
self.btn_prev.clicked.connect(self.navigate_brf)
self.btn_next = QToolButtonNormal(icons.get_icon('go_next'))
self.btn_next.clicked.connect(self.navigate_brf)
self.current_brf = QSpinBox()
self.current_brf.setRange(0, 99)
self.current_brf.setAlignment(Qt.AlignCenter)
self.current_brf.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.current_brf.setCorrectionMode(
QAbstractSpinBox.CorrectToNearestValue)
self.current_brf.valueChanged.connect(self.navigate_brf)
self.current_brf.setValue(0)
self.total_brf = QLabel('/ 0')
# ---- Language button
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.plot_brf)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Toolbar
# Generate the buttons :
self.btn_del = QToolButtonNormal(icons.get_icon('clear_search'))
self.btn_del.setToolTip('Delete current BRF results')
self.btn_del.clicked.connect(self.del_brf)
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save current BRF graph...')
btn_save.clicked.connect(self.select_savefig_path)
self.btn_setp = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_setp.setToolTip('Show graph layout parameters...')
self.btn_setp.clicked.connect(self.toggle_graphpannel)
# Generate the layout :
self.tbar = myqt.QFrameLayout()
buttons = [
btn_save, self.btn_del,
VSep(), self.btn_prev, self.current_brf, self.total_brf,
self.btn_next,
VSep(), self.btn_setp, self.btn_language
]
for btn in buttons:
if isinstance(btn, QLayout):
self.tbar.addLayout(btn, 1, self.tbar.columnCount())
else:
self.tbar.addWidget(btn, 1, self.tbar.columnCount())
row = self.tbar.columnCount()
self.tbar.addWidget(HSep(), 0, 0, 1, row + 1)
self.tbar.setColumnStretch(row, 100)
self.tbar.setContentsMargins(10, 0, 10, 10) # (l, t, r, b)
# ---- Graph Canvas
self.fig_frame = QFrame()
self.fig_frame.setFrameStyle(StyleDB().frame)
self.fig_frame.setObjectName("figframe")
self.fig_frame.setStyleSheet("#figframe {background-color:white;}")
self.brf_canvas = FigureCanvasQTAgg(BRFFigure())
fflay = QGridLayout(self.fig_frame)
fflay.setContentsMargins(0, 0, 0, 0) # (left, top, right, bottom)
fflay.addWidget(self.tbar, 1, 0)
fflay.addWidget(self.brf_canvas, 0, 0)
# ---- Graph Options Panel
self.graph_opt_panel = BRFOptionsPanel()
self.graph_opt_panel.sig_graphconf_changed.connect(self.plot_brf)
# ---- Main Layout
ml = QGridLayout(self)
ml.addWidget(self.fig_frame, 0, 2)
ml.addWidget(self.graph_opt_panel, 0, 3)
ml.setColumnStretch(1, 100)
# ---- Toolbar Handlers
def toggle_graphpannel(self):
if self.graph_opt_panel.isVisible() is True:
# Hide the panel.
self.graph_opt_panel.setVisible(False)
self.btn_setp.setAutoRaise(True)
self.btn_setp.setToolTip('Show graph layout parameters...')
w = self.size().width() - self.graph_opt_panel.size().width()
self.setFixedWidth(w)
else:
# Show the panel.
self.graph_opt_panel.setVisible(True)
self.btn_setp.setAutoRaise(False)
self.btn_setp.setToolTip('Hide graph layout parameters...')
w = self.size().width() + self.graph_opt_panel.size().width()
self.setFixedWidth(w)
def navigate_brf(self):
if self.sender() == self.btn_prev:
cur_num = self.current_brf.value() - 1
elif self.sender() == self.btn_next:
cur_num = self.current_brf.value() + 1
elif self.sender() == self.current_brf:
cur_num = self.current_brf.value()
self.current_brf.setValue(cur_num)
self.update_brfnavigate_state()
def del_brf(self):
"""Delete the graph and data of the currently selected result."""
index = self.current_brf.value() - 1
name = self.wldset.get_brfAt(index)
self.wldset.del_brf(name)
self.update_brfnavigate_state()
def new_brf_added(self):
self.current_brf.setMaximum(self.wldset.brf_count())
self.current_brf.setValue(self.wldset.brf_count())
self.update_brfnavigate_state()
def update_brfnavigate_state(self):
count = self.wldset.brf_count()
self.total_brf.setText('/ %d' % count)
self.current_brf.setMinimum(min(count, 1))
self.current_brf.setMaximum(count)
curnt = self.current_brf.value()
self.tbar.setEnabled(count > 0)
self.btn_prev.setEnabled(curnt > 1)
self.btn_next.setEnabled(curnt < count)
self.btn_del.setEnabled(count > 0)
self.plot_brf()
def select_savefig_path(self):
"""
Opens a dialog to select a file path where to save the brf figure.
"""
ddir = osp.join(self.__save_ddir, 'brf_%s' % self.wldset['Well'])
dialog = QFileDialog()
fname, ftype = dialog.getSaveFileName(self, "Save Figure", ddir,
'*.pdf;;*.svg')
ftype = ftype.replace('*', '')
if fname:
self.__save_ddir = osp.dirname(fname)
if not fname.endswith(ftype):
fname = fname + ftype
self.save_brf_fig(fname)
def save_brf_fig(self, fname):
"""Saves the current BRF figure to fname."""
self.brf_canvas.figure.savefig(fname)
# ---- Others
def set_wldset(self, wldset):
self.wldset = wldset
if wldset is None:
self.setEnabled(False)
else:
self.setEnabled(True)
self.update_brfnavigate_state()
def plot_brf(self):
self.brf_canvas.figure.set_language(self.btn_language.language)
if self.wldset.brf_count() == 0:
self.brf_canvas.figure.empty_BRF()
else:
name = self.wldset.get_brfAt(self.current_brf.value() - 1)
lag, A, err, date_start, date_end = self.wldset.get_brf(name)
well = self.wldset['Well']
if self.graph_opt_panel.show_ebar is False:
err = []
msize = self.graph_opt_panel.markersize
draw_line = self.graph_opt_panel.draw_line
ymin = self.graph_opt_panel.ymin
ymax = self.graph_opt_panel.ymax
yscale = self.graph_opt_panel.yscale
xmin = self.graph_opt_panel.xmin
xmax = self.graph_opt_panel.xmax
xscale = self.graph_opt_panel.xscale
time_units = self.graph_opt_panel.time_units
date0 = '%02d/%02d/%04d' % (date_start[2], date_start[1],
date_start[0])
date1 = '%02d/%02d/%04d' % (date_end[2], date_end[1], date_end[0])
self.brf_canvas.figure.plot_BRF(lag, A, err, date0, date1, well,
msize, draw_line, [ymin, ymax],
[xmin, xmax], time_units, xscale,
yscale)
self.brf_canvas.draw()
def show(self):
super(BRFViewer, self).show()
qr = self.frameGeometry()
if self.parentWidget():
parent = self.parentWidget()
wp = parent.frameGeometry().width()
hp = parent.frameGeometry().height()
cp = parent.mapToGlobal(QPoint(wp / 2, hp / 2))
else:
cp = QDesktopWidget().availableGeometry().center()
qr.moveCenter(cp)
self.move(qr.topLeft())
self.fig_frame.setFixedSize(self.fig_frame.size())
self.setFixedSize(self.size())
self.raise_()
if self.windowState() == Qt.WindowMinimized:
# Window is minimised. Restore it.
self.setWindowState(Qt.WindowNoState)
def __initUI__(self):
self.setWindowTitle('Weather Averages')
self.setWindowIcon(icons.get_icon('master'))
# ---- Toolbar
# Initialize the widgets :
menu_save = QMenu()
menu_save.addAction('Save normals graph as...', self.save_graph)
menu_save.addAction('Save normals table as...', self.save_normals)
btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save normals')
btn_save.setMenu(menu_save)
btn_save.setPopupMode(QToolButton.InstantPopup)
btn_save.setStyleSheet("QToolButton::menu-indicator {image: none;}")
self.btn_export = ExportWeatherButton()
self.btn_export.setIconSize(icons.get_iconsize('normal'))
btn_showStats = QToolButtonNormal(icons.get_icon('showGrid'))
btn_showStats.setToolTip(
"Show the monthly weather normals data table.")
btn_showStats.clicked.connect(self.show_monthly_grid)
btn_language = LangToolButton()
btn_language.setToolTip(
"Set the language of the text shown in the graph.")
btn_language.sig_lang_changed.connect(self.set_lang)
btn_language.setIconSize(icons.get_iconsize('normal'))
# Instantiate and define a layout for the year range widget :
self.year_rng = RangeSpinBoxes(1000, 9999)
self.year_rng.setRange(1800, datetime.now().year)
self.year_rng.sig_range_changed.connect(self.update_normals)
btn_expand = QToolButtonVRectSmall(icons.get_icon('expand_range_vert'))
btn_expand.clicked.connect(self.expands_year_range)
btn_expand.setToolTip("Set the maximal possible year range.")
lay_expand = QGridLayout()
lay_expand.addWidget(self.year_rng.spb_upper, 0, 0)
lay_expand.addWidget(btn_expand, 0, 1)
lay_expand.setContentsMargins(0, 0, 0, 0)
lay_expand.setSpacing(1)
qgrid = QHBoxLayout(self.year_rng)
qgrid.setContentsMargins(0, 0, 0, 0)
qgrid.addWidget(QLabel('Year Range :'))
qgrid.addWidget(self.year_rng.spb_lower)
qgrid.addWidget(QLabel('to'))
qgrid.addLayout(lay_expand)
# Generate the layout of the toolbar :
toolbar_widget = ToolBarWidget()
for btn in [
btn_save, self.btn_export, btn_showStats, btn_language, None,
self.year_rng
]:
toolbar_widget.addWidget(btn)
# ---- Main Layout
# Initialize the widgets :
self.fig_weather_normals = FigWeatherNormals()
self.grid_weather_normals = GridWeatherNormals()
self.grid_weather_normals.hide()
# Generate the layout :
mainGrid = QGridLayout()
row = 0
mainGrid.addWidget(toolbar_widget, row, 0)
row += 1
mainGrid.addWidget(self.fig_weather_normals, row, 0)
row += 1
mainGrid.addWidget(self.grid_weather_normals, row, 0)
mainGrid.setContentsMargins(10, 10, 10, 10) # (L, T, R, B)
mainGrid.setSpacing(10)
mainGrid.setRowStretch(row, 500)
mainGrid.setColumnStretch(0, 500)
self.setLayout(mainGrid)
class BRFViewer(QWidget):
"""
Window that is used to show all the results produced with for the
currently selected water level dataset.
"""
def __init__(self, wldset=None, parent=None):
super(BRFViewer, self).__init__(parent)
self.__save_ddir = None
self.setWindowTitle('BRF Results Viewer')
self.setWindowIcon(icons.get_icon('master'))
self.setWindowFlags(Qt.Window | Qt.CustomizeWindowHint
| Qt.WindowMinimizeButtonHint
| Qt.WindowCloseButtonHint)
self.__initGUI__()
self.set_wldset(wldset)
def __initGUI__(self):
# ---- Navigator
self.btn_prev = QToolButtonNormal(icons.get_icon('go_previous'))
self.btn_prev.clicked.connect(self.navigate_brf)
self.btn_next = QToolButtonNormal(icons.get_icon('go_next'))
self.btn_next.clicked.connect(self.navigate_brf)
self.current_brf = QSpinBox()
self.current_brf.setRange(0, 99)
self.current_brf.setAlignment(Qt.AlignCenter)
self.current_brf.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.current_brf.setCorrectionMode(
QAbstractSpinBox.CorrectToNearestValue)
self.current_brf.valueChanged.connect(self.navigate_brf)
self.current_brf.setValue(0)
self.total_brf = QLabel('/ 0')
# ---- Language button
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.plot_brf)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Toolbar
# Generate the buttons :
self.btn_del = QToolButtonNormal(icons.get_icon('delete_data'))
self.btn_del.setToolTip('Delete current BRF results')
self.btn_del.clicked.connect(self.del_brf)
self.btn_del_all = QToolButtonNormal(icons.get_icon('close_all'))
self.btn_del_all.setToolTip(
'Delete all BRF results for the current water level dataset.')
self.btn_del_all.clicked.connect(self.del_all_brf)
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save current BRF graph as...')
btn_save.clicked.connect(self.select_savefig_path)
self.btn_export = QToolButtonNormal(icons.get_icon('export_data'))
self.btn_export.setToolTip('Export data to file.')
self.btn_export.clicked.connect(self.select_export_brfdata_filepath)
self.btn_copy = QToolButtonNormal('copy_clipboard')
self.btn_copy.setToolTip('Copy figure to clipboard as image.')
self.btn_copy.clicked.connect(self.copyfig_to_clipboard)
self.btn_setp = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_setp.setToolTip('Show graph layout parameters...')
self.btn_setp.clicked.connect(self.toggle_graphpannel)
# Generate the layout :
self.tbar = myqt.QFrameLayout()
buttons = [
btn_save, self.btn_copy, self.btn_export, self.btn_del,
self.btn_del_all,
VSep(), self.btn_prev, self.current_brf, self.total_brf,
self.btn_next,
VSep(), self.btn_setp, self.btn_language
]
for btn in buttons:
if isinstance(btn, QLayout):
self.tbar.addLayout(btn, 1, self.tbar.columnCount())
else:
self.tbar.addWidget(btn, 1, self.tbar.columnCount())
row = self.tbar.columnCount()
self.tbar.addWidget(HSep(), 0, 0, 1, row + 1)
self.tbar.setColumnStretch(row, 100)
self.tbar.setContentsMargins(10, 0, 10, 10) # (l, t, r, b)
# ---- Graph Canvas
self.fig_frame = QFrame()
self.fig_frame.setFrameStyle(StyleDB().frame)
self.fig_frame.setObjectName("figframe")
self.fig_frame.setStyleSheet("#figframe {background-color:white;}")
self.brf_canvas = FigureCanvasQTAgg(BRFFigure())
fflay = QGridLayout(self.fig_frame)
fflay.setContentsMargins(0, 0, 0, 0) # (left, top, right, bottom)
fflay.addWidget(self.tbar, 1, 0)
fflay.addWidget(self.brf_canvas, 0, 0)
# ---- Graph Options Panel
self.graph_opt_panel = BRFOptionsPanel()
self.graph_opt_panel.sig_graphconf_changed.connect(self.plot_brf)
# ---- Main Layout
ml = QGridLayout(self)
ml.addWidget(self.fig_frame, 0, 2)
ml.addWidget(self.graph_opt_panel, 0, 3)
ml.setColumnStretch(1, 100)
@property
def savedir(self):
"""Return a path where figures and files are saved by default."""
if self.__save_ddir is None or not osp.exists(self.__save_ddir):
try:
savedir = self.wldset.dirname
except AttributeError:
savedir = osp.dirname(__rootdir__)
finally:
return savedir
else:
return self.__save_ddir
# ---- Toolbar Handlers
def toggle_graphpannel(self):
if self.graph_opt_panel.isVisible() is True:
# Hide the panel.
self.graph_opt_panel.setVisible(False)
self.btn_setp.setAutoRaise(True)
self.btn_setp.setToolTip('Show graph layout parameters...')
w = self.size().width() - self.graph_opt_panel.size().width()
self.setFixedWidth(w)
else:
# Show the panel.
self.graph_opt_panel.setVisible(True)
self.btn_setp.setAutoRaise(False)
self.btn_setp.setToolTip('Hide graph layout parameters...')
w = self.size().width() + self.graph_opt_panel.size().width()
self.setFixedWidth(w)
def navigate_brf(self):
if self.sender() == self.btn_prev:
cur_num = self.current_brf.value() - 1
elif self.sender() == self.btn_next:
cur_num = self.current_brf.value() + 1
elif self.sender() == self.current_brf:
cur_num = self.current_brf.value()
self.current_brf.setValue(cur_num)
self.update_brfnavigate_state()
def del_brf(self):
"""Delete the graph and data of the currently selected result."""
index = self.current_brf.value() - 1
name = self.wldset.get_brfname_at(index)
self.wldset.del_brf(name)
self.update_brfnavigate_state()
def del_all_brf(self):
"""Delete all the graphs and BRF results for the current dataset."""
msg = ("Do you want to delete all BRF results that were evaluated "
"for dataset <i>{}</i>?"
"<br><br>"
"All data will be lost permanently.").format(self.wldset.name)
btn = QMessageBox.Yes | QMessageBox.No
reply = QMessageBox.question(self, 'Delete all BRF results', msg, btn)
if reply == QMessageBox.Yes:
for name in self.wldset.saved_brf():
self.wldset.del_brf(name)
self.update_brfnavigate_state()
def new_brf_added(self):
self.current_brf.setMaximum(self.wldset.brf_count())
self.current_brf.setValue(self.wldset.brf_count())
self.update_brfnavigate_state()
def update_brfnavigate_state(self):
count = self.wldset.brf_count()
self.total_brf.setText('/ %d' % count)
self.current_brf.setMinimum(min(count, 1))
self.current_brf.setMaximum(count)
curnt = self.current_brf.value()
self.tbar.setEnabled(count > 0)
self.btn_prev.setEnabled(curnt > 1)
self.btn_next.setEnabled(curnt < count)
self.btn_del.setEnabled(count > 0)
self.plot_brf()
def select_savefig_path(self):
"""
Opens a dialog to select a file path where to save the brf figure.
"""
ddir = osp.join(self.savedir, 'brf_%s' % self.wldset['Well'])
dialog = QFileDialog()
fname, ftype = dialog.getSaveFileName(self, "Save Figure", ddir,
'*.pdf;;*.svg')
ftype = ftype.replace('*', '')
if fname:
if not osp.samefile(osp.dirname(ddir), osp.dirname(fname)):
self.__save_ddir = osp.dirname(fname)
if not fname.endswith(ftype):
fname = fname + ftype
self.save_brf_fig(fname)
def save_brf_fig(self, fname):
"""Saves the current BRF figure to fname."""
self.brf_canvas.figure.savefig(fname)
def copyfig_to_clipboard(self):
"""Saves the current BRF figure to the clipboard."""
buf = io.BytesIO()
self.save_brf_fig(buf)
QApplication.clipboard().setImage(QImage.fromData(buf.getvalue()))
buf.close()
def select_export_brfdata_filepath(self):
"""
Open a dialog to select a file path where to save the brf data.
"""
fname = 'brf_' + self.wldset['Well']
if self.wldset['Well ID']:
fname += '_%s' % self.wldset['Well ID']
ddir = osp.join(self.savedir, fname)
dialog = QFileDialog()
fname, ftype = dialog.getSaveFileName(self, "Export Data", ddir,
"*.xlsx;;*.xls;;*.csv")
ftype = ftype.replace('*', '')
if fname:
if not osp.samefile(osp.dirname(ddir), osp.dirname(fname)):
self.__save_ddir = osp.dirname(fname)
if not fname.endswith(ftype):
fname = fname + ftype
self.export_brf_data(fname)
def export_brf_data(self, fname):
"""Export the current BRF data to to file."""
self.wldset.export_brf_to_csv(fname, self.current_brf.value() - 1)
# ---- Others
def set_wldset(self, wldset):
self.wldset = wldset
self.setEnabled(wldset is not None)
if wldset is not None:
self.update_brfnavigate_state()
def plot_brf(self):
self.brf_canvas.figure.set_language(self.btn_language.language)
if self.wldset.brf_count() == 0:
self.brf_canvas.figure.empty_BRF()
else:
name = self.wldset.get_brfname_at(self.current_brf.value() - 1)
databrf = self.wldset.get_brf(name)
well = self.wldset['Well']
msize = self.graph_opt_panel.markersize
draw_line = self.graph_opt_panel.draw_line
ymin = self.graph_opt_panel.ymin
ymax = self.graph_opt_panel.ymax
yscale = self.graph_opt_panel.yscale
xmin = self.graph_opt_panel.xmin
xmax = self.graph_opt_panel.xmax
xscale = self.graph_opt_panel.xscale
time_units = self.graph_opt_panel.time_units
date0 = databrf.date_start.strftime(format='%d/%m/%y %H:%M')
date1 = databrf.date_end.strftime(format='%d/%m/%y %H:%M')
self.brf_canvas.figure.plot_BRF(databrf['Lag'].values,
databrf['SumA'].values,
databrf['sdA'].values, date0,
date1, well, msize, draw_line,
[ymin, ymax], [xmin, xmax],
time_units, xscale, yscale)
self.brf_canvas.draw()
def show(self):
super(BRFViewer, self).show()
qr = self.frameGeometry()
if self.parentWidget():
parent = self.parentWidget()
wp = parent.frameGeometry().width()
hp = parent.frameGeometry().height()
cp = parent.mapToGlobal(QPoint(wp / 2, hp / 2))
else:
cp = QDesktopWidget().availableGeometry().center()
qr.moveCenter(cp)
self.move(qr.topLeft())
self.fig_frame.setFixedSize(self.fig_frame.size())
self.setFixedSize(self.size())
self.raise_()
if self.windowState() == Qt.WindowMinimized:
# Window is minimised. Restore it.
self.setWindowState(Qt.WindowNoState)
def __initGUI__(self):
# ---- Navigator
self.btn_prev = QToolButtonNormal(icons.get_icon('go_previous'))
self.btn_prev.clicked.connect(self.navigate_brf)
self.btn_next = QToolButtonNormal(icons.get_icon('go_next'))
self.btn_next.clicked.connect(self.navigate_brf)
self.current_brf = QSpinBox()
self.current_brf.setRange(0, 99)
self.current_brf.setAlignment(Qt.AlignCenter)
self.current_brf.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.current_brf.setCorrectionMode(
QAbstractSpinBox.CorrectToNearestValue)
self.current_brf.valueChanged.connect(self.navigate_brf)
self.current_brf.setValue(0)
self.total_brf = QLabel('/ 0')
# ---- Language button
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.plot_brf)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Toolbar
# Generate the buttons :
self.btn_del = QToolButtonNormal(icons.get_icon('delete_data'))
self.btn_del.setToolTip('Delete current BRF results')
self.btn_del.clicked.connect(self.del_brf)
self.btn_del_all = QToolButtonNormal(icons.get_icon('close_all'))
self.btn_del_all.setToolTip(
'Delete all BRF results for the current water level dataset.')
self.btn_del_all.clicked.connect(self.del_all_brf)
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save current BRF graph as...')
btn_save.clicked.connect(self.select_savefig_path)
self.btn_export = QToolButtonNormal(icons.get_icon('export_data'))
self.btn_export.setToolTip('Export data to file.')
self.btn_export.clicked.connect(self.select_export_brfdata_filepath)
self.btn_copy = QToolButtonNormal('copy_clipboard')
self.btn_copy.setToolTip('Copy figure to clipboard as image.')
self.btn_copy.clicked.connect(self.copyfig_to_clipboard)
self.btn_setp = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_setp.setToolTip('Show graph layout parameters...')
self.btn_setp.clicked.connect(self.toggle_graphpannel)
# Generate the layout :
self.tbar = myqt.QFrameLayout()
buttons = [btn_save, self.btn_copy, self.btn_export, self.btn_del,
self.btn_del_all, VSep(), self.btn_prev, self.current_brf,
self.total_brf, self.btn_next, VSep(), self.btn_setp,
self.btn_language]
for btn in buttons:
if isinstance(btn, QLayout):
self.tbar.addLayout(btn, 1, self.tbar.columnCount())
else:
self.tbar.addWidget(btn, 1, self.tbar.columnCount())
row = self.tbar.columnCount()
self.tbar.addWidget(HSep(), 0, 0, 1, row+1)
self.tbar.setColumnStretch(row, 100)
self.tbar.setContentsMargins(10, 0, 10, 10) # (l, t, r, b)
# ---- Graph Canvas
self.fig_frame = QFrame()
self.fig_frame.setFrameStyle(StyleDB().frame)
self.fig_frame.setObjectName("figframe")
self.fig_frame.setStyleSheet("#figframe {background-color:white;}")
self.brf_canvas = FigureCanvasQTAgg(BRFFigure())
fflay = QGridLayout(self.fig_frame)
fflay.setContentsMargins(0, 0, 0, 0) # (left, top, right, bottom)
fflay.addWidget(self.tbar, 1, 0)
fflay.addWidget(self.brf_canvas, 0, 0)
# ---- Graph Options Panel
self.graph_opt_panel = BRFOptionsPanel()
self.graph_opt_panel.sig_graphconf_changed.connect(self.plot_brf)
# ---- Main Layout
ml = QGridLayout(self)
ml.addWidget(self.fig_frame, 0, 2)
ml.addWidget(self.graph_opt_panel, 0, 3)
ml.setColumnStretch(1, 100)
class BRFViewer(QWidget):
"""
Window that is used to show all the results produced with for the
currently selected water level dataset.
"""
def __init__(self, wldset=None, parent=None):
super(BRFViewer, self).__init__(parent)
self.__save_ddir = None
self.setWindowTitle('BRF Results Viewer')
self.setWindowIcon(icons.get_icon('master'))
self.setWindowFlags(Qt.Window |
Qt.CustomizeWindowHint |
Qt.WindowMinimizeButtonHint |
Qt.WindowCloseButtonHint)
self.__initGUI__()
self.set_wldset(wldset)
def __initGUI__(self):
# ---- Navigator
self.btn_prev = QToolButtonNormal(icons.get_icon('go_previous'))
self.btn_prev.clicked.connect(self.navigate_brf)
self.btn_next = QToolButtonNormal(icons.get_icon('go_next'))
self.btn_next.clicked.connect(self.navigate_brf)
self.current_brf = QSpinBox()
self.current_brf.setRange(0, 99)
self.current_brf.setAlignment(Qt.AlignCenter)
self.current_brf.setButtonSymbols(QAbstractSpinBox.NoButtons)
self.current_brf.setCorrectionMode(
QAbstractSpinBox.CorrectToNearestValue)
self.current_brf.valueChanged.connect(self.navigate_brf)
self.current_brf.setValue(0)
self.total_brf = QLabel('/ 0')
# ---- Language button
self.btn_language = LangToolButton()
self.btn_language.setToolTip(
"Set the language of the text shown in the graph.")
self.btn_language.sig_lang_changed.connect(self.plot_brf)
self.btn_language.setIconSize(icons.get_iconsize('normal'))
# ---- Toolbar
# Generate the buttons :
self.btn_del = QToolButtonNormal(icons.get_icon('delete_data'))
self.btn_del.setToolTip('Delete current BRF results')
self.btn_del.clicked.connect(self.del_brf)
self.btn_del_all = QToolButtonNormal(icons.get_icon('close_all'))
self.btn_del_all.setToolTip(
'Delete all BRF results for the current water level dataset.')
self.btn_del_all.clicked.connect(self.del_all_brf)
self.btn_save = btn_save = QToolButtonNormal(icons.get_icon('save'))
btn_save.setToolTip('Save current BRF graph as...')
btn_save.clicked.connect(self.select_savefig_path)
self.btn_export = QToolButtonNormal(icons.get_icon('export_data'))
self.btn_export.setToolTip('Export data to file.')
self.btn_export.clicked.connect(self.select_export_brfdata_filepath)
self.btn_copy = QToolButtonNormal('copy_clipboard')
self.btn_copy.setToolTip('Copy figure to clipboard as image.')
self.btn_copy.clicked.connect(self.copyfig_to_clipboard)
self.btn_setp = QToolButtonNormal(icons.get_icon('page_setup'))
self.btn_setp.setToolTip('Show graph layout parameters...')
self.btn_setp.clicked.connect(self.toggle_graphpannel)
# Generate the layout :
self.tbar = myqt.QFrameLayout()
buttons = [btn_save, self.btn_copy, self.btn_export, self.btn_del,
self.btn_del_all, VSep(), self.btn_prev, self.current_brf,
self.total_brf, self.btn_next, VSep(), self.btn_setp,
self.btn_language]
for btn in buttons:
if isinstance(btn, QLayout):
self.tbar.addLayout(btn, 1, self.tbar.columnCount())
else:
self.tbar.addWidget(btn, 1, self.tbar.columnCount())
row = self.tbar.columnCount()
self.tbar.addWidget(HSep(), 0, 0, 1, row+1)
self.tbar.setColumnStretch(row, 100)
self.tbar.setContentsMargins(10, 0, 10, 10) # (l, t, r, b)
# ---- Graph Canvas
self.fig_frame = QFrame()
self.fig_frame.setFrameStyle(StyleDB().frame)
self.fig_frame.setObjectName("figframe")
self.fig_frame.setStyleSheet("#figframe {background-color:white;}")
self.brf_canvas = FigureCanvasQTAgg(BRFFigure())
fflay = QGridLayout(self.fig_frame)
fflay.setContentsMargins(0, 0, 0, 0) # (left, top, right, bottom)
fflay.addWidget(self.tbar, 1, 0)
fflay.addWidget(self.brf_canvas, 0, 0)
# ---- Graph Options Panel
self.graph_opt_panel = BRFOptionsPanel()
self.graph_opt_panel.sig_graphconf_changed.connect(self.plot_brf)
# ---- Main Layout
ml = QGridLayout(self)
ml.addWidget(self.fig_frame, 0, 2)
ml.addWidget(self.graph_opt_panel, 0, 3)
ml.setColumnStretch(1, 100)
@property
def savedir(self):
"""Return a path where figures and files are saved by default."""
if self.__save_ddir is None or not osp.exists(self.__save_ddir):
try:
savedir = self.wldset.dirname
except AttributeError:
savedir = osp.dirname(__rootdir__)
finally:
return savedir
else:
return self.__save_ddir
# ---- Toolbar Handlers
def toggle_graphpannel(self):
if self.graph_opt_panel.isVisible() is True:
# Hide the panel.
self.graph_opt_panel.setVisible(False)
self.btn_setp.setAutoRaise(True)
self.btn_setp.setToolTip('Show graph layout parameters...')
w = self.size().width() - self.graph_opt_panel.size().width()
self.setFixedWidth(w)
else:
# Show the panel.
self.graph_opt_panel.setVisible(True)
self.btn_setp.setAutoRaise(False)
self.btn_setp.setToolTip('Hide graph layout parameters...')
w = self.size().width() + self.graph_opt_panel.size().width()
self.setFixedWidth(w)
def navigate_brf(self):
if self.sender() == self.btn_prev:
cur_num = self.current_brf.value() - 1
elif self.sender() == self.btn_next:
cur_num = self.current_brf.value() + 1
elif self.sender() == self.current_brf:
cur_num = self.current_brf.value()
self.current_brf.setValue(cur_num)
self.update_brfnavigate_state()
def del_brf(self):
"""Delete the graph and data of the currently selected result."""
index = self.current_brf.value()-1
name = self.wldset.get_brfname_at(index)
self.wldset.del_brf(name)
self.update_brfnavigate_state()
def del_all_brf(self):
"""Delete all the graphs and BRF results for the current dataset."""
msg = ("Do you want to delete all BRF results that were evaluated "
"for dataset <i>{}</i>?"
"<br><br>"
"All data will be lost permanently."
).format(self.wldset.name)
btn = QMessageBox.Yes | QMessageBox.No
reply = QMessageBox.question(self, 'Delete all BRF results', msg, btn)
if reply == QMessageBox.Yes:
for name in self.wldset.saved_brf():
self.wldset.del_brf(name)
self.update_brfnavigate_state()
def new_brf_added(self):
self.current_brf.setMaximum(self.wldset.brf_count())
self.current_brf.setValue(self.wldset.brf_count())
self.update_brfnavigate_state()
def update_brfnavigate_state(self):
count = self.wldset.brf_count()
self.total_brf.setText('/ %d' % count)
self.current_brf.setMinimum(min(count, 1))
self.current_brf.setMaximum(count)
curnt = self.current_brf.value()
self.tbar.setEnabled(count > 0)
self.btn_prev.setEnabled(curnt > 1)
self.btn_next.setEnabled(curnt < count)
self.btn_del.setEnabled(count > 0)
self.plot_brf()
def select_savefig_path(self):
"""
Opens a dialog to select a file path where to save the brf figure.
"""
ddir = osp.join(self.savedir, 'brf_%s' % self.wldset['Well'])
dialog = QFileDialog()
fname, ftype = dialog.getSaveFileName(
self, "Save Figure", ddir, '*.pdf;;*.svg')
ftype = ftype.replace('*', '')
if fname:
if not osp.samefile(osp.dirname(ddir), osp.dirname(fname)):
self.__save_ddir = osp.dirname(fname)
if not fname.endswith(ftype):
fname = fname + ftype
self.save_brf_fig(fname)
def save_brf_fig(self, fname):
"""Saves the current BRF figure to fname."""
self.brf_canvas.figure.savefig(fname)
def copyfig_to_clipboard(self):
"""Saves the current BRF figure to the clipboard."""
buf = io.BytesIO()
self.save_brf_fig(buf)
QApplication.clipboard().setImage(QImage.fromData(buf.getvalue()))
buf.close()
def select_export_brfdata_filepath(self):
"""
Open a dialog to select a file path where to save the brf data.
"""
fname = 'brf_' + self.wldset['Well']
if self.wldset['Well ID']:
fname += '_%s' % self.wldset['Well ID']
ddir = osp.join(self.savedir, fname)
dialog = QFileDialog()
fname, ftype = dialog.getSaveFileName(
self, "Export Data", ddir, "*.xlsx;;*.xls;;*.csv")
ftype = ftype.replace('*', '')
if fname:
if not osp.samefile(osp.dirname(ddir), osp.dirname(fname)):
self.__save_ddir = osp.dirname(fname)
if not fname.endswith(ftype):
fname = fname + ftype
self.export_brf_data(fname)
def export_brf_data(self, fname):
"""Export the current BRF data to to file."""
self.wldset.export_brf_to_csv(fname, self.current_brf.value()-1)
# ---- Others
def set_wldset(self, wldset):
self.wldset = wldset
self.setEnabled(wldset is not None)
if wldset is not None:
self.update_brfnavigate_state()
def plot_brf(self):
self.brf_canvas.figure.set_language(self.btn_language.language)
if self.wldset.brf_count() == 0:
self.brf_canvas.figure.empty_BRF()
else:
name = self.wldset.get_brfname_at(self.current_brf.value()-1)
databrf = self.wldset.get_brf(name)
well = self.wldset['Well']
msize = self.graph_opt_panel.markersize
draw_line = self.graph_opt_panel.draw_line
ymin = self.graph_opt_panel.ymin
ymax = self.graph_opt_panel.ymax
yscale = self.graph_opt_panel.yscale
xmin = self.graph_opt_panel.xmin
xmax = self.graph_opt_panel.xmax
xscale = self.graph_opt_panel.xscale
time_units = self.graph_opt_panel.time_units
date0 = databrf.date_start.strftime(format='%d/%m/%y %H:%M')
date1 = databrf.date_end.strftime(format='%d/%m/%y %H:%M')
self.brf_canvas.figure.plot_BRF(
databrf['Lag'].values, databrf['SumA'].values,
databrf['sdA'].values, date0, date1, well, msize, draw_line,
[ymin, ymax], [xmin, xmax], time_units, xscale, yscale)
self.brf_canvas.draw()
def show(self):
super(BRFViewer, self).show()
qr = self.frameGeometry()
if self.parentWidget():
parent = self.parentWidget()
wp = parent.frameGeometry().width()
hp = parent.frameGeometry().height()
cp = parent.mapToGlobal(QPoint(wp/2, hp/2))
else:
cp = QDesktopWidget().availableGeometry().center()
qr.moveCenter(cp)
self.move(qr.topLeft())
self.fig_frame.setFixedSize(self.fig_frame.size())
self.setFixedSize(self.size())
self.raise_()
if self.windowState() == Qt.WindowMinimized:
# Window is minimised. Restore it.
self.setWindowState(Qt.WindowNoState)