GuiShell/main_windows.py

import os
import queue
import re
import sys
import threading
from PySide2.QtWidgets import QApplication, QMainWindow
from PySide2.QtUiTools import QUiLoader
from PySide2.QtWidgets import *
from PySide2.QtCore import *
from PySide2 import QtCore
from PySide2.QtGui import *
from ui.ui_MainWindows import Ui_MainWindow
from ui.Analyze import AnalyzeWindow
from sub_windows import SubWindows,LineEditWin,FormWindow, get_sub_win_list,get_sub_win
from ui.About import AboutWindow
from ui.Elaborate import Elaborate
from ui.Read_Timing_Constraint import ReadTimeWindow
from ui.ReportException import ReportExceptionWindow
from ui.ReportClocks import ReportClocks
import GuiType


class Stream(QObject):
    """Redirects console output to text widget."""
    newText = Signal(str)
    def __init__(self):
        super().__init__()
        self.stdout_bak = sys.stdout
        self.buffer = self.stdout_bak.buffer
        self.close = self.stdout_bak.close
        self.closed = self.stdout_bak.closed
        self.detach = self.stdout_bak.detach
        self.encoding = self.stdout_bak.encoding
        self.errors = self.stdout_bak.errors
        self.fileno = self.stdout_bak.fileno
        self.flush = self.stdout_bak.flush
        self.isatty = self.stdout_bak.isatty
        self.line_buffering = self.stdout_bak.line_buffering
        self.mode = self.stdout_bak.mode
        self.name = self.stdout_bak.name
        self.newlines = self.stdout_bak.newlines
        self.read = self.stdout_bak.read
        self.readable = self.stdout_bak.readable
        self.readline = self.stdout_bak.readline
        self.readlines = self.stdout_bak.readlines
        self.reconfigure = self.stdout_bak.reconfigure
        self.seek = self.stdout_bak.seek
        self.seekable = self.stdout_bak.seekable
        self.tell = self.stdout_bak.tell
        self.truncate = self.stdout_bak.truncate
        self.writable = self.stdout_bak.writable
        self.write_through = self.stdout_bak.write_through
        self.writelines = self.stdout_bak.writelines
        
    def write(self, text:str):
        self.stdout_bak.write(str(text))
        self.stdout_bak.flush()
        if len(text) == 0:
            return
        if text.isspace():
            return
        self.newText.emit(str(text+'\n'))


class MessageItem(QTreeWidgetItem):
    def __init__(self, parent, msg:GuiType._GuiMessage):
      super().__init__(parent)
      self.setText(0, msg.Text)
      self.msg = msg

class MessagesWidgetItem(QTreeWidgetItem):
    def __init__(self, parent, messages:GuiType._GuiMessages, level):
        super().__init__(parent)
        summary = messages.GetSummary()
        self.setText(0, level + f"({messages.Size})")
        for id_, size, text in summary:
            sumary_item = QTreeWidgetItem(self)
            sumary_item.setText(0, f"{id_}({size}):{text}")
            
            for msg in messages.GetMsgByID(id_):
                msg:GuiType._GuiMessage
                MessageItem(sumary_item, msg)
        

class MainWindow(Ui_MainWindow, QMainWindow):
    textEdit_sig = Signal(str)
    sig = Signal(object)
    def __init__(self, GUI) -> None:
        super().__init__()
        self.setupUi(self)
        # 这里显示一下是为了能获取到 mdiArea 的尺寸，给 sub_windows 使用
        self.GUI = GUI
        self.setWindowTitle(GUI.Title)
        self.show()
        
        sys.stdout = Stream()
        sys.stdout.newText.connect(self.outputWritten)
        sys.stderr = sys.stdout
        
        self.btn_shell.setText(GUI.Prompt)
        self.btn_shell.clicked.connect(self.click_btn_sell)
        self.tabWidget.currentChanged.connect(self.table_widget_change)
        self.tableHistory.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.tableHistory.customContextMenuRequested.connect(self.history_right_click_menu)
        self.MessageTreeView.customContextMenuRequested.connect(self.message_right_click_menu)
        self.status_label = QLabel("Ready")
        self.statusbar.addWidget(self.status_label)
        self.loader = QUiLoader()
        
        
        # -- MENUE --
        # file
        self.actionStop_GUI.triggered.connect(GUI.Stop)
        self.actionExit.triggered.connect(GUI.Exit)
        # Edit
        self.actionCopy.triggered.connect(self.action_copy)
        self.actionPast.triggered.connect(self.action_paste)
        self.actionSelect_all.triggered.connect(self.action_select_all)
        # View
        self.actionIncr_Font
        self.actionDecr_Font
        self.actionToolbar.triggered.connect(lambda: self.toolBar.setHidden(not self.actionToolbar.isChecked()))
        self.actionSidebar.triggered.connect(lambda: self.sub_win.side_bar.mdisubwin.setHidden(not self.actionSidebar.isChecked()))
        self.actionConsole.triggered.connect(lambda: self.tabWidget.setHidden(not self.actionConsole.isChecked()))
        self.actionStatusBar.triggered.connect(lambda: self.statusbar.setHidden(not self.actionStatusBar.isChecked()))
        self.actionSave_as_Default
        self.actionPreferences
        # Design
        self.actionAnalyze.triggered.connect(self.show_analyze_window)
        self.actionElaborate.triggered.connect(self.elaborate_window)
        self.actionRead_Timing_Constraint.triggered.connect(self.read_timing_constraint)
        self.actionRead_Power_Intent
        self.actionReload_Design.triggered.connect(self.reload_design)
        # Windows
        # 先实例一个 ActionGroup，等到创建动态窗口的时候，才会在 菜单栏-Window 中添加Action
        self.action_win_group = QActionGroup(self)
        # 获取分隔符，方便 菜单栏-Window-Action 插入到分隔符之前
        self.action_win_separator = self.menuWindow.actions()[0]
        action_tabgroup = QActionGroup(self)
        self.actionLog.setActionGroup(action_tabgroup)
        self.actionHistory.setActionGroup(action_tabgroup)
        self.actionMessages.setActionGroup(action_tabgroup)
        self.actionLog.triggered.connect(self.action_tab_widget)
        self.actionHistory.triggered.connect(self.action_tab_widget)
        self.actionMessages.triggered.connect(self.action_tab_widget)
        # Report
        self.actionReport_Exceptions.triggered.connect(self.reportException)
        self.actionReport_Clocks.triggered.connect(self.reportClocks)
        self.actionReport_Virtual_Timing
        
        self.actionAbout.triggered.connect(self.about)
        self.textEdit.document().contentsChanged.connect(self.textEdit_change_size)
        self.textEdit.document().contentsChange.connect(self.textEdit_change_contents)
                
        self.sub_win = SubWindows(self)
        self.sub_win.sig.connect(self.sub_win.add_windows)
    
    def action_copy(self):
        # 获得文本光标所在控件
        widget:QTextBrowser = self.focusWidget()
        if type(widget) not in (QTextBrowser, QTextEdit):
            return
        cur:QTextCursor = widget.textCursor()
        self.copy(cur.selectedText())

    def action_paste(self):
        clipboard = QApplication.clipboard()
        # 获得文本光标所在控件
        widget:QTextBrowser = self.focusWidget()
        if type(widget) not in (QTextBrowser, QTextEdit):
            return
        widget.textCursor().insertText(clipboard.text())

    
    def action_select_all(self):
        # 获得文本光标所在控件
        widget:QTextBrowser = self.focusWidget()
        if type(widget) not in (QTextBrowser, QTextEdit):
            return
        widget.selectAll()
    
    def action_tab_widget(self):
        action_sender = self.sender()
        action = {
            self.actionLog: 0, 
            self.actionHistory: 1, 
            self.actionMessages: 2, 
        }
        index = action.get(action_sender)
        if index != None:
            self.tabWidget.setCurrentIndex(index)
    
    def read_timing_constraint(self):
        ReadTimeWindow(self, os.getcwd())
    
    def table_widget_change(self, index):
        member = {
            0: None,
            1: self.show_history,
            2: self.show_message
        }
        method = member.get(index)
        if method:
            method()
    
    def show_history(self):
        self.tableHistory.clear()
        blank_row = self.tableHistory.height() / self.tableHistory.verticalHeader().height() + 2
        self.tableHistory.setRowCount(blank_row)
        for row_text in self.GUI.GetHistory():
            end_row = self.tableHistory.rowCount()
            self.tableHistory.insertRow(end_row)
            self.tableHistory.setItem(end_row, 0, QTableWidgetItem(row_text))
        self.tableHistory.scrollToBottom()
    
    def copy(self, text):
        clipboard = QApplication.clipboard()
        clipboard.setText(text)
     
    def history_right_click_menu(self):
        itemPos = self.tableHistory.mapFromGlobal(QCursor().pos())
        item = self.tableHistory.itemAt(itemPos)
        if type(item) != QTableWidgetItem:
            return
        pop_menu  = QMenu()
        pop_menu.addAction(QAction('Copy', pop_menu))
        pop_menu.triggered.connect(lambda: self.copy(self.tableHistory.currentItem().text()))
        pop_menu.exec_(QCursor.pos())

    def message_right_click_menu(self):
        itemPos = self.MessageTreeView.mapFromGlobal(QCursor().pos())
        item:MessageItem = self.MessageTreeView.itemAt(itemPos)
        if type(item) != MessageItem:
            return
        pop_menu  = QMenu()
        pop_menu.addAction(QAction('Crossprobe', pop_menu))
        pop_menu.addAction(QAction('Properties', pop_menu))
        pop_menu.triggered.connect(lambda action: self.message_popmenu_action(action, item))
        pop_menu.exec_(QCursor.pos())
    
    def message_popmenu_action(self, action, item):
        if action.text() == 'Crossprobe':
            win_line_text = item.msg.CrossProbe()
            win = get_sub_win(self.mdiArea, LineEditWin, win_line_text.Title)
            # 如果窗口已经存在，关闭窗口，以便重新更新数据
            if win:
                win.close()
            LineEditWin(self, win_line_text)
        elif action.text() == 'Properties':
            win_form = item.msg.Properties()
            # 如果窗口已经存在，关闭窗口，以便重新更新数据
            win = get_sub_win(self.mdiArea, FormWindow, win_form.Title)
            if win:
                win.close()
            FormWindow(self, win_form)
    
    
    def show_message(self):
        for _ in range(self.MessageTreeView.topLevelItemCount()):
            self.MessageTreeView.takeTopLevelItem(0)
        MessagesWidgetItem(self.MessageTreeView, self.GUI.GetErrors(), "Errors")
        MessagesWidgetItem(self.MessageTreeView, self.GUI.GetCriticalWarnings(), "CriticalWarnings")
        MessagesWidgetItem(self.MessageTreeView, self.GUI.GetWarnings(), "Warnings")
        MessagesWidgetItem(self.MessageTreeView, self.GUI.GetInfos(), "Infos")
        
    
    def about(self):
        d = AboutWindow(self, self.GUI.About)
        d.exec_()
    
    def show_analyze_window(self):
        d = AnalyzeWindow(self, self.GUI.Shell_Execute)
        d.exec_()
    
    def elaborate_window(self):
        d = Elaborate(self)
        d.exec_()
    
    def reportException(self):
        d = ReportExceptionWindow(self)
        d.exec_()
    
    def reportClocks(self):
        d = ReportClocks(self)
        d.exec_()
    
    def outputWritten(self, text):
        cursor = self.LogText.textCursor()
        cursor.movePosition(QTextCursor.End)
        cursor.insertText(text)
        self.LogText.setTextCursor(cursor)
        self.LogText.ensureCursorVisible()
    
    def textEdit_change_size(self):
        height = self.textEdit.document().size().height()
        self.textEdit.setMinimumHeight(height)
    
    def click_btn_sell(self):
        self.textEdit.append('\n')
    
    def textEdit_change_contents(self, from_index, charsRemoved, charsAdded):
        if not charsAdded:
            return
        stdin = self.textEdit.toPlainText()
        if not stdin:
            return
        # 捕获回车键
        if stdin[from_index] != '\n':
            return
        # 回车时，判断字符串代码是否结束
        if self.GUI.Shell_IsCmdFinished(stdin):
            # 删除回车字符
            stdin = stdin[:from_index] + stdin[from_index+1:]
            print(stdin)
            thread = threading.Thread(target=self.GUI.Shell_Execute, args=(stdin,))
            thread.start()
            self.textEdit.setText('')
            self.textEdit_change_size()
    
    def reload_design(self):
        thread = threading.Thread(target=self.GUI.ReloadDesign)
        thread.start()

    def add_sub_win(self):
        print('add sub')
        self.t = self.loader.load("ui/TimingAnalyze.ui")
        self.mdiArea.addSubWindow(self.t)
        self.t.show()