class LabelShow(QWidget):
"""Present single label color scheme"""
remove_labels = Signal(str)
edit_labels = Signal([str, list], [list])
selected = Signal(str)
def __init__(self, name: str, label: list, removable, parent=None):
super().__init__(parent)
self.label = label
self.name = name
self.removable = removable
self.radio_btn = QRadioButton()
self.label_show = _LabelShow(np.array(label, dtype=np.uint8))
self.remove_btn = QToolButton()
self.remove_btn.setIcon(_icon_selector.close_icon)
self.edit_btn = QToolButton()
self.edit_btn.setIcon(_icon_selector.edit_icon)
if removable:
self.remove_btn.setToolTip("Remove colormap")
else:
self.remove_btn.setToolTip("This colormap is protected")
self.edit_btn.setToolTip("Create new label schema base on this")
layout = QHBoxLayout()
layout.addWidget(self.radio_btn)
layout.addWidget(self.label_show, 1)
layout.addWidget(self.remove_btn)
layout.addWidget(self.edit_btn)
self.remove_btn.setEnabled(removable)
self.setLayout(layout)
self.remove_btn.clicked.connect(self.remove_fun)
self.edit_btn.clicked.connect(self.edit_fun)
self.radio_btn.clicked.connect(self.selected_fun)
def set_checked(self, val):
self.radio_btn.setChecked(val)
if self.removable:
self.remove_btn.setDisabled(val)
@Slot()
def remove_fun(self):
if self.remove_btn.isEnabled():
self.remove_labels.emit(self.name)
@Slot()
def edit_fun(self):
self.edit_labels.emit(self.name, deepcopy(self.label))
self.edit_labels[list].emit(deepcopy(self.label))
@Slot(bool)
def selected_fun(self):
self.selected.emit(self.name)
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, 10, 10)
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(10, 10, 10, 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.btnSettings = QToolButton(self.titleBar)
self.btnSettings.setObjectName('btnSettings')
self.btnSettings.setSizePolicy(spButtons)
self.vboxFrame.addWidget(self.titleBar)
self.windowContent = QWidget(self.windowFrame)
self.vboxFrame.addWidget(self.windowContent)
self.vboxWindow.addWidget(self.windowFrame)
if True:
self.hboxTitle.addWidget(self.btnClose)
self.hboxTitle.addWidget(self.btnMinimize)
self.hboxTitle.addWidget(self.btnMaximize)
self.hboxTitle.addWidget(self.btnRestore)
self.hboxTitle.addWidget(self.lblTitle)
self.hboxTitle.addWidget(self.btnSettings)
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 True: #QT_VERSION >= (5,):
self.setAttribute(Qt.WA_TranslucentBackground, True)
self.setAttribute(Qt.WA_NoSystemBackground, True)
# set stylesheet
with open(_FL_STYLESHEET) as stylesheet:
self.setStyleSheet(stylesheet.read())
# automatically connect slots
QMetaObject.connectSlotsByName(self)
def paintEvent(self, e):
height = self.height()
width = self.width()
# draw background
painter = QPainter(self)
painter.setRenderHint(QPainter.Antialiasing)
painter.setPen(Qt.transparent)
painter.setBrush(BACKGROUND)
painter.setOpacity(1)
path = QPainterPath()
path.setFillRule(Qt.WindingFill)
path.addRoundedRect(QRectF(8, 0, width - 8, height), 15, 15)
painter.drawPath(path.simplified())
# draw sidebar
painter.setBrush(SIDEBAR)
painter.setOpacity(1)
path = QPainterPath()
path.setFillRule(Qt.WindingFill)
# we draw a fixed-width toolbar background
path.addRoundedRect(QRectF(8, 0, 250 - 8, height), 15, 15)
path.addRect(QRectF(200, 0, 50, 50))
path.addRect(QRectF(200, height - 50, 50, 50))
painter.drawPath(path.simplified())
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):
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(150, 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 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()
