class ModernWindow(QWidget):
""" Modern window.
Args:
w (QWidget): Main widget.
parent (QWidget, optional): Parent widget.
"""
def __init__(self, w, parent=None):
QWidget.__init__(self, parent)
self._w = w
self.setupUi()
contentLayout = QHBoxLayout()
contentLayout.setContentsMargins(0, 0, 0, 0)
contentLayout.addWidget(w)
self.windowContent.setLayout(contentLayout)
self.setWindowTitle(w.windowTitle())
self.setGeometry(w.geometry())
# Adding attribute to clean up the parent window when the child is closed
self._w.setAttribute(Qt.WA_DeleteOnClose, True)
self._w.destroyed.connect(self.__child_was_closed)
def setupUi(self):
# create title bar, content
self.vboxWindow = QVBoxLayout(self)
self.vboxWindow.setContentsMargins(0, 0, 0, 0)
self.windowFrame = QWidget(self)
self.windowFrame.setObjectName('windowFrame')
self.vboxFrame = QVBoxLayout(self.windowFrame)
self.vboxFrame.setContentsMargins(0, 0, 0, 0)
self.titleBar = WindowDragger(self, self.windowFrame)
self.titleBar.setObjectName('titleBar')
self.titleBar.setSizePolicy(
QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Fixed))
self.hboxTitle = QHBoxLayout(self.titleBar)
self.hboxTitle.setContentsMargins(0, 0, 0, 0)
self.hboxTitle.setSpacing(0)
self.lblTitle = QLabel('Title')
self.lblTitle.setObjectName('lblTitle')
self.lblTitle.setAlignment(Qt.AlignCenter)
spButtons = QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.btnMinimize = QToolButton(self.titleBar)
self.btnMinimize.setObjectName('btnMinimize')
self.btnMinimize.setSizePolicy(spButtons)
self.btnRestore = QToolButton(self.titleBar)
self.btnRestore.setObjectName('btnRestore')
self.btnRestore.setSizePolicy(spButtons)
self.btnMaximize = QToolButton(self.titleBar)
self.btnMaximize.setObjectName('btnMaximize')
self.btnMaximize.setSizePolicy(spButtons)
self.btnClose = QToolButton(self.titleBar)
self.btnClose.setObjectName('btnClose')
self.btnClose.setSizePolicy(spButtons)
self.vboxFrame.addWidget(self.titleBar)
self.windowContent = QWidget(self.windowFrame)
self.vboxFrame.addWidget(self.windowContent)
self.vboxWindow.addWidget(self.windowFrame)
if PLATFORM == "Darwin":
self.hboxTitle.addWidget(self.btnClose)
self.hboxTitle.addWidget(self.btnMinimize)
self.hboxTitle.addWidget(self.btnRestore)
self.hboxTitle.addWidget(self.btnMaximize)
self.hboxTitle.addWidget(self.lblTitle)
else:
self.hboxTitle.addWidget(self.lblTitle)
self.hboxTitle.addWidget(self.btnMinimize)
self.hboxTitle.addWidget(self.btnRestore)
self.hboxTitle.addWidget(self.btnMaximize)
self.hboxTitle.addWidget(self.btnClose)
# set window flags
self.setWindowFlags(Qt.Window | Qt.FramelessWindowHint
| Qt.WindowSystemMenuHint
| Qt.WindowCloseButtonHint
| Qt.WindowMinimizeButtonHint
| Qt.WindowMaximizeButtonHint)
if QT_VERSION >= (5, ):
self.setAttribute(Qt.WA_TranslucentBackground)
# set stylesheet
with open(_FL_STYLESHEET) as stylesheet:
self.setStyleSheet(stylesheet.read())
# automatically connect slots
QMetaObject.connectSlotsByName(self)
def __child_was_closed(self):
self._w = None # The child was deleted, remove the reference to it and close the parent window
self.close()
def closeEvent(self, event):
if not self._w:
event.accept()
else:
self._w.close()
event.setAccepted(self._w.isHidden())
def setWindowTitle(self, title):
""" Set window title.
Args:
title (str): Title.
"""
super(ModernWindow, self).setWindowTitle(title)
self.lblTitle.setText(title)
def _setWindowButtonState(self, hint, state):
btns = {
Qt.WindowCloseButtonHint: self.btnClose,
Qt.WindowMinimizeButtonHint: self.btnMinimize,
Qt.WindowMaximizeButtonHint: self.btnMaximize
}
button = btns.get(hint)
maximized = bool(self.windowState() & Qt.WindowMaximized)
if button == self.btnMaximize: # special rules for max/restore
self.btnRestore.setEnabled(state)
self.btnMaximize.setEnabled(state)
if maximized:
self.btnRestore.setVisible(state)
self.btnMaximize.setVisible(False)
else:
self.btnMaximize.setVisible(state)
self.btnRestore.setVisible(False)
else:
button.setEnabled(state)
allButtons = [
self.btnClose, self.btnMinimize, self.btnMaximize, self.btnRestore
]
if True in [b.isEnabled() for b in allButtons]:
for b in allButtons:
b.setVisible(True)
if maximized:
self.btnMaximize.setVisible(False)
else:
self.btnRestore.setVisible(False)
self.lblTitle.setContentsMargins(0, 0, 0, 0)
else:
for b in allButtons:
b.setVisible(False)
self.lblTitle.setContentsMargins(0, 2, 0, 0)
def setWindowFlag(self, Qt_WindowType, on=True):
buttonHints = [
Qt.WindowCloseButtonHint, Qt.WindowMinimizeButtonHint,
Qt.WindowMaximizeButtonHint
]
if Qt_WindowType in buttonHints:
self._setWindowButtonState(Qt_WindowType, on)
else:
QWidget.setWindowFlag(self, Qt_WindowType, on)
def setWindowFlags(self, Qt_WindowFlags):
buttonHints = [
Qt.WindowCloseButtonHint, Qt.WindowMinimizeButtonHint,
Qt.WindowMaximizeButtonHint
]
for hint in buttonHints:
self._setWindowButtonState(hint, bool(Qt_WindowFlags & hint))
QWidget.setWindowFlags(self, Qt_WindowFlags)
@Slot()
def on_btnMinimize_clicked(self):
self.setWindowState(Qt.WindowMinimized)
@Slot()
def on_btnRestore_clicked(self):
if self.btnMaximize.isEnabled() or self.btnRestore.isEnabled():
self.btnRestore.setVisible(False)
self.btnRestore.setEnabled(False)
self.btnMaximize.setVisible(True)
self.btnMaximize.setEnabled(True)
self.setWindowState(Qt.WindowNoState)
@Slot()
def on_btnMaximize_clicked(self):
if self.btnMaximize.isEnabled() or self.btnRestore.isEnabled():
self.btnRestore.setVisible(True)
self.btnRestore.setEnabled(True)
self.btnMaximize.setVisible(False)
self.btnMaximize.setEnabled(False)
self.setWindowState(Qt.WindowMaximized)
@Slot()
def on_btnClose_clicked(self):
self.close()
@Slot()
def on_titleBar_doubleClicked(self):
if not bool(self.windowState() & Qt.WindowMaximized):
self.on_btnMaximize_clicked()
else:
self.on_btnRestore_clicked()
class ChipEditor(QWidget):
def __init__(self):
super().__init__()
self.viewer = ChipSceneViewer()
actionsLayout = QHBoxLayout()
self._actionsWidget = QWidget(self.viewer)
self._actionsWidget.setLayout(actionsLayout)
self._lockButton = QToolButton()
self._lockButton.setText("Done")
self._lockButton.clicked.connect(lambda: self.SetEditing(False))
self._editButton = QToolButton()
self._editButton.setText("Edit")
self._editButton.clicked.connect(lambda: self.SetEditing(True))
self._plusButton = QToolButton()
self._plusButton.setText("+")
self._plusButton.setPopupMode(QToolButton.InstantPopup)
actionsLayout.addWidget(self._plusButton)
actionsLayout.addWidget(self._lockButton)
actionsLayout.addWidget(self._editButton)
menu = QMenu(self._plusButton)
menu.addAction("Valve").triggered.connect(self.AddValve)
menu.addAction("Program Preset...").triggered.connect(
self.SelectProgramPreset)
menu.addAction("Image...").triggered.connect(self.BrowseForImage)
self._plusButton.setMenu(menu)
layout = QHBoxLayout()
layout.addWidget(self.viewer)
self.setLayout(layout)
self._mode = Mode.VIEWING
AppGlobals.Instance().onChipOpened.connect(self.LoadChip)
def AddValve(self):
newValve = Valve()
newValve.solenoidNumber = AppGlobals.Chip().NextSolenoidNumber()
AppGlobals.Chip().valves.append(newValve)
AppGlobals.Instance().onChipModified.emit()
self.viewer.CenterItem(self.viewer.AddItem(ValveChipItem(newValve)))
def BrowseForImage(self):
filename, filterType = QFileDialog.getOpenFileName(
self,
"Browse for Image",
filter="Images (*.png *.bmp *.jpg *.jpeg *.tif *.tiff)")
if filename:
newImage = Image(Path(filename))
AppGlobals.Chip().images.append(newImage)
AppGlobals.Instance().onChipModified.emit()
self.viewer.CenterItem(self.viewer.AddItem(
ImageChipItem(newImage)))
def SelectProgramPreset(self):
if not AppGlobals.Chip().programs:
return
presetSelection, confirmed = QInputDialog.getItem(
self, "Program Preset", "Program",
[program.name for program in AppGlobals.Chip().programs], 0, False)
if confirmed:
selected = [
program for program in AppGlobals.Chip().programs
if program.name == presetSelection
][0]
newPreset = ProgramPreset(selected)
AppGlobals.Chip().programPresets.append(newPreset)
AppGlobals.Instance().onChipModified.emit()
self.viewer.CenterItem(
self.viewer.AddItem(ProgramPresetItem(newPreset)))
def SetEditing(self, editing):
self.viewer.SetEditing(editing)
self._lockButton.setVisible(editing)
self._plusButton.setVisible(editing)
self._editButton.setVisible(not editing)
AppGlobals.Chip().editingMode = editing
self.FloatWidgets()
def resizeEvent(self, event: PySide6.QtGui.QResizeEvent) -> None:
super().resizeEvent(event)
self.FloatWidgets()
def FloatWidgets(self):
self._actionsWidget.adjustSize()
self._actionsWidget.move(self.viewer.rect().topRight() -
self._actionsWidget.rect().topRight())
def LoadChip(self):
self.viewer.RemoveAll()
for valve in AppGlobals.Chip().valves:
self.viewer.AddItem(ValveChipItem(valve))
for image in AppGlobals.Chip().images:
self.viewer.AddItem(ImageChipItem(image))
for preset in AppGlobals.Chip().programPresets:
self.viewer.AddItem(ProgramPresetItem(preset))
self.viewer.Recenter()
self.SetEditing(AppGlobals.Chip().editingMode)