HYTOS

# coding: utf-8

""" This is license module """

import sys

import os

from functools import partial

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from PyQt5.QtSvg import *

from PyQt5.QtCore import QTranslator

from PyQt5.QtCore import QLocale

# PyQtRibbon: import, and error msg if not installed

try:

    from PyQtRibbon.FileMenu import QFileMenu, QFileMenuPanel

    from PyQtRibbon.RecentFilesManager import QRecentFilesManager

    from PyQtRibbon.Ribbon import QRibbon, QRibbonTab, QRibbonSection

except (ImportError, NameError):

    errormsg = 'You haven\'t installed PyQtRibbon, or your installation of it is broken. Please download or fix it.'

    raise Exception(errormsg)

class AppRibbon(QRibbon):

    """ Class that represents App ribbon """

    def __init__(self):

        """ Creates and initializes the App Ribbon """

        from AppDocData import AppDocData

        QRibbon.__init__(self)

        self._panes = {}

        self._advanced_mode = False

        app_doc_data = AppDocData.instance()

        configs = app_doc_data.getAppConfigs('app', 'mode')

        self._advanced_mode = True if configs and 1 == len(configs) and 'advanced' == configs[0].value else False

        # Set up the file menu

        self.fileMenu = AppRibbonFileMenu(self._advanced_mode)

        self.setFileMenu(self.fileMenu)

        # Add tabs

        self.add_home_tab()

        self.add_edit_tab()

        if self._advanced_mode:

            self.add_data_tab()

        self.add_view_tab()

        if self._advanced_mode:

            self.add_tool_tab()

        self.add_help_tab()

        self.add_tab_shortcut()

    @property

    def advanced_mode(self) -> bool:

        return self._advanced_mode

    def add_tab_shortcut(self):

        """ add shortcut to each tabs """

        pane = self._panes['Home File']

        self.connect_shortcut(pane)

        pane = self._panes['Edit']

        self.connect_shortcut(pane)

        pane = self._panes['Help']

        self.connect_shortcut(pane)

        pane = self._panes['View']

        self.connect_shortcut(pane)

        if 'Data' in self._panes:

            pane = self._panes['Data']

            self.connect_shortcut(pane)

        if 'Tool' in self._panes:

            pane = self._panes['Tool']

            self.connect_shortcut(pane)

    def connect_shortcut(self, parent):

        from App import App

        main_wnd = App.mainWnd()

        shortcut = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_S), parent)

        shortcut.activated.connect(main_wnd.actionSaveCliked)

        shortcut = QShortcut(QKeySequence(Qt.Key_F5), parent)

        shortcut.activated.connect(main_wnd.recognize)

        shortcut = QShortcut(QKeySequence(Qt.Key_L), parent)

        shortcut.activated.connect(main_wnd.onPlaceLine)

        shortcut = QShortcut(QKeySequence(Qt.Key_T), parent)

        shortcut.activated.connect(main_wnd.onAreaOcr)

        shortcut = QShortcut(QKeySequence(Qt.Key_V), parent)

        shortcut.activated.connect(main_wnd.onValidation)

        shortcut = QShortcut(QKeySequence(Qt.Key_Z), parent)

        shortcut.activated.connect(main_wnd.onAreaZoom)

        shortcut = QShortcut(QKeySequence(Qt.Key_W), parent)

        shortcut.activated.connect(main_wnd.change_display_colors)

        shortcut = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_F), parent)

        shortcut.activated.connect(main_wnd.findReplaceTextClicked)

        shortcut = QShortcut(QKeySequence(Qt.Key_F1), parent)

        shortcut.activated.connect(main_wnd.on_help)

        shortcut = QShortcut(QKeySequence(Qt.Key_1), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_1))

        shortcut = QShortcut(QKeySequence(Qt.Key_2), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_2))

        shortcut = QShortcut(QKeySequence(Qt.Key_3), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_3))

        shortcut = QShortcut(QKeySequence(Qt.Key_4), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_4))

        shortcut = QShortcut(QKeySequence(Qt.Key_5), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_5))

        shortcut = QShortcut(QKeySequence(Qt.Key_6), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_6))

        shortcut = QShortcut(QKeySequence(Qt.Key_7), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, Qt.Key_7))

        shortcut = QShortcut(QKeySequence(96), parent)

        shortcut.activated.connect(partial(main_wnd.on_view_toggle, 96))

    def get_pane(self, name: str) -> QWidget:

        return self._panes[name] if name in self._panes else None

    def add_home_tab(self) -> None:

        """ Adds the Home Tab """

        from App import App

        from AppDocData import AppDocData

        from AppRibbonPanes import QHomeFilePane, QHomePane, QHomeVisualizationPane

        from LineTypeConditions import LineTypeConditions

        from DisplayColors import DisplayColors

        from DisplayColors import DisplayOptions

        try:

            tab = self.addTab(self.tr('Home'))

            app_doc_data = AppDocData.instance()

            main_wnd = App.mainWnd()

            # File Section

            cSection = tab.addSection(self.tr('File'))

            pane = QHomeFilePane()

            self._panes['Home File'] = pane

            pane.ui.toolButtonFileOpen.setEnabled(False)

            #pane.ui.toolButtonFileOpen.clicked.connect(partial(main_wnd.open_image_drawing, None))

            #shortcut = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_O), pane.ui.toolButtonFileOpen)

            #shortcut.activated.connect(partial(main_wnd.open_image_drawing, None))

            pane.ui.toolButtonFileSave.clicked.connect(main_wnd.actionSaveCliked)

            #shortcut = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_S), pane.ui.toolButtonFileSave)

            #shortcut.activated.connect(main_wnd.actionSaveCliked)

            cSection.addCustomWidget(pane)

            # Home Section

            cSection = tab.addSection(self.tr('Home'))

            pane = QHomePane()

            self._panes['Home'] = pane

            pane.ui.toolButtonRecognition.clicked.connect(main_wnd.recognize)

            #shortcut = QShortcut(QKeySequence(Qt.Key_F5), pane.ui.toolButtonRecognition)

            #shortcut.activated.connect(main_wnd.recognize)

            pane.ui.toolButtonStreamline.clicked.connect(main_wnd.on_streamline)

            pane.ui.toolButtonLinkAttribute.clicked.connect(main_wnd.connect_attributes)

            pane.ui.toolButtonInitialize.clicked.connect(main_wnd.on_initialize_scene)

            pane.ui.toolButtonUndo.clicked.connect(main_wnd._scene.undo_stack.undo)

            pane.ui.toolButtonRedo.clicked.connect(main_wnd._scene.undo_stack.redo)

            pane.ui.toolButtonLine.clicked.connect(main_wnd.onPlaceLine)

            #shortcut = QShortcut(QKeySequence(Qt.Key_L), pane.ui.toolButtonLine)

            #shortcut.activated.connect(main_wnd.onPlaceLine)

            """fill line type"""

            for condition in LineTypeConditions.items():

                pane.ui.comboBoxLineType.addItem(condition.name)

            configs = app_doc_data.getConfigs('Line', 'Default Type')

            value = configs[0].value if 1 == len(configs) else ''

            if value:

                at = pane.ui.comboBoxLineType.findText(value)

                pane.ui.comboBoxLineType.setCurrentIndex(at)

            else:

                at = pane.ui.comboBoxLineType.findText('Secondary')

                pane.ui.comboBoxLineType.setCurrentIndex(at)

            """up to here"""

            pane.ui.toolButtonOCR.clicked.connect(main_wnd.onAreaOcr)

            #shortcut = QShortcut(QKeySequence(Qt.Key_T), pane.ui.toolButtonOCR)

            #shortcut.activated.connect(main_wnd.onAreaOcr)

            pane.ui.toolButtonVendor.clicked.connect(main_wnd.onVendor)

            pane.ui.comboBoxPackage.addItems(['Equipment Package', 'Vendor Package'])

            pane.ui.toolButtonValidate.clicked.connect(main_wnd.onValidation)

            #shortcut = QShortcut(QKeySequence(Qt.Key_V), pane.ui.toolButtonValidate)

            #shortcut.activated.connect(main_wnd.onValidation)

            pane.ui.toolButtonRotateCW.clicked.connect(main_wnd.onRotate)

            cSection.addCustomWidget(pane)

            # Visualization Section

            cSection = tab.addSection(self.tr('Visualization'))

            pane = QHomeVisualizationPane()

            self._panes['Home Visualization'] = pane

            pane.ui.radioButtonByGroup.toggled.connect(main_wnd.display_colors)

            pane.ui.radioButtonByType.toggled.connect(main_wnd.display_colors)

            pane.ui.radioButtonByStreamNo.toggled.connect(main_wnd.display_colors)

            if DisplayColors.instance().option == DisplayOptions.DisplayByLineNo:

                pane.ui.radioButtonByGroup.setChecked(True)

            elif DisplayColors.instance().option == DisplayOptions.DisplayByLineType:

                pane.ui.radioButtonByType.setChecked(True)

            elif DisplayColors.instance().option == DisplayOptions.DisplayByStreamNo:

                pane.ui.radioButtonByStreamNo.setChecked(True)

            pane.ui.toolButtonZoom.clicked.connect(main_wnd.onAreaZoom)

            #shortcut = QShortcut(QKeySequence(Qt.Key_Z), pane.ui.toolButtonZoom)

            #shortcut.activated.connect(main_wnd.onAreaZoom)

            pane.ui.toolButtonFitWindow.clicked.connect(main_wnd.fitWindow)

            pane.ui.toolButtonLockAxis.clicked.connect(main_wnd.on_toggle_lock_axis)

            cSection.addCustomWidget(pane)

        except Exception as ex:

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

    def add_edit_tab(self) -> None:

        """ Adds the Edit Tab """

        from App import App

        from AppRibbonPanes import QEditPane

        try:

            tab = self.addTab(self.tr('Edit'))  # "Edit"

            main_wnd = App.mainWnd()

            # Edit Section

            cSection = tab.addSection(self.tr('Edit'))

            pane = QEditPane()

            self._panes['Edit'] = pane

            pane.ui.toolButtonFindReplaceText.clicked.connect(main_wnd.findReplaceTextClicked)

            #shortcut = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_F), pane.ui.toolButtonFindReplaceText)

            #shortcut.activated.connect(main_wnd.findReplaceTextClicked)

            pane.ui.toolButtonTextDataList.clicked.connect(main_wnd.showTextDataList)

            pane.ui.toolButtonSymbolReplaceInsert.clicked.connect(main_wnd.replaceInsertSymbolClicked)

            pane.ui.toolButtonConnectLineToSymbol.clicked.connect(main_wnd.on_connect_line_to_symbol)

            pane.ui.toolButtonOCRUnknown.clicked.connect(main_wnd.on_ocr_unknown_items)

            cSection.addCustomWidget(pane)

        except Exception as ex:

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

    def add_data_tab(self) -> None:

        """Adds the Data Tab"""

        from App import App

        from AppRibbonPanes import QDataPane

        try:

            tab = self.addTab(self.tr('Data'))  # "Data"

            main_wnd = App.mainWnd()

            # File Section

            cSection = tab.addSection(self.tr('Data'))

            pane = QDataPane()

            self._panes['Data'] = pane

            pane.ui.toolButtonHMBData.clicked.connect(main_wnd.onHMBData)

            pane.ui.toolButtonEngInfoList.clicked.connect(main_wnd.showItemDataList)

            pane.ui.toolButtonSpecialItemTypes.clicked.connect(main_wnd.on_show_special_item_types)

            pane.ui.toolButtonOPCRelation.clicked.connect(main_wnd.on_show_opc_relation)

            pane.ui.toolButtonCodeTable.clicked.connect(main_wnd.onShowCodeTable)

            pane.ui.toolButtonCustomCodeTable.clicked.connect(main_wnd.onShowCustomCodeTable)

            pane.ui.toolButtonReplaceCodeTable.clicked.connect(main_wnd.onShowReplaceCodeTable)

            pane.ui.toolButtonGlobalValidation.clicked.connect(main_wnd.on_validation_global_clicked)

            pane.ui.toolButtonOCRTraining.clicked.connect(main_wnd.oCRTrainingClicked)

            pane.ui.toolButtonSymbolTraining.clicked.connect(main_wnd.symbolTrainingClicked)

            pane.ui.toolButtonMakeLabelData.clicked.connect(main_wnd.on_make_label_data)

            pane.ui.toolButtonLineList.clicked.connect(main_wnd.on_line_list)

            cSection.addCustomWidget(pane)

        except Exception as ex:

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

    def add_view_tab(self) -> None:

        """Adds the View Tab"""

        from App import App

        from AppRibbonPanes import QViewPane

        try:

            tab = self.addTab(self.tr('View'))

            main_wnd = App.mainWnd()

            # View Section

            cSection = tab.addSection(self.tr('View'))

            pane = QViewPane()

            self._panes['View'] = pane

            pane.ui.toolButtonViewImageDrawing.clicked.connect(main_wnd.onViewImageDrawing)

            pane.ui.toolButtonViewText.clicked.connect(main_wnd.onViewText)

            pane.ui.toolButtonViewSymbol.clicked.connect(main_wnd.onViewSymbol)

            pane.ui.toolButtonViewLine.clicked.connect(main_wnd.onViewLine)

            pane.ui.toolButtonViewUnknown.clicked.connect(main_wnd.onViewUnknown)

            pane.ui.toolButtonViewInconsistency.clicked.connect(main_wnd.onViewInconsistency)

            pane.ui.toolButtonViewVendorArea.clicked.connect(main_wnd.onViewVendorArea)

            pane.ui.toolButtonViewDrawingOnly.clicked.connect(main_wnd.onViewDrawingOnly)

            cSection.addCustomWidget(pane)

        except Exception as ex:

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

    def add_tool_tab(self) -> None:

        """Adds the Tool Tab"""

        from App import App

        from AppRibbonPanes import QToolPane

        try:

            tab = self.addTab(self.tr('Tool'))

            main_wnd = App.mainWnd()

            # Tool Section

            cSection = tab.addSection(self.tr('Tool'))

            pane = QToolPane()

            self._panes['Tool'] = pane

            pane.ui.toolButtonPDFToImage.clicked.connect(main_wnd.onConvertPDFToImage)

            pane.ui.toolButtonImportTextFromCAD.clicked.connect(main_wnd.on_import_text_from_cad)

            pane.ui.toolButtonImportTextFromCADforInstrument.clicked.connect(main_wnd.on_import_text_from_cad_for_instrument)

            pane.ui.toolButtonSymbolThicknessReinforcement.clicked.connect(main_wnd.onSymbolThickness)

            pane.ui.toolButtonDataTransfer.clicked.connect(main_wnd.on_show_data_transfer)

            pane.ui.toolButtonDataExport.clicked.connect(main_wnd.on_show_data_export)

            pane.ui.toolButtonConnection.clicked.connect(main_wnd.on_ext_app_connection)

            pane.ui.toolButtonExtApps.clicked.connect(main_wnd.on_execute_ext_app)

            cSection.addCustomWidget(pane)

        except Exception as ex:

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

    def add_help_tab(self) -> None:

        """ Adds the Help Tab """

        from App import App

        from AppRibbonPanes import QHelpPane

        try:

            tab = self.addTab('Help')

            main_wnd = App.mainWnd()

            # Help Section

            cSection = tab.addSection('Help')

            pane = QHelpPane()

            self._panes['Help'] = pane

            pane.ui.toolButtonHelp.clicked.connect(main_wnd.on_help)

            #shortcut = QShortcut(QKeySequence(Qt.Key_F1), pane.ui.toolButtonHelp)

            #shortcut.activated.connect(main_wnd.on_help)

            pane.ui.toolButtonReadme.clicked.connect(main_wnd.on_readme)

            cSection.addCustomWidget(pane)

        except Exception as ex:

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

class AppRibbonFileMenu(QFileMenu):

    """ Widget that represents the file menu for the ribbon """

    def __init__(self, advanced_mode: bool):

        """ Creates and initializes the menu """

        from App import App

        from AppRibbonPanes import QConfigPane

        QFileMenu.__init__(self)

        self._panes = {}

        main_wnd = App.mainWnd()

        if advanced_mode:

            self.addButton(icon=':/newPrefix/setup_area.png', title=self.tr('Area Setup'), handler=main_wnd.areaConfiguration)

            self.addButton(icon=':/newPrefix/setting.png', title=self.tr('Configuration'), handler=main_wnd.configuration)

            self.addSeparator()

        pane = QConfigPane()

        self._addBtn(pane)

        self._panes['Config'] = pane

        self.addSeparator()

        self.addButton(icon=':/newPrefix/Exit.svg', title=self.tr('Exit'), handler=main_wnd.close)

    def get_pane(self, name: str) -> QWidget:

        return self._panes[name] if name in self._panes else None

    def handleRecentFileClicked(self, path):

        """ Handles recent files being clicked """

        pass