class UI_equipment(UI_equip):
"""Divider equipment edition dialog"""
Equipment = Divider()
def __init__(self, equipment=None, salidas=0, parent=None):
"""
equipment: Initial equipment instance to model
salidas: Stream Output number to equipment
"""
super().__init__(Divider, entrada=False, parent=parent)
# Calculate tab
lyt_Calc = QtWidgets.QGridLayout(self.tabCalculo)
lyt_Calc.addWidget(QtWidgets.QLabel(QtWidgets.QApplication.translate(
"pychemqt", "Separation")), 1, 1, 1, 1)
self.criterio = QtWidgets.QComboBox()
for txt in self.Equipment.TEXT_CRITERIO:
self.criterio.addItem(txt)
self.criterio.currentIndexChanged.connect(self.criterio_Changed)
lyt_Calc.addWidget(self.criterio, 1, 2, 1, 1)
self.fracciones = Tabla(1, horizontalHeader=[True], stretch=False)
self.fracciones.setItemDelegateForColumn(0, CellEditor(self))
lyt_Calc.addWidget(self.fracciones, 2, 1, 1, 2)
lyt_Calc.addWidget(QtWidgets.QLabel(QtWidgets.QApplication.translate(
"pychemqt", "Pressure lost")), 3, 1, 1, 1)
self.deltaP = Entrada_con_unidades(Pressure, value=0)
self.deltaP.valueChanged.connect(partial(self.changeParams, "deltaP"))
lyt_Calc.addWidget(self.deltaP, 3, 2, 1, 1)
if equipment and salidas:
equipment(salidas=salidas)
elif equipment:
salidas = equipment.kwargs["salidas"]
else:
self.Equipment = Divider(salidas=salidas)
self.fracciones.setRowCount(salidas)
for i in range(salidas):
itm = QtWidgets.QTableWidgetItem("%i" % i)
itm.setTextAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.fracciones.setItem(0, i, itm)
self.fracciones.setRowHeight(i, 20)
widget = UI_corriente.Ui_corriente(readOnly=True)
self.Salida.addTab(widget, str(i+1))
self.criterio_Changed(0)
self.fracciones.editingFinished.connect(
partial(self.changeParams, "split"))
self.setEquipment(equipment)
def criterio_Changed(self, int):
if int:
item = QtWidgets.QTableWidgetItem(QtWidgets.QApplication.translate(
"pychemqt", "Flow")+", "+MassFlow.text())
self.fracciones.setHorizontalHeaderItem(0, item)
self.fracciones.item(self.fracciones.rowCount()-1, 0).setFlags(
QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled |
QtCore.Qt.ItemIsSelectable)
else:
item = QtWidgets.QTableWidgetItem(QtWidgets.QApplication.translate(
"pychemqt", "Flow")+", "+MassFlow.text())
self.fracciones.setHorizontalHeaderItem(0, item)
self.fracciones.item(self.fracciones.rowCount()-1, 0).setFlags(
QtCore.Qt.NoItemFlags)
self.changeParams("criterio", int)
def changeParams(self, parametro, valor=None):
if parametro == "split":
valor = self.fracciones.getColumn(0, False)
if self.criterio.currentIndex() == 0:
if len(valor)+1 < self.fracciones.rowCount():
return
elif len(valor)+1 == self.fracciones.rowCount():
valor.append(1-sum(valor))
elif len(valor) == self.fracciones.rowCount():
valor[-1] = 1-sum(valor[:-1])
self.calculo(**{parametro: valor})
def rellenar(self):
UI_equip.rellenar(self)
if self.Equipment.status == 1 and self.criterio.currentIndex() == 1:
self.entrada.setCorriente(self.Equipment.entrada)
def rellenarInput(self):
UI_equip.rellenarInput(self)
self.fracciones.setColumn(0, self.Equipment.kwargs["split"])
class UI_equipment(UI_equip):
"""Divider equipment edition dialog"""
Equipment = Divider()
def __init__(self, equipment=None, salidas=0, parent=None):
"""
equipment: Initial equipment instance to model
salidas: Stream Output number to equipment
"""
super().__init__(Divider, entrada=False, parent=parent)
# Calculate tab
lyt_Calc = QtWidgets.QGridLayout(self.tabCalculo)
lyt_Calc.addWidget(QtWidgets.QLabel(QtWidgets.QApplication.translate(
"pychemqt", "Separation")), 1, 1, 1, 1)
self.criterio = QtWidgets.QComboBox()
for txt in self.Equipment.TEXT_CRITERIO:
self.criterio.addItem(txt)
self.criterio.currentIndexChanged.connect(self.criterio_Changed)
lyt_Calc.addWidget(self.criterio, 1, 2, 1, 1)
self.fracciones = Tabla(1, horizontalHeader=[True], stretch=False)
self.fracciones.setItemDelegateForColumn(0, CellEditor(self))
lyt_Calc.addWidget(self.fracciones, 2, 1, 1, 2)
lyt_Calc.addWidget(QtWidgets.QLabel(QtWidgets.QApplication.translate(
"pychemqt", "Pressure lost")), 3, 1, 1, 1)
self.deltaP = Entrada_con_unidades(Pressure, value=0)
self.deltaP.valueChanged.connect(partial(self.changeParams, "deltaP"))
lyt_Calc.addWidget(self.deltaP, 3, 2, 1, 1)
if equipment and salidas:
equipment(salidas=salidas)
elif equipment:
salidas = equipment.kwargs["salidas"]
else:
self.Equipment = Divider(salidas=salidas)
self.fracciones.setRowCount(salidas)
for i in range(salidas):
itm = QtWidgets.QTableWidgetItem("%i" % i)
itm.setTextAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.fracciones.setItem(0, i, itm)
self.fracciones.setRowHeight(i, 20)
widget = UI_corriente.Ui_corriente(readOnly=True)
self.Salida.addTab(widget, str(i+1))
self.criterio_Changed(0)
self.fracciones.editingFinished.connect(
partial(self.changeParams, "split"))
self.setEquipment(equipment)
def criterio_Changed(self, int):
if int:
item = QtWidgets.QTableWidgetItem(QtWidgets.QApplication.translate(
"pychemqt", "Flow")+", "+MassFlow.text())
self.fracciones.setHorizontalHeaderItem(0, item)
self.fracciones.item(self.fracciones.rowCount()-1, 0).setFlags(
QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled |
QtCore.Qt.ItemIsSelectable)
else:
item = QtWidgets.QTableWidgetItem(QtWidgets.QApplication.translate(
"pychemqt", "Flow")+", "+MassFlow.text())
self.fracciones.setHorizontalHeaderItem(0, item)
self.fracciones.item(self.fracciones.rowCount()-1, 0).setFlags(
QtCore.Qt.NoItemFlags)
self.changeParams("criterio", int)
def changeParams(self, parametro, valor=None):
if parametro == "split":
valor = self.fracciones.getColumn(0, False)
if self.criterio.currentIndex() == 0:
if len(valor)+1 < self.fracciones.rowCount():
return
elif len(valor)+1 == self.fracciones.rowCount():
valor.append(1-sum(valor))
elif len(valor) == self.fracciones.rowCount():
valor[-1] = 1-sum(valor[:-1])
self.calculo(**{parametro: valor})
def rellenar(self):
UI_equip.rellenar(self)
if self.Equipment.status == 1 and self.criterio.currentIndex() == 1:
self.entrada.setCorriente(self.Equipment.entrada)
def rellenarInput(self):
UI_equip.rellenarInput(self)
self.fracciones.setColumn(0, self.Equipment.kwargs["split"])
class InputTableWidget(QtWidgets.QWidget):
"""Table data input dialog"""
def __init__(self, columnas, data=None, t=[], property=[],
horizontalHeader=[], title="", DIPPR=False, hasTc=0,
Tc=None, eq=1, unit=[], parent=None):
"""
data: mrray with original data
t: values for x column, generally temperature
property: values for 2...n columns
horizontalHeader: List with column title
DIPPR: boolean to show DIPPR widget
hasTc: boolean to show critical temperature (some DIPPR eq need it)
Tc: value for critical temperature
eq: Value for DIPPR equation
unit: List of unidades classes for column definition
"""
super(InputTableWidget, self).__init__(parent)
self.columnas = columnas
self.title = title
self.unit = unit
gridLayout = QtWidgets.QGridLayout(self)
gridLayout.setContentsMargins(0, 0, 0, 0)
openButton = QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"]+"/images/button/fileOpen.png")), "")
openButton.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Load data from a file"))
openButton.clicked.connect(self.open)
gridLayout.addWidget(openButton, 1, 1)
saveButton = QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"]+"/images/button/fileSave.png")), "")
saveButton.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Save data to a file"))
saveButton.clicked.connect(self.save)
gridLayout.addWidget(saveButton, 1, 2)
clearButton = QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"]+"/images/button/clear.png")), "")
clearButton.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Clear data"))
clearButton.clicked.connect(self.delete)
gridLayout.addWidget(clearButton, 1, 3)
gridLayout.addItem(QtWidgets.QSpacerItem(
0, 0, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Fixed), 1, 4)
self.tabla = Tabla(self.columnas, horizontalHeader=horizontalHeader,
verticalHeader=False, stretch=False)
self.tabla.setConnected()
if unit:
hHeader = []
for unit, title in zip(self.unit, horizontalHeader):
hHeader.append("%s, %s" % (title, unit.text()))
self.tabla.setHorizontalHeaderLabels(hHeader)
self.tabla.horizontalHeader().sectionClicked.connect(self.editUnit)
if data:
self.tabla.setData(data)
self.tabla.addRow()
elif t and property:
self.tabla.setColumn(0, t)
self.tabla.setColumn(1, property)
gridLayout.addWidget(self.tabla, 2, 1, 1, 4)
if DIPPR:
self.eqDIPPR = eqDIPPR(eq)
gridLayout.addWidget(self.eqDIPPR, 3, 1, 1, 4)
self.eqDIPPR.eqDIPPR.valueChanged.connect(self.showTc)
self.labelTc = QtWidgets.QLabel("Tc: ", self)
gridLayout.addWidget(self.labelTc, 4, 1)
self.tc = Entrada_con_unidades(Temperature, value=Tc)
gridLayout.addWidget(self.tc, 4, 2, 1, 3)
self.showTc(1)
def showTc(self, value):
"""Show/hide Tc widget"""
self.labelTc.setVisible(value in (7, 9))
self.tc.setVisible(value in (7, 9))
def open(self):
"""Load data from a test file"""
fname, ext = QtWidgets.QFileDialog.getOpenFileName(
self,
QtWidgets.QApplication.translate("pychemqt", "Open text file"),
"./")
if fname:
try:
# Numpy raise error if use the fname directly and find a
# non-latin1 character, inclusive in comment lines
with open(fname, "rb") as file:
data = loadtxt(file)
self.delete()
self.tabla.setData(data)
except ValueError as er:
# Raise a error msg if the file can load by loadtxt, the user
# can select any type of file and the input error is possible
title = QtWidgets.QApplication.translate(
"pychemqt", "Failed to load file")
msg = fname + "\n" + er.args[0]
QtWidgets.QMessageBox.critical(self, title, msg)
def save(self):
"""Save currend data of table to a file"""
fname, ext = QtWidgets.QFileDialog.getSaveFileName(
self,
QtWidgets.QApplication.translate("pychemqt", "Save data to file"),
"./")
if fname:
with open(fname, 'w') as file:
file.write("#"+self.title+"\n")
file.write("#")
for i in range(self.tabla.columnCount()):
item = self.tabla.horizontalHeaderItem(i)
file.write(item.text()+"\t")
file.write("\n")
data = self.data
for fila in range(len(data)):
for columna in range(self.tabla.columnCount()):
file.write(str(data[fila][columna])+"\t")
file.write("\n")
def delete(self):
"""Clear table"""
self.tabla.setRowCount(0)
self.tabla.clearContents()
self.tabla.addRow()
@property
def data(self):
return self.tabla.getData()
def column(self, column, magnitud=None, unit="conf"):
"""
column: column to get
magnitud: magnitud to get the values
unit: unit of the values in table"""
data = self.tabla.getColumn(column)
if self.unit:
magnitud = self.unit[column]
tx = self.tabla.horizontalHeaderItem(column).text().split(", ")[-1]
unit = magnitud.__units__[magnitud.__text__.index(tx)]
if magnitud is not None:
data = [magnitud(x, unit) for x in data]
return data
def editUnit(self, col):
"""Show dialog to config input unit"""
unit = self.unit[col]
widget = QtWidgets.QComboBox(self.tabla)
for txt in unit.__text__:
widget.addItem(txt)
txt = self.tabla.horizontalHeaderItem(col).text().split(", ")[-1]
widget.setCurrentText(txt)
# Define Unit combobox geometry
size = self.tabla.horizontalHeader().sectionSize(col)
pos = self.tabla.horizontalHeader().sectionPosition(col)
h = self.tabla.horizontalHeader().height()
geometry = QtCore.QRect(pos, 0, size, h)
widget.setGeometry(geometry)
widget.currentIndexChanged["int"].connect(
partial(self.updateHeader, col))
widget.show()
widget.showPopup()
def updateHeader(self, col, index):
"""Change the text in header"""
widget = self.sender()
txt = self.tabla.horizontalHeaderItem(col).text()
newtxt = "%s, %s" % (txt.split(",")[0], widget.currentText())
self.tabla.setHorizontalHeaderItem(
col, QtWidgets.QTableWidgetItem(newtxt))
widget.close()
