def makeControlWindow(self): tb = QToolBox() tb.addItem(self.makeIpContainer(), "Connection and Run-State") tb.addItem(self.makeDisplayedParameterControl(), "Displayed Parameters") tb.addItem(self.makeSweepControl(), "Sweep Parameters") tb.addItem(self.makeCallButtonCtrl(), "Calibration") tb.setMinimumWidth(220) tb.setMaximumWidth(220) layout = QVBoxLayout() layout.addWidget(tb) layout.addStretch(1) return layout
def makeControlWindow(self):
tb = QToolBox()
tb.addItem(self.makeIpContainer(), "Connection and Run-State")
tb.addItem(self.makeDisplayedParameterControl(),
"Displayed Parameters")
tb.addItem(self.makeSweepControl(), "Sweep Parameters")
tb.addItem(self.makeCallButtonCtrl(), "Calibration")
tb.setMinimumWidth(220)
tb.setMaximumWidth(220)
layout = QVBoxLayout()
layout.addWidget(tb)
layout.addStretch(1)
return layout
class UITlvInfo(QGroupBox):
__codecinfo_ = None
__sdpinfo_ = None
__fileinfo_ = None
def __init__(self, title: str):
super(UITlvInfo, self).__init__(title)
self.__initUI()
def __initUI(self):
vlayout = QVBoxLayout(self)
self.__toolBox_ = QToolBox()
vlayout.addWidget(self.__toolBox_)
# 设置左侧导航栏 toolBox 初始化时的宽度
self.__toolBox_.setStyleSheet("QToolBoxButton { min-width:180px}")
# 设置左侧导航栏 toolBox 在左右拉拽时的最小宽度
self.__toolBox_.setMinimumWidth(100)
# 设置软件启动时默认打开导航栏的第几个 Item；这里设置的是打开第1个 Item。
self.__toolBox_.setCurrentIndex(0)
# self.__toolBox_.setFrameShape(QToolBox.Shape.StyledPanel) # 为toolbox增加边框
# self.__toolBox_.set
self.setFileInfoText(info)
self.setTlvSdpText(sdp)
self.setTlvCodecText('')
def setFileInfoText(self, text: str):
if self.__fileinfo_ is None:
self.__fileinfo_ = QTextBrowser()
self.__toolBox_.addItem(self.__fileinfo_, "File Info")
self.__fileinfo_.setText(text)
def setTlvSdpText(self, text):
if self.__sdpinfo_ is None:
self.__sdpinfo_ = QTextBrowser()
self.__toolBox_.addItem(self.__sdpinfo_, 'Sdp Info')
self.__sdpinfo_.setText(text)
def setTlvCodecText(self, text):
if self.__codecinfo_ is None:
self.__codecinfo_ = QTextBrowser()
self.__toolBox_.addItem(self.__codecinfo_, 'Codec Info')
self.__codecinfo_.setText(text)
def __init__(self):
super(MainForm, self).__init__()
# 主窗口初始化时实现主窗口布局
self.setupUi(self)
toolBox = QToolBox()
# 设置左侧导航栏 toolBox 初始化时的宽度
toolBox.setStyleSheet("QToolBoxButton { min-width:210px}")
# 设置左侧导航栏 toolBox 在左右拉拽时的最小宽度
toolBox.setMinimumWidth(100)
toolBox.addItem(QPushButton("Tab Content 1"), "Tab &1")
toolBox.addItem(QLabel("Tab Content 2"), "生产动态分析测试与建模")
toolBox.addItem(QPlainTextEdit('Text 3'), QIcon('snapshot.png'), "设置")
toolBox.setCurrentIndex(0) # 设置软件启动时默认打开导航栏的第几个 Item；这里设置的是打开第1个 Item。
self.verticalLayout.addWidget(toolBox)
# # 下面两行为设置 QSplitter 分割器伸缩大小因子，但是这样设置全屏后导航栏放大了不好看
# self.splitter.setStretchFactor(0, 30) # 分割器左边第1个占比例30%
# self.splitter.setStretchFactor(1, 70) # 分割器左边第1个占比例70%
# 下面一行为设置 QSplitter 分割器伸缩大小因子，但是这样设置全屏后导航栏放大了比较好看；不清楚原因。
self.splitter.setStretchFactor(
0, 0
) # 此函数用于设定：控件是否可伸缩。第一个参数用于指定控件的序号。第二个函数大于0时，表示控件可伸缩，小于0时，表示控件不可伸缩。
self.splitter.setStretchFactor(
1, 1
) # 此函数用于设定：控件是否可伸缩。第一个参数用于指定控件的序号。第二个函数大于0时，表示控件可伸缩，小于0时，表示控件不可伸缩。
# 设置 QSplitter 分割器各部分最小化时的情况，设置为“False”意味着左右拉动分隔栏时各部分不会消失；此设置也可以在界面设计时在 QtDesigner 里设置。
self.splitter.setChildrenCollapsible(False)
# 设置 QSplitter 分割器随主窗口自适应大小变化。此设置也可以在界面设计时在 QtDesigner 里设置。
self.splitter.setAutoFillBackground(True)
# 主窗口初始化时实例化子窗口
self.child = ChildrenForm()
# 在主窗口的QSplitter里添加子窗口
self.splitter.addWidget(self.child)
class MainWindow(QMainWindow):
InsertTextButton = 10
def __init__(self):
super(MainWindow, self).__init__()
self.createActions()
self.createMenus()
self.createToolBox()
self.scene = DiagramScene(self.itemMenu)
self.scene.setSceneRect(QRectF(0, 0, 5000, 5000))
self.scene.itemInserted.connect(self.itemInserted)
self.scene.textInserted.connect(self.textInserted)
self.scene.itemSelected.connect(self.itemSelected)
self.createToolbars()
layout = QHBoxLayout()
layout.addWidget(self.toolBox)
self.view = QGraphicsView(self.scene)
layout.addWidget(self.view)
self.widget = QWidget()
self.widget.setLayout(layout)
self.setCentralWidget(self.widget)
self.setWindowTitle("Diagramscene")
def backgroundButtonGroupClicked(self, button):
buttons = self.backgroundButtonGroup.buttons()
for myButton in buttons:
if myButton != button:
button.setChecked(False)
text = button.text()
if text == "Blue Grid":
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background1.png')))
elif text == "White Grid":
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background2.png')))
elif text == "Gray Grid":
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background3.png')))
else:
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background4.png')))
self.scene.update()
self.view.update()
def buttonGroupClicked(self, id):
buttons = self.buttonGroup.buttons()
for button in buttons:
if self.buttonGroup.button(id) != button:
button.setChecked(False)
if id == self.InsertTextButton:
self.scene.setMode(DiagramScene.InsertText)
else:
self.scene.setItemType(id)
self.scene.setMode(DiagramScene.InsertItem)
def deleteItem(self):
for item in self.scene.selectedItems():
if isinstance(item, DiagramItem):
item.removeArrows()
self.scene.removeItem(item)
def pointerGroupClicked(self, i):
self.scene.setMode(self.pointerTypeGroup.checkedId())
def bringToFront(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() >= zValue and isinstance(item, DiagramItem)):
zValue = item.zValue() + 0.1
selectedItem.setZValue(zValue)
def sendToBack(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() <= zValue and isinstance(item, DiagramItem)):
zValue = item.zValue() - 0.1
selectedItem.setZValue(zValue)
def itemInserted(self, item):
self.pointerTypeGroup.button(DiagramScene.MoveItem).setChecked(True)
self.scene.setMode(self.pointerTypeGroup.checkedId())
self.buttonGroup.button(item.diagramType).setChecked(False)
def textInserted(self, item):
self.buttonGroup.button(self.InsertTextButton).setChecked(False)#
self.scene.setMode(self.pointerTypeGroup.checkedId())
def currentFontChanged(self, font):
self.handleFontChange()
def fontSizeChanged(self, font):
self.handleFontChange()
def sceneScaleChanged(self, scale):
newScale = float(scale[:scale.index("%")]) / 100.0
oldMatrix = self.view.transform()
self.view.resetTransform()
self.view.translate(oldMatrix.dx(), oldMatrix.dy())
self.view.scale(newScale, newScale)
def textColorChanged(self):
self.textAction = self.sender()
self.fontColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/textpointer.png',
QColor(self.textAction.data())))
self.textButtonTriggered()
def itemColorChanged(self):
self.fillAction = self.sender()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon( ':/images/floodfill.png',
QColor(self.fillAction.data())))
self.fillButtonTriggered()
def lineColorChanged(self):
self.lineAction = self.sender()
self.lineColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/linecolor.png',
QColor(self.lineAction.data())))
self.lineButtonTriggered()
def textButtonTriggered(self):
self.scene.setTextColor(QColor(self.textAction.data()))
def fillButtonTriggered(self):
self.scene.setItemColor(QColor(self.fillAction.data()))
def lineButtonTriggered(self):
self.scene.setLineColor(QColor(self.lineAction.data()))
def handleFontChange(self):
font = self.fontCombo.currentFont()
font.setPointSize(int(self.fontSizeCombo.currentText()))
if self.boldAction.isChecked():
font.setWeight(QFont.Bold)
else:
font.setWeight(QFont.Normal)
font.setItalic(self.italicAction.isChecked())
font.setUnderline(self.underlineAction.isChecked())
self.scene.setFont(font)
def itemSelected(self, item):
font = item.font()
color = item.defaultTextColor()
self.fontCombo.setCurrentFont(font)
self.fontSizeCombo.setEditText(str(font.pointSize()))
self.boldAction.setChecked(font.weight() == QFont.Bold)
self.italicAction.setChecked(font.italic())
self.underlineAction.setChecked(font.underline())
def about(self):
QMessageBox.about(self, "About Diagram Scene",
"The <b>Diagram Scene</b> example shows use of the graphics framework.")
def createToolBox(self):
self.buttonGroup = QButtonGroup()
self.buttonGroup.setExclusive(False)
self.buttonGroup.buttonClicked[int].connect(self.buttonGroupClicked)
layout = QGridLayout()
layout.addWidget(self.createCellWidget("Conditional", DiagramItem.Conditional),
0, 0)
layout.addWidget(self.createCellWidget("Process", DiagramItem.Step), 0,
1)
layout.addWidget(self.createCellWidget("Input/Output", DiagramItem.Io),
1, 0)
textButton = QToolButton()
textButton.setCheckable(True)
self.buttonGroup.addButton(textButton, self.InsertTextButton)
textButton.setIcon(QIcon(QPixmap(':/images/textpointer.png').scaled(30, 30)))
textButton.setIconSize(QSize(50, 50))
textLayout = QGridLayout()
textLayout.addWidget(textButton, 0, 0, Qt.AlignHCenter)
textLayout.addWidget(QLabel("Text"), 1, 0, Qt.AlignCenter)
textWidget = QWidget()
textWidget.setLayout(textLayout)
layout.addWidget(textWidget, 1, 1)
layout.setRowStretch(3, 10)
layout.setColumnStretch(2, 10)
itemWidget = QWidget()
itemWidget.setLayout(layout)
self.backgroundButtonGroup = QButtonGroup()
self.backgroundButtonGroup.buttonClicked.connect(self.backgroundButtonGroupClicked)
backgroundLayout = QGridLayout()
backgroundLayout.addWidget(self.createBackgroundCellWidget("Blue Grid",
':/images/background1.png'), 0, 0)
backgroundLayout.addWidget(self.createBackgroundCellWidget("White Grid",
':/images/background2.png'), 0, 1)
backgroundLayout.addWidget(self.createBackgroundCellWidget("Gray Grid",
':/images/background3.png'), 1, 0)
backgroundLayout.addWidget(self.createBackgroundCellWidget("No Grid",
':/images/background4.png'), 1, 1)
backgroundLayout.setRowStretch(2, 10)
backgroundLayout.setColumnStretch(2, 10)
backgroundWidget = QWidget()
backgroundWidget.setLayout(backgroundLayout)
self.toolBox = QToolBox()
self.toolBox.setSizePolicy(QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Ignored))
self.toolBox.setMinimumWidth(itemWidget.sizeHint().width())
self.toolBox.addItem(itemWidget, "Basic Flowchart Shapes")
self.toolBox.addItem(backgroundWidget, "Backgrounds")
def createActions(self):
self.toFrontAction = QAction(
QIcon(':/images/bringtofront.png'), "Bring to &Front",
self, shortcut="Ctrl+F", statusTip="Bring item to front",
triggered=self.bringToFront)
self.sendBackAction = QAction(
QIcon(':/images/sendtoback.png'), "Send to &Back", self,
shortcut="Ctrl+B", statusTip="Send item to back",
triggered=self.sendToBack)
self.deleteAction = QAction(QIcon(':/images/delete.png'),
"&Delete", self, shortcut="Delete",
statusTip="Delete item from diagram",
triggered=self.deleteItem)
self.exitAction = QAction("E&xit", self, shortcut="Ctrl+X",
statusTip="Quit Scenediagram example", triggered=self.close)
self.boldAction = QAction(QIcon(':/images/bold.png'),
"Bold", self, checkable=True, shortcut="Ctrl+B",
triggered=self.handleFontChange)
self.italicAction = QAction(QIcon(':/images/italic.png'),
"Italic", self, checkable=True, shortcut="Ctrl+I",
triggered=self.handleFontChange)
self.underlineAction = QAction(
QIcon(':/images/underline.png'), "Underline", self,
checkable=True, shortcut="Ctrl+U",
triggered=self.handleFontChange)
self.aboutAction = QAction("A&bout", self, shortcut="Ctrl+B",
triggered=self.about)
def createMenus(self):
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.addAction(self.exitAction)
self.itemMenu = self.menuBar().addMenu("&Item")
self.itemMenu.addAction(self.deleteAction)
self.itemMenu.addSeparator()
self.itemMenu.addAction(self.toFrontAction)
self.itemMenu.addAction(self.sendBackAction)
self.aboutMenu = self.menuBar().addMenu("&Help")
self.aboutMenu.addAction(self.aboutAction)
def createToolbars(self):
self.editToolBar = self.addToolBar("Edit")
self.editToolBar.addAction(self.deleteAction)
self.editToolBar.addAction(self.toFrontAction)
self.editToolBar.addAction(self.sendBackAction)
self.fontCombo = QFontComboBox()
self.fontCombo.currentFontChanged.connect(self.currentFontChanged)
self.fontSizeCombo = QComboBox()
self.fontSizeCombo.setEditable(True)
for i in range(8, 30, 2):
self.fontSizeCombo.addItem(str(i))
validator = QIntValidator(2, 64, self)
self.fontSizeCombo.setValidator(validator)
self.fontSizeCombo.currentIndexChanged.connect(self.fontSizeChanged)
self.fontColorToolButton = QToolButton()
self.fontColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.fontColorToolButton.setMenu(
self.createColorMenu(self.textColorChanged, Qt.black))
self.textAction = self.fontColorToolButton.menu().defaultAction()
self.fontColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/textpointer.png',
Qt.black))
self.fontColorToolButton.setAutoFillBackground(True)
self.fontColorToolButton.clicked.connect(self.textButtonTriggered)
self.fillColorToolButton = QToolButton()
self.fillColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.fillColorToolButton.setMenu(
self.createColorMenu(self.itemColorChanged, Qt.white))
self.fillAction = self.fillColorToolButton.menu().defaultAction()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/floodfill.png',
Qt.white))
self.fillColorToolButton.clicked.connect(self.fillButtonTriggered)
self.lineColorToolButton = QToolButton()
self.lineColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.lineColorToolButton.setMenu(
self.createColorMenu(self.lineColorChanged, Qt.black))
self.lineAction = self.lineColorToolButton.menu().defaultAction()
self.lineColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/linecolor.png',
Qt.black))
self.lineColorToolButton.clicked.connect(self.lineButtonTriggered)
self.textToolBar = self.addToolBar("Font")
self.textToolBar.addWidget(self.fontCombo)
self.textToolBar.addWidget(self.fontSizeCombo)
self.textToolBar.addAction(self.boldAction)
self.textToolBar.addAction(self.italicAction)
self.textToolBar.addAction(self.underlineAction)
self.colorToolBar = self.addToolBar("Color")
self.colorToolBar.addWidget(self.fontColorToolButton)
self.colorToolBar.addWidget(self.fillColorToolButton)
self.colorToolBar.addWidget(self.lineColorToolButton)
pointerButton = QToolButton()
pointerButton.setCheckable(True)
pointerButton.setChecked(True)
pointerButton.setIcon(QIcon(':/images/pointer.png'))
linePointerButton = QToolButton()
linePointerButton.setCheckable(True)
linePointerButton.setIcon(QIcon(':/images/linepointer.png'))
self.pointerTypeGroup = QButtonGroup()
self.pointerTypeGroup.addButton(pointerButton, DiagramScene.MoveItem)
self.pointerTypeGroup.addButton(linePointerButton,
DiagramScene.InsertLine)
self.pointerTypeGroup.buttonClicked[int].connect(self.pointerGroupClicked)
self.sceneScaleCombo = QComboBox()
self.sceneScaleCombo.addItems(["50%", "75%", "100%", "125%", "150%"])
self.sceneScaleCombo.setCurrentIndex(2)
self.sceneScaleCombo.currentIndexChanged[str].connect(self.sceneScaleChanged)
self.pointerToolbar = self.addToolBar("Pointer type")
self.pointerToolbar.addWidget(pointerButton)
self.pointerToolbar.addWidget(linePointerButton)
self.pointerToolbar.addWidget(self.sceneScaleCombo)
def createBackgroundCellWidget(self, text, image):
button = QToolButton()
button.setText(text)
button.setIcon(QIcon(image))
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.backgroundButtonGroup.addButton(button)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createCellWidget(self, text, diagramType):
item = DiagramItem(diagramType, self.itemMenu)
icon = QIcon(item.image())
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.buttonGroup.addButton(button, diagramType)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createColorMenu(self, slot, defaultColor):
colors = [Qt.black, Qt.white, Qt.red, Qt.blue, Qt.yellow]
names = ["black", "white", "red", "blue", "yellow"]
colorMenu = QMenu(self)
for color, name in zip(colors, names):
action = QAction(self.createColorIcon(color), name, self,
triggered=slot)
action.setData(QColor(color))
colorMenu.addAction(action)
if color == defaultColor:
colorMenu.setDefaultAction(action)
return colorMenu
def createColorToolButtonIcon(self, imageFile, color):
pixmap = QPixmap(50, 80)
pixmap.fill(Qt.transparent)
painter = QPainter(pixmap)
image = QPixmap(imageFile)
target = QRect(0, 0, 50, 60)
source = QRect(0, 0, 42, 42)
painter.fillRect(QRect(0, 60, 50, 80), color)
painter.drawPixmap(target, image, source)
painter.end()
return QIcon(pixmap)
def createColorIcon(self, color):
pixmap = QPixmap(20, 20)
painter = QPainter(pixmap)
painter.setPen(Qt.NoPen)
painter.fillRect(QRect(0, 0, 20, 20), color)
painter.end()
return QIcon(pixmap)
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(970, 650)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.horizontalLayout = QtWidgets.QHBoxLayout(self.centralwidget)
self.horizontalLayout.setObjectName("horizontalLayout")
self.splitter = QtWidgets.QSplitter(self.centralwidget)
self.splitter.setOrientation(QtCore.Qt.Horizontal)
self.splitter.setHandleWidth(1)
self.splitter.setObjectName("splitter")
self.horizontalLayout.addWidget(self.splitter)
MainWindow.setCentralWidget(self.centralwidget)
# 文件树形目录浏览器
self.tree = QTreeView()
self.model = QtWidgets.QFileSystemModel()
self.model.setRootPath('')
self.nameFile = ["*.png", "*.jpg", "*.jpeg"]
self.model.setNameFilterDisables(False)
self.model.setNameFilters(self.nameFile)
self.tree.setModel(self.model)
self.tree.setAnimated(False)
self.tree.setIndentation(20)
self.tree.setWindowTitle("Dir View")
self.tree.setSortingEnabled(True)
self.tree.setColumnHidden(1, True)
self.tree.setColumnHidden(2, True)
self.tree.setColumnHidden(3, True)
self.tree.doubleClicked.connect(self.openFileFromTreeList)
# 图库管理器
self.tree_2 = QTreeWidget()
self.tree_2.setColumnCount(1)
# 设置表头信息：隐藏表头
self.tree_2.setHeaderHidden(1)
# 设置root和root2为self.tree的子树，所以root和root2就是跟节点
root = QTreeWidgetItem(self.tree_2)
root2 = QTreeWidgetItem(self.tree_2)
# 设置根节点的名称
root.setText(0, '第一节点')
root2.setText(0, '第二节点')
# 为root节点设置子结点
child1 = QTreeWidgetItem(root)
child1.setText(0, 'child1')
child1.setText(1, 'name1')
child2 = QTreeWidgetItem(root)
# 设置child2节点的图片
child2.setText(0, 'child2')
child2.setText(1, 'name2')
# 实例化QToolBox
self.toolBox = QToolBox()
# 设置左侧导航栏 toolBox 在左右拉拽时的最小宽度
self.toolBox.setMinimumWidth(100)
# 给QToolBox添加两个子项目
self.toolBox.addItem(self.tree, "文件资源")
self.toolBox.addItem(self.tree_2, "图库")
# 给QSplitter添加第一个窗体（QToolBox）
self.splitter.addWidget(self.toolBox)
#菜单栏
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(10, 10, 820, 30))
self.menubar.setObjectName("menubar")
MainWindow.setMenuBar(self.menubar)
self.menu1_file = QtWidgets.QMenu(self.menubar)
self.menu1_file.setObjectName("menu1_file")
self.menu2_edit = QtWidgets.QMenu(self.menubar)
self.menu2_edit.setObjectName("menu2_edit")
self.menu3_imageLib = QtWidgets.QMenu(self.menubar)
self.menu3_imageLib.setObjectName("menu3_imageLib")
self.menu4_image = QtWidgets.QMenu(self.menubar)
self.menu4_image.setObjectName("menu4_image")
self.menu6_user = QtWidgets.QMenu(self.menubar)
self.menu6_user.setObjectName("menu6_user")
self.menu5_help = QtWidgets.QMenu(self.menubar)
self.menu5_help.setObjectName("menu5_help")
MainWindow.setMenuBar(self.menubar)
#menu Action
#################action1######################
self.action_1_1 = QtWidgets.QAction(MainWindow)
self.action_1_1.setObjectName("action_1_1") #打开
self.action_1_1.triggered.connect(self.openfile)
self.action_1_2 = QtWidgets.QAction(MainWindow)
self.action_1_2.setObjectName("action_1_2") #保存
self.action_1_3 = QtWidgets.QAction(MainWindow)
self.action_1_3.setObjectName("action_1_3") #退出
#################action2######################
self.action_2_1 = QtWidgets.QAction(MainWindow)
self.action_2_1.setObjectName("action_2_1") #图片编辑
self.action_2_1.triggered.connect(self.openChildWindow)
#################action3######################
self.action_3_1 = QtWidgets.QAction(MainWindow)
self.action_3_1.setObjectName("action_3_1") #图库管理器
self.action_3_1.triggered.connect(self.imageLibUi)
#################action4######################
self.action_4_1 = QtWidgets.QAction(MainWindow)
self.action_4_1.setObjectName("action_4_1") # 添加图片
self.action_4_2 = QtWidgets.QAction(MainWindow)
self.action_4_2.setObjectName("action_4_2") # 删除图片
self.action_4_3 = QtWidgets.QAction(MainWindow)
self.action_4_3.setObjectName("action_4_3") # 重命名
#################action5######################
self.action_5 = QtWidgets.QAction(MainWindow)
self.action_5.setObjectName("action_5") #帮助
#################action6######################
self.action_6 = QtWidgets.QAction(MainWindow)
self.action_6.setObjectName("action_6") #用户管理
self.action_6.triggered.connect(self.userManager)
#给menu添加Action
self.menu1_file.addAction(self.action_1_1) #打开
self.menu1_file.addAction(self.action_1_2) #保存
self.menu1_file.addAction(self.action_1_3) #退出
self.menu2_edit.addAction(self.action_2_1) #图片编辑
self.menu3_imageLib.addAction(self.action_3_1) #图库管理器
self.menu4_image.addAction(self.action_4_1) #添加图片
self.menu4_image.addAction(self.action_4_2) #删除图片
self.menu4_image.addAction(self.action_4_3) #重命名
self.menu6_user.addAction(self.action_6) #用户管理
self.menu5_help.addAction(self.action_5) #帮助
self.menubar.addAction(self.menu1_file.menuAction())
self.menubar.addAction(self.menu2_edit.menuAction())
self.menubar.addAction(self.menu3_imageLib.menuAction())
self.menubar.addAction(self.menu4_image.menuAction())
self.menubar.addAction(self.menu5_help.menuAction())
self.menubar.addAction(self.menu6_user.menuAction())
#Icon
icon = QtGui.QIcon()
icon2 = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap("Icon/openfile.png"), QtGui.QIcon.Normal,
QtGui.QIcon.Off)
icon2.addPixmap(QtGui.QPixmap("Icon/edit.png"), QtGui.QIcon.Normal,
QtGui.QIcon.Off)
self.action_1_1.setIcon(icon)
self.action_2_1.setIcon(icon2)
#工具栏
self.toolBar = QtWidgets.QToolBar(MainWindow)
self.toolBar.setInputMethodHints(QtCore.Qt.ImhHiddenText
| QtCore.Qt.ImhSensitiveData)
self.toolBar.setObjectName("toolBar")
MainWindow.addToolBar(QtCore.Qt.TopToolBarArea, self.toolBar)
self.toolBar.addAction(self.action_1_1)
self.toolBar.addAction(self.action_2_1)
#状态栏
self.statusbar = QtWidgets.QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
#初始化默认窗体
#初始化默认窗体（图片显示窗体）
self.imageView = ImageView()
# 在主窗口里添加子窗口
self.splitter.addWidget(self.imageView)
#是普通用户还是管理员
self
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
# 创建默认存放文件的文件夹：
cur_dir = "C:"
folder_name = 'InfraredImage'
if os.path.isdir("C:/InfraredImage"):
print("Already exist!")
else:
os.mkdir(os.path.join(cur_dir, folder_name))
os.mkdir("C:/InfraredImage/TmpImg")
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "红外图像质量客观评价系统V1.0"))
#导航栏名字
self.menu1_file.setTitle(_translate("MainWindow", "文件"))
self.menu3_imageLib.setTitle(_translate("MainWindow", "图库管理"))
self.menu4_image.setTitle(_translate("MainWindow", "图片管理"))
self.menu2_edit.setTitle(_translate("MainWindow", "编辑"))
self.menu6_user.setTitle(_translate("MainWindow", "用户管理"))
self.menu5_help.setTitle(_translate("MainWindow", "帮助"))
#action名字
self.action_1_1.setText(_translate("MainWindow", "打开"))
self.action_1_1.setShortcut(_translate("MainWindow", "Ctrl+O"))
self.action_1_2.setText(_translate("MainWindow", "保存"))
self.action_1_2.setShortcut(_translate("MainWindow", "Ctrl+S"))
self.action_1_3.setText(_translate("MainWindow", "退出"))
self.action_1_3.setShortcut(_translate("MainWindow", "Ctrl+E"))
self.action_2_1.setText(_translate("MainWindow", "编辑图片"))
self.action_3_1.setText(_translate("MainWindow", "图库管理器"))
self.action_4_1.setText(_translate("MainWindow", "添加图片"))
self.action_4_2.setText(_translate("MainWindow", "删除图片"))
self.action_4_3.setText(_translate("MainWindow", "重命名"))
self.action_5.setText(_translate("MainWindow", "Help ?"))
self.action_6.setText(_translate("MainWindow", "用户管理器"))
self.statusbar.showMessage("Ready")
##########################功能函数区###############################
def openfile(self):
# 设置文件扩展名过滤,注意用双分号间隔
fileName1, filetype = QFileDialog.getOpenFileName(
None, "选取文件", "C:/", "All Files (*);;Image (*.png *.jpg *.jpeg)")
#显示图片，并初始化图片基本信息
self.openImage(fileName1)
# 从文件资源目录树打开图片
def openFileFromTreeList(self, Qmodelidx):
filePath = self.model.filePath(Qmodelidx)
self.openImage(filePath)
# 打开图片,并初始化图片基本信息
def openImage(self, filePath):
self.action_2_1.setEnabled(True)
self.splitter.widget(1).setParent(None)
self.splitter.insertWidget(1, self.imageView)
img = QtGui.QPixmap(filePath).scaled(
self.imageView.graphicsView.width(),
self.imageView.graphicsView.height())
self.imageView.graphicsView.setPixmap(img)
self.newPic = Image.open(filePath)
self.statusbar.showMessage("打开图片")
self.imageView.listWidget.item(1).setText("大小： " +
str(self.newPic.size))
self.imageView.listWidget.item(2).setText("色彩通道： " + self.newPic.mode)
self.imageView.listWidget.item(3).setText("格式： " + self.newPic.format)
self.imageView.listWidget.item(4).setText(
"像素： " + str(self.newPic.getextrema()))
self.imageView.listWidget.item(5).setText("文件： " + str(filePath))
# # 要把新打开的图片，和所有修改过的图片，都存进tempPic.jpg里面，作为一个中间变量。
self.newPic.save("C:/InfraredImage/TmpImg/tempPic.png")
self.newPic = "C:/InfraredImage/TmpImg/tempPic.png"
# 打开图片处理子窗口函
def openChildWindow(self):
childUi = childWindow()
if childUi.exec_() == QtWidgets.QDialog.Accepted:
pass
#打开图库管理窗口
def imageLibUi(self):
self.imageLibUi = ImageLibUi()
# 把QSplitter的指定位置的窗体从QSplitter中剥离
self.splitter.widget(1).setParent(None)
# 在QSplitter的指定位置载入新窗体
self.splitter.insertWidget(1, self.imageLibUi)
# 打开图库管理窗口时将图片编辑器设为不可用
self.action_2_1.setEnabled(False)
#用户管理
def userManager(self):
self.userWindow = UserManagerUi()
# 把QSplitter的指定位置的窗体从QSplitter中剥离
self.splitter.widget(1).setParent(None)
# 在QSplitter的指定位置载入新窗体
self.splitter.insertWidget(1, self.userWindow)
# 打开图库管理窗口时将图片编辑器设为不可用
self.action_2_1.setEnabled(False)
class MainWindow(QMainWindow):
InsertTextButton = 10
def __init__(self):
super(MainWindow, self).__init__()
self.createActions()
self.createMenus()
self.createToolBox()
self.scene = DiagramScene(self.itemMenu)
self.scene.setSceneRect(QRectF(0, 0, 5000, 5000))
self.scene.itemInserted.connect(self.itemInserted)
self.scene.textInserted.connect(self.textInserted)
self.scene.itemSelected.connect(self.itemSelected)
self.createToolbars()
layout = QHBoxLayout()
layout.addWidget(self.toolBox)
self.view = QGraphicsView(self.scene)
layout.addWidget(self.view)
self.widget = QWidget()
self.widget.setLayout(layout)
self.setCentralWidget(self.widget)
self.setWindowTitle("Diagramscene")
def backgroundButtonGroupClicked(self, button):
buttons = self.backgroundButtonGroup.buttons()
for myButton in buttons:
if myButton != button:
button.setChecked(False)
text = button.text()
if text == "Blue Grid":
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background1.png')))
elif text == "White Grid":
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background2.png')))
elif text == "Gray Grid":
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background3.png')))
else:
self.scene.setBackgroundBrush(QBrush(QPixmap(':/images/background4.png')))
self.scene.update()
self.view.update()
def buttonGroupClicked(self, id):
buttons = self.buttonGroup.buttons()
for button in buttons:
if self.buttonGroup.button(id) != button:
button.setChecked(False)
if id == self.InsertTextButton:
self.scene.setMode(DiagramScene.InsertText)
else:
self.scene.setItemType(id)
self.scene.setMode(DiagramScene.InsertItem)
def deleteItem(self):
for item in self.scene.selectedItems():
if isinstance(item, DiagramItem):
item.removeArrows()
self.scene.removeItem(item)
def pointerGroupClicked(self, i):
self.scene.setMode(self.pointerTypeGroup.checkedId())
def bringToFront(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() >= zValue and isinstance(item, DiagramItem)):
zValue = item.zValue() + 0.1
selectedItem.setZValue(zValue)
def sendToBack(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() <= zValue and isinstance(item, DiagramItem)):
zValue = item.zValue() - 0.1
selectedItem.setZValue(zValue)
def itemInserted(self, item):
self.pointerTypeGroup.button(DiagramScene.MoveItem).setChecked(True)
self.scene.setMode(self.pointerTypeGroup.checkedId())
self.buttonGroup.button(item.diagramType).setChecked(False)
def textInserted(self, item):
self.buttonGroup.button(self.InsertTextButton).setChecked(False)
self.scene.setMode(self.pointerTypeGroup.checkedId())
def currentFontChanged(self, font):
self.handleFontChange()
def fontSizeChanged(self, font):
self.handleFontChange()
def sceneScaleChanged(self, scale):
newScale = scale.left(scale.indexOf("%")).toDouble()[0] / 100.0
oldMatrix = self.view.matrix()
self.view.resetMatrix()
self.view.translate(oldMatrix.dx(), oldMatrix.dy())
self.view.scale(newScale, newScale)
def textColorChanged(self):
self.textAction = self.sender()
self.fontColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/textpointer.png',
QColor(self.textAction.data())))
self.textButtonTriggered()
def itemColorChanged(self):
self.fillAction = self.sender()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon( ':/images/floodfill.png',
QColor(self.fillAction.data())))
self.fillButtonTriggered()
def lineColorChanged(self):
self.lineAction = self.sender()
self.lineColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/linecolor.png',
QColor(self.lineAction.data())))
self.lineButtonTriggered()
def textButtonTriggered(self):
self.scene.setTextColor(QColor(self.textAction.data()))
def fillButtonTriggered(self):
self.scene.setItemColor(QColor(self.fillAction.data()))
def lineButtonTriggered(self):
self.scene.setLineColor(QColor(self.lineAction.data()))
def handleFontChange(self):
font = self.fontCombo.currentFont()
font.setPointSize(self.fontSizeCombo.currentText().toInt()[0])
if self.boldAction.isChecked():
font.setWeight(QFont.Bold)
else:
font.setWeight(QFont.Normal)
font.setItalic(self.italicAction.isChecked())
font.setUnderline(self.underlineAction.isChecked())
self.scene.setFont(font)
def itemSelected(self, item):
font = item.font()
color = item.defaultTextColor()
self.fontCombo.setCurrentFont(font)
self.fontSizeCombo.setEditText(str(font.pointSize()))
self.boldAction.setChecked(font.weight() == QFont.Bold)
self.italicAction.setChecked(font.italic())
self.underlineAction.setChecked(font.underline())
def about(self):
QMessageBox.about(self, "About Diagram Scene",
"The <b>Diagram Scene</b> example shows use of the graphics framework.")
def createToolBox(self):
self.buttonGroup = QButtonGroup()
self.buttonGroup.setExclusive(False)
self.buttonGroup.buttonClicked[int].connect(self.buttonGroupClicked)
layout = QGridLayout()
layout.addWidget(self.createCellWidget("Conditional", DiagramItem.Conditional),
0, 0)
layout.addWidget(self.createCellWidget("Process", DiagramItem.Step), 0,
1)
layout.addWidget(self.createCellWidget("Input/Output", DiagramItem.Io),
1, 0)
textButton = QToolButton()
textButton.setCheckable(True)
self.buttonGroup.addButton(textButton, self.InsertTextButton)
textButton.setIcon(QIcon(QPixmap(':/images/textpointer.png').scaled(30, 30)))
textButton.setIconSize(QSize(50, 50))
textLayout = QGridLayout()
textLayout.addWidget(textButton, 0, 0, Qt.AlignHCenter)
textLayout.addWidget(QLabel("Text"), 1, 0, Qt.AlignCenter)
textWidget = QWidget()
textWidget.setLayout(textLayout)
layout.addWidget(textWidget, 1, 1)
layout.setRowStretch(3, 10)
layout.setColumnStretch(2, 10)
itemWidget = QWidget()
itemWidget.setLayout(layout)
self.backgroundButtonGroup = QButtonGroup()
self.backgroundButtonGroup.buttonClicked.connect(self.backgroundButtonGroupClicked)
backgroundLayout = QGridLayout()
backgroundLayout.addWidget(self.createBackgroundCellWidget("Blue Grid",
':/images/background1.png'), 0, 0)
backgroundLayout.addWidget(self.createBackgroundCellWidget("White Grid",
':/images/background2.png'), 0, 1)
backgroundLayout.addWidget(self.createBackgroundCellWidget("Gray Grid",
':/images/background3.png'), 1, 0)
backgroundLayout.addWidget(self.createBackgroundCellWidget("No Grid",
':/images/background4.png'), 1, 1)
backgroundLayout.setRowStretch(2, 10)
backgroundLayout.setColumnStretch(2, 10)
backgroundWidget = QWidget()
backgroundWidget.setLayout(backgroundLayout)
self.toolBox = QToolBox()
self.toolBox.setSizePolicy(QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Ignored))
self.toolBox.setMinimumWidth(itemWidget.sizeHint().width())
self.toolBox.addItem(itemWidget, "Basic Flowchart Shapes")
self.toolBox.addItem(backgroundWidget, "Backgrounds")
def createActions(self):
self.toFrontAction = QAction(
QIcon(':/images/bringtofront.png'), "Bring to &Front",
self, shortcut="Ctrl+F", statusTip="Bring item to front",
triggered=self.bringToFront)
self.sendBackAction = QAction(
QIcon(':/images/sendtoback.png'), "Send to &Back", self,
shortcut="Ctrl+B", statusTip="Send item to back",
triggered=self.sendToBack)
self.deleteAction = QAction(QIcon(':/images/delete.png'),
"&Delete", self, shortcut="Delete",
statusTip="Delete item from diagram",
triggered=self.deleteItem)
self.exitAction = QAction("E&xit", self, shortcut="Ctrl+X",
statusTip="Quit Scenediagram example", triggered=self.close)
self.boldAction = QAction(QIcon(':/images/bold.png'),
"Bold", self, checkable=True, shortcut="Ctrl+B",
triggered=self.handleFontChange)
self.italicAction = QAction(QIcon(':/images/italic.png'),
"Italic", self, checkable=True, shortcut="Ctrl+I",
triggered=self.handleFontChange)
self.underlineAction = QAction(
QIcon(':/images/underline.png'), "Underline", self,
checkable=True, shortcut="Ctrl+U",
triggered=self.handleFontChange)
self.aboutAction = QAction("A&bout", self, shortcut="Ctrl+B",
triggered=self.about)
def createMenus(self):
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.addAction(self.exitAction)
self.itemMenu = self.menuBar().addMenu("&Item")
self.itemMenu.addAction(self.deleteAction)
self.itemMenu.addSeparator()
self.itemMenu.addAction(self.toFrontAction)
self.itemMenu.addAction(self.sendBackAction)
self.aboutMenu = self.menuBar().addMenu("&Help")
self.aboutMenu.addAction(self.aboutAction)
def createToolbars(self):
self.editToolBar = self.addToolBar("Edit")
self.editToolBar.addAction(self.deleteAction)
self.editToolBar.addAction(self.toFrontAction)
self.editToolBar.addAction(self.sendBackAction)
self.fontCombo = QFontComboBox()
self.fontCombo.currentFontChanged.connect(self.currentFontChanged)
self.fontSizeCombo = QComboBox()
self.fontSizeCombo.setEditable(True)
for i in range(8, 30, 2):
self.fontSizeCombo.addItem(str(i))
validator = QIntValidator(2, 64, self)
self.fontSizeCombo.setValidator(validator)
self.fontSizeCombo.currentIndexChanged.connect(self.fontSizeChanged)
self.fontColorToolButton = QToolButton()
self.fontColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.fontColorToolButton.setMenu(
self.createColorMenu(self.textColorChanged, Qt.black))
self.textAction = self.fontColorToolButton.menu().defaultAction()
self.fontColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/textpointer.png',
Qt.black))
self.fontColorToolButton.setAutoFillBackground(True)
self.fontColorToolButton.clicked.connect(self.textButtonTriggered)
self.fillColorToolButton = QToolButton()
self.fillColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.fillColorToolButton.setMenu(
self.createColorMenu(self.itemColorChanged, Qt.white))
self.fillAction = self.fillColorToolButton.menu().defaultAction()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/floodfill.png',
Qt.white))
self.fillColorToolButton.clicked.connect(self.fillButtonTriggered)
self.lineColorToolButton = QToolButton()
self.lineColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.lineColorToolButton.setMenu(
self.createColorMenu(self.lineColorChanged, Qt.black))
self.lineAction = self.lineColorToolButton.menu().defaultAction()
self.lineColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/linecolor.png',
Qt.black))
self.lineColorToolButton.clicked.connect(self.lineButtonTriggered)
self.textToolBar = self.addToolBar("Font")
self.textToolBar.addWidget(self.fontCombo)
self.textToolBar.addWidget(self.fontSizeCombo)
self.textToolBar.addAction(self.boldAction)
self.textToolBar.addAction(self.italicAction)
self.textToolBar.addAction(self.underlineAction)
self.colorToolBar = self.addToolBar("Color")
self.colorToolBar.addWidget(self.fontColorToolButton)
self.colorToolBar.addWidget(self.fillColorToolButton)
self.colorToolBar.addWidget(self.lineColorToolButton)
pointerButton = QToolButton()
pointerButton.setCheckable(True)
pointerButton.setChecked(True)
pointerButton.setIcon(QIcon(':/images/pointer.png'))
linePointerButton = QToolButton()
linePointerButton.setCheckable(True)
linePointerButton.setIcon(QIcon(':/images/linepointer.png'))
self.pointerTypeGroup = QButtonGroup()
self.pointerTypeGroup.addButton(pointerButton, DiagramScene.MoveItem)
self.pointerTypeGroup.addButton(linePointerButton,
DiagramScene.InsertLine)
self.pointerTypeGroup.buttonClicked[int].connect(self.pointerGroupClicked)
self.sceneScaleCombo = QComboBox()
self.sceneScaleCombo.addItems(["50%", "75%", "100%", "125%", "150%"])
self.sceneScaleCombo.setCurrentIndex(2)
self.sceneScaleCombo.currentIndexChanged[str].connect(self.sceneScaleChanged)
self.pointerToolbar = self.addToolBar("Pointer type")
self.pointerToolbar.addWidget(pointerButton)
self.pointerToolbar.addWidget(linePointerButton)
self.pointerToolbar.addWidget(self.sceneScaleCombo)
def createBackgroundCellWidget(self, text, image):
button = QToolButton()
button.setText(text)
button.setIcon(QIcon(image))
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.backgroundButtonGroup.addButton(button)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createCellWidget(self, text, diagramType):
item = DiagramItem(diagramType, self.itemMenu)
icon = QIcon(item.image())
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.buttonGroup.addButton(button, diagramType)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createColorMenu(self, slot, defaultColor):
colors = [Qt.black, Qt.white, Qt.red, Qt.blue, Qt.yellow]
names = ["black", "white", "red", "blue", "yellow"]
colorMenu = QMenu(self)
for color, name in zip(colors, names):
action = QAction(self.createColorIcon(color), name, self,
triggered=slot)
action.setData(QColor(color))
colorMenu.addAction(action)
if color == defaultColor:
colorMenu.setDefaultAction(action)
return colorMenu
def createColorToolButtonIcon(self, imageFile, color):
pixmap = QPixmap(50, 80)
pixmap.fill(Qt.transparent)
painter = QPainter(pixmap)
image = QPixmap(imageFile)
target = QRect(0, 0, 50, 60)
source = QRect(0, 0, 42, 42)
painter.fillRect(QRect(0, 60, 50, 80), color)
painter.drawPixmap(target, image, source)
painter.end()
return QIcon(pixmap)
def createColorIcon(self, color):
pixmap = QPixmap(20, 20)
painter = QPainter(pixmap)
painter.setPen(Qt.NoPen)
painter.fillRect(QRect(0, 0, 20, 20), color)
painter.end()
return QIcon(pixmap)
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.createActions()
self.createMenus()
self.createToolBox()
self.createStatusBar()
self.scene = ITGScene(self.itemMenu)
self.scene.setSceneRect(QRectF(0, 0, 5000, 5000))
self.scene.itemInserted.connect(self.itemInserted)
self.scene.itemSelected.connect(self.itemSelected)
# ================================================================
# THIS IS THE PROBLEM SECTION
# THIS CAUSES TypeError: native Qt signal is not callable
# To reproduce the error, run the program, click on one of the
# shapes, then click on the scene area
# ================================================================
self.scene.selectionChanged.connect(self.print_item_info) # this one doesn't work
self.createToolbars()
layout = QHBoxLayout()
layout.addWidget(self.toolBox)
self.view = QGraphicsView(self.scene) # view and scene established together
layout.addWidget(self.view)
self.widget = QWidget()
self.widget.setLayout(layout)
self.setCentralWidget(self.widget)
self.setWindowTitle("ITG: Illegitimate Template Generator")
#@pyqtSlot() # Adding this does not change the error
def print_item_info(self):
for item in self.scene.selectedItems():
print(item)
def buttonGroupClicked(self, id):
buttons = self.buttonGroup.buttons()
for button in buttons:
if self.buttonGroup.button(id) != button:
button.setChecked(False)
self.scene.setItemType(id)
self.scene.setMode(ITGScene.InsertItem)
self.updateStatusBar()
def deleteItem(self):
for item in self.scene.selectedItems():
self.scene.removeItem(item)
def pointerGroupClicked(self, i):
self.scene.setMode(self.pointerTypeGroup.checkedId())
self.updateStatusBar()
def bringToFront(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() >= zValue and isinstance(item, ITGItem)):
zValue = item.zValue() + 0.1
selectedItem.setZValue(zValue)
def sendToBack(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() <= zValue and isinstance(item, ITGItem)):
zValue = item.zValue() - 0.1
selectedItem.setZValue(zValue)
def itemInserted(self, item):
self.pointerTypeGroup.button(ITGScene.MoveItem).setChecked(True)
self.scene.setMode(self.pointerTypeGroup.checkedId())
self.updateStatusBar()
self.buttonGroup.button(item.ITGType).setChecked(False)
def sceneScaleChanged(self, scale):
newScale = scale.left(scale.indexOf("%")).toDouble()[0] / 100.0
oldMatrix = self.view.matrix()
self.view.resetMatrix()
self.view.translate(oldMatrix.dx(), oldMatrix.dy())
self.view.scale(newScale, newScale)
def itemColorChanged(self):
self.fillAction = self.sender()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon( ':/images/floodfill.png',
QColor(self.fillAction.data())))
self.fillButtonTriggered()
def fillButtonTriggered(self):
self.scene.setItemColor(QColor(self.fillAction.data()))
def itemSelected(self, item):
font = item.font()
color = item.defaultTextColor()
self.fontCombo.setCurrentFont(font)
self.fontSizeCombo.setEditText(str(font.pointSize()))
self.boldAction.setChecked(font.weight() == QFont.Bold)
self.italicAction.setChecked(font.italic())
self.underlineAction.setChecked(font.underline())
def about(self):
QMessageBox.about(self, "About ITG Scene",
"The <b>ITG Scene</b> example shows use of the graphics framework.")
def createToolBox(self):
self.buttonGroup = QButtonGroup()
self.buttonGroup.setExclusive(False)
self.buttonGroup.buttonClicked[int].connect(self.buttonGroupClicked)
layout = QGridLayout()
layout.addWidget(self.createCellWidget("Diamond", ITGItem.Diamond),
0, 0)
layout.addWidget(self.createCellWidget("Square", ITGItem.Square), 0,
1)
layout.addWidget(self.createCellWidget("Parallelogram", ITGItem.Parallelogram),
1, 0)
layout.setRowStretch(3, 10)
layout.setColumnStretch(2, 10)
itemWidget = QWidget()
itemWidget.setLayout(layout)
self.backgroundButtonGroup = QButtonGroup()
self.toolBox = QToolBox()
self.toolBox.setSizePolicy(QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Ignored))
self.toolBox.setMinimumWidth(itemWidget.sizeHint().width())
self.toolBox.addItem(itemWidget, "Basic Components")
def createActions(self):
self.toFrontAction = QAction(
QIcon(':/images/bringtofront.png'), "Bring to &Front",
self, shortcut="Ctrl+F", statusTip="Bring item to front",
triggered=self.bringToFront)
self.sendBackAction = QAction(
QIcon(':/images/sendtoback.png'), "Send to &Back", self,
shortcut="Ctrl+B", statusTip="Send item to back",
triggered=self.sendToBack)
self.deleteAction = QAction(QIcon(':/images/delete.png'),
"&Delete", self, shortcut="Delete",
statusTip="Delete item from ITG",
triggered=self.deleteItem)
self.exitAction = QAction("E&xit", self, shortcut="Ctrl+X",
statusTip="Quit SceneITG example", triggered=self.close)
self.aboutAction = QAction("A&bout", self, shortcut="Ctrl+B",
triggered=self.about)
def createMenus(self):
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.addAction(self.exitAction)
self.itemMenu = self.menuBar().addMenu("&Item")
self.itemMenu.addAction(self.deleteAction)
self.itemMenu.addSeparator()
self.itemMenu.addAction(self.toFrontAction)
self.itemMenu.addAction(self.sendBackAction)
self.aboutMenu = self.menuBar().addMenu("&Help")
self.aboutMenu.addAction(self.aboutAction)
def createToolbars(self):
self.editToolBar = self.addToolBar("Edit")
self.editToolBar.addAction(self.deleteAction)
self.editToolBar.addAction(self.toFrontAction)
self.editToolBar.addAction(self.sendBackAction)
self.fillColorToolButton = QToolButton()
self.fillColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.fillColorToolButton.setMenu(
self.createColorMenu(self.itemColorChanged, Qt.red))
self.fillAction = self.fillColorToolButton.menu().defaultAction()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/floodfill.png',
Qt.red))
self.fillColorToolButton.clicked.connect(self.fillButtonTriggered)
self.colorToolBar = self.addToolBar("Color")
self.colorToolBar.addWidget(self.fillColorToolButton)
pointerButton = QToolButton()
pointerButton.setCheckable(True)
pointerButton.setChecked(True)
pointerButton.setIcon(QIcon(':/images/pointer.png'))
self.pointerTypeGroup = QButtonGroup()
self.pointerTypeGroup.addButton(pointerButton, ITGScene.MoveItem)
self.pointerTypeGroup.buttonClicked[int].connect(self.pointerGroupClicked)
self.sceneScaleCombo = QComboBox()
self.sceneScaleCombo.addItems(["50%", "75%", "100%", "125%", "150%"])
self.sceneScaleCombo.setCurrentIndex(2)
self.sceneScaleCombo.currentIndexChanged[str].connect(self.sceneScaleChanged)
self.pointerToolbar = self.addToolBar("Pointer type")
self.pointerToolbar.addWidget(pointerButton)
self.pointerToolbar.addWidget(self.sceneScaleCombo)
def createBackgroundCellWidget(self, text, image):
button = QToolButton()
button.setText(text)
button.setIcon(QIcon(image))
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.backgroundButtonGroup.addButton(button)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createCellWidget(self, text, ITGType):
item = ITGItem(ITGType, self.itemMenu)
icon = QIcon(item.image())
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.buttonGroup.addButton(button, ITGType)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createColorMenu(self, slot, defaultColor):
colors = [Qt.black, Qt.white, Qt.red, Qt.blue, Qt.yellow]
names = ["black", "white", "red", "blue", "yellow"]
colorMenu = QMenu(self)
for color, name in zip(colors, names):
action = QAction(self.createColorIcon(color), name, self,
triggered=slot)
action.setData(QColor(color))
colorMenu.addAction(action)
if color == defaultColor:
colorMenu.setDefaultAction(action)
return colorMenu
def createColorToolButtonIcon(self, imageFile, color):
pixmap = QPixmap(50, 80)
pixmap.fill(Qt.transparent)
painter = QPainter(pixmap)
image = QPixmap(imageFile)
target = QRect(0, 0, 50, 60)
source = QRect(0, 0, 42, 42)
painter.fillRect(QRect(0, 60, 50, 80), color)
painter.drawPixmap(target, image, source)
painter.end()
return QIcon(pixmap)
def createColorIcon(self, color):
pixmap = QPixmap(20, 20)
painter = QPainter(pixmap)
painter.setPen(Qt.NoPen)
painter.fillRect(QRect(0, 0, 20, 20), color)
painter.end()
return QIcon(pixmap)
def createStatusBar(self):
self.sbar = self.statusBar()
self.lbl_mode = QLabel('Mode: not set')
self.sbar.addPermanentWidget(self.lbl_mode)
self.lbl_selection = QLabel('Sel: none')
self.sbar.addPermanentWidget(self.lbl_selection)
def updateStatusBar(self):
self.lbl_mode.setText('Mode: ' + self.scene.getMode())
#! This section is not working!!!
if len(self.scene.selectedItems()) == 0:
self.lbl_selection.setText('Sel: none')
elif len(self.scene.selectedItems()) == 1:
self.lbl_selection.setText('Sel: ' + self.scene.selectedItems()[0].get_ITGType_as_str())
elif len(self.scene.selectedItems()) > 1:
self.lbl_selection.setText('Sel: <multiple>')
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.create_main_window_actions([
QActionProperties(
name=main_window_constants.EXPORT_ACTION_NAME,
icon=QIcon(':/icons/export'),
text='&Export',
shortcut='Ctrl+E',
status_tip='Export to Python code',
triggered=self.export_diagram,
),
QActionProperties(
name=main_window_constants.DELETE_ACTON_NAME,
icon=QIcon(':/icons/delete'),
text='&Delete',
shortcut='Delete',
status_tip='Delete item from diagram',
triggered=self.delete_item,
),
QActionProperties(
name=main_window_constants.TO_FRONT_ACTION_NAME,
icon=QIcon(':/icons/bring_to_front'),
text='Bring to &Front',
shortcut='Ctrl+F',
status_tip='Bring item to front',
triggered=self.bring_to_front,
),
QActionProperties(
name=main_window_constants.TO_BACK_ACTION_NAME,
icon=QIcon(':/icons/send_to_back'),
text='Send to &Back',
shortcut='Ctrl+B',
status_tip='Send item to back',
triggered=self.send_to_back,
),
])
self.create_file_menu()
self.create_edit_menu()
self.create_frameworks_toolbar()
self.create_edit_diagram_toolbar()
self.create_pointer_toolbar()
self.create_diagram_scene_and_view()
self.create_framework_toolbox()
self.layout = QHBoxLayout()
self.layout.addWidget(self.framework_toolbox)
self.layout.addWidget(self.view)
self.widget = QWidget()
self.widget.setLayout(self.layout)
self.setWindowTitle(main_window_constants.MAIN_WINDOW_TITLE)
self.setCentralWidget(self.widget)
# Create methods.
def create_main_window_actions(
self, main_window_actions: List[QActionProperties]):
self.main_window_actions = dict()
for main_window_action in main_window_actions:
self.main_window_actions[main_window_action.name] = QAction(
main_window_action.icon,
main_window_action.text,
self,
shortcut=main_window_action.shortcut,
statusTip=main_window_action.status_tip,
triggered=main_window_action.triggered,
)
def create_file_menu(self):
self.file_menu = self.menuBar().addMenu(
main_window_constants.FILE_MENU_NAME)
self.file_menu.addAction(
self.main_window_actions[main_window_constants.EXPORT_ACTION_NAME])
def create_edit_menu(self):
self.edit_menu = self.menuBar().addMenu(
main_window_constants.EDIT_MENU_NAME)
self.edit_menu.addAction(
self.main_window_actions[main_window_constants.DELETE_ACTON_NAME])
self.edit_menu.addSeparator()
self.edit_menu.addAction(self.main_window_actions[
main_window_constants.TO_FRONT_ACTION_NAME])
self.edit_menu.addAction(self.main_window_actions[
main_window_constants.TO_BACK_ACTION_NAME])
def create_frameworks_toolbar(self):
self.frameworks_label = QLabel(main_window_constants.FRAMEWORKS_LABEL)
self.frameworks_combobox = QComboBox()
self.frameworks_combobox.setEditable(False)
for framework in frameworks_utils.get_sorted_frameworks_list():
self.frameworks_combobox.addItem(framework)
self.frameworks_toolbar = self.addToolBar(
main_window_constants.FRAMEWORKS_TOOLBAR_NAME)
self.frameworks_toolbar.addWidget(self.frameworks_label)
self.frameworks_toolbar.addWidget(self.frameworks_combobox)
def create_edit_diagram_toolbar(self):
self.edit_diagram_toolbar = self.addToolBar(
main_window_constants.DIAGRAM_EDIT_TOOLBAR_NAME)
self.edit_diagram_toolbar.addAction(
self.main_window_actions[main_window_constants.DELETE_ACTON_NAME])
self.edit_diagram_toolbar.addAction(self.main_window_actions[
main_window_constants.TO_FRONT_ACTION_NAME])
self.edit_diagram_toolbar.addAction(self.main_window_actions[
main_window_constants.TO_BACK_ACTION_NAME])
def create_pointer_toolbar(self):
pointer_button = QToolButton()
pointer_button.setCheckable(True)
pointer_button.setChecked(True)
pointer_button.setIcon(QIcon(':/icons/pointer'))
line_pointer_button = QToolButton()
line_pointer_button.setCheckable(True)
line_pointer_button.setIcon(QIcon(':/icons/line_pointer'))
self.pointer_type_group = QButtonGroup()
self.pointer_type_group.addButton(pointer_button,
DiagramScene.move_item)
self.pointer_type_group.addButton(line_pointer_button,
DiagramScene.insert_line)
self.pointer_type_group.buttonClicked[int].connect(
self.pointer_group_clicked)
scene_scale_combo = QComboBox()
scene_scale_combo.addItems(main_window_constants.DIAGRAM_SCENE_SCALES)
scene_scale_combo.setCurrentIndex(
main_window_constants.DIAGRAM_SCENE_SCALES.index(
main_window_constants.DIAGRAM_SCENE_DEFAULT_SCALE))
scene_scale_combo.currentIndexChanged[str].connect(
self.scene_scale_changed)
self.pointer_toolbar = self.addToolBar(
main_window_constants.POINTER_TYPE_TOOLBAR_NAME)
self.pointer_toolbar.addWidget(pointer_button)
self.pointer_toolbar.addWidget(line_pointer_button)
self.pointer_toolbar.addWidget(scene_scale_combo)
def create_diagram_scene_and_view(self):
self.scene = DiagramScene(self.edit_menu)
self.scene.setSceneRect(
QRectF(0, 0, main_window_constants.DIAGRAM_SCENE_SIZE,
main_window_constants.DIAGRAM_SCENE_SIZE))
self.scene.item_inserted.connect(self.item_inserted)
self.view = QGraphicsView(self.scene)
def create_framework_toolbox(self):
framework_layers = frameworks_utils.get_framework_layers(
self.get_selected_framework())
self.framework_layers_button_group = QButtonGroup()
self.framework_layers_button_group.setExclusive(False)
self.framework_layers_button_group.buttonClicked[int].connect(
self.framework_layers_button_group_clicked)
layout = QGridLayout()
for framework_layer in framework_layers:
layout.addWidget(
self.create_framework_layer_widget(framework_layer()))
layout.setRowStretch(3, 10)
layout.setColumnStretch(2, 10)
item_widget = QWidget()
item_widget.setLayout(layout)
self.framework_toolbox = QToolBox()
self.framework_toolbox.setSizePolicy(
QSizePolicy(
QSizePolicy.Maximum,
QSizePolicy.Ignored,
), )
self.framework_toolbox.setMinimumWidth(item_widget.sizeHint().width())
self.framework_toolbox.addItem(item_widget,
main_window_constants.LAYERS)
# Callback methods.
def export_diagram(self):
nodes = self.get_nodes_from_scene()
if len(nodes) == 0:
return
edges = self.get_edges_from_scene()
nodes_mapping = self.create_nodes_mapping(nodes)
uni_graph = graph_utils.create_graph_from_qt_elements(
nodes, edges, nodes_mapping)
bi_graph = graph_utils.create_graph_from_qt_elements(
nodes, edges, nodes_mapping, is_bi_directional=True)
is_one_connected_component = graph_utils.is_one_connected_component(
bi_graph)
graph_topological_sort = graph_utils.create_graph_topological_sort(
uni_graph)
root_nodes = graph_utils.get_root_nodes(uni_graph)
is_all_root_nodes_are_input_layers = graph_utils.is_all_root_nodes_are_input_layers(
nodes, root_nodes)
if not is_one_connected_component:
self.show_model_graph_eval_error_msg(
main_window_constants.MODEL_GRAPH_MULTIPLE_COMPONENTS_ERROR_MSG
)
elif graph_topological_sort is None:
self.show_model_graph_eval_error_msg(
main_window_constants.MODEL_GRAPH_CYCLE_ERROR_MSG)
elif not is_all_root_nodes_are_input_layers:
self.show_model_graph_eval_error_msg(
main_window_constants.
MODEL_GRAPH_ROOT_NODE_IS_NOT_INPUT_ERROR_MSG)
else:
input_definitions = self.build_input_definitions(
map(nodes.__getitem__, root_nodes))
layer_definitions = self.build_layer_definitions(
nodes, graph_topological_sort, root_nodes)
model_connections = self.build_model_connections(
nodes, uni_graph, graph_topological_sort, root_nodes)
framework_template = frameworks_utils.get_formatted_framework_template(
self.get_selected_framework(),
input_definitions,
layer_definitions,
model_connections,
)
file_path, _ = QFileDialog.getSaveFileName(
self,
'Export Model As',
'talzeeq.py',
'Python Language (*.py);;'
'All files (*.*)',
)
if file_path:
with open(file_path, 'w') as fp:
fp.write(framework_template)
def delete_item(self):
for item in self.scene.selectedItems():
if isinstance(item, DiagramItem):
item.remove_arrows()
self.scene.removeItem(item)
def bring_to_front(self):
for selected_item in self.scene.selectedItems():
z_value = 0
for item in selected_item.collidingItems():
if item.zValue() >= z_value and isinstance(item, DiagramItem):
z_value = item.zValue() + 0.1
selected_item.setZValue(z_value)
def send_to_back(self):
for selected_item in self.scene.selectedItems():
z_value = 0
for item in selected_item.collidingItems():
if item.zValue() <= z_value and isinstance(item, DiagramItem):
z_value = item.zValue() - 0.1
selected_item.setZValue(z_value)
def pointer_group_clicked(self, index: int):
self.scene.set_mode(self.pointer_type_group.checkedId())
def scene_scale_changed(self, scale: str):
new_scale = int(
scale[:scale.index(main_window_constants.
DIAGRAM_SCENE_SCALE_PERCENT)]) / 100.0
old_transform = self.view.transform()
self.view.resetTransform()
self.view.translate(old_transform.dx(), old_transform.dy())
self.view.scale(new_scale, new_scale)
def item_inserted(self, item: DiagramItem):
self.pointer_type_group.button(DiagramScene.move_item).setChecked(True)
self.scene.set_mode(self.pointer_type_group.checkedId())
layer_index = frameworks_utils.get_framework_layer_index(
self.get_selected_framework(),
item.framework_layer.__class__,
)
self.framework_layers_button_group.button(layer_index).setChecked(
False)
def framework_layers_button_group_clicked(self, id: int):
buttons = self.framework_layers_button_group.buttons()
for button in buttons:
if self.framework_layers_button_group.button(id) != button:
button.setChecked(False)
if self.framework_layers_button_group.button(id).isChecked():
self.scene.set_item_type(id)
self.scene.set_framework_name(self.get_selected_framework())
self.scene.set_mode(DiagramScene.insert_item)
else:
self.scene.set_mode(DiagramScene.move_item)
# Helper methods.
def get_selected_framework(self) -> str:
return str(self.frameworks_combobox.currentText())
def get_nodes_from_scene(self) -> List[DiagramItem]:
return list(
filter(lambda item: isinstance(item, DiagramItem),
self.scene.items()))
def get_edges_from_scene(self) -> List[Arrow]:
return list(
filter(lambda item: isinstance(item, Arrow), self.scene.items()))
def create_nodes_mapping(
self, nodes: List[DiagramItem]) -> Dict[DiagramItem, int]:
return {node: index for index, node in enumerate(nodes)}
def show_model_graph_eval_error_msg(self, message: str):
msg = QMessageBox()
msg.setIcon(QMessageBox.Critical)
msg.setText(main_window_constants.MODEL_GRAPH_EVAL_ERROR_MSG_TEXT)
msg.setInformativeText(message)
msg.setWindowTitle(
main_window_constants.MODEL_GRAPH_EVAL_ERROR_MSG_TEXT)
msg.exec_()
def build_input_definitions(self, root_nodes: List[DiagramItem]) -> str:
input_definitions = list()
for node in root_nodes:
input_definitions.append(
node.get_framework_layer().layer_definition())
return ', '.join(input_definitions)
def build_layer_definitions(
self,
nodes: List[DiagramItem],
graph_topological_sort: List[int],
root_nodes: List[int],
) -> str:
layer_definitions = list()
for element in graph_topological_sort:
if element not in root_nodes:
layer_definitions.append(
nodes[element].get_framework_layer().layer_definition())
return '\n'.join(layer_definitions)
def build_model_connections(
self,
nodes: List[DiagramItem],
graph: List[List[int]],
graph_topological_sort: List[int],
root_nodes: List[int],
) -> str:
model_connections = list()
for element in graph_topological_sort:
parents = list()
is_root = list()
for node in range(len(graph)):
if element in graph[node]:
parents.append(nodes[node].get_framework_layer())
is_root.append(node in root_nodes)
layer_connections = nodes[element].get_framework_layer(
).layer_connections(parents, is_root)
if layer_connections:
model_connections.append(layer_connections)
return '\n'.join(model_connections)
def create_framework_layer_widget(
self, framework_layer: LayerInterface) -> QWidget:
button = QToolButton()
button.setText(framework_layer.layer_name())
button.setCheckable(True)
self.framework_layers_button_group.addButton(
button,
frameworks_utils.get_framework_layer_index(
self.get_selected_framework(),
framework_layer.__class__,
),
)
layout = QVBoxLayout()
layout.addWidget(button)
widget = QWidget()
widget.setLayout(layout)
return widget
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.createActions()
self.createMenus()
self.createToolBox()
self.createStatusBar()
self.scene = ITGScene(self.itemMenu)
self.scene.setSceneRect(QRectF(0, 0, 5000, 5000))
self.scene.itemInserted.connect(self.itemInserted)
self.scene.itemSelected.connect(self.itemSelected)
# ================================================================
# THIS IS THE PROBLEM SECTION
# THIS CAUSES TypeError: native Qt signal is not callable
# To reproduce the error, run the program, click on one of the
# shapes, then click on the scene area
# ================================================================
self.scene.selectionChanged.connect(
self.print_item_info) # this one doesn't work
self.createToolbars()
layout = QHBoxLayout()
layout.addWidget(self.toolBox)
self.view = QGraphicsView(
self.scene) # view and scene established together
layout.addWidget(self.view)
self.widget = QWidget()
self.widget.setLayout(layout)
self.setCentralWidget(self.widget)
self.setWindowTitle("ITG: Illegitimate Template Generator")
#@pyqtSlot() # Adding this does not change the error
def print_item_info(self):
for item in self.scene.selectedItems():
print(item)
def buttonGroupClicked(self, id):
buttons = self.buttonGroup.buttons()
for button in buttons:
if self.buttonGroup.button(id) != button:
button.setChecked(False)
self.scene.setItemType(id)
self.scene.setMode(ITGScene.InsertItem)
self.updateStatusBar()
def deleteItem(self):
for item in self.scene.selectedItems():
self.scene.removeItem(item)
def pointerGroupClicked(self, i):
self.scene.setMode(self.pointerTypeGroup.checkedId())
self.updateStatusBar()
def bringToFront(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() >= zValue and isinstance(item, ITGItem)):
zValue = item.zValue() + 0.1
selectedItem.setZValue(zValue)
def sendToBack(self):
if not self.scene.selectedItems():
return
selectedItem = self.scene.selectedItems()[0]
overlapItems = selectedItem.collidingItems()
zValue = 0
for item in overlapItems:
if (item.zValue() <= zValue and isinstance(item, ITGItem)):
zValue = item.zValue() - 0.1
selectedItem.setZValue(zValue)
def itemInserted(self, item):
self.pointerTypeGroup.button(ITGScene.MoveItem).setChecked(True)
self.scene.setMode(self.pointerTypeGroup.checkedId())
self.updateStatusBar()
self.buttonGroup.button(item.ITGType).setChecked(False)
def sceneScaleChanged(self, scale):
newScale = scale.left(scale.indexOf("%")).toDouble()[0] / 100.0
oldMatrix = self.view.matrix()
self.view.resetMatrix()
self.view.translate(oldMatrix.dx(), oldMatrix.dy())
self.view.scale(newScale, newScale)
def itemColorChanged(self):
self.fillAction = self.sender()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/floodfill.png',
QColor(self.fillAction.data())))
self.fillButtonTriggered()
def fillButtonTriggered(self):
self.scene.setItemColor(QColor(self.fillAction.data()))
def itemSelected(self, item):
font = item.font()
color = item.defaultTextColor()
self.fontCombo.setCurrentFont(font)
self.fontSizeCombo.setEditText(str(font.pointSize()))
self.boldAction.setChecked(font.weight() == QFont.Bold)
self.italicAction.setChecked(font.italic())
self.underlineAction.setChecked(font.underline())
def about(self):
QMessageBox.about(
self, "About ITG Scene",
"The <b>ITG Scene</b> example shows use of the graphics framework."
)
def createToolBox(self):
self.buttonGroup = QButtonGroup()
self.buttonGroup.setExclusive(False)
self.buttonGroup.buttonClicked[int].connect(self.buttonGroupClicked)
layout = QGridLayout()
layout.addWidget(self.createCellWidget("Diamond", ITGItem.Diamond), 0,
0)
layout.addWidget(self.createCellWidget("Square", ITGItem.Square), 0, 1)
layout.addWidget(
self.createCellWidget("Parallelogram", ITGItem.Parallelogram), 1,
0)
layout.setRowStretch(3, 10)
layout.setColumnStretch(2, 10)
itemWidget = QWidget()
itemWidget.setLayout(layout)
self.backgroundButtonGroup = QButtonGroup()
self.toolBox = QToolBox()
self.toolBox.setSizePolicy(
QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Ignored))
self.toolBox.setMinimumWidth(itemWidget.sizeHint().width())
self.toolBox.addItem(itemWidget, "Basic Components")
def createActions(self):
self.toFrontAction = QAction(QIcon(':/images/bringtofront.png'),
"Bring to &Front",
self,
shortcut="Ctrl+F",
statusTip="Bring item to front",
triggered=self.bringToFront)
self.sendBackAction = QAction(QIcon(':/images/sendtoback.png'),
"Send to &Back",
self,
shortcut="Ctrl+B",
statusTip="Send item to back",
triggered=self.sendToBack)
self.deleteAction = QAction(QIcon(':/images/delete.png'),
"&Delete",
self,
shortcut="Delete",
statusTip="Delete item from ITG",
triggered=self.deleteItem)
self.exitAction = QAction("E&xit",
self,
shortcut="Ctrl+X",
statusTip="Quit SceneITG example",
triggered=self.close)
self.aboutAction = QAction("A&bout",
self,
shortcut="Ctrl+B",
triggered=self.about)
def createMenus(self):
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.addAction(self.exitAction)
self.itemMenu = self.menuBar().addMenu("&Item")
self.itemMenu.addAction(self.deleteAction)
self.itemMenu.addSeparator()
self.itemMenu.addAction(self.toFrontAction)
self.itemMenu.addAction(self.sendBackAction)
self.aboutMenu = self.menuBar().addMenu("&Help")
self.aboutMenu.addAction(self.aboutAction)
def createToolbars(self):
self.editToolBar = self.addToolBar("Edit")
self.editToolBar.addAction(self.deleteAction)
self.editToolBar.addAction(self.toFrontAction)
self.editToolBar.addAction(self.sendBackAction)
self.fillColorToolButton = QToolButton()
self.fillColorToolButton.setPopupMode(QToolButton.MenuButtonPopup)
self.fillColorToolButton.setMenu(
self.createColorMenu(self.itemColorChanged, Qt.red))
self.fillAction = self.fillColorToolButton.menu().defaultAction()
self.fillColorToolButton.setIcon(
self.createColorToolButtonIcon(':/images/floodfill.png', Qt.red))
self.fillColorToolButton.clicked.connect(self.fillButtonTriggered)
self.colorToolBar = self.addToolBar("Color")
self.colorToolBar.addWidget(self.fillColorToolButton)
pointerButton = QToolButton()
pointerButton.setCheckable(True)
pointerButton.setChecked(True)
pointerButton.setIcon(QIcon(':/images/pointer.png'))
self.pointerTypeGroup = QButtonGroup()
self.pointerTypeGroup.addButton(pointerButton, ITGScene.MoveItem)
self.pointerTypeGroup.buttonClicked[int].connect(
self.pointerGroupClicked)
self.sceneScaleCombo = QComboBox()
self.sceneScaleCombo.addItems(["50%", "75%", "100%", "125%", "150%"])
self.sceneScaleCombo.setCurrentIndex(2)
self.sceneScaleCombo.currentIndexChanged[str].connect(
self.sceneScaleChanged)
self.pointerToolbar = self.addToolBar("Pointer type")
self.pointerToolbar.addWidget(pointerButton)
self.pointerToolbar.addWidget(self.sceneScaleCombo)
def createBackgroundCellWidget(self, text, image):
button = QToolButton()
button.setText(text)
button.setIcon(QIcon(image))
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.backgroundButtonGroup.addButton(button)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createCellWidget(self, text, ITGType):
item = ITGItem(ITGType, self.itemMenu)
icon = QIcon(item.image())
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.buttonGroup.addButton(button, ITGType)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
def createColorMenu(self, slot, defaultColor):
colors = [Qt.black, Qt.white, Qt.red, Qt.blue, Qt.yellow]
names = ["black", "white", "red", "blue", "yellow"]
colorMenu = QMenu(self)
for color, name in zip(colors, names):
action = QAction(self.createColorIcon(color),
name,
self,
triggered=slot)
action.setData(QColor(color))
colorMenu.addAction(action)
if color == defaultColor:
colorMenu.setDefaultAction(action)
return colorMenu
def createColorToolButtonIcon(self, imageFile, color):
pixmap = QPixmap(50, 80)
pixmap.fill(Qt.transparent)
painter = QPainter(pixmap)
image = QPixmap(imageFile)
target = QRect(0, 0, 50, 60)
source = QRect(0, 0, 42, 42)
painter.fillRect(QRect(0, 60, 50, 80), color)
painter.drawPixmap(target, image, source)
painter.end()
return QIcon(pixmap)
def createColorIcon(self, color):
pixmap = QPixmap(20, 20)
painter = QPainter(pixmap)
painter.setPen(Qt.NoPen)
painter.fillRect(QRect(0, 0, 20, 20), color)
painter.end()
return QIcon(pixmap)
def createStatusBar(self):
self.sbar = self.statusBar()
self.lbl_mode = QLabel('Mode: not set')
self.sbar.addPermanentWidget(self.lbl_mode)
self.lbl_selection = QLabel('Sel: none')
self.sbar.addPermanentWidget(self.lbl_selection)
def updateStatusBar(self):
self.lbl_mode.setText('Mode: ' + self.scene.getMode())
#! This section is not working!!!
if len(self.scene.selectedItems()) == 0:
self.lbl_selection.setText('Sel: none')
elif len(self.scene.selectedItems()) == 1:
self.lbl_selection.setText(
'Sel: ' + self.scene.selectedItems()[0].get_ITGType_as_str())
elif len(self.scene.selectedItems()) > 1:
self.lbl_selection.setText('Sel: <multiple>')
class MainWindow(QMainWindow):
InsertTextButton = 10
items = {-2: "source", -3: "channel", -4: "sink"}
def __init__(self):
import _diagramscene_rc
super(MainWindow, self).__init__()
self.config_manipulations = FlumeConfig(self)
properties_generator.dump_props()
self.create_actions()
self.create_menus()
self.create_tool_box()
self.clicked_button_id = 0
self.scene = DiagramScene(self.item_menu)
self.scene.setSceneRect(QRectF(0, 0, 5000, 5000))
self.scene.itemInserted.connect(self.item_inserted)
self.scene.textInserted.connect(self.text_inserted)
self.scene.itemSelected.connect(self.item_selected)
self.create_tool_bars()
# self.scene.enable_grid()
layout = QHBoxLayout()
layout.addWidget(self.tool_box)
self.view = QGraphicsView(self.scene)
self.view.centerOn(0, 0)
layout.addWidget(self.view)
self.widget = QWidget()
self.widget.setLayout(layout)
self.setCentralWidget(self.widget)
self.setWindowTitle("The Flume Illustrator")
# noinspection PyAttributeOutsideInit,PyArgumentList
def create_actions(self):
self.to_front_action = QAction(QIcon(':/images/bringtofront.png'),
"Bring to &Front", self, shortcut="Ctrl+F",
statusTip="Bring item to front", triggered=self.bring_to_front)
self.send_back_action = QAction(QIcon(':/images/sendtoback.png'),
"Send to &Back", self, shortcut="Ctrl+B",
statusTip="Send item to back", triggered=self.send_to_back)
self.bold_action = QAction(QIcon(':/images/bold.png'),
"Bold", self, checkable=True, shortcut="Ctrl+B",
triggered=self.handle_font_change)
self.italic_action = QAction(QIcon(':/images/italic.png'),
"Italic", self, checkable=True, shortcut="Ctrl+I",
triggered=self.handle_font_change)
self.underline_action = QAction(QIcon(':/images/underline.png'),
"Underline", self, checkable=True, shortcut="Ctrl+U",
triggered=self.handle_font_change)
self.delete_action = QAction(QIcon(':/images/delete.png'),
"Delete", self, shortcut="Delete", statusTip='Delete item from diagram',
triggered=self.delete_item)
self.exit_action = QAction("Exit", self, shortcut="Ctrl+X",
statusTip="Quit program", triggered=self.close)
self.about_action = QAction("About", self, shortcut="Ctrl+B",
triggered=self.about)
self.load_config_action = QAction("Load", self, shortcut="Ctrl+O",
statusTip="Load config file", triggered=self.config_manipulations.load_config)
self.enable_grid_action = QAction("Enable grid", self, checkable=True, triggered=self.enable_grid)
# noinspection PyAttributeOutsideInit
def create_menus(self):
self.file_menu = self.menuBar().addMenu("File")
self.file_menu.addAction(self.load_config_action)
self.file_menu.addAction(self.exit_action)
self.item_menu = self.menuBar().addMenu("Item")
self.item_menu.addAction(self.delete_action)
self.item_menu.addSeparator()
self.item_menu.addAction(self.to_front_action)
self.item_menu.addAction(self.send_back_action)
self.about_menu = self.menuBar().addMenu("Help")
self.about_menu.addAction(self.about_action)
# noinspection PyAttributeOutsideInit,PyUnresolvedReferences
def create_tool_box(self):
self.button_group = QButtonGroup()
self.button_group.setExclusive(False)
self.button_group.buttonClicked[int].connect(self.button_group_clicked)
layout = QGridLayout()
layout.addWidget(self.create_cell_widget("Source", "source"), 0, 0)
layout.addWidget(self.create_cell_widget("Channel", "channel"), 0, 1)
layout.addWidget(self.create_cell_widget("Sink", "sink"), 1, 0)
text_button = QToolButton()
text_button.setCheckable(True)
self.button_group.addButton(text_button, self.InsertTextButton)
text_button.setIcon(QIcon(QPixmap(':/images/textpointer.png').scaled(30, 30)))
text_button.setIconSize(QSize(50, 50))
text_layout = QGridLayout()
text_layout.addWidget(text_button, 0, 0, Qt.AlignHCenter)
text_layout.addWidget(QLabel("Text"), 1, 0, Qt.AlignCenter)
text_widget = QWidget()
text_widget.setLayout(text_layout)
layout.addWidget(text_widget, 1, 1)
layout.setRowStretch(3, 10)
layout.setColumnStretch(2, 10)
item_widget = QWidget()
item_widget.setLayout(layout)
self.tool_box = QToolBox()
self.tool_box.setSizePolicy(QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Ignored))
self.tool_box.setMinimumWidth(item_widget.sizeHint().width())
self.tool_box.addItem(item_widget, "Basic Flume Items")
# noinspection PyAttributeOutsideInit,PyUnresolvedReferences
def create_tool_bars(self):
self.edit_tool_bar = self.addToolBar("Edit")
self.edit_tool_bar.addAction(self.delete_action)
self.edit_tool_bar.addAction(self.to_front_action)
self.edit_tool_bar.addAction(self.send_back_action)
self.edit_tool_bar.addAction(self.enable_grid_action)
self.font_combo = QFontComboBox()
self.font_combo.currentFontChanged.connect(self.current_font_changed)
self.font_size_combo = QComboBox()
self.font_size_combo.setEditable(True)
for i in range(8, 30, 2):
self.font_size_combo.addItem(str(i))
validator = QIntValidator(2, 64, self)
self.font_size_combo.setValidator(validator)
self.font_size_combo.currentIndexChanged.connect(self.font_size_changed)
self.font_color_tool_button = QToolButton()
self.font_color_tool_button.setPopupMode(QToolButton.MenuButtonPopup)
self.font_color_tool_button.setMenu(
self.create_color_menu(self.text_color_changed, Qt.black))
self.text_action = self.font_color_tool_button.menu().defaultAction()
self.font_color_tool_button.setIcon(
self.create_color_tool_button_icon(':/images/textpointer.png',
Qt.black))
self.font_color_tool_button.setAutoFillBackground(True)
self.font_color_tool_button.clicked.connect(self.text_button_triggered)
self.fill_color_tool_button = QToolButton()
self.fill_color_tool_button.setPopupMode(QToolButton.MenuButtonPopup)
self.fill_color_tool_button.setMenu(
self.create_color_menu(self.item_color_changed, Qt.white))
self.fillAction = self.fill_color_tool_button.menu().defaultAction()
self.fill_color_tool_button.setIcon(
self.create_color_tool_button_icon(':/images/floodfill.png',
Qt.white))
self.fill_color_tool_button.clicked.connect(self.fill_button_triggered)
self.line_color_tool_button = QToolButton()
self.line_color_tool_button.setPopupMode(QToolButton.MenuButtonPopup)
self.line_color_tool_button.setMenu(
self.create_color_menu(self.line_color_changed, Qt.black))
self.lineAction = self.line_color_tool_button.menu().defaultAction()
self.line_color_tool_button.setIcon(
self.create_color_tool_button_icon(':/images/linecolor.png',
Qt.black))
self.line_color_tool_button.clicked.connect(self.line_button_triggered)
self.text_tool_bar = self.addToolBar("Font")
self.text_tool_bar.addWidget(self.font_combo)
self.text_tool_bar.addWidget(self.font_size_combo)
self.text_tool_bar.addAction(self.bold_action)
self.text_tool_bar.addAction(self.italic_action)
self.text_tool_bar.addAction(self.underline_action)
self.color_tool_bar = self.addToolBar("Color")
self.color_tool_bar.addWidget(self.font_color_tool_button)
self.color_tool_bar.addWidget(self.fill_color_tool_button)
self.color_tool_bar.addWidget(self.line_color_tool_button)
self.loading_tool_bar = self.addToolBar("Load")
self.loading_tool_bar.addAction(self.load_config_action)
pointer_button = QToolButton()
pointer_button.setCheckable(True)
pointer_button.setChecked(True)
pointer_button.setIcon(QIcon(":/images/pointer.png"))
line_pointer_button = QToolButton()
line_pointer_button.setCheckable(True)
line_pointer_button.setIcon(QIcon(":/images/linepointer.png"))
self.pointer_type_group = QButtonGroup()
self.pointer_type_group.addButton(pointer_button, DiagramScene.MoveItem)
self.pointer_type_group.addButton(line_pointer_button, DiagramScene.InsertLine)
self.pointer_type_group.buttonClicked[int].connect(self.pointer_group_clicked)
self.scene_scale_combo = QComboBox()
self.scene_scale_combo.addItems(["50%", "75%", "100%", "125%", "150%"])
self.scene_scale_combo.setCurrentIndex(2)
self.scene_scale_combo.currentIndexChanged[str].connect(self.scene_scale_changed)
self.pointer_tool_bar = self.addToolBar("Pointer type")
self.pointer_tool_bar.addWidget(pointer_button)
self.pointer_tool_bar.addWidget(line_pointer_button)
self.pointer_tool_bar.addWidget(self.scene_scale_combo)
def button_group_clicked(self, button_id):
buttons = self.button_group.buttons()
self.clicked_button_id = button_id
for button in buttons:
if self.button_group.button(button_id) != button:
button.setChecked(False)
if button_id == self.InsertTextButton:
self.scene.set_mode(DiagramScene.InsertText)
else:
self.scene.set_item_type(self.items[button_id])
self.scene.set_mode(DiagramScene.InsertItem)
def delete_item(self):
for item in self.scene.selectedItems():
if isinstance(item, FlumeDiagramItem):
item.remove_arrows()
self.scene.removeItem(item)
# noinspection PyTypeChecker,PyCallByClass
def about(self):
# noinspection PyArgumentList
QMessageBox.about(self, "About Flume Illustrator", "The Flume illustrator shows config-file details")
def pointer_group_clicked(self):
self.scene.set_mode(self.pointer_type_group.checkedId())
def bring_to_front(self):
if not self.scene.selectedItems():
return
selected_item = self.scene.selectedItems()[0]
overlap_items = selected_item.collidingItems()
z_value = 0
for item in overlap_items:
if item.zValue() >= z_value and isinstance(item, FlumeDiagramItem):
z_value = item.zValue() + 0.1
selected_item.setZValue(z_value)
def send_to_back(self):
if not self.scene.selectedItems():
return
selected_item = self.scene.selectedItems()[0]
overlap_items = selected_item.collidingItems()
z_value = 0
for item in overlap_items:
if item.zValue() <= z_value and isinstance(item, FlumeDiagramItem):
z_value = item.zValue() - 0.1
selected_item.setZValue(z_value)
def scene_scale_changed(self, scale):
new_scale = float(scale[:scale.index("%")]) / 100
old_transform = self.view.transform()
self.view.resetTransform()
self.view.translate(old_transform.dx(), old_transform.dy())
self.view.scale(new_scale, new_scale)
def item_inserted(self, diagram_type):
self.pointer_type_group.button(DiagramScene.MoveItem).setChecked(True)
self.scene.set_mode(self.scene.DefaultMode)
self.button_group.button(self.clicked_button_id).setChecked(False)
def text_inserted(self, item):
self.button_group.button(self.InsertTextButton).setChecked(False)
self.scene.set_mode(self.pointer_type_group.checkedId())
def current_font_changed(self, font):
self.handle_font_change()
def font_size_changed(self, font=None):
self.handle_font_change()
def text_color_changed(self):
self.text_action = self.sender()
self.font_color_tool_button.setIcon(
self.create_color_tool_button_icon(':/images/textpointer.png',
QColor(self.text_action.data())))
self.text_button_triggered()
def item_color_changed(self):
self.fillAction = self.sender()
self.fill_color_tool_button.setIcon(
self.create_color_tool_button_icon(':/images/floodfill.png',
QColor(self.fillAction.data())))
self.fill_button_triggered()
def line_color_changed(self):
self.lineAction = self.sender()
self.line_color_tool_button.setIcon(
self.create_color_tool_button_icon(':/images/linecolor.png',
QColor(self.lineAction.data())))
self.line_button_triggered()
def text_button_triggered(self):
self.scene.set_text_color(QColor(self.text_action.data()))
def fill_button_triggered(self):
self.scene.set_item_color(QColor(self.fillAction.data()))
def line_button_triggered(self):
self.scene.set_line_color(QColor(self.lineAction.data()))
def handle_font_change(self):
font = self.font_combo.currentFont()
font.setPointSize(int(self.font_size_combo.currentText()))
if self.bold_action.isChecked():
font.setWeight(QFont.Bold)
else:
font.setWeight(QFont.Normal)
font.setItalic(self.italic_action.isChecked())
font.setUnderline(self.underline_action.isChecked())
self.scene.setFont(font)
def item_selected(self, item):
font = item.font()
self.font_combo.setCurrentFont(font)
self.font_size_combo.setEditText(str(font.pointSize()))
self.bold_action.setChecked(font.weight() == QFont.Bold)
self.italic_action.setChecked(font.italic())
self.underline_action.setChecked(font.underline())
def create_cell_widget(self, text, diagram_type):
item = FlumeObject(diagram_type, "")
icon = QIcon(item.pictogram.image())
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.button_group.addButton(button) # , diagram_type
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignHCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
# noinspection PyArgumentList
def create_color_menu(self, slot, default_color):
colors = [Qt.black, Qt.white, Qt.red, Qt.blue, Qt.yellow]
names = ["black", "white", "red", "blue", "yellow"]
color_menu = QMenu(self)
for color, name in zip(colors, names):
action = QAction(self.create_color_icon(color), name, self,
triggered=slot)
action.setData(QColor(color))
color_menu.addAction(action)
if color == default_color:
color_menu.setDefaultAction(action)
return color_menu
@staticmethod
def create_color_tool_button_icon(image_file, color):
pixmap = QPixmap(50, 80)
pixmap.fill(Qt.transparent)
painter = QPainter(pixmap)
image = QPixmap(image_file)
target = QRect(0, 0, 50, 60)
source = QRect(0, 0, 42, 42)
painter.fillRect(QRect(0, 60, 50, 80), color)
painter.drawPixmap(target, image, source)
painter.end()
return QIcon(pixmap)
@staticmethod
def create_color_icon(color):
pixmap = QPixmap(20, 20)
painter = QPainter(pixmap)
painter.setPen(Qt.NoPen)
painter.fillRect(QRect(0, 0, 20, 20), color)
painter.end()
return QIcon(pixmap)
def enable_grid(self):
if self.enable_grid_action.isChecked():
color = Qt.black
else:
color = Qt.white
for i in range(50):
for j in range(50):
self.scene.addEllipse(i * 100, j * 100, 2, 2, QPen(color))
