parouby
/
Pamhyr2
mirror of https://gitlab.com/pamhyr/pamhyr2
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
Pamhyr2/src/View/Geometry/PlotRKZ.py

375 lines
10 KiB
Python
Raw Blame History

# PlotRKC.py -- Pamhyr
# Copyright (C) 2023-2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
import logging
from tools import timer
from View.Tools.PamhyrPlot import PamhyrPlot
from PyQt5.QtWidgets import QApplication
from PyQt5.QtCore import (
Qt, QItemSelectionModel,
QItemSelection, QItemSelectionRange,
)
logger = logging.getLogger()
class PlotRKZ(PamhyrPlot):
def __init__(self, canvas=None, trad=None,
study=None, data=None, toolbar=None,
table=None, parent=None):
self._study = study
super(PlotRKZ, self).__init__(
canvas=canvas,
trad=trad,
data=data,
toolbar=toolbar,
table=table,
parent=parent
)
self._isometric_axis = False
self.line_rk_zgl = []
self.line_rk_zmin = None
self.line_rk_zmin_zmax = None
self.line_rk_zmin_zmax_highlight = None
self.label_x = self._trad["unit_rk"]
self.label_y = self._trad["unit_depth"]
self.before_plot_selected = None
self.plot_selected = None
self.after_plot_selected = None
self.parent = parent
self._table = table
self._colors = []
def onpick(self, event):
if event.mouseevent.inaxes != self.canvas.axes:
return
if event.mouseevent.button.value != 1:
return
modifiers = QApplication.keyboardModifiers()
if modifiers not in [Qt.ControlModifier,
Qt.NoModifier,
Qt.ShiftModifier]:
return
ind, point = self._closest_rk(event)
if self.parent._table is None:
return
self.parent._table.blockSignals(True)
if modifiers == Qt.ControlModifier:
rows = list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
)
if ind in rows:
rows.remove(ind)
else:
rows.append(ind)
self._select_in_table(rows)
elif modifiers == Qt.ShiftModifier:
rows = sorted(list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
))
if len(rows) > 0:
i1 = min(rows[0], rows[-1], ind)
i2 = max(rows[0], rows[-1], ind)
else:
i1 = ind
i2 = ind
self._select_range_in_table(i1, i2)
else:
self._select_in_table([ind])
self.parent._table.blockSignals(False)
return
def _closest_rk(self, event):
s = event.artist.get_segments()
x = [i[0, 0] for i in s]
mx = event.mouseevent.xdata
points = enumerate(x)
def dist_mouse(point):
x = point[1]
d = abs(mx - x)
return d
closest = min(
points, key=dist_mouse
)
return closest
def _select_in_table(self, ind):
if self._table is None:
return
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection()
if len(ind) == 0:
return
for i in ind:
index.append(QItemSelectionRange(self._table.model().index(i, 0)))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
if len(ind) > 0:
self._table.scrollTo(self._table.model().index(ind[-1], 0))
def _select_range_in_table(self, ind1, ind2):
if self._table is not None:
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection(self._table.model().index(ind1, 0),
self._table.model().index(ind2, 0))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
self._table.scrollTo(self._table.model().index(ind2, 0))
@timer
def draw(self):
self.init_axes()
if self.data is None:
return
if len(self.data.profiles) == 0:
return
profiles_defined = any(
filter(
lambda profile: len(profile) > 0,
self.data.profiles
)
)
if not profiles_defined:
self._init = False
return
self.draw_z_line()
self.draw_lr()
self.draw_gl()
self.draw_bottom()
self.draw_profiles_hs(self._data)
self.idle()
self._init = True
def draw_z_line(self):
rk = self.data.get_rk_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self._colors, self._style = self.color_highlight()
self.line_rk_zmin_zmax = self.canvas.axes.vlines(
x=rk, ymin=z_min, ymax=z_max,
color=self._colors,
linestyle=self._style,
lw=1.,
picker=10,
)
def color_highlight(self):
rows = sorted(list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
))
colors = [self.color_plot for row in range(len(self._data))]
style = ["-" for row in range(len(self._data))]
if len(rows) > 0:
for row in rows:
colors[row] = self.color_plot_current
if rows[0] > 0:
colors[rows[0]-1] = self.color_plot_previous
style[rows[0]-1] = "--"
if rows[-1] < len(self._data)-1:
colors[rows[-1]+1] = self.color_plot_next
style[rows[-1]+1] = "--"
return colors, style
def draw_lr(self):
rk = self.data.get_rk_complete_profiles()
lz = []
rz = []
self.line_lr = []
for z in self.data.get_z():
lz.append(z[0])
rz.append(z[-1])
line = self.canvas.axes.plot(
rk, lz,
color=self.color_plot_river_bottom,
linestyle="dotted",
lw=1.,
)
self.line_lr.append(line)
line = self.canvas.axes.plot(
rk, rz,
color=self.color_plot_river_bottom,
linestyle="dotted",
lw=1.,
)
self.line_lr.append(line)
def draw_gl(self):
rk = self.data.get_rk_complete_profiles()
ind = 0
self.line_rk_zgl = []
for z in self.data.get_guidelines_z():
# Is incomplete guideline?
if len(z) != len(rk):
continue
self.line_rk_zgl.append(
self.canvas.axes.plot(
rk, z, lw=1.,
color=self.colors[ind % len(self.colors)],
linestyle=self.linestyle[ind // len(self.colors)]
)
)
ind += 1
def draw_bottom(self):
rk = self.data.get_rk_complete_profiles()
z_min = self.data.get_z_min()
self.line_rk_zmin, = self.canvas.axes.plot(
rk, z_min,
linestyle=":", lw=1.5,
color='lightgrey'
)
def draw_profiles_hs(self, reach):
lhs = filter(
lambda hs: hs._input_reach.reach is reach,
filter(
lambda hs: hs._input_reach is not None,
self._study.river.hydraulic_structures.lst
)
)
for hs in lhs:
if not hs.enabled:
continue
x = hs.input_rk
if x is not None:
z_min = reach.get_z_min()
z_max = reach.get_z_max()
self.canvas.axes.plot(
[x, x],
[min(z_min), max(z_max)],
linestyle="--",
lw=1.,
color=self.color_plot_previous,
)
self.canvas.axes.annotate(
" > " + hs.name,
(x, max(z_max)),
horizontalalignment='left',
verticalalignment='top',
annotation_clip=True,
fontsize=9, color=self.color_plot_previous,
)
@timer
def update(self):
if not self._init:
self.draw()
return
self.update_lr()
self.update_gl()
self.update_current()
self.update_idle()
def update_current(self):
if self._current_data_update:
self._colors, self._style = self.color_highlight()
self.line_rk_zmin_zmax.set_colors(self._colors)
self.line_rk_zmin_zmax.set_linestyle(self._style)
def update_gl(self):
if self._current_data_update:
return
rk = self.data.get_rk_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self.line_rk_zmin.set_data(rk, z_min)
# TODO comprendre à quoi sert ce bout de code
# ========>
# self.line_rk_zmin_zmax.remove()
# self._colors, self._style = self.color_highlight()
# self.line_rk_zmin_zmax = self.canvas.axes.vlines(
# x=rk,
# ymin=z_min,
# ymax=z_max,
# color=self._colors,
# linestyle = self._style,
# lw=1.
# )
# <========
z_complete = self.data.get_guidelines_z()
try:
for i in range(len(self.line_rk_zgl)):
self.line_rk_zgl[i][0].set_data(
rk, z_complete[i]
)
except Exception as e:
logger.warning(f"Failed to update graphic RKZ: {e}")
def update_lr(self):
for line in self.line_lr:
line[0].remove()
self.draw_lr()
Reference in New Issue View Git Blame Copy Permalink