#!/usr/bin/python3
# -*- coding: utf-8 -*-
'''Pychemqt, Chemical Engineering Process simulator
Copyright (C) 2009-2017, Juan José Gómez Romera <jjgomera@gmail.com>
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 3 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, see <http://www.gnu.org/licenses/>.'''
###############################################################################
# Library to configure process flow diagram (PFD)
#
# - Widget: PFD configuration
# - Dialog: Dialog tool for standalone use
# - ConfLine: Composite widget with line format configuration tools
# - ConfLineDialog: ConfLine dialog for standalone use
###############################################################################
import os
from PyQt5 import QtCore, QtGui, QtWidgets
from UI.widgets import ColorSelector, Entrada_con_unidades, PFDLineCombo
from tools import UI_confResolution
class Widget(QtWidgets.QDialog):
"""Flow Diagram configuration"""
def __init__(self, config=None, parent=None):
super(Widget, self).__init__(parent)
lyt = QtWidgets.QGridLayout(self)
lyt.setContentsMargins(0, 0, 0, 0)
scroll = QtWidgets.QScrollArea()
scroll.setFrameStyle(QtWidgets.QFrame.NoFrame)
lyt.addWidget(scroll)
dlg = QtWidgets.QWidget()
layout = QtWidgets.QGridLayout(dlg)
layout.addWidget(QtWidgets.QLabel(
QtWidgets.QApplication.translate("pychemqt", "Input color")), 1, 1)
self.ColorButtonEntrada = ColorSelector()
layout.addWidget(self.ColorButtonEntrada, 1, 2)
layout.addWidget(QtWidgets.QLabel(QtWidgets.QApplication.translate(
"pychemqt", "Output color:")), 2, 1)
self.ColorButtonSalida = ColorSelector()
layout.addWidget(self.ColorButtonSalida, 2, 2)
group = QtWidgets.QGroupBox(
QtWidgets.QApplication.translate("pychemqt", "Line format"))
layout.addWidget(group, 3, 1, 1, 3)
lyt = QtWidgets.QHBoxLayout(group)
self.lineFormat = ConfLine()
lyt.addWidget(self.lineFormat)
group = QtWidgets.QGroupBox(
QtWidgets.QApplication.translate("pychemqt", "PFD resolution"))
layout.addWidget(group, 4, 1, 1, 3)
lyt = QtWidgets.QHBoxLayout(group)
self.resolution = UI_confResolution.UI_confResolution_widget(config)
lyt.addWidget(self.resolution)
layout.addItem(QtWidgets.QSpacerItem(
10, 0, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Expanding), 14, 1, 1, 4)
scroll.setWidget(dlg)
if config and config.has_section("PFD"):
self.ColorButtonEntrada.setColor(
config.get("PFD", 'Color_Entrada'))
self.ColorButtonSalida.setColor(config.get("PFD", 'Color_Salida'))
self.lineFormat.ColorButtonLine.setColor(
config.get("PFD", 'Color_Stream'))
self.lineFormat.groupJoint.button(
(config.getint("PFD", 'Union')+2)*-1).setChecked(True)
self.lineFormat.mitterLimit.setValue(
config.getfloat("PFD", 'Miter_limit'))
self.lineFormat.groupCap.button(
(config.getint("PFD", 'Punta')+2)*-1).setChecked(True)
self.lineFormat.guion.setCurrentIndex(
config.getint("PFD", 'Guion'))
self.lineFormat.dashOffset.setValue(
config.getfloat("PFD", 'Dash_offset'))
self.lineFormat.width.setValue(config.getfloat("PFD", 'Width'))
def value(self, config):
if not config.has_section("PFD"):
config.add_section("PFD")
config = self.resolution.value(config)
config.set("PFD", "Color_Entrada",
self.ColorButtonEntrada.color.name())
config.set("PFD", "Color_Salida", self.ColorButtonSalida.color.name())
config.set("PFD", "Color_Stream",
self.lineFormat.ColorButtonLine.color.name())
config.set("PFD", "Width", str(self.lineFormat.width.value))
config.set("PFD", "Union",
str(abs(self.lineFormat.groupJoint.checkedId())-2))
config.set("PFD", "Miter_limit",
str(self.lineFormat.mitterLimit.value))
config.set("PFD", "Punta",
str(abs(self.lineFormat.groupCap.checkedId())-2))
config.set("PFD", "Guion", str(self.lineFormat.guion.currentIndex()))
config.set("PFD", "Dash_offset", str(self.lineFormat.dashOffset.value))
return config
class Dialog(QtWidgets.QDialog):
"""Dialog to config thermal method calculations"""
def __init__(self, config=None, parent=None):
super(Dialog, self).__init__(parent)
self.setWindowTitle(QtWidgets.QApplication.translate(
"pychemqt", "PFD configuration"))
layout = QtWidgets.QVBoxLayout(self)
self.widget = Widget(config)
layout.addWidget(self.widget)
self.buttonBox = QtWidgets.QDialogButtonBox(
QtWidgets.QDialogButtonBox.Cancel | QtWidgets.QDialogButtonBox.Ok)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
layout.addWidget(self.buttonBox)
def value(self, config):
"""Function result for wizard"""
config = self.widget.value(config)
return config
class ConfLine(QtWidgets.QWidget):
"""Composite widget with line format configuration tools"""
join = [QtCore.Qt.MiterJoin, QtCore.Qt.BevelJoin, QtCore.Qt.RoundJoin]
cap = [QtCore.Qt.FlatCap, QtCore.Qt.RoundCap, QtCore.Qt.SquareCap]
line = [QtCore.Qt.SolidLine, QtCore.Qt.DashLine, QtCore.Qt.DotLine,
QtCore.Qt.DashDotLine, QtCore.Qt.DashDotDotLine]
def __init__(self, pen=None, parent=None):
super(ConfLine, self).__init__(parent)
lyt = QtWidgets.QVBoxLayout(self)
lyt1 = QtWidgets.QHBoxLayout()
lyt1.addWidget(QtWidgets.QLabel(
QtWidgets.QApplication.translate("pychemqt", "Line")))
self.ColorButtonLine = ColorSelector()
self.ColorButtonLine.setToolTip(
QtWidgets.QApplication.translate("pychemqt", "Default line color"))
lyt1.addWidget(self.ColorButtonLine)
self.width = Entrada_con_unidades(
float, width=50, decimales=1, spinbox=True, step=0.1,
textounidad="px")
self.width.entrada.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Line Width"))
lyt1.addWidget(self.width)
lyt.addLayout(lyt1)
lyt2 = QtWidgets.QHBoxLayout()
lyt2.addWidget(QtWidgets.QLabel(
QtWidgets.QApplication.translate("pychemqt", "Join")))
self.mitterLimit = Entrada_con_unidades(
float, width=50, decimales=1, spinbox=True, step=0.1)
self.mitterLimit.entrada.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Mitter Limit"))
lyt2.addWidget(self.mitterLimit)
toolJoinMitter = QtWidgets.QToolButton()
toolJoinMitter.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"] +
os.path.join("images", "button", "stroke-join-miter.png"))))
toolJoinMitter.setIconSize(QtCore.QSize(24, 24))
toolJoinMitter.setCheckable(True)
toolJoinMitter.setToolTip(QtWidgets.QApplication.translate(
"pychemqt",
"Join mitter: The triangular notch between the two lines is not "
"filled"))
lyt2.addWidget(toolJoinMitter)
toolJoinBevel = QtWidgets.QToolButton()
toolJoinBevel.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"] +
os.path.join("images", "button", "stroke-join-bevel.png"))))
toolJoinBevel.setIconSize(QtCore.QSize(24, 24))
toolJoinBevel.setCheckable(True)
toolJoinBevel.setToolTip(QtWidgets.QApplication.translate(
"pychemqt",
"Join bevel: The triangular notch between the two lines is "
"filled"))
lyt2.addWidget(toolJoinBevel)
toolJoinRound = QtWidgets.QToolButton()
toolJoinRound.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"] +
os.path.join("images", "button", "stroke-join-round.png"))))
toolJoinRound.setIconSize(QtCore.QSize(24, 24))
toolJoinRound.setCheckable(True)
toolJoinRound.setToolTip(QtWidgets.QApplication.translate(
"pychemqt",
"Join round: A circular arc between the two lines is filled"))
lyt2.addWidget(toolJoinRound)
self.groupJoint = QtWidgets.QButtonGroup()
self.groupJoint.addButton(toolJoinMitter)
self.groupJoint.addButton(toolJoinBevel)
self.groupJoint.addButton(toolJoinRound)
self.groupJoint.buttonClicked["int"].connect(self.mitterlimitEnabled)
lyt.addLayout(lyt2)
lyt3 = QtWidgets.QHBoxLayout()
lyt3.addWidget(QtWidgets.QLabel(
QtWidgets.QApplication.translate("pychemqt", "Cap")))
toolCapFlat = QtWidgets.QToolButton()
toolCapFlat.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"] +
os.path.join("images", "button", "stroke-cap-butt.png"))))
toolCapFlat.setIconSize(QtCore.QSize(24, 24))
toolCapFlat.setCheckable(True)
toolCapFlat.setToolTip(QtWidgets.QApplication.translate(
"pychemqt",
"Flat Cap: A square line end that does not cover the end point of "
"the line"))
lyt3.addWidget(toolCapFlat)
toolCapRound = QtWidgets.QToolButton()
toolCapRound.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"] +
os.path.join("images", "button", "stroke-cap-round.png"))))
toolCapRound.setIconSize(QtCore.QSize(24, 24))
toolCapRound.setCheckable(True)
toolCapRound.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Round Cap: A rounded line end"))
lyt3.addWidget(toolCapRound)
toolCapSquare = QtWidgets.QToolButton()
toolCapSquare.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.environ["pychemqt"] +
os.path.join("images", "button", "stroke-cap-square.png"))))
toolCapSquare.setIconSize(QtCore.QSize(24, 24))
toolCapSquare.setCheckable(True)
toolCapSquare.setToolTip(QtWidgets.QApplication.translate(
"pychemqt",
"Square Cap: A square line end that covers the end point and "
"extends beyond it by half the line width"))
lyt3.addWidget(toolCapSquare)
self.groupCap = QtWidgets.QButtonGroup()
self.groupCap.addButton(toolCapFlat)
self.groupCap.addButton(toolCapRound)
self.groupCap.addButton(toolCapSquare)
lyt.addLayout(lyt3)
lyt4 = QtWidgets.QHBoxLayout()
lyt4.addWidget(QtWidgets.QLabel(
QtWidgets.QApplication.translate("pychemqt", "Dash")))
self.guion = PFDLineCombo()
lyt4.addWidget(self.guion)
self.dashOffset = Entrada_con_unidades(
float, width=50, decimales=1, spinbox=True, step=0.1)
self.dashOffset.entrada.setToolTip(QtWidgets.QApplication.translate(
"pychemqt", "Dash offset"))
lyt4.addWidget(self.dashOffset)
lyt.addLayout(lyt4)
lyt.addItem(QtWidgets.QSpacerItem(
0, 0, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Expanding))
if pen:
self.ColorButtonLine.setColor(pen.color().name())
self.groupJoint.button((self.join.index(
pen.joinStyle())+2)*-1).setChecked(True)
self.mitterLimit.setValue(pen.miterLimit())
self.groupCap.button((self.cap.index(
pen.capStyle())+2)*-1).setChecked(True)
self.guion.setCurrentIndex(self.line.index(pen.style()))
self.dashOffset.setValue(pen.dashOffset())
self.width.setValue(pen.widthF())
def mitterlimitEnabled(self, id):
self.mitterLimit.setEnabled(id == -2)
def pen(self):
"""Return a QPen with the live configuration"""
pen = QtGui.QPen(QtGui.QColor(self.ColorButtonLine.color.name()))
pen.setWidthF(self.width.value)
pen.setJoinStyle(self.join[abs(self.groupJoint.checkedId())-2])
pen.setMiterLimit(self.mitterLimit.value)
pen.setCapStyle(self.cap[abs(self.groupCap.checkedId())-2])
pen.setStyle(self.line[self.guion.currentIndex()])
pen.setDashOffset(self.dashOffset.value)
return pen
class ConfLineDialog(QtWidgets.QDialog, ConfLine):
"""Dialog derived ConfLine for standalone use"""
def __init__(self, pen=None, parent=None):
super(ConfLineDialog, self).__init__(pen)
self.setWindowTitle(
QtWidgets.QApplication.translate("pychemqt", "Edit format line"))
buttonBox = QtWidgets.QDialogButtonBox(
QtWidgets.QDialogButtonBox.Ok | QtWidgets.QDialogButtonBox.Cancel)
buttonBox.accepted.connect(self.accept)
buttonBox.rejected.connect(self.reject)
self.layout().addWidget(buttonBox)
if __name__ == "__main__":
import sys
from configparser import ConfigParser
app = QtWidgets.QApplication(sys.argv)
conf_dir = os.path.expanduser('~') + "/.pychemqt/"
config = ConfigParser()
config.read(conf_dir+"pychemqtrc")
Dialog = Dialog(config)
Dialog.show()
sys.exit(app.exec_())
