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 widgetReacciones(QtWidgets.QWidget):
"""Widget con la tabla de reacciones y los botones para modificar la lista de reacciones"""
changed = QtCore.pyqtSignal()
reacciones=[]
reaccion=None
activo=None
ajuste=None
def __init__(self, parent=None):
super(widgetReacciones, self).__init__(parent)
self.indices, self.nombres, M=getComponents()
gridLayout = QtWidgets.QGridLayout(self)
self.TablaReacciones=Tabla(5, horizontalHeader=[QtWidgets.QApplication.translate("pychemqt", "Reaction"), "ΔHr, %s" %unidades.MolarEnthalpy(None).text(), QtWidgets.QApplication.translate("pychemqt", "Type"), QtWidgets.QApplication.translate("pychemqt", "Phase"), QtWidgets.QApplication.translate("pychemqt", "Description")], dinamica=False, verticalHeader=True, orientacion=QtCore.Qt.AlignLeft)
self.TablaReacciones.setMinimumWidth(500)
self.TablaReacciones.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
self.TablaReacciones.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)
self.TablaReacciones.horizontalHeader().setStretchLastSection(True)
self.TablaReacciones.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
self.TablaReacciones.itemSelectionChanged.connect(self.actualizarBotones)
gridLayout.addWidget(self.TablaReacciones,1,1,6,4)
self.botonAbrir=QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(os.environ["pychemqt"]+"/images/button/fileOpen.png")), QtWidgets.QApplication.translate("pychemqt", "Open"))
self.botonAbrir.clicked.connect(self.botonAbrirClicked)
gridLayout.addWidget(self.botonAbrir,1,5)
self.botonGuardar=QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(os.environ["pychemqt"]+"/images/button/fileSave.png")), QtWidgets.QApplication.translate("pychemqt", "Save"))
self.botonGuardar.clicked.connect(self.botonGuardarClicked)
self.botonGuardar.setSizePolicy(QtWidgets.QSizePolicy.Fixed,QtWidgets.QSizePolicy.Fixed)
self.botonGuardar.setEnabled(False)
gridLayout.addWidget(self.botonGuardar,2,5)
self.botonNew=QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(os.environ["pychemqt"]+"/images/button/fileNew.png")), QtWidgets.QApplication.translate("pychemqt", "New"))
self.botonNew.clicked.connect(self.botonNewClicked)
gridLayout.addWidget(self.botonNew,3,5)
self.botonEdit=QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(os.environ["pychemqt"]+"/images/button/editor.png")), QtWidgets.QApplication.translate("pychemqt", "Edit"))
self.botonEdit.setEnabled(False)
self.botonEdit.setCheckable(True)
self.botonEdit.clicked.connect(self.botonEditClicked)
gridLayout.addWidget(self.botonEdit,4,5)
self.botonDelete=QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(os.environ["pychemqt"]+"/images/button/editDelete.png")), QtWidgets.QApplication.translate("pychemqt", "Delete"))
self.botonDelete.setEnabled(False)
self.botonDelete.clicked.connect(self.botonDeleteClicked)
gridLayout.addWidget(self.botonDelete,5,5)
self.botonClear=QtWidgets.QPushButton(QtGui.QIcon(QtGui.QPixmap(os.environ["pychemqt"]+"/images/button/clear.png")), QtWidgets.QApplication.translate("pychemqt", "Clear"))
self.botonClear.clicked.connect(self.botonClearClicked)
gridLayout.addWidget(self.botonClear,6,5)
gridLayout.addItem(QtWidgets.QSpacerItem(10,10,QtWidgets.QSizePolicy.Expanding,QtWidgets.QSizePolicy.Expanding),10,1)
def actualizarBotones(self, bool=True):
self.botonEdit.setEnabled(bool)
self.botonDelete.setEnabled(bool)
def botonAbrirClicked(self):
fname = str(QtWidgets.QFileDialog.getOpenFileName(self, QtWidgets.QApplication.translate("pychemqt", "Open reaction file"), "./", QtWidgets.QApplication.translate("pychemqt", "reaction file")+" (*.rec);;"+QtWidgets.QApplication.translate("pychemqt", "All files")+" (*.*)")[0])
if fname:
with open(fname, "r") as archivo:
reacciones=pickle.load(archivo)
print(reacciones)
self.reacciones=reacciones
self.botonGuardar.setEnabled(True)
for fila, reaccion in enumerate(reacciones):
self.TablaReacciones.addRow()
self.TablaReacciones.setValue(fila, 0, reaccion.text)
self.TablaReacciones.setValue(fila, 1, "%0.4e" %reaccion.Hr.config(), QtCore.Qt.AlignRight)
self.TablaReacciones.setValue(fila, 2, str(reaccion.tipo+1)+" - "+reaction.Reaction.TEXT_TYPE[reaccion.tipo])
self.TablaReacciones.setValue(fila, 3, reaction.Reaction.TEXT_PHASE[reaccion.fase])
self.TablaReacciones.item(fila, 4).setFlags(QtCore.Qt.ItemIsEditable|QtCore.Qt.ItemIsEnabled|QtCore.Qt.ItemIsSelectable)
for i in range(4):
self.TablaReacciones.resizeColumnToContents(i)
self.changed.emit()
def botonGuardarClicked(self):
fname = str(QtWidgets.QFileDialog.getSaveFileName(self, QtWidgets.QApplication.translate("pychemqt", "Save reaction to file"), "./", QtWidgets.QApplication.translate("pychemqt", "reaction file")+" (*.rec)")[0])
if fname:
if fname.split(".")[-1]!="rec":
fname+=".rec"
pickle.dump(self.reacciones, open(fname, "w"))
def botonNewClicked(self):
dialog=UI_reacciones(parent=self)
if dialog.exec_():
pass
def botonEditClicked(self, bool):
if bool:
indice=self.TablaReacciones.currentRow()
reaccion=self.reacciones[indice]
dialogo=UI_reacciones(reaccion, self)
dialogo.exec_()
# self.rellenar(self.reaccion)
# self.activo=indice
else:
self.botonAddClicked(self.activo, False)
self.reacciones[self.activo]=self.reaccion
self.TablaReacciones.setCurrentCell(self.activo, 0)
self.activo=-1
self.changed.emit()
self.botonNew.setEnabled(not bool)
self.botonDelete.setEnabled(not bool)
self.botonClear.setEnabled(not bool)
self.botonAdd.setEnabled(not bool)
self.botonAbrir.setEnabled(not bool)
self.botonGuardar.setEnabled(not bool)
def botonDeleteClicked(self):
indice=self.TablaReacciones.currentRow()
self.TablaReacciones.removeRow(indice)
del self.reacciones[indice]
self.TablaReacciones.clearSelection()
self.actualizarBotones(False)
self.changed.emit()
def botonClearClicked(self):
if self.reacciones:
self.reacciones=[]
self.TablaReacciones.setRowCount(0)
self.botonGuardar.setEnabled(False)
def botonAddClicked(self, fila, add=True):
if add:
fila=self.TablaReacciones.rowCount()
self.TablaReacciones.addRow()
self.TablaReacciones.setValue(fila, 0, self.Formula.text())
self.TablaReacciones.setValue(fila, 1, "%0.4e" %self.Hr.value.config(), QtCore.Qt.AlignRight)
self.TablaReacciones.setValue(fila, 2, str(self.tipo.currentIndex()+1)+" - "+self.tipo.currentText())
self.TablaReacciones.setValue(fila, 3, self.Fase.currentText())
self.TablaReacciones.item(fila, 4).setFlags(QtCore.Qt.ItemIsEditable|QtCore.Qt.ItemIsEnabled|QtCore.Qt.ItemIsSelectable)
for i in range(4):
self.TablaReacciones.resizeColumnToContents(i)
self.reacciones.insert(fila, self.reaccion)
self.botonGuardar.setEnabled(True)
self.changed.emit()
class widgetReacciones(QtWidgets.QWidget):
"""Widget con la tabla de reacciones y los botones para modificar la lista de reacciones"""
changed = QtCore.pyqtSignal()
reacciones = []
reaccion = None
activo = None
ajuste = None
def __init__(self, parent=None):
super(widgetReacciones, self).__init__(parent)
self.indices, self.nombres, M = getComponents()
gridLayout = QtWidgets.QGridLayout(self)
self.TablaReacciones = Tabla(
5,
horizontalHeader=[
QtWidgets.QApplication.translate("pychemqt", "Reaction"),
"ΔHr, %s" % unidades.MolarEnthalpy(None).text(),
QtWidgets.QApplication.translate("pychemqt", "Type"),
QtWidgets.QApplication.translate("pychemqt", "Phase"),
QtWidgets.QApplication.translate("pychemqt", "Description")
],
dinamica=False,
verticalHeader=True,
orientacion=QtCore.Qt.AlignLeft)
self.TablaReacciones.setMinimumWidth(500)
self.TablaReacciones.setSelectionBehavior(
QtWidgets.QAbstractItemView.SelectRows)
self.TablaReacciones.setSelectionMode(
QtWidgets.QAbstractItemView.SingleSelection)
self.TablaReacciones.horizontalHeader().setStretchLastSection(True)
self.TablaReacciones.setEditTriggers(
QtWidgets.QAbstractItemView.NoEditTriggers)
self.TablaReacciones.itemSelectionChanged.connect(
self.actualizarBotones)
gridLayout.addWidget(self.TablaReacciones, 1, 1, 6, 4)
self.botonAbrir = QtWidgets.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/fileOpen.png")),
QtWidgets.QApplication.translate("pychemqt", "Open"))
self.botonAbrir.clicked.connect(self.botonAbrirClicked)
gridLayout.addWidget(self.botonAbrir, 1, 5)
self.botonGuardar = QtWidgets.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/fileSave.png")),
QtWidgets.QApplication.translate("pychemqt", "Save"))
self.botonGuardar.clicked.connect(self.botonGuardarClicked)
self.botonGuardar.setSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
self.botonGuardar.setEnabled(False)
gridLayout.addWidget(self.botonGuardar, 2, 5)
self.botonNew = QtWidgets.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/fileNew.png")),
QtWidgets.QApplication.translate("pychemqt", "New"))
self.botonNew.clicked.connect(self.botonNewClicked)
gridLayout.addWidget(self.botonNew, 3, 5)
self.botonEdit = QtWidgets.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/editor.png")),
QtWidgets.QApplication.translate("pychemqt", "Edit"))
self.botonEdit.setEnabled(False)
self.botonEdit.setCheckable(True)
self.botonEdit.clicked.connect(self.botonEditClicked)
gridLayout.addWidget(self.botonEdit, 4, 5)
self.botonDelete = QtWidgets.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/editDelete.png")),
QtWidgets.QApplication.translate("pychemqt", "Delete"))
self.botonDelete.setEnabled(False)
self.botonDelete.clicked.connect(self.botonDeleteClicked)
gridLayout.addWidget(self.botonDelete, 5, 5)
self.botonClear = QtWidgets.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/clear.png")),
QtWidgets.QApplication.translate("pychemqt", "Clear"))
self.botonClear.clicked.connect(self.botonClearClicked)
gridLayout.addWidget(self.botonClear, 6, 5)
gridLayout.addItem(
QtWidgets.QSpacerItem(10, 10, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Expanding), 10, 1)
def actualizarBotones(self, bool=True):
self.botonEdit.setEnabled(bool)
self.botonDelete.setEnabled(bool)
def botonAbrirClicked(self):
fname = str(
QtWidgets.QFileDialog.getOpenFileName(
self,
QtWidgets.QApplication.translate("pychemqt",
"Open reaction file"), "./",
QtWidgets.QApplication.translate("pychemqt", "reaction file") +
" (*.rec);;" +
QtWidgets.QApplication.translate("pychemqt", "All files") +
" (*.*)")[0])
if fname:
with open(fname, "r") as archivo:
reacciones = pickle.load(archivo)
print(reacciones)
self.reacciones = reacciones
self.botonGuardar.setEnabled(True)
for fila, reaccion in enumerate(reacciones):
self.TablaReacciones.addRow()
self.TablaReacciones.setValue(fila, 0, reaccion.text)
self.TablaReacciones.setValue(fila, 1,
"%0.4e" % reaccion.Hr.config(),
QtCore.Qt.AlignRight)
self.TablaReacciones.setValue(
fila, 2,
str(reaccion.tipo + 1) + " - " +
reaction.Reaction.TEXT_TYPE[reaccion.tipo])
self.TablaReacciones.setValue(
fila, 3, reaction.Reaction.TEXT_PHASE[reaccion.fase])
self.TablaReacciones.item(
fila, 4).setFlags(QtCore.Qt.ItemIsEditable
| QtCore.Qt.ItemIsEnabled
| QtCore.Qt.ItemIsSelectable)
for i in range(4):
self.TablaReacciones.resizeColumnToContents(i)
self.changed.emit()
def botonGuardarClicked(self):
fname = str(
QtWidgets.QFileDialog.getSaveFileName(
self,
QtWidgets.QApplication.translate("pychemqt",
"Save reaction to file"),
"./",
QtWidgets.QApplication.translate("pychemqt", "reaction file") +
" (*.rec)")[0])
if fname:
if fname.split(".")[-1] != "rec":
fname += ".rec"
pickle.dump(self.reacciones, open(fname, "w"))
def botonNewClicked(self):
dialog = UI_reacciones(parent=self)
if dialog.exec_():
pass
def botonEditClicked(self, bool):
if bool:
indice = self.TablaReacciones.currentRow()
reaccion = self.reacciones[indice]
dialogo = UI_reacciones(reaccion, self)
dialogo.exec_()
# self.rellenar(self.reaccion)
# self.activo=indice
else:
self.botonAddClicked(self.activo, False)
self.reacciones[self.activo] = self.reaccion
self.TablaReacciones.setCurrentCell(self.activo, 0)
self.activo = -1
self.changed.emit()
self.botonNew.setEnabled(not bool)
self.botonDelete.setEnabled(not bool)
self.botonClear.setEnabled(not bool)
self.botonAdd.setEnabled(not bool)
self.botonAbrir.setEnabled(not bool)
self.botonGuardar.setEnabled(not bool)
def botonDeleteClicked(self):
indice = self.TablaReacciones.currentRow()
self.TablaReacciones.removeRow(indice)
del self.reacciones[indice]
self.TablaReacciones.clearSelection()
self.actualizarBotones(False)
self.changed.emit()
def botonClearClicked(self):
if self.reacciones:
self.reacciones = []
self.TablaReacciones.setRowCount(0)
self.botonGuardar.setEnabled(False)
def botonAddClicked(self, fila, add=True):
if add:
fila = self.TablaReacciones.rowCount()
self.TablaReacciones.addRow()
self.TablaReacciones.setValue(fila, 0, self.Formula.text())
self.TablaReacciones.setValue(fila, 1,
"%0.4e" % self.Hr.value.config(),
QtCore.Qt.AlignRight)
self.TablaReacciones.setValue(
fila, 2,
str(self.tipo.currentIndex() + 1) + " - " +
self.tipo.currentText())
self.TablaReacciones.setValue(fila, 3, self.Fase.currentText())
self.TablaReacciones.item(fila,
4).setFlags(QtCore.Qt.ItemIsEditable
| QtCore.Qt.ItemIsEnabled
| QtCore.Qt.ItemIsSelectable)
for i in range(4):
self.TablaReacciones.resizeColumnToContents(i)
self.reacciones.insert(fila, self.reaccion)
self.botonGuardar.setEnabled(True)
self.changed.emit()
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):
"""Spreadsheet interaction equipment edition dialog"""
Equipment = Spreadsheet()
def __init__(self, equipment=None, project=None, parent=None):
"""
equipment: Initial equipment instance to model
"""
super().__init__(Spreadsheet, entrada=True, salida=True,
calculo=False, parent=parent)
self.project = project
# Calculate tab
layout = QtWidgets.QGridLayout(self.Entrada)
label = QtWidgets.QApplication.translate(
"pychemqt", "Spreadsheet path")+":"
msg = QtWidgets.QApplication.translate(
"pychemqt", "Select Spreadsheet")
patrones = []
if os.environ["ezodf"]:
patrones.append(QtWidgets.QApplication.translate(
"pychemqt", "Libreoffice spreadsheet files")+" (*.ods)")
if os.environ["xlwt"]:
patrones.append(QtWidgets.QApplication.translate(
"pychemqt", "Microsoft Excel 97/2000/XP/2003 XML")+" (*.xls)")
if os.environ["openpyxl"]:
patrones.append(QtWidgets.QApplication.translate(
"pychemqt", "Microsoft Excel 2007/2010 XML")+" (*.xlsx)")
patron = ";;".join(patrones)
self.filename = PathConfig(label, msg=msg, patron=patron)
self.filename.valueChanged.connect(self.changeSpreadsheet)
layout.addWidget(self.filename, 1, 1)
header = [QtWidgets.QApplication.translate("pychemqt", "Entity"),
QtWidgets.QApplication.translate("pychemqt", "Variable"),
QtWidgets.QApplication.translate("pychemqt", "Unit value"),
QtWidgets.QApplication.translate("pychemqt", "Sheet"),
QtWidgets.QApplication.translate("pychemqt", "Cell")]
self.datamap = Tabla(
5, filas=1, dinamica=True, horizontalHeader=header,
verticalHeader=False, orientacion=QtCore.Qt.AlignLeft,
delegate=None, delegateforRow=TableDelegate, parent=self)
self.datamap.setEnabled(False)
self.datamap.cellChanged.connect(self.cellChanged)
self.datamap.rowFinished.connect(self.addRow)
layout.addWidget(self.datamap, 2, 1)
layout.addItem(QtWidgets.QSpacerItem(
10, 10, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Expanding), 10, 1)
entitys = []
for stream in list(self.project.streams.keys()):
entitys.append("s%i" % stream)
for equip in list(self.project.items.keys()):
if equip[0] == "e":
entitys.append(equip)
self.datamap.itemDelegateForRow(0).setItemsByIndex(0, entitys)
self.entitys = entitys
if equipment:
self.setEquipment(equipment)
def changeSpreadsheet(self, path):
self.datamap.setEnabled(bool(path))
self.changeParams("filename", str(path))
self.datamap.blockSignals(True)
self.datamap.clear()
self.datamap.blockSignals(False)
spreadsheet = ezodf.opendoc(path)
sheets = [name for name in spreadsheet.sheets.names()]
self.datamap.itemDelegateForRow(0).setItemsByIndex(3, sheets)
def rellenarInput(self):
self.blockSignals(True)
self.datamap.itemDelegateForRow(
self.datamap.rowCount()-1).setItemsByIndex(0, self.entitys)
if self.Equipment.status:
self.datamap.setEnabled(True)
self.filename.setText(self.Equipment.kwargs["filename"])
self.datamap.itemDelegateForRow(0).setItemsByIndex(
3, self.Equipment.sheets)
self.datamap.blockSignals(True)
self.datamap.clear()
if self.Equipment.kwargs["datamap"]:
for i, data in enumerate(self.Equipment.kwargs["datamap"]):
self.datamap.addRow()
self.datamap.itemDelegateForRow(i).setItemsByIndex(
0, self.entitys)
self.datamap.itemDelegateForRow(i).setItemsByIndex(
3, self.Equipment.sheets)
self.datamap.setItem(
i, 0, QtWidgets.QTableWidgetItem(data["entity"]))
self.datamap.setItem(
i, 1, QtWidgets.QTableWidgetItem(data["property"]))
self.datamap.setItem(
i, 2, QtWidgets.QTableWidgetItem(data["unit"]))
self.datamap.setItem(
i, 3, QtWidgets.QTableWidgetItem(data["sheet"]))
self.datamap.setItem(
i, 4, QtWidgets.QTableWidgetItem(data["cell"]))
self.datamap.itemDelegateForRow(
self.datamap.rowCount()-1).setItemsByIndex(0, self.entitys)
self.datamap.itemDelegateForRow(
self.datamap.rowCount()-1).setItemsByIndex(
3, self.Equipment.sheets)
self.datamap.blockSignals(False)
self.blockSignals(False)
def rellenar(self):
self.rellenarInput()
self.status.setState(self.Equipment.status, self.Equipment.msg)
def cellChanged(self, i, j):
obj = self.project.getObject(str(self.datamap.item(i, 0).text()))
properties = [prop[0] for prop in obj.propertiesNames()]
if j == 0: # Entity cambiado, cambiar variables disponibles
self.datamap.itemDelegateForRow(i).setItemsByIndex(1, properties)
editor = QtWidgets.QComboBox()
editor.addItems(self.datamap.itemDelegateForRow(i).items[1])
self.datamap.setColumnWidth(1, editor.sizeHint().width())
elif j == 1: # Variable cambiada, cambiar unidades disponibles
value = self.datamap.item(i, 1).text()
ind = properties.index(value)
if obj.propertiesUnit()[ind] == str:
self.datamap.itemDelegateForRow(i).setItemsByIndex(2, [" "])
self.datamap.item(i, 2).setText(" ")
else:
self.datamap.itemDelegateForRow(i).setItemsByIndex(
2, obj.propertiesNames()[ind][2].__text__)
elif j == 3:
self.datamap.item(i, 4).setText("")
def addRow(self, fila):
datamap = self.Equipment.kwargs["datamap"][:]
data = {}
data["entity"] = str(fila[0])
data["property"] = str(fila[1])
data["unit"] = str(fila[2])
data["sheet"] = str(fila[3])
data["cell"] = str(fila[4])
datamap.append(data)
self.changeParams("datamap", datamap)
class UI_equipment(UI_equip):
"""Spreadsheet interaction equipment edition dialog"""
Equipment = Spreadsheet()
def __init__(self, equipment=None, project=None, parent=None):
"""
equipment: Initial equipment instance to model
"""
super(UI_equipment, self).__init__(Spreadsheet,
entrada=True,
salida=True,
calculo=False,
parent=parent)
self.project = project
# Calculate tab
layout = QtGui.QGridLayout(self.entrada)
label = QtGui.QApplication.translate("pychemqt",
"Spreadsheet path") + ":"
msg = QtGui.QApplication.translate("pychemqt", "Select Spreadsheet")
patrones = QtCore.QStringList()
if os.environ["ezodf"]:
patrones.append(
QtGui.QApplication.translate(
"pychemqt", "Libreoffice spreadsheet files") + " (*.ods)")
# if os.environ["xlwt"]:
# patrones.append(QtGui.QApplication.translate(
# "pychemqt", "Microsoft Excel 97/2000/XP/2003 XMLL")+ " (*.xls)")
if os.environ["openpyxl"]:
patrones.append(
QtGui.QApplication.translate(
"pychemqt", "Microsoft Excel 2007/2010 XML") + " (*.xlsx)")
patron = patrones.join(";;")
self.filename = PathConfig(label, msg=msg, patron=patron)
self.filename.valueChanged.connect(self.changeSpreadsheet)
layout.addWidget(self.filename, 1, 1)
header = [
QtGui.QApplication.translate("pychemqt", "Entity"),
QtGui.QApplication.translate("pychemqt", "Variable"),
QtGui.QApplication.translate("pychemqt", "Unit value"),
QtGui.QApplication.translate("pychemqt", "Sheet"),
QtGui.QApplication.translate("pychemqt", "Cell")
]
self.datamap = Tabla(5,
filas=1,
dinamica=True,
horizontalHeader=header,
verticalHeader=False,
orientacion=QtCore.Qt.AlignLeft,
num=False,
delegateforRow=TableDelegate,
parent=self)
self.datamap.setEnabled(False)
self.datamap.cellChanged.connect(self.cellChanged)
self.datamap.rowFinished.connect(self.addRow)
layout.addWidget(self.datamap, 2, 1)
layout.addItem(
QtGui.QSpacerItem(10, 10, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding), 10, 1)
entitys = []
for stream in self.project.streams.keys():
entitys.append("s%i" % stream)
for equip in self.project.items.keys():
if equip[0] == "e":
entitys.append(equip)
self.datamap.itemDelegateForRow(0).setItemsByIndex(0, entitys)
self.entitys = entitys
if equipment:
self.setEquipment(equipment)
def changeSpreadsheet(self, path):
self.datamap.setEnabled(bool(path))
self.changeParams("filename", str(path))
self.datamap.blockSignals(True)
self.datamap.clear()
self.datamap.blockSignals(False)
spreadsheet = ezodf.opendoc(path)
sheets = [name for name in spreadsheet.sheets.names()]
self.datamap.itemDelegateForRow(0).setItemsByIndex(3, sheets)
def rellenarInput(self):
self.blockSignals(True)
self.datamap.itemDelegateForRow(self.datamap.rowCount() -
1).setItemsByIndex(0, self.entitys)
if self.Equipment.status:
self.datamap.setEnabled(True)
self.filename.setText(self.Equipment.kwargs["filename"])
self.datamap.itemDelegateForRow(0).setItemsByIndex(
3, self.Equipment.sheets)
self.datamap.blockSignals(True)
self.datamap.clear()
if self.Equipment.kwargs["datamap"]:
for i, data in enumerate(self.Equipment.kwargs["datamap"]):
self.datamap.addRow()
self.datamap.itemDelegateForRow(i).setItemsByIndex(
0, self.entitys)
self.datamap.itemDelegateForRow(i).setItemsByIndex(
3, self.Equipment.sheets)
self.datamap.setItem(i, 0,
QtGui.QTableWidgetItem(data["entity"]))
self.datamap.setItem(i, 1,
QtGui.QTableWidgetItem(data["property"]))
self.datamap.setItem(i, 2,
QtGui.QTableWidgetItem(data["unit"]))
self.datamap.setItem(i, 3,
QtGui.QTableWidgetItem(data["sheet"]))
self.datamap.setItem(i, 4,
QtGui.QTableWidgetItem(data["cell"]))
self.datamap.itemDelegateForRow(self.datamap.rowCount() -
1).setItemsByIndex(
0, self.entitys)
self.datamap.itemDelegateForRow(self.datamap.rowCount() -
1).setItemsByIndex(
3, self.Equipment.sheets)
self.datamap.blockSignals(False)
self.blockSignals(False)
def rellenar(self):
self.rellenarInput()
self.status.setState(self.Equipment.status, self.Equipment.msg)
def cellChanged(self, i, j):
obj = self.project.getObject(str(self.datamap.item(i, 0).text()))
properties = [prop[0] for prop in obj.propertiesNames()]
if j == 0: # Entity cambiado, cambiar variables disponibles
self.datamap.itemDelegateForRow(i).setItemsByIndex(1, properties)
editor = QtGui.QComboBox()
editor.addItems(self.datamap.itemDelegateForRow(i).items[1])
self.datamap.setColumnWidth(1, editor.sizeHint().width())
elif j == 1: # Variable cambiada, cambiar unidades disponibles
value = self.datamap.item(i, 1).text()
ind = properties.index(value)
if obj.propertiesUnit()[ind] == str:
self.datamap.itemDelegateForRow(i).setItemsByIndex(2, [" "])
self.datamap.item(i, 2).setText(" ")
else:
self.datamap.itemDelegateForRow(i).setItemsByIndex(
2,
obj.propertiesNames()[ind][2].__text__)
elif j == 3:
self.datamap.item(i, 4).setText("")
def addRow(self, fila):
datamap = self.Equipment.kwargs["datamap"][:]
data = {}
data["entity"] = str(fila[0])
data["property"] = unicode(fila[1])
data["unit"] = unicode(fila[2])
data["sheet"] = unicode(fila[3])
data["cell"] = str(fila[4])
datamap.append(data)
self.changeParams("datamap", datamap)
