import threading from PySide2 import QtCore from PySide2.QtUiTools import QUiLoader from PySide2.QtWidgets import * from PySide2.QtCore import * from PySide2.QtGui import * from PySide2.QtUiTools import QUiLoader from ui.ui_Form import Ui_Form from ui.ui_SideBar import Ui_SideBar from ui.lineNumberWidget import LineNumberWidget from ui.ui_LineEdit import Ui_LineEdit from ui.Analyze import AnalyzeWindow import GuiType as GuiType def get_sub_win(mdiArea:QMdiArea, class_type, title=''): for mid_sub_win in mdiArea.subWindowList(): sub_win = mid_sub_win.widget() if type(sub_win) == class_type: if title: if sub_win.windowTitle() == title: return mid_sub_win else: return mid_sub_win return def get_sub_win_list(mdiArea:QMdiArea, class_type): win_list = [] for mid_sub_win in mdiArea.subWindowList(): sub_win = mid_sub_win.widget() if type(sub_win) == class_type: win_list.append(mid_sub_win) return win_list def ensure_subwin_in_midarea(mdiArea:QMdiArea, win:QWidget, pos:QPoint=None): if pos ==None: pos = win.pos() if pos.x() in range(int(win.width()*0.3),mdiArea.width()-win.width()): x = pos.x()-win.width()*0.3 elif pos.x() < 0: x = 0 else: x = mdiArea.width() - win.width() if pos.y() in range(int(win.height()*0.3), mdiArea.height()-win.height()): y = pos.y()-win.height()*0.3 elif pos.y() < 0: y = 0 else: y = mdiArea.height() - win.height() return QPoint(x, y) # 动态窗口，初始化时会添加动作到 菜单栏 - window 中 class MdiActionWindow(QMainWindow): def __init__(self, mainWindow): super().__init__(mainWindow) self.mainWindow = mainWindow self.regist_action(mainWindow) # 修改窗口标题时同步修改menu-Action信息 self.windowTitleChanged.connect(lambda: self.menuWindow_action.setText(self.windowTitle())) def closeEvent(self, event: QCloseEvent): self.mainWindow.menuWindow.removeAction(self.menuWindow_action) return super().closeEvent(event) def regist_action(self, mainWindow): action = QAction(mainWindow) action.setText(self.windowTitle()) mainWindow.menuWindow.insertAction(mainWindow.action_win_separator,action) action.setActionGroup(mainWindow.action_win_group) action.setCheckable(True) action.setChecked(True) action.triggered.connect(self.active_win) self.menuWindow_action = action def active_win(self): self.hide() self.show() class FormWindow(Ui_Form, MdiActionWindow): def __init__(self, mainWindow, instan:GuiType._GuiWinForm, pos:QPoint=None) -> None: super().__init__(mainWindow) self.setupUi(self) mdiArea:QMdiArea = mainWindow.mdiArea self.instan = instan self.setWindowTitle(instan.Title) self.reportButton.clicked.connect(self.report_virtual_timing) self.reportButton.setHidden(not instan._hasReportVirtualTimingButton) if instan._hasCrossProbeMenu: self.tableWidget.customContextMenuRequested.connect(self.right_click_menu) self.tableWidget.setModel(FormWindow.get_form_model(self.instan.Form)) self.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeToContents) self.mdisubwin:QMdiSubWindow = mainWindow.mdiArea.addSubWindow(self) self.mdisubwin.setAttribute(Qt.WA_DeleteOnClose) self.mdisubwin.setWindowIcon(QIcon(":/icon/resource/todo-line.png")) # mdisubwin.move(mdiArea.x()+(mdiArea.width()-self.width())/2 + self.width()*0.8,mdiArea.y()+(mdiArea.height()-self.height())/2) self.mdisubwin.show() width = FormWindow.get_table_width(self.tableWidget, self) self.mdisubwin.resize(width, self.height()) # 必须要show之后才能 ensure_subwin_in_midarea，因为没有显示时Form窗口大小是不正确的 # 移动到鼠标 pos 位置 if pos: self.mdisubwin.move(ensure_subwin_in_midarea(mdiArea, self, pos)) def get_form_model(form:GuiType._GuiForm): model = QStandardItemModel(0, len(form.FixedRow)) model.setHorizontalHeaderLabels(form.FixedRow) for row in form.Rows: row_items=[QStandardItem(text) for text in row] model.appendRow(row_items) return model def get_table_width(tableWidget:QTableWidget, win:MdiActionWindow): # 表格与窗口的边距宽度 width = 28 for i in range(tableWidget.horizontalHeader().count()): width += tableWidget.columnWidth(i) if width < win.width(): width = win.width() return width def right_click_menu(self, pos): pop_menu = QMenu() pop_menu.addAction(QAction('Crossprobe', pop_menu)) pop_menu.triggered.connect(self.crossprobe) pop_menu.exec_(QCursor.pos()) def get_row(tableWidget:QTableView): form_row = [] for col in range(tableWidget.model().columnCount()): form_row.append(tableWidget.model().index(tableWidget.currentIndex().row(), col).data()) return form_row def crossprobe(self, action): form_row = FormWindow.get_row(self.tableWidget) line_text_win = self.instan.CrossProbe(form_row) LineEditWin(self.mainWindow, line_text_win) def report_virtual_timing(self): if self.tableWidget.currentIndex().row() == -1: return form_row = FormWindow.get_row(self.tableWidget) thread = threading.Thread(target=self.instan.ReportVirtualTiming, args=(form_row, self.mainWindow.GUI,)) thread.start() class MultiFormWin(Ui_Form, MdiActionWindow): def __init__(self, mainWindow, instan:GuiType._GuiWinFormMulti) -> None: super().__init__(mainWindow) self.setupUi(self) self.init_multi_form_ui() self.setWindowTitle(instan.Title) MultiFormWin.set_form_multi(self.tableWidget, self.sheets, instan.Forms) self.instan = instan self.sheets.activated.connect(lambda index: MultiFormWin.activate_table_sheet(self.tableWidget, self.instan.Forms, index)) self.reportButton.clicked.connect(self.report_virtual_timing) mdiArea:QMdiArea = mainWindow.mdiArea mdisub = mdiArea.addSubWindow(self) width = self.tableWidget.horizontalHeader().width() * self.tableWidget.horizontalHeader().count() + 25 mdisub.setWindowIcon(QIcon(":/icon/resource/todo-line.png")) mdisub.resize(width, self.height()) mdisub.show() def set_form_multi(tableWidget:QTableView, sheets:QComboBox, multi_form:GuiType._GuiFormMulti): sheets.clear() for sheet_name in multi_form.Names: sheets.addItem(sheet_name) if sheets.count() > 0: form = multi_form.GetForm(sheets.currentIndex()) model = FormWindow.get_form_model(form) tableWidget.setModel(model) def activate_table_sheet(tableWidget, form_multi_instan, index): model = FormWindow.get_form_model(form_multi_instan.GetForm(index)) tableWidget.setModel(model) def crossprobe(self, action): form_row = FormWindow.get_row(self.tableWidget) form_name = self.sheets.currentText() line_text_win = self.instan.CrossProbe(form_name, form_row) LineEditWin(self.mainWindow, line_text_win) def report_virtual_timing(self): if self.tableWidget.currentIndex().row() == -1: return form_row = FormWindow.get_row(self.tableWidget) thread = threading.Thread(target=self.instan.ReportVirtualTiming, args=(self.sheets.currentText(), form_row, self.mainWindow.GUI,)) thread.start() def init_multi_form_ui(self): self.horizontalLayout_2 = QHBoxLayout() self.horizontalLayout_2.setObjectName(u"horizontalLayout_2") self.sheets = QComboBox(self.centralwidget) self.sheets.setObjectName(u"sheets") self.horizontalLayout_2.addWidget(self.sheets) self.horizontalSpacer_3 = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum) self.horizontalLayout_2.addItem(self.horizontalSpacer_3) self.horizontalLayout_2.setStretch(0, 3) self.horizontalLayout_2.setStretch(1, 6) self.verticalLayout.insertLayout(0,self.horizontalLayout_2) self.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch) self.tableWidget.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu) self.tableWidget.customContextMenuRequested.connect(self.right_click_menu) def right_click_menu(self, pos): pop_menu = QMenu() pop_menu.addAction(QAction('Crossprobe', pop_menu)) pop_menu.triggered.connect(self.crossprobe) pop_menu.exec_(QCursor.pos()) class TreeWidgetItem(QTreeWidgetItem): def __init__(self, parent: int, tree_row:GuiType._GuiTreeRow): super().__init__(parent) self.tree_row = tree_row class TextBrowserWin(Ui_LineEdit, MdiActionWindow): def __init__(self, mainWindow, instance:GuiType._GuiWinText) -> None: super().__init__(mainWindow) self.setupUi(self) self.setWindowTitle(instance.Title) self.textBrowser.append(instance.Text) self.instance = instance self.mdisubwin:QMdiSubWindow = mainWindow.mdiArea.addSubWindow(self) self.mdisubwin.setAttribute(Qt.WA_DeleteOnClose) self.mdisubwin.setWindowIcon(QIcon(":/icon/resource/todo-line.png")) self.mdisubwin.resize(self.width(), mainWindow.mdiArea.height()) self.mdisubwin.show() self.mdisubwin.move(ensure_subwin_in_midarea(mainWindow.mdiArea, self)) class LineEditWin(Ui_LineEdit, MdiActionWindow): def __init__(self, mainWindow, instance:GuiType._GuiWinLinedText) -> None: super().__init__(mainWindow) self.setupUi(self) self.setWindowTitle(instance.Title) self.textBrowser.textChanged.connect(self.__line_widget_line_count_changed) self.lineWidget = LineNumberWidget(self.textBrowser) self.horizontalLayout.insertWidget(0,self.lineWidget) self.instance = instance self.set_text(instance) mdiArea:QMdiArea = mainWindow.mdiArea self.mdisubwin:QMdiSubWindow = mdiArea.addSubWindow(self) self.mdisubwin.setAttribute(Qt.WA_DeleteOnClose) self.mdisubwin.setWindowIcon(QIcon(":/icon/resource/todo-line.png")) self.mdisubwin.resize(self.width(), mdiArea.height()) self.mdisubwin.show() self.go_to_line(self.instance.GotoLine) def set_text(self, instance:GuiType._GuiWinLinedText): for text in instance.LinedText: self.textBrowser.append(text) def go_to_line(self, line): if line > self.lineWidget.line_count(): line = self.lineWidget.line_count() cur = self.textBrowser.textCursor() cur.movePosition(QTextCursor.Start) self.textBrowser.setTextCursor(cur) position = self.textBrowser.document().findBlockByLineNumber(line-1).position() if position < self.textBrowser.height()/2: position = 0 elif position < self.textBrowser.document().lastBlock().position() - self.textBrowser.height()/2: position += self.textBrowser.height()/2 else: position = self.textBrowser.document().lastBlock().position() cur.setPosition(position, QTextCursor.MoveAnchor) self.textBrowser.setTextCursor(cur) self.lineWidget.highlightLineNumber(line) def __line_widget_line_count_changed(self): if self.lineWidget: n = int(self.textBrowser.document().lineCount()) self.lineWidget.changeLineCount(n) class SideBarWin(Ui_SideBar, MdiActionWindow): def __init__(self, ui, instance:GuiType._GuiSideBar) -> None: super().__init__(ui) self.setupUi(self) self.ui = ui self.instance = instance self.init_win_data(instance) self.form_multi_instan = None self.treeWidget.itemExpanded.connect(self.expand_tree) self.treeWidget.itemCollapsed.connect(self.collapse_tree) self.treeWidget.currentItemChanged.connect(self.item_change) self.treeWidget.customContextMenuRequested.connect(self.right_click_menu) self.sheets.activated.connect(lambda index: MultiFormWin.activate_table_sheet(self.tableView, self.form_multi_instan, index)) self.tableView.customContextMenuRequested.connect(self.right_click_menu) self.tableView.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch) mdiArea:QMdiArea = ui.mdiArea self.mdiArea:QMdiArea = mdiArea self.mdisubwin:QMdiSubWindow = mdiArea.addSubWindow(self) self.mdisubwin.setAttribute(Qt.WA_DeleteOnClose) self.mdisubwin.setWindowIcon(QIcon(":/icon/resource/node-tree.png")) # self.mdisubwin 是用于管理子窗口的，在子窗口中，是由 mdiArea 来管理显示/隐藏、大小、关闭等 self.mdisubwin.resize(self.width(), mdiArea.height()) self.mdisubwin.show() def closeEvent(self, event: QCloseEvent): event.ignore() self.mdisubwin.hide() self.menuWindow_action.setChecked(False) def right_click_menu(self, pos): sender = self.sender() pop_menu = QMenu() pop_menu.addAction(QAction('Crossprobe', pop_menu)) pop_menu.addAction(QAction('Properties', pop_menu)) pop_menu.triggered.connect(lambda action: self.right_click_menu_action(action, sender)) pop_menu.exec_(QCursor.pos()) def right_click_menu_action(self, action:QAction, sender): if action.text() == 'Crossprobe': self.add_crossprobe_win(sender) elif action.text() == 'Properties': self.add_properties_win(sender) def add_crossprobe_win(self, sender): tree_names, formname, formrow = self.get_win_argument(sender) line_text = self.instance.CrossProbe(tree_names, formname, formrow) property_win:QMdiSubWindow = get_sub_win(self.mdiArea, LineEditWin, line_text.Title) # 窗口已经存在 if property_win: property_win.close() self.lineEdit_win = LineEditWin(self.ui, line_text) def add_properties_win(self, sender:QWidget): tree_names, formname, formrow = self.get_win_argument(sender) win_form_instan = self.instance.Properties(tree_names, formname, formrow) property_win:QMdiSubWindow = get_sub_win(self.mdiArea, FormWindow, win_form_instan.Title) # 窗口已经存在 if property_win: property_win.close() mouse_pos = QCursor().pos() pos = self.mdiArea.mapFromGlobal(mouse_pos) form_window = FormWindow(self.ui, win_form_instan, pos) def get_win_argument(self, sender): item = self.treeWidget.currentItem() tree_names = self.get_tree_names(item) if type(sender) is QTreeWidget: formname = None formrow = [] elif type(sender) is QTableView: formname = self.sheets.currentText() formrow = [] for column in range(self.tableView.model().columnCount()): data = self.tableView.model().index(self.tableView.currentIndex().row(), column).data() formrow.append(data) return tree_names, formname, formrow # 侧边栏只有一个，只需要 update 即可 def init_win_data(self, instance:GuiType._GuiSideBar): if self.treeWidget.topLevelItemCount(): for i in range(self.treeWidget.topLevelItemCount()): # 获取这个 item 并删除 self.treeWidget.takeTopLevelItem(i) self.ui.setWindowTitle(self.ui.GUI.Title) self.instance = instance self.setWindowTitle(self.instance.Title) self.set_tree(self.instance.Tree) # 添加一行树节点，该节点包含 n 列数据 def add_tree_row(self, parent, tree_row:GuiType._GuiTreeRow): item = TreeWidgetItem(parent, tree_row) item.setText(0,tree_row.Name) for i in range(0, len(tree_row.Row)): item.setText(i+1,tree_row.Row[i]) return item def get_tree_names(self, tree_item:TreeWidgetItem): tree_names = [tree_item.text(0)] parent = tree_item.parent() while 1: if parent: tree_names.append(parent.text(0)) parent = parent.parent() else: break return tree_names[::-1] def item_change(self, item:TreeWidgetItem, column): if not item: return self.form_multi_instan = self.instance.GetForms(self.get_tree_names(item)) MultiFormWin.set_form_multi(self.tableView, self.sheets, self.form_multi_instan) def collapse_tree(self, item:TreeWidgetItem): # 取出并删除所有子节点 childrens = item.takeChildren() if not childrens: return if item.tree_row.HasSubTreeRows: # 只添加一个子节点，让父节点变成可展开形式 sub_tree_row:GuiType._GuiTreeRow = item.tree_row.GetSubTreeRows()[0] self.add_tree_row(item, sub_tree_row) def expand_tree(self, item:TreeWidgetItem): # 取出所有 Tree row item.takeChildren() # 加载所有 Tree row if item.tree_row.HasSubTreeRows: sub_tree_rows = item.tree_row.GetSubTreeRows() for row in sub_tree_rows: self.set_tree_row(item, row) def set_tree_row(self, parent:TreeWidgetItem, tree_row:GuiType._GuiTreeRow): tree_widget_item = self.add_tree_row(parent, tree_row) if tree_row.HasSubTreeRows: # 只添加一个子节点，让父节点变成可展开形式 sub_tree_row:GuiType._GuiTreeRow = tree_row.GetSubTreeRows()[0] self.add_tree_row(tree_widget_item, sub_tree_row) return tree_widget_item # 树形控件设置数据 def set_tree(self, tree:GuiType._GuiTree): # 树 header_item = self.treeWidget.headerItem() for i in range(len(tree.FixedRow)): header_item.setText(i, tree.FixedRow[i]) for tree_row in tree.TreeRows: # 根节点 self.set_tree_row(self.treeWidget, tree_row) # 默认选择第一行 if self.treeWidget.topLevelItemCount(): self.treeWidget.setItemSelected(self.treeWidget.topLevelItem(0), True) class SubWindows(QObject): sig = Signal(object) def __init__(self, parent): super().__init__(parent) self.ui = parent self.side_bar = SideBarWin(self.ui, self.ui.GUI.SideBar) self.info = {} def add_win_form_multi(self, instance:GuiType._GuiWinFormMulti): form_multi_win = MultiFormWin(self.ui, instance) form_multi_win.setWindowTitle(instance.Title) form_multi_win.sheets.currentIndexChanged.connect(lambda:self.change_table_sheet(form_multi_win)) self.set_form_multi_table(form_multi_win,instance.Forms) self.ui.mdiArea.activeSubWindow().resize(self.ui.mdiArea.width()/2,self.ui.mdiArea.height()) self.ui.mdiArea.activeSubWindow().move(form_multi_win.width(),0) def set_form_multi_table(self, win:MultiFormWin, multi_form:GuiType._GuiFormMulti): multi_form:GuiType._GuiFormMulti for i in range(len(multi_form.Names)): sheet_name = multi_form.Names[i] form:GuiType._GuiForm = multi_form.GetForm(i) model = QStandardItemModel(0, len(form.FixedRow)) model.setHorizontalHeaderLabels(form.FixedRow) for row in form.Rows: row_items=[QStandardItem(text) for text in row] model.appendRow(row_items) win.sheets.addItem(sheet_name) win.models.append(model) if win.sheets.count() > 0: win.sheets.setCurrentIndex(0) win.tableView.setModel(win.models[0]) def change_table_sheet(self, win:MultiFormWin): if not win.models: return win.tableView.setModel(win.models[win.sheets.currentIndex()]) def add_windows(self, instance): win_info = { GuiType._GuiWinFormMulti: lambda: MultiFormWin(self.ui, instance), GuiType._GuiSideBar: lambda: self.side_bar.init_win_data(instance), GuiType._GuiWinForm: lambda: FormWindow(self.ui, instance), GuiType._GuiWinText: lambda: TextBrowserWin(self.ui, instance), } win_info.get(type(instance))() return if isinstance(instance, GuiType._GuiWinFormMulti): MultiFormWin(self.ui, instance) elif isinstance(instance, GuiType._GuiSideBar): self.side_bar.init_win_data(instance) if __name__ == '__main__': m = GuiType._GuiWinFormMulti('multi', 1)