import json from PyQt5.QtWidgets import QMainWindow, QDialog, QTabWidget, QPushButton, QSpacerItem, QSizePolicy\ ,QMessageBox, QHBoxLayout, QWidget, QTabBar, QProxyStyle, QStyle from PyQt5.QtGui import QIcon from PyQt5.QtCore import QSize from UI.DeviceDialogWidget import DeviceDialog from model.ProjectModel.DeviceManage import DevicesManange from utils.DBModels.BaseModel import * from UI.AreaTabWidget import AreaTabWidget from utils.DBModels.DeviceModels import DeviceDB from model.ProjectModel.DeviceManage import DevicesManange from UI.DeviceDialogWidget import DeviceDialog from PyQt5.QtWidgets import QApplication, QStyleOptionTab, QTabBar, QWidget from PyQt5.QtGui import QPainter, QPen, QColor, QPainterPath, QPolygonF, QFontMetrics, QTextOption, QFont from PyQt5.QtCore import Qt, QRectF, QPointF, QSizeF class NewChromeTabStyle(QProxyStyle): def drawControl(self, element, option, painter, widget): # painter.save() # painter.setPen(Qt.red); # painter.restore(); margin = 4 if element == QStyle.CE_TabBarTabLabel: tab = option # print(type(tab)) textRect = QRectF(self.subElementRect(QStyle.SE_TabBarTabText, tab, widget)) painter.save() if (tab.state & QStyle.State_Selected): painter.setPen(QPen(QColor("black"))) elif (tab.state & QStyle.State_MouseOver): painter.setPen(QPen(QColor("black"))) else: painter.setPen(QPen(QColor("black"))) option = QTextOption() # 设置对齐方式，Qt.AlignLeft 和 Qt.AlignVCenter 分别对应左对齐和垂直居中 font = QFont(); font.setPointSize(20); painter.setFont(font); option.setAlignment(Qt.AlignCenter | Qt.AlignVCenter) # 设置换行模式为不换行 option.setWrapMode(QTextOption.NoWrap) painter.drawText(textRect, tab.text, option) painter.restore() elif element == QStyle.CE_TabBarTabShape: tab = option painter.setRenderHint(QPainter.Antialiasing); shapeRect = QRectF(tab.rect); shapeRect.setY(shapeRect.y() + margin) arcLen = shapeRect.height() / 2; if (tab.state & QStyle.State_Selected): painter.save(); pen = QPen(Qt.white); pen.setWidthF(0.5); painter.setPen(pen); painter.restore(); painter.save(); shapeRect.setX(shapeRect.x() - arcLen); shapeRect.setWidth(shapeRect.width() + arcLen *3/ 5); path = QPainterPath(); arcRect1 = QRectF(shapeRect.x(), shapeRect.y() + shapeRect.height() / 3, arcLen, shapeRect.height() * 2 / 3); path.moveTo(QPointF(arcRect1.x(), arcRect1.y() + arcRect1.height())); path.lineTo(QPointF(arcRect1.x() + arcRect1.width() / 2, arcRect1.y() + arcRect1.height())); # // Starting from 270°, returning to the arc of 90° path.arcTo(arcRect1, 270.0, 90.0); path.lineTo(QPointF(arcRect1.x() + arcRect1.width(), arcRect1.y())); arcRect2 = QRectF(shapeRect.x() + arcRect1.width(), shapeRect.y(), arcLen, shapeRect.height() * 2 / 3); # // Starting from 180°, returning to the arc of -90° path.arcTo(arcRect2, 180.0, -90.0); path.lineTo(QPointF(arcRect2.x() + shapeRect.width() - 2 * (arcRect2.x() + arcRect2.width()) + arcRect2.width() / 2, arcRect2.y())); arcRect3 = QRectF(shapeRect.x() + shapeRect.width() - 2 * arcRect1.width(), shapeRect.y(), arcRect1.width(), shapeRect.height() * 2 / 3); path.arcTo(arcRect3, 90.0, -90.0); path.lineTo(QPointF(arcRect3.x() + arcRect3.width(), arcRect3.y() + arcRect3.height())); arcRect4 = QRectF(arcRect3.x() + arcRect3.width(), arcRect3.y() + arcRect3.height() / 2, arcRect1.width(), shapeRect.height() * 2 / 3); path.arcTo(arcRect4, 180.0, 90.0); path.lineTo(QPointF(arcRect1.x(), arcRect1.y() + arcRect1.height())); painter.setPen(Qt.NoPen); painter.setBrush(Qt.white); polygon = path.toFillPolygon(); painter.drawPolygon(polygon); painter.setBrush(QColor('#D3E3FD')); magPoint = shapeRect.topLeft() magPoint.setY(magPoint.y() - margin) painter.drawRect(QRectF(magPoint, QSizeF(shapeRect.width(), margin))) painter.restore(); elif (tab.state & QStyle.State_MouseOver): painter.save(); # 设置画笔颜色和宽度 painter.setBrush(QColor('#D3E3FD')); painter.setPen(Qt.NoPen) magPoint = shapeRect.topLeft() magPoint.setY(magPoint.y() - margin) painter.drawRect(QRectF(magPoint, QSizeF(shapeRect.width(), shapeRect.height() + 4))) shapeRect.setX(shapeRect.x()); shapeRect.setWidth(shapeRect.width() - 15); shapeRect.setHeight(shapeRect.height() - 4) painter.setPen(QColor('#A8C7FA')) # 设置填充颜色（如果需要） painter.setBrush(QColor('#A8C7FA')) # 圆角矩形的参数：左上角坐标x, y，宽度w，高度h，以及圆角半径r radius = 5 painter.drawRoundedRect(shapeRect, radius, radius) painter.restore(); else: painter.save(); shapeRect.setX(shapeRect.x() - arcLen); shapeRect.setWidth(shapeRect.width() + arcLen / 2); pen = QPen(QColor('#D3E3FD')); pen.setWidthF(0.1); painter.setPen(pen) painter.drawLine(shapeRect.bottomLeft(),shapeRect.bottomRight()); # painter.drawLine(QPointF(shapeRect.x() + arcLen*2/3 -1, shapeRect.height() / 3), # QPointF(shapeRect.x() + arcLen*2/3 -1, shapeRect.height() * 2 / 3)); painter.setBrush(QColor('#D3E3FD')); painter.drawRect(shapeRect) magPoint = shapeRect.topLeft() magPoint.setY(magPoint.y() - margin) painter.drawRect(QRectF(magPoint, QSizeF(shapeRect.width(), margin))) painter.restore(); # painter.save(); # painter.restore(); else: super(NewChromeTabStyle, self).drawControl(element, option, painter, widget); class DeviceTabBar(QTabBar): def __init__(self): super().__init__() self.setStyle(NewChromeTabStyle()) def tabSizeHint(self, index): newSize = QTabBar().tabSizeHint(index) width = len(self.tabText(index)) * 20 + 100 newSize.setWidth(width) newSize.setHeight(35) # tabRect = QTabBar().tabRect(index) # padding = 50 # tabRect.adjust(padding, padding, padding, padding) return newSize class DeviceTab(QTabWidget): def __init__(self, dockWidget, devicesManange): super().__init__() self.dockWidget = dockWidget self.devicesManange = devicesManange self.setObjectName('deviceWidget') self.initUI() def initUI(self): # self.tabBar = DeviceTabBar() self.setTabBar(DeviceTabBar()) #获取DP、PA协议和主从模式 self.dataTypeAndModel = self.dockWidget.windowTitle() self.proType = self.dataTypeAndModel[0:2] self.masterSlaveModel = self.dataTypeAndModel[2:4] self.setObjectName('deviceTabWidget') self.addDeviceButton = QPushButton('添加设备') self.addDeviceButton.setObjectName('deviceAddButton') self.addDeviceButton.setIcon(QIcon('Static/add.png')) self.addDeviceButton.setFlat(True) self.addDeviceButton.clicked.connect(self.addDeviceWidget) self.setCornerWidget(self.addDeviceButton) self.setTabsClosable(True) self.tabCloseRequested.connect(self.closeTab) self.state = True self.initWidget() def closeTab(self, index): reply = QMessageBox.question(self, 'Confirmation', '确定删掉此设备吗?', QMessageBox.Yes | QMessageBox.No, QMessageBox.No) if reply == QMessageBox.Yes: if index != -1: deviceName = self.tabText(index) + self.dockWidget.windowTitle() # print(deviceName,index,222) self.removeTab(index) self.devicesManange.delDevice(deviceName = deviceName ) DeviceDB.deleteDevice(deviceName = deviceName) else: return if self.count() == 0: self.creatInitWidget() def addDeviceWidget(self,init = False,deviceName = None): if deviceName is None: dialog = DeviceDialog(self.dataTypeAndModel) if dialog.exec_() == QDialog.Accepted: self.deviceName, pvUpperLimit, pvLowerLimit, pvUnit = dialog.getParameters() DeviceDB().addDevice(deviceName = self.deviceName, proType = self.proType , masterSlaveModel = self.masterSlaveModel, pvUpperLimit=pvUpperLimit, pvLowerLimit=pvLowerLimit, pvUnit=pvUnit) if init: self.removeTab(0) tabIndex = self.count() self.tabBar().setHidden(False) self.addTab(AreaTabWidget(self),str(self.deviceName[:-4])) self.setCurrentIndex(tabIndex) self.devicesManange.addDevice(proType=self.proType, masterSlaveModel = self.masterSlaveModel, deviceName = self.deviceName) else: return else: self.deviceName = deviceName self.titleName = self.deviceName[:-4] self.tabBar().setHidden(False) self.addTab(AreaTabWidget(self),str(self.titleName)) def initWidget(self): alldevices = DevicesManange.getAllDevice() titleName = self.dockWidget.windowTitle() if alldevices: for devices in alldevices: deviceName = devices[0] if titleName in deviceName: # print(devices,333) self.devicesManange.addDevice(proType = devices[1], masterSlaveModel = devices[2], deviceName = devices[0]) self.addDeviceWidget(deviceName=deviceName) self.state = False if self.state: self.creatInitWidget() def creatInitWidget(self): widget = QWidget() widget.setObjectName('initWidget') layout = QHBoxLayout() addButton = QPushButton('添加设备') addButton.setObjectName('initAddDeviceButton') icon = QIcon('Static/add.png') iconSize = QSize(50,50) addButton.setIcon(icon) addButton.setIconSize(iconSize) addButton.clicked.connect(lambda: self.addDeviceWidget(True)) layout.addWidget(QWidget()) layout.addWidget(addButton) layout.addWidget(QWidget()) widget.setLayout(layout) self.addTab(widget,'') self.tabBar().setHidden(True)