# -*- coding: utf-8 -*- #------------------------------------------------------------------------------- # This file is part of Code_Saturne, a general-purpose CFD tool. # # Copyright (C) 1998-2015 EDF S.A. # # 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 2 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, write to the Free Software Foundation, Inc., 51 Franklin # Street, Fifth Floor, Boston, MA 02110-1301, USA. #------------------------------------------------------------------------------- """ This module defines the GroundwaterLaw model data management. This module contains the following classes: - GroundwaterLawView """ #------------------------------------------------------------------------------- # Library modules import #------------------------------------------------------------------------------- import sys, logging #------------------------------------------------------------------------------- # Third-party modules #------------------------------------------------------------------------------- from code_saturne.Base.QtCore import * from code_saturne.Base.QtGui import * from code_saturne.Base.QtWidgets import * #------------------------------------------------------------------------------- # Application modules import #------------------------------------------------------------------------------- from code_saturne.model.Common import GuiParam from code_saturne.Base.QtPage import ComboModel, DoubleValidator from code_saturne.Base.QtPage import to_qvariant, from_qvariant, to_text_string from code_saturne.Pages.GroundwaterLawForm import Ui_GroundwaterLawForm from code_saturne.model.LocalizationModel import LocalizationModel, Zone from code_saturne.Pages.QMegEditorView import QMegEditorView from code_saturne.model.GroundwaterLawModel import GroundwaterLawModel from code_saturne.model.GroundwaterModel import GroundwaterModel from code_saturne.model.DefineUserScalarsModel import DefineUserScalarsModel from code_saturne.model.NotebookModel import NotebookModel #------------------------------------------------------------------------------- # log config #------------------------------------------------------------------------------- logging.basicConfig() log = logging.getLogger("GroundwaterLawView") log.setLevel(GuiParam.DEBUG) #------------------------------------------------------------------------------- # StandarItemModel class to display Head Losses Zones in a QTreeView #------------------------------------------------------------------------------- class StandardItemModelGroundwaterLaw(QStandardItemModel): def __init__(self): QStandardItemModel.__init__(self) self.headers = [self.tr("Label"), self.tr("Zone"), self.tr("Selection criteria")] self.setColumnCount(len(self.headers)) self.dataDarcyLawZones = [] def data(self, index, role): if not index.isValid(): return to_qvariant() if role == Qt.DisplayRole: return to_qvariant(self.dataDarcyLawZones[index.row()][index.column()]) return to_qvariant() def flags(self, index): if not index.isValid(): return Qt.ItemIsEnabled else: return Qt.ItemIsEnabled | Qt.ItemIsSelectable def headerData(self, section, orientation, role): if orientation == Qt.Horizontal and role == Qt.DisplayRole: return to_qvariant(self.headers[section]) return to_qvariant() def setData(self, index, value, role): self.dataChanged.emit(index, index) return True def insertItem(self, label, name, local): line = [label, name, local] self.dataDarcyLawZones.append(line) row = self.rowCount() self.setRowCount(row+1) def getItem(self, row): return self.dataDarcyLawZones[row] #------------------------------------------------------------------------------- # Main view class #------------------------------------------------------------------------------- class GroundwaterLawView(QWidget, Ui_GroundwaterLawForm): def __init__(self, parent, case): """ Constructor """ QWidget.__init__(self, parent) Ui_GroundwaterLawForm.__init__(self) self.setupUi(self) self.case = case self.case.undoStopGlobal() self.mdl = GroundwaterLawModel(self.case) self.notebook = NotebookModel(self.case) self.list_scalars = [] self.m_sca = DefineUserScalarsModel(self.case) for s in self.m_sca.getUserScalarNameList(): self.list_scalars.append((s, self.tr("Additional scalar"))) # Create the Page layout. # Model and QTreeView self.modelGroundwaterLaw = StandardItemModelGroundwaterLaw() self.treeView.setModel(self.modelGroundwaterLaw) # Combo model self.modelGroundwaterLawType = ComboModel(self.comboBoxType, 2, 1) self.modelGroundwaterLawType.addItem(self.tr("User law"), 'user') self.modelGroundwaterLawType.addItem(self.tr("Van Genuchten law"), 'VanGenuchten') self.modelNameDiff = ComboModel(self.comboBoxNameDiff,1,1) # Set up validators self.lineEditKs.setValidator(DoubleValidator(self.lineEditKs)) self.lineEditKsXX.setValidator(DoubleValidator(self.lineEditKsXX)) self.lineEditKsYY.setValidator(DoubleValidator(self.lineEditKsYY)) self.lineEditKsZZ.setValidator(DoubleValidator(self.lineEditKsZZ)) self.lineEditKsXY.setValidator(DoubleValidator(self.lineEditKsXY)) self.lineEditKsXZ.setValidator(DoubleValidator(self.lineEditKsXZ)) self.lineEditKsYZ.setValidator(DoubleValidator(self.lineEditKsYZ)) self.lineEditThetas.setValidator(DoubleValidator(self.lineEditThetas)) self.lineEditThetar.setValidator(DoubleValidator(self.lineEditThetar)) self.lineEditN.setValidator(DoubleValidator(self.lineEditN)) self.lineEditL.setValidator(DoubleValidator(self.lineEditL)) self.lineEditAlpha.setValidator(DoubleValidator(self.lineEditAlpha)) self.lineEditLongitudinal.setValidator(DoubleValidator(self.lineEditLongitudinal)) self.lineEditTransverse.setValidator(DoubleValidator(self.lineEditTransverse)) self.lineEditKsSaturated.setValidator(DoubleValidator(self.lineEditKsSaturated)) self.lineEditKsSaturatedXX.setValidator(DoubleValidator(self.lineEditKsSaturatedXX)) self.lineEditKsSaturatedYY.setValidator(DoubleValidator(self.lineEditKsSaturatedYY)) self.lineEditKsSaturatedZZ.setValidator(DoubleValidator(self.lineEditKsSaturatedZZ)) self.lineEditKsSaturatedXY.setValidator(DoubleValidator(self.lineEditKsSaturatedXY)) self.lineEditKsSaturatedXZ.setValidator(DoubleValidator(self.lineEditKsSaturatedXZ)) self.lineEditKsSaturatedYZ.setValidator(DoubleValidator(self.lineEditKsSaturatedYZ)) self.lineEditThetasSaturated.setValidator(DoubleValidator(self.lineEditThetasSaturated)) self.lineEditDispersion.setValidator(DoubleValidator(self.lineEditDispersion)) self.lineEditSoilDensity.setValidator(DoubleValidator(self.lineEditSoilDensity)) self.lineEditDiffusivity.setValidator(DoubleValidator(self.lineEditDiffusivity)) self.lineEditKd.setValidator(DoubleValidator(self.lineEditKd)) self.lineEditkplus.setValidator(DoubleValidator(self.lineEditkplus)) self.lineEditkminus.setValidator(DoubleValidator(self.lineEditkminus)) self.scalar = "" scalar_list = self.m_sca.getUserScalarNameList() for s in self.m_sca.getScalarsVarianceList(): if s in scalar_list: scalar_list.remove(s) if scalar_list != []: self.scalar = scalar_list[0] for scalar in scalar_list: self.modelNameDiff.addItem(scalar) # Connections self.treeView.pressed[QModelIndex].connect(self.slotSelectGroundwaterLawZones) self.comboBoxType.activated[str].connect(self.slotGroundwaterLaw) self.lineEditKs.textChanged[str].connect(self.slotKs) self.lineEditKsXX.textChanged[str].connect(self.slotKsXX) self.lineEditKsYY.textChanged[str].connect(self.slotKsYY) self.lineEditKsZZ.textChanged[str].connect(self.slotKsZZ) self.lineEditKsXY.textChanged[str].connect(self.slotKsXY) self.lineEditKsXZ.textChanged[str].connect(self.slotKsXZ) self.lineEditKsYZ.textChanged[str].connect(self.slotKsYZ) self.lineEditThetas.textChanged[str].connect(self.slotThetas) self.lineEditThetar.textChanged[str].connect(self.slotThetar) self.lineEditN.textChanged[str].connect(self.slotN) self.lineEditL.textChanged[str].connect(self.slotL) self.lineEditAlpha.textChanged[str].connect(self.slotAlpha) self.lineEditLongitudinal.textChanged[str].connect(self.slotLongitudinal) self.lineEditTransverse.textChanged[str].connect(self.slotTransverse) self.lineEditKsSaturated.textChanged[str].connect(self.slotKsSat) self.lineEditKsSaturatedXX.textChanged[str].connect(self.slotKsXXSat) self.lineEditKsSaturatedYY.textChanged[str].connect(self.slotKsYYSat) self.lineEditKsSaturatedZZ.textChanged[str].connect(self.slotKsZZSat) self.lineEditKsSaturatedXY.textChanged[str].connect(self.slotKsXYSat) self.lineEditKsSaturatedXZ.textChanged[str].connect(self.slotKsXZSat) self.lineEditKsSaturatedYZ.textChanged[str].connect(self.slotKsYZSat) self.lineEditThetasSaturated.textChanged[str].connect(self.slotThetasSat) self.lineEditDispersion.textChanged[str].connect(self.slotDispersion) self.lineEditSoilDensity.textChanged[str].connect(self.slotSoilDensity) self.pushButtonUserLaw.clicked.connect(self.slotFormula) self.comboBoxNameDiff.activated[str].connect(self.slotNameDiff) self.lineEditDiffusivity.textChanged[str].connect(self.slotDiffusivity) self.lineEditKd.textChanged[str].connect(self.slotKd) self.lineEditkplus.textChanged[str].connect(self.slotkplus) self.lineEditkminus.textChanged[str].connect(self.slotkminus) # Initialize Widgets self.entriesNumber = -1 d = self.mdl.getNameAndLocalizationZone() liste=[] liste=list(d.items()) t=[] for t in liste : NamLoc=t[1] Lab=t[0] self.modelGroundwaterLaw.insertItem(Lab, NamLoc[0],NamLoc[1]) self.forgetStandardWindows() self.case.undoStartGlobal() @pyqtSlot("QModelIndex") def slotSelectGroundwaterLawZones(self, index): label, name, local = self.modelGroundwaterLaw.getItem(index.row()) self.entriesNumber = index.row() if hasattr(self, "modelScalars"): del self.modelScalars log.debug("slotSelectGroundwaterLawZones label %s " % label ) self.groupBoxGroundProperties.show() # ground properties self.groupBoxSoilDensity.show() value = self.mdl.getSoilDensity(name) self.lineEditSoilDensity.setText(str(value)) if GroundwaterModel(self.case).getUnsaturatedZone() == "true": self.groupBoxType.show() choice = self.mdl.getGroundwaterLawModel(name) self.modelGroundwaterLawType.setItem(str_model=choice) if choice == "user": self.groupBoxVanGenuchten.hide() self.groupBoxUser.show() exp = self.mdl.getGroundwaterLawFormula(name) if exp: self.pushButtonUserLaw.setStyleSheet("background-color: green") self.pushButtonUserLaw.setToolTip(exp) else: self.pushButtonUserLaw.setStyleSheet("background-color: red") else: self.groupBoxVanGenuchten.show() self.groupBoxUser.hide() self.initializeVanGenuchten(name) else: self.groupBoxSaturated.show() if GroundwaterModel(self.case).getPermeabilityType() == "anisotropic": self.labelKsSaturated.hide() self.lineEditKsSaturated.hide() value = self.mdl.getValue(name, "ks_xx") self.lineEditKsSaturatedXX.setText(str(value)) value = self.mdl.getValue(name, "ks_yy") self.lineEditKsSaturatedYY.setText(str(value)) value = self.mdl.getValue(name, "ks_zz") self.lineEditKsSaturatedZZ.setText(str(value)) value = self.mdl.getValue(name, "ks_xy") self.lineEditKsSaturatedXY.setText(str(value)) value = self.mdl.getValue(name, "ks_xz") self.lineEditKsSaturatedXZ.setText(str(value)) value = self.mdl.getValue(name, "ks_yz") self.lineEditKsSaturatedYZ.setText(str(value)) else: self.labelKsSaturatedXX.hide() self.labelKsSaturatedYY.hide() self.labelKsSaturatedZZ.hide() self.labelKsSaturatedXY.hide() self.labelKsSaturatedXZ.hide() self.labelKsSaturatedYZ.hide() self.lineEditKsSaturatedXX.hide() self.lineEditKsSaturatedYY.hide() self.lineEditKsSaturatedZZ.hide() self.lineEditKsSaturatedXY.hide() self.lineEditKsSaturatedXZ.hide() self.lineEditKsSaturatedYZ.hide() value = self.mdl.getValue(name, "ks") self.lineEditKsSaturated.setText(str(value)) value = self.mdl.getValue(name, "thetas") self.lineEditThetasSaturated.setText(str(value)) # solute properties scal = self.scalar if scal == "": self.groupBoxSoluteProperties.hide() else : self.groupBoxSoluteProperties.show() self.modelNameDiff.setItem(str_model=str(scal)) value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'diffusivity') self.lineEditDiffusivity.setText(str(value)) value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'kd') self.lineEditKd.setText(str(value)) # chemistry model if GroundwaterModel(self.case).getChemistryModel(scal) == "EK": self.lineEditkplus.show() self.label_kplus.show() self.lineEditkminus.show() self.label_kminus.show() value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'kplus') self.lineEditkplus.setText(str(value)) value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'kminus') self.lineEditkminus.setText(str(value)) else: self.lineEditkplus.hide() self.label_kplus.hide() self.lineEditkminus.hide() self.label_kminus.hide() if GroundwaterModel(self.case).getDispersionType() == 'anisotropic': self.groupBoxIsotropicDispersion.hide() self.groupBoxAnisotropicDispersion.show() value = self.mdl.getDispersionCoefficient(name, "longitudinal") self.lineEditLongitudinal.setText(str(value)) value = self.mdl.getDispersionCoefficient(name, "transverse") self.lineEditTransverse.setText(str(value)) else: self.groupBoxIsotropicDispersion.show() self.groupBoxAnisotropicDispersion.hide() value = self.mdl.getDispersionCoefficient(name, "isotropic") self.lineEditDispersion.setText(str(value)) def initializeVanGenuchten(self, name): """ initialize variables for groupBoxVanGenuchten """ value = self.mdl.getValue(name, "thetas") self.lineEditThetas.setText(str(value)) value = self.mdl.getValue(name, "thetar") self.lineEditThetar.setText(str(value)) value = self.mdl.getValue(name, "n") self.lineEditN.setText(str(value)) value = self.mdl.getValue(name, "l") self.lineEditL.setText(str(value)) value = self.mdl.getValue(name, "alpha") self.lineEditAlpha.setText(str(value)) if GroundwaterModel(self.case).getPermeabilityType() == "anisotropic": self.labelKs.hide() self.lineEditKs.hide() value = self.mdl.getValue(name, "ks_xx") self.lineEditKsXX.setText(str(value)) value = self.mdl.getValue(name, "ks_yy") self.lineEditKsYY.setText(str(value)) value = self.mdl.getValue(name, "ks_zz") self.lineEditKsZZ.setText(str(value)) value = self.mdl.getValue(name, "ks_xy") self.lineEditKsXY.setText(str(value)) value = self.mdl.getValue(name, "ks_xz") self.lineEditKsXZ.setText(str(value)) value = self.mdl.getValue(name, "ks_yz") self.lineEditKsYZ.setText(str(value)) else: self.labelKsXX.hide() self.labelKsYY.hide() self.labelKsZZ.hide() self.labelKsXY.hide() self.labelKsXZ.hide() self.labelKsYZ.hide() self.lineEditKsXX.hide() self.lineEditKsYY.hide() self.lineEditKsZZ.hide() self.lineEditKsXY.hide() self.lineEditKsXZ.hide() self.lineEditKsYZ.hide() value = self.mdl.getValue(name, "ks") self.lineEditKs.setText(str(value)) def forgetStandardWindows(self): """ For forget standard windows """ self.groupBoxGroundProperties.hide() self.groupBoxType.hide() self.groupBoxUser.hide() self.groupBoxVanGenuchten.hide() self.groupBoxSaturated.hide() self.groupBoxSoluteProperties.hide() @pyqtSlot(str) def slotGroundwaterLaw(self, text): """ Method to call 'getState' with correct arguements for 'rho' """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) choice = self.modelGroundwaterLawType.dicoV2M[str(text)] self.mdl.setGroundwaterLawModel(name, choice) if choice == "user": self.groupBoxVanGenuchten.hide() self.groupBoxUser.show() exp = self.mdl.getGroundwaterLawFormula(name) if exp: self.pushButtonUserLaw.setStyleSheet("background-color: green") self.pushButtonUserLaw.setToolTip(exp) else: self.pushButtonUserLaw.setStyleSheet("background-color: red") else: self.groupBoxVanGenuchten.show() self.groupBoxUser.hide() self.initializeVanGenuchten(name) @pyqtSlot(str) def slotKs(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKs.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks", val) @pyqtSlot(str) def slotKsXX(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsXX.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_xx", val) @pyqtSlot(str) def slotKsYY(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsYY.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_yy", val) @pyqtSlot(str) def slotKsZZ(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsZZ.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_zz", val) @pyqtSlot(str) def slotKsXY(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsXY.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_xy", val) @pyqtSlot(str) def slotKsXZ(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsXZ.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_xz", val) @pyqtSlot(str) def slotKsYZ(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsYZ.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_yz", val) @pyqtSlot(str) def slotThetas(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditThetas.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "thetas", val) @pyqtSlot(str) def slotKsSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturated.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks", val) @pyqtSlot(str) def slotKsXXSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturatedXX.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_xx", val) @pyqtSlot(str) def slotKsYYSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturatedYY.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_yy", val) @pyqtSlot(str) def slotKsZZSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturatedZZ.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_zz", val) @pyqtSlot(str) def slotKsXYSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturatedXY.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_xy", val) @pyqtSlot(str) def slotKsXZSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturatedXZ.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_xz", val) @pyqtSlot(str) def slotKsYZSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKsSaturatedYZ.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "ks_yz", val) @pyqtSlot(str) def slotThetasSat(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditThetasSaturated.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "thetas", val) @pyqtSlot(str) def slotThetar(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditThetar.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "thetar", val) @pyqtSlot(str) def slotN(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditN.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "n", val) @pyqtSlot(str) def slotL(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditL.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "l", val) @pyqtSlot(str) def slotAlpha(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditAlpha.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setValue(name, "alpha", val) @pyqtSlot(str) def slotLongitudinal(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditLongitudinal.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setDispersionCoefficient(name, "longitudinal", val) @pyqtSlot(str) def slotTransverse(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditTransverse.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setDispersionCoefficient(name, "transverse", val) @pyqtSlot(str) def slotDispersion(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditDispersion.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setDispersionCoefficient(name, "isotropic", val) @pyqtSlot(str) def slotSoilDensity(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditSoilDensity.validator().state == QValidator.Acceptable: val = float(text) self.mdl.setSoilDensity(name, val) @pyqtSlot() def slotFormula(self): """ User formula for Groundwater functions """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) exp, req, sym = self.mdl.getGroundwaterLawFormulaComponents(name) exa = """#example: \n""" + self.mdl.getDefaultGroundwaterLawFormula() dialog = QMegEditorView(parent = self, function_type = 'vol', zone_name = label, variable_name = 'capacity+saturation+permeability', expression = exp, required = req, symbols = sym, examples = exa) if dialog.exec_(): result = dialog.get_result() log.debug("slotFormula -> %s" % str(result)) self.mdl.setGroundwaterLawFormula(name, str(result)) self.pushButtonUserLaw.setStyleSheet("background-color: green") self.pushButtonUserLaw.setToolTip(result) @pyqtSlot(str) def slotNameDiff(self, text): """ Method to choose the scalar which properties shall be changed """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) log.debug("slotNameDiff -> %s" % (text)) self.scalar = str(text) scal = self.scalar value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'diffusivity') self.lineEditDiffusivity.setText(str(value)) value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'kd') self.lineEditKd.setText(str(value)) if GroundwaterModel(self.case).getChemistryModel(scal) == "EK": self.lineEditkplus.show() self.label_kplus.show() self.lineEditkminus.show() self.label_kminus.show() value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'kplus') self.lineEditkplus.setText(str(value)) value = self.mdl.getGroundWaterScalarPropertyByZone(scal, name, 'kminus') self.lineEditkminus.setText(str(value)) else: self.lineEditkplus.hide() self.label_kplus.hide() self.lineEditkminus.hide() self.label_kminus.hide() @pyqtSlot(str) def slotDiffusivity(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditDiffusivity.validator().state == QValidator.Acceptable: val = float(text) scal = self.scalar self.mdl.setGroundWaterScalarPropertyByZone(scal, name, 'diffusivity', val) @pyqtSlot(str) def slotKd(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditKd.validator().state == QValidator.Acceptable: val = float(text) scal = self.scalar self.mdl.setGroundWaterScalarPropertyByZone(scal, name, 'kd', val) @pyqtSlot(str) def slotkplus(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditkplus.validator().state == QValidator.Acceptable: val = float(text) scal = self.scalar self.mdl.setGroundWaterScalarPropertyByZone(scal, name, 'kplus', val) @pyqtSlot(str) def slotkminus(self, text): """ """ label, name, local = self.modelGroundwaterLaw.getItem(self.entriesNumber) if self.lineEditkminus.validator().state == QValidator.Acceptable: val = float(text) scal = self.scalar self.mdl.setGroundWaterScalarPropertyByZone(scal, name, 'kminus', val) def tr(self, text): """ Translation """ return text #------------------------------------------------------------------------------- # Testing part #------------------------------------------------------------------------------- if __name__ == "__main__": pass #------------------------------------------------------------------------------- # End #-------------------------------------------------------------------------------