class Entrada_Datos(QtGui.QDialog):
"""Table data input dialog"""
def __init__(self,
data=None,
t=[],
property=[],
horizontalHeader=[],
title="",
help=False,
helpFile="",
DIPPR=False,
tc=0,
tcValue=None,
eq=1,
parent=None):
"""
title: window title
data: mrray with original data
t: values for x column, generally temperature
property: values for 2...n columns
horizontalHeader: List with column title
help: boolean to show help button
helpFile: Path for help file, file or url
DIPPR: boolean to show DIPPR widget
tc: boolean to show critical temperature (same DIPPR eq need it)
tcValue: value for critical temperature
eq: Value for DIPPR equation
"""
super(Entrada_Datos, self).__init__(parent)
self.setWindowTitle(title)
self.columnas = len(horizontalHeader)
self.horizontalHeader = horizontalHeader
self.title = title
self.helpFile = helpFile
gridLayout = QtGui.QGridLayout(self)
self.botonAbrir = QtGui.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/fileOpen.png")),
QtGui.QApplication.translate("pychemqt", "Open"))
self.botonAbrir.clicked.connect(self.Abrir)
gridLayout.addWidget(self.botonAbrir, 1, 1)
self.botonGuardar = QtGui.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/fileSave.png")),
QtGui.QApplication.translate("pychemqt", "Save"))
self.botonGuardar.clicked.connect(self.Guardar)
gridLayout.addWidget(self.botonGuardar, 1, 2)
self.botonDelete = QtGui.QPushButton(
QtGui.QIcon(
QtGui.QPixmap(os.environ["pychemqt"] +
"/images/button/clear.png")),
QtGui.QApplication.translate("pychemqt", "Clear"))
self.botonDelete.clicked.connect(self.Borrar)
gridLayout.addWidget(self.botonDelete, 1, 3)
gridLayout.addItem(
QtGui.QSpacerItem(0, 0, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding), 1, 4)
self.tabla = Tabla(self.columnas,
horizontalHeader=horizontalHeader,
verticalHeader=False,
stretch=False)
self.tabla.setConnected()
if data:
self.tabla.setMatrix(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)
if tc:
lyt = QtGui.QHBoxLayout()
self.labelTc = QtGui.QLabel("Tc: ", self)
lyt.addWidget(self.labelTc)
self.tc = Entrada_con_unidades(Temperature, value=tcValue)
lyt.addWidget(self.tc)
lyt.addItem(
QtGui.QSpacerItem(0, 0, QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding))
gridLayout.addItem(lyt, 4, 1, 1, 4)
self.showTc(1)
if help:
botones = QtGui.QDialogButtonBox.Help | \
QtGui.QDialogButtonBox.Cancel | QtGui.QDialogButtonBox.Ok
else:
botones = QtGui.QDialogButtonBox.Cancel | QtGui.QDialogButtonBox.Ok
self.boton = QtGui.QDialogButtonBox(botones)
self.boton.accepted.connect(self.accept)
self.boton.rejected.connect(self.reject)
self.boton.helpRequested.connect(self.ayuda)
gridLayout.addWidget(self.boton, 5, 1, 1, 4)
def showTc(self, value):
self.labelTc.setVisible(value in (7, 9))
self.tc.setVisible(value in (7, 9))
def Abrir(self):
fname = unicode(
QtGui.QFileDialog.getOpenFileName(
self, QtGui.QApplication.translate("pychemqt",
"Open text file"), "./"))
if fname:
data = loadtxt(fname)
self.tabla.setMatrix(data)
self.tabla.addRow()
def Guardar(self):
fname = unicode(
QtGui.QFileDialog.getSaveFileName(
self,
QtGui.QApplication.translate("pychemqt", "Save data to file"),
"./"))
if fname:
with open(fname, 'w') as file:
file.write("#" + self.title + "\n")
file.write("#")
try:
for i in self.horizontalHeader:
file.write(i + "\t")
except UnicodeEncodeError:
pass
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 Borrar(self):
"""Clear table"""
self.tabla.setRowCount(1)
self.tabla.clearContents()
def ayuda(self):
"""Show help file"""
Dialog = HelpView(self.windowTitle(), QtCore.QUrl(self.helpFile))
Dialog.exec_()
@property
def data(self):
return self.tabla.getMatrix()
