# -*- coding: utf-8 -*- #------------------------------------------------------------------------------- # This file is part of Code_Saturne, a general-purpose CFD tool. # # Copyright (C) 1998-2019 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 contains the following classes: - BoundaryConditionsVelocityInletView """ #------------------------------------------------------------------------------- # Standard modules #------------------------------------------------------------------------------- import 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.Pages.BoundaryConditionsVelocityInletForm import Ui_BoundaryConditionsVelocityInletForm from code_saturne.model.Common import GuiParam from code_saturne.Base.QtPage import DoubleValidator, ComboModel, from_qvariant from code_saturne.model.LocalizationModel import LocalizationModel, Zone from code_saturne.model.Boundary import Boundary from code_saturne.model.CompressibleModel import CompressibleModel from code_saturne.model.GasCombustionModel import GasCombustionModel from code_saturne.Pages.QMegEditorView import QMegEditorView from code_saturne.model.NotebookModel import NotebookModel #------------------------------------------------------------------------------- # log config #------------------------------------------------------------------------------- logging.basicConfig() log = logging.getLogger("BoundaryConditionsVelocityInletView") log.setLevel(GuiParam.DEBUG) #------------------------------------------------------------------------------- # Main class #------------------------------------------------------------------------------- class BoundaryConditionsVelocityInletView(QWidget, Ui_BoundaryConditionsVelocityInletForm): """ Boundary condition for velocity in inlet, without particular physics. """ def __init__(self, parent): """ Constructor """ QWidget.__init__(self, parent) Ui_BoundaryConditionsVelocityInletForm.__init__(self) self.setupUi(self) self.thermodynamic_list = ['Pressure', 'Density', 'Temperature', 'Energy'] def setup(self, case): """ Setup the widget """ self.case = case self.__boundary = None self.case.undoStopGlobal() self.mdl = CompressibleModel(self.case) self.gas = GasCombustionModel(self.case) self.notebook = NotebookModel(self.case) # Connections self.comboBoxVelocity.activated[str].connect(self.__slotChoiceVelocity) self.lineEditVelocity.textChanged[str].connect(self.__slotVelocityValue) self.comboBoxDirection.activated[str].connect(self.__slotChoiceDirection) self.lineEditDirectionX.textChanged[str].connect(self.__slotDirX) self.lineEditDirectionY.textChanged[str].connect(self.__slotDirY) self.lineEditDirectionZ.textChanged[str].connect(self.__slotDirZ) self.comboBoxTypeInlet.activated[str].connect(self.__slotInletType) self.checkBoxPressure.clicked.connect(self.__slotPressure) self.checkBoxDensity.clicked.connect(self.__slotDensity) self.checkBoxTemperature.clicked.connect(self.__slotTemperature) self.checkBoxEnergy.clicked.connect(self.__slotEnergy) self.lineEditPressure.textChanged[str].connect(self.__slotPressureValue) self.lineEditDensity.textChanged[str].connect(self.__slotDensityValue) self.lineEditTotalPressure.textChanged[str].connect(self.__slotTotalPressure) self.lineEditTotalEnthalpy.textChanged[str].connect(self.__slotTotalEnthalpy) self.lineEditTemperature.textChanged[str].connect(self.__slotTemperatureValue) self.lineEditEnergy.textChanged[str].connect(self.__slotEnergyValue) self.comboBoxTypeInletGasComb.activated[str].connect(self.__slotInletTypeGasComb) self.lineEditTemperatureGasComb.textChanged[str].connect(self.__slotTemperatureGasComb) self.lineEditFraction.textChanged[str].connect(self.__slotMeanMixtureFraction) # Combo models self.modelVelocity = ComboModel(self.comboBoxVelocity, 6, 1) self.modelVelocity.addItem(self.tr("norm"), 'norm') self.modelVelocity.addItem(self.tr("mass flow rate"), 'flow1') self.modelVelocity.addItem(self.tr("volumic flow rate"), 'flow2') self.modelVelocity.addItem(self.tr("norm (user law)"), 'norm_formula') self.modelVelocity.addItem(self.tr("mass flow rate (user law)"), 'flow1_formula') self.modelVelocity.addItem(self.tr("volumic flow rate (user law)"), 'flow2_formula') self.modelDirection = ComboModel(self.comboBoxDirection, 3, 1) self.modelDirection.addItem(self.tr("normal direction to the inlet"), 'normal') self.modelDirection.addItem(self.tr("specified coordinates"), 'coordinates') self.modelDirection.addItem(self.tr("user profile"), 'formula') self.modelTypeInlet = ComboModel(self.comboBoxTypeInlet, 2, 1) self.modelTypeInlet.addItem(self.tr("imposed inlet"), 'imposed_inlet') self.modelTypeInlet.addItem(self.tr("subsonic inlet (imposed total pressure and total enthalpy)"), \ 'subsonic_inlet_PH') self.modelTypeInletGasComb = ComboModel(self.comboBoxTypeInletGasComb, 2, 1) model = self.gas.getGasCombustionModel() if model == 'lwp' or model =='ebu': self.modelTypeInletGasComb.addItem(self.tr("Unburned gas"), 'unburned') self.modelTypeInletGasComb.addItem(self.tr("Burned gas"), 'burned') elif model == 'd3p': self.modelTypeInletGasComb.addItem(self.tr("Oxydant"), 'oxydant') self.modelTypeInletGasComb.addItem(self.tr("Fuel"), 'fuel') # Validators validatorVelocity = DoubleValidator(self.lineEditVelocity) validatorX = DoubleValidator(self.lineEditDirectionX) validatorY = DoubleValidator(self.lineEditDirectionY) validatorZ = DoubleValidator(self.lineEditDirectionZ) validatorP = DoubleValidator(self.lineEditPressure, min = 0.0) validatorD = DoubleValidator(self.lineEditDensity, min = 0.0) validatorT = DoubleValidator(self.lineEditTemperature, min = 0.0) validatorE = DoubleValidator(self.lineEditEnergy, min = 0.0) validatorP2 = DoubleValidator(self.lineEditTotalPressure, min = 0.0) validatorH2 = DoubleValidator(self.lineEditTotalEnthalpy, min = 0.0) validatorTemp = DoubleValidator(self.lineEditTemperatureGasComb, min=0.) validatorFrac = DoubleValidator(self.lineEditFraction, min=0., max=1.) # Apply validators self.lineEditVelocity.setValidator(validatorVelocity) self.lineEditDirectionX.setValidator(validatorX) self.lineEditDirectionY.setValidator(validatorY) self.lineEditDirectionZ.setValidator(validatorZ) self.lineEditPressure.setValidator(validatorP) self.lineEditDensity.setValidator(validatorD) self.lineEditTemperature.setValidator(validatorT) self.lineEditEnergy.setValidator(validatorE) self.lineEditTotalPressure.setValidator(validatorP2) self.lineEditTotalEnthalpy.setValidator(validatorH2) self.lineEditTemperatureGasComb.setValidator(validatorTemp) self.lineEditFraction.setValidator(validatorFrac) self.pushButtonVelocityFormula.clicked.connect(self.__slotVelocityFormula) self.pushButtonDirectionFormula.clicked.connect(self.__slotDirectionFormula) self.case.undoStartGlobal() def showWidget(self, boundary): """ Show the widget """ self.__boundary = boundary # Initialize velocity choice = self.__boundary.getVelocityChoice() self.modelVelocity.setItem(str_model=choice) self.__updateLabel() if choice[-7:] == "formula": self.pushButtonVelocityFormula.setEnabled(True) self.lineEditVelocity.setEnabled(False) else: self.pushButtonVelocityFormula.setEnabled(False) self.lineEditVelocity.setEnabled(True) v = self.__boundary.getVelocity() self.lineEditVelocity.setText(str(v)) # Initialize direction choice = self.__boundary.getDirectionChoice() self.modelDirection.setItem(str_model=choice) text = self.modelDirection.dicoM2V[choice] if choice == "formula": self.pushButtonDirectionFormula.setEnabled(True) self.frameDirectionCoordinates.hide() elif choice == "coordinates": self.pushButtonDirectionFormula.setEnabled(False) self.frameDirectionCoordinates.show() v = self.__boundary.getDirection('direction_x') self.lineEditDirectionX.setText(str(v)) v = self.__boundary.getDirection('direction_y') self.lineEditDirectionY.setText(str(v)) v = self.__boundary.getDirection('direction_z') self.lineEditDirectionZ.setText(str(v)) elif choice == "normal": self.pushButtonDirectionFormula.setEnabled(False) self.frameDirectionCoordinates.hide() self.initialize() def initialize(self): """ Initialize widget for compressible """ self.comboBoxVelocity.show() self.lineEditVelocity.show() self.labelUnitVelocity.show() self.pushButtonVelocityFormula.show() # Initialize thermodynamic value if self.mdl.getCompressibleModel() != 'off': inlet_type = self.__boundary.getInletType() self.modelTypeInlet.setItem(str_model = inlet_type) self.__boundary.setInletType(inlet_type) if inlet_type == 'imposed_inlet': self.groupBoxThermodynamic.show() self.frameDensity.hide() for name in self.thermodynamic_list: __checkBox = getattr(self, "checkBox" + name) __lineEdit = getattr(self, "lineEdit" + name) __checkBox.setChecked(False) __checkBox.setEnabled(False) __lineEdit.setEnabled(False) __lineEdit.clear() box_list = self.__boundary.getCheckedBoxList() if len(box_list) == 0: for name in self.thermodynamic_list: __checkBox = getattr(self, "checkBox" + name) __lineEdit = getattr(self, "lineEdit" + name) __checkBox.setEnabled(True) elif len(box_list) == 1: for name in self.thermodynamic_list: __checkBox = getattr(self, "checkBox" + name) __lineEdit = getattr(self, "lineEdit" + name) __checkBox.setEnabled(True) box = box_list[0] if box == 'Temperature': self.checkBoxEnergy.setEnabled(False) elif box == 'Energy': self.checkBoxTemperature.setEnabled(False) __checkBox = getattr(self, "checkBox" + box) __checkBox.setChecked(True) __lineEdit = getattr(self, "lineEdit" + box) __lineEdit.setEnabled(True) v1 = self.__boundary.getListValue()[0] __lineEdit.setText(str(v1)) elif len(box_list) == 2: v1,v2 = self.__boundary.getListValue() for name in box_list: __checkBox = getattr(self, "checkBox" + name) __lineEdit = getattr(self, "lineEdit" + name) __checkBox.setEnabled(True) __checkBox.setChecked(True) __lineEdit.setEnabled(True) if v1 >= 0.: __lineEdit.setText(str(v1)) else: __lineEdit.setText(str(v2)) v1 = -1. elif inlet_type == 'subsonic_inlet_PH': self.comboBoxVelocity.hide() self.lineEditVelocity.hide() self.labelUnitVelocity.hide() self.pushButtonVelocityFormula.hide() self.groupBoxThermodynamic.hide() self.frameDensity.show() pressure = self.__boundary.getThermoValue('total_pressure') self.lineEditTotalPressure.setText(str(pressure)) enthalpy = self.__boundary.getThermoValue('enthalpy') self.lineEditTotalEnthalpy.setText(str(enthalpy)) else: self.groupBoxCompressible.hide() # Initialize temperature and mean mixture fraction model = self.gas.getGasCombustionModel() if model != 'off': self.groupBoxGasCombustion.show() inlet_type = self.__boundary.getInletGasCombustionType() self.modelTypeInletGasComb.setItem(str_model = inlet_type) if model == 'd3p': self.lineEditTemperatureGasComb.hide() self.labelTemperature_2.hide() self.labelUnitTemp.hide() self.lineEditFraction.setEnabled(False) f = self.__boundary.setMeanMixtureFraction(1) self.lineEditFraction.setText(str(1) if inlet_type == 'oxydant' else str(0)) else : self.lineEditTemperatureGasComb.show() self.labelTemperature_2.show() self.labelUnitTemp.show() t = self.__boundary.getGasCombustionTemperature() self.lineEditTemperatureGasComb.setText(str(t)) self.lineEditFraction.setEnabled(True) f = self.__boundary.getMeanMixtureFraction() self.lineEditFraction.setText(str(f)) else: self.groupBoxGasCombustion.hide() self.show() def hideWidget(self): """ Hide all """ self.hide() def getVelocityExpression(self, boundary): """ Get the mathematical expression for velocity """ exp = boundary.getVelocity() c = boundary.getVelocityChoice() if c == 'norm_formula': n = 'u_norm' elif c == 'flow1_formula': n = 'q_m' elif c == 'flow2_formula': n = 'q_v' return exp, n @pyqtSlot(str) def __slotChoiceVelocity(self, text): """ Private slot. Input the velocity boundary type choice (norm, ). @type text: C{QString} @param text: velocity boundary type choice. """ c = self.modelVelocity.dicoV2M[str(text)] log.debug("slotChoiceVelocity: %s " % c) self.__boundary.setVelocityChoice(c) if c[-7:] == "formula": self.pushButtonVelocityFormula.setEnabled(True) exp = self.__boundary.getVelocity() if exp: self.pushButtonVelocityFormula.setStyleSheet("background-color: green") self.pushButtonVelocityFormula.setToolTip(exp) else: self.pushButtonVelocityFormula.setStyleSheet("background-color: red") self.lineEditVelocity.setEnabled(False) self.lineEditVelocity.setText("") else: self.pushButtonVelocityFormula.setEnabled(False) self.pushButtonVelocityFormula.setStyleSheet("background-color: None") self.lineEditVelocity.setEnabled(True) v = self.__boundary.getVelocity() self.lineEditVelocity.setText(str(v)) self.__updateLabel() def __updateLabel(self): """ Update the unit for the velocity specification. """ c = self.__boundary.getVelocityChoice() if c in ('norm', 'norm_formula'): self.labelUnitVelocity.setText(str('m/s')) elif c in ('flow1', 'flow1_formula'): self.labelUnitVelocity.setText(str('kg/s')) elif c in ('flow2', 'flow2_formula'): self.labelUnitVelocity.setText(str('m3/s')) @pyqtSlot(str) def __slotVelocityValue(self, text): """ Private slot. New value associated to the velocity boundary type. @type text: C{QString} @param text: value """ if self.lineEditVelocity.validator().state == QValidator.Acceptable: v = from_qvariant(text, float) self.__boundary.setVelocity(v) @pyqtSlot() def __slotVelocityFormula(self): """ """ exp = self.__boundary.getVelocity() c = self.__boundary.getVelocityChoice() if c == 'norm_formula': exa = "u_norm = 1.0;" req = [('u_norm', 'Norm of the velocity')] elif c == 'flow1_formula': exa = "q_m = 1.0;" req = [('q_m', 'mass flow rate')] elif c == 'flow2_formula': exa = "q_v = 1.0;" req = [('q_v', 'volumic flow rate')] sym = [('x', "X face's gravity center"), ('y', "Y face's gravity center"), ('z', "Z face's gravity center"), ('dt', 'time step'), ('t', 'current time'), ('iter', 'number of iteration'), ('surface', 'Boundary zone surface')] for (nme, val) in self.notebook.getNotebookList(): sym.append((nme, 'value (notebook) = ' + str(val))) dialog = QMegEditorView(parent = self, function_type = "bnd", zone_name = self.__boundary._label, variable_name = "velocity", expression = exp, required = req, symbols = sym, condition = c, examples = exa) if dialog.exec_(): result = dialog.get_result() log.debug("slotFormulaVelocity -> %s" % str(result)) self.__boundary.setVelocity(str(result)) self.pushButtonVelocityFormula.setStyleSheet("background-color: green") self.pushButtonVelocityFormula.setToolTip(result) @pyqtSlot(str) def __slotChoiceDirection(self, text): """ Input the direction type choice. """ c = self.modelDirection.dicoV2M[str(text)] log.debug("slotChoiceVelocity: %s " % c) self.__boundary.setDirectionChoice(c) if c == "formula": self.pushButtonDirectionFormula.setEnabled(True) exp = self.__boundary.getDirection('direction_formula') if exp: self.pushButtonDirectionFormula.setStyleSheet("background-color: green") self.pushButtonDirectionFormula.setToolTip(exp) else: self.pushButtonDirectionFormula.setStyleSheet("background-color: red") self.frameDirectionCoordinates.hide() elif c == "coordinates": self.pushButtonDirectionFormula.setEnabled(False) self.pushButtonDirectionFormula.setStyleSheet("background-color: None") self.frameDirectionCoordinates.show() v = self.__boundary.getDirection('direction_x') self.lineEditDirectionX.setText(str(v)) v = self.__boundary.getDirection('direction_y') self.lineEditDirectionY.setText(str(v)) v = self.__boundary.getDirection('direction_z') self.lineEditDirectionZ.setText(str(v)) elif c == "normal": self.pushButtonDirectionFormula.setEnabled(False) self.pushButtonDirectionFormula.setStyleSheet("background-color: None") self.frameDirectionCoordinates.hide() @pyqtSlot(str) def __slotDirX(self, text): """ INPUT value into direction of inlet flow """ if self.sender().validator().state == QValidator.Acceptable: value = from_qvariant(text, float) self.__boundary.setDirection('direction_x', value) @pyqtSlot(str) def __slotDirY(self, text): """ INPUT value into direction of inlet flow """ if self.sender().validator().state == QValidator.Acceptable: value = from_qvariant(text, float) self.__boundary.setDirection('direction_y', value) @pyqtSlot(str) def __slotDirZ(self, text): """ INPUT value into direction of inlet flow """ if self.sender().validator().state == QValidator.Acceptable: value = from_qvariant(text, float) self.__boundary.setDirection('direction_z', value) @pyqtSlot() def __slotDirectionFormula(self): """ """ exp = self.__boundary.getDirection('direction_formula') req = [('dir_x', 'Direction of the flow along X'), ('dir_y', 'Direction of the flow along Y'), ('dir_z', 'Direction of the flow along Z')] exa = "dir_x = 3.0;\ndir_y = 1.0;\ndir_z = 0.0;\n" sym = [('x', "X face's gravity center"), ('y', "Y face's gravity center"), ('z', "Z face's gravity center"), ('dt', 'time step'), ('t', 'current time'), ('iter', 'number of iteration'), ('surface', 'Boundary zone surface')] for (nme, val) in self.notebook.getNotebookList(): sym.append((nme, 'value (notebook) = ' + str(val))) dialog = QMegEditorView(parent = self, function_type = "bnd", zone_name = self.__boundary._label, variable_name = "direction", expression = exp, required = req, symbols = sym, condition = "formula", examples = exa) if dialog.exec_(): result = dialog.get_result() log.debug("slotFormulaDirection -> %s" % str(result)) self.__boundary.setDirection('direction_formula', str(result)) self.pushButtonDirectionFormula.setToolTip(result) self.pushButtonDirectionFormula.setStyleSheet("background-color: green") @pyqtSlot(str) def __slotInletType(self, text): """ INPUT inlet type : 'oxydant'/'fuel' or 'burned'/'unburned' """ value = self.modelTypeInlet.dicoV2M[str(text)] log.debug("__slotInletType value = %s " % value) self.__boundary.setInletType(value) self.initialize() @pyqtSlot() def __slotPressure(self): """ Pressure selected or not for the initialisation. """ if self.checkBoxPressure.isChecked(): self.__boundary.setThermoStatus('pressure', "on") else: self.__boundary.setThermoStatus('pressure', "off") self.initialize() @pyqtSlot() def __slotDensity(self): """ Density selected or not for the initialisation. """ if self.checkBoxDensity.isChecked(): self.__boundary.setThermoStatus('density', "on") else: self.__boundary.setThermoStatus('density', "off") self.initialize() @pyqtSlot() def __slotTemperature(self): """ Temperature selected or not for the initialisation. """ if self.checkBoxTemperature.isChecked(): self.__boundary.setThermoStatus('temperature', "on") else: self.__boundary.setThermoStatus('temperature', "off") self.initialize() @pyqtSlot() def __slotEnergy(self): """ Energy selected or not for the initialisation. """ if self.checkBoxEnergy.isChecked(): self.__boundary.setThermoStatus('energy', "on") else: self.__boundary.setThermoStatus('energy', "off") self.initialize() @pyqtSlot(str) def __slotPressureValue(self, text): """ INPUT inlet Pressure """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setThermoValue('pressure', t) @pyqtSlot(str) def __slotDensityValue(self, text): """ INPUT inlet Density """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setThermoValue('density', t) @pyqtSlot(str) def __slotTemperatureValue(self, text): """ INPUT inlet Temperature """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setThermoValue('temperature', t) @pyqtSlot(str) def __slotEnergyValue(self, text): """ INPUT inlet Energy """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setThermoValue('energy', t) @pyqtSlot(str) def __slotTotalPressure(self, text): """ INPUT inlet total pressure """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setThermoValue('total_pressure', t) @pyqtSlot(str) def __slotTotalEnthalpy(self, text): """ INPUT inlet total enthalpy """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setThermoValue('enthalpy', t) @pyqtSlot(str) def __slotTemperatureGasComb(self, text): """ INPUT inlet temperature """ if self.sender().validator().state == QValidator.Acceptable: t = from_qvariant(text, float) self.__boundary.setGasCombustionTemperature(t) @pyqtSlot(str) def __slotMeanMixtureFraction(self, text): """ INPUT inlet mean mixutre fraction """ if self.sender().validator().state == QValidator.Acceptable: f = from_qvariant(text, float) self.__boundary.setMeanMixtureFraction(f) @pyqtSlot(str) def __slotInletTypeGasComb(self, text): """ INPUT inlet type : 'oxydant'/'fuel' or 'burned'/'unburned' """ value = self.modelTypeInletGasComb.dicoV2M[str(text)] log.debug("__slotInletTypeGasComb value = %s " % value) self.__boundary.setInletGasCombustionType(value) self.initialize() def tr(self, text): """ Translation """ return text #------------------------------------------------------------------------------- # End #-------------------------------------------------------------------------------