mirror of https://gitlab.com/pamhyr/pamhyr2
130 lines
3.6 KiB
Python
130 lines
3.6 KiB
Python
# CustomPlotValuesSelectionDialog.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 -*-
|
|
|
|
from View.Tools.PamhyrWindow import PamhyrDialog
|
|
|
|
from PyQt5.QtWidgets import (
|
|
QRadioButton, QCheckBox, QVBoxLayout,
|
|
)
|
|
|
|
from View.Results.translate import ResultsTranslate
|
|
|
|
|
|
class CustomExportAdisDialog(PamhyrDialog):
|
|
_pamhyr_ui = "CustomExportAdisDialog"
|
|
_pamhyr_name = "Custom Plot Selection"
|
|
|
|
def __init__(self, pollutants, parent=None):
|
|
trad = ResultsTranslate()
|
|
super(CustomExportAdisDialog, self).__init__(
|
|
title=trad[self._pamhyr_name],
|
|
options=[],
|
|
trad=trad,
|
|
parent=parent
|
|
)
|
|
|
|
if pollutants is not None:
|
|
self.pollutants = pollutants
|
|
if "total_sediment" in self.pollutants:
|
|
self.pollutants.remove("total_sediment")
|
|
else:
|
|
self.pollutants = pollutants
|
|
|
|
self._available_values_x = self._trad.get_dict("values_x")
|
|
self._available_values_y = self._trad.get_dict("values_y_pol")
|
|
|
|
self.setup_radio_buttons_x()
|
|
self.setup_radio_buttons_pol()
|
|
self.setup_check_boxes()
|
|
|
|
self.value = None
|
|
|
|
def setup_radio_buttons_x(self):
|
|
self._radio = []
|
|
layout = self.find(QVBoxLayout, "verticalLayout_x")
|
|
|
|
for value in self._available_values_x:
|
|
btn = QRadioButton(
|
|
self._available_values_x[value],
|
|
parent=self
|
|
)
|
|
self._radio.append((value, btn))
|
|
layout.addWidget(btn)
|
|
|
|
self._radio[0][1].setChecked(True)
|
|
layout.addStretch()
|
|
|
|
def setup_radio_buttons_pol(self):
|
|
self._radio2 = []
|
|
layout = self.find(QVBoxLayout, "verticalLayout_pol")
|
|
|
|
for value in self.pollutants:
|
|
btn = QRadioButton(
|
|
value,
|
|
parent=self
|
|
)
|
|
self._radio2.append((value, btn))
|
|
layout.addWidget(btn)
|
|
|
|
self._radio2[0][1].setChecked(True)
|
|
layout.addStretch()
|
|
|
|
def setup_check_boxes(self):
|
|
self._check = []
|
|
layout = self.find(QVBoxLayout, "verticalLayout_y")
|
|
|
|
for value in self._available_values_y:
|
|
btn = QCheckBox(
|
|
self._available_values_y[value],
|
|
parent=self
|
|
)
|
|
self._check.append((value, btn))
|
|
layout.addWidget(btn)
|
|
|
|
self._check[0][1].setChecked(True)
|
|
layout.addStretch()
|
|
|
|
def accept(self):
|
|
x = next(
|
|
filter(
|
|
lambda r: r[1].isChecked(),
|
|
self._radio
|
|
)
|
|
)[0]
|
|
|
|
y = list(
|
|
map(
|
|
lambda b: b[0],
|
|
filter(
|
|
lambda b: b[1].isChecked(),
|
|
self._check
|
|
)
|
|
)
|
|
)
|
|
|
|
pol = next(
|
|
filter(
|
|
lambda r: r[1].isChecked(),
|
|
self._radio2
|
|
)
|
|
)[0]
|
|
|
|
self.value = x, y, pol
|
|
|
|
super().accept()
|