GuiShell/sub_windows.py

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):
    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)
    
    
class FormWindow(Ui_Form, QMainWindow):
    def __init__(self, mdiArea:QMdiArea, instan:GuiType._GuiWinForm, pos:QPoint=None) -> None:
        super().__init__(mdiArea)
        self.setupUi(self)
        self.instan = instan
        self.setWindowTitle(instan.Title)
        self.tableView.setModel(FormWindow.get_form_model(self.instan.Form))
        mdisubwin = mdiArea.addSubWindow(self)
        mdisubwin.setAttribute(Qt.WA_DeleteOnClose)
        
        # mdisubwin.move(mdiArea.x()+(mdiArea.width()-self.width())/2 + self.width()*0.8,mdiArea.y()+(mdiArea.height()-self.height())/2)
        mdisubwin.show()
        # 必须要show之后才能 ensure_subwin_in_midarea，因为没有显示时Form窗口大小是不正确的
        # 移动到鼠标 pos 位置
        if pos:
            mdisubwin.move(ensure_subwin_in_midarea(mdiArea, self, pos))
        self.reportButton.clicked.connect(lambda: print(self.geometry()))
    
    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


class MultiFormWin(FormWindow):
    def __init__(self, mdiArea:QMdiArea, instan:GuiType._GuiWinFormMulti) -> None:
        super().__init__()
        self.setupUi(self)
        self.setParent(mdiArea)
        self.init_multi_form_ui(mdiArea)
        self.reportButton.clicked.connect(self.report_table)
        self.instan = instan
 
    def set_form_multi(self, multi_form:GuiType._GuiFormMulti):
        self.sheets.clear()
        for sheet_name in multi_form.Names:
            self.sheets.addItem(sheet_name)
        if self.sheets.count() > 0:
            form = multi_form.GetForm(self.sheets.currentIndex())
            model = FormWindow.get_form_model(form)
            self.tableView.setModel(model)

    def report_table(self):
        print("report test")
            
    def init_multi_form_ui(self,mdiArea:QMdiArea):
        self.horizontalLayout_2 = QHBoxLayout()
        self.horizontalLayout_2.setObjectName(u"horizontalLayout_2")
        self.sheets = QComboBox(self.centralwidget)
        self.sheets.setObjectName(u"sheets")
        sizePolicy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Expanding)
        sizePolicy.setHorizontalStretch(0)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(self.sheets.sizePolicy().hasHeightForWidth())
        self.sheets.setSizePolicy(sizePolicy)

        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)
        mdiArea.addSubWindow(self)
        # mdiArea.currentSubWindow
        self.show()


class TreeWidgetItem(QTreeWidgetItem):
    def __init__(self, parent: int, tree_row:GuiType._GuiTreeRow):
        super().__init__(parent)    
        self.tree_row = tree_row

class LineEditWin(Ui_LineEdit, QMainWindow):
    def __init__(self, mdiArea:QMdiArea, instance:GuiType._GuiWinLinedText) -> None:
        super().__init__()
        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)

        self.mdisubwin = mdiArea.addSubWindow(self)
        self.mdisubwin.setAttribute(Qt.WA_DeleteOnClose)
        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, QMainWindow):
    def __init__(self, ui, instance:GuiType._GuiSideBar) -> None:
        super().__init__()
        self.setupUi(self)
        self.ui = ui
        mdiArea:QMdiArea = ui.mdiArea
        self.mdiArea = mdiArea
        # self.mdisubwin 是用于管理子窗口的，在子窗口中，是由 mdiArea 来管理显示/隐藏、大小、关闭等
        self.mdisubwin = mdiArea.addSubWindow(self)
        self.mdisubwin.setAttribute(Qt.WA_DeleteOnClose)
        self.mdisubwin.resize(self.width(), mdiArea.height())
        self.mdisubwin.show()
        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(self.activate_table_sheet)
        self.tableView.customContextMenuRequested.connect(self.right_click_menu)
    
    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)
        self.lineEdit_win = LineEditWin(self.ui.mdiArea, 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.mdiArea, 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)
        self.mdisubwin.update()

    # 添加一行树节点，该节点包含 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):
        self.form_multi_instan = self.instance.GetForms(self.get_tree_names(item))
        self.set_form_multi(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 activate_table_sheet(self, index):
        model = FormWindow.get_form_model(self.form_multi_instan.GetForm(index))
        self.tableView.setModel(model)
    
    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)
    
    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)

    # def get_form_model(self, 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 set_form_multi(self, multi_form:GuiType._GuiFormMulti):
        self.sheets.clear()
        for sheet_name in multi_form.Names:
            self.sheets.addItem(sheet_name)
        if self.sheets.count() > 0:
            form = multi_form.GetForm(self.sheets.currentIndex())
            model = FormWindow.get_form_model(form)
            self.tableView.setModel(model)
        

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.mdiArea)
        # form_multi_win:Ui_form_multi
        # self.ui.mdiArea.addSubWindow(form_multi_win)
        form_multi_win.setWindowTitle(instance.Title)
        # form_multi_win.sheets.currentIndexChanged.connect(lambda:self.change_table_sheet(form_multi_win))
        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.sheets.currentText()
            # win
            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):
        if isinstance(instance, GuiType._GuiWinFormMulti):
            self.add_win_form_multi(instance)
        elif isinstance(instance, GuiType._GuiSideBar):
            self.side_bar.init_win_data(instance)
            

if __name__ == '__main__':            
    m = GuiType._GuiWinFormMulti('multi', 1)