HYTOS

# coding: utf-8

import sys

import os

import subprocess

from functools import partial

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\Commands')

import CreateCommand

import CropCommand

import AreaOcrCommand

import CreateSymbolCommand

import AreaZoomCommand

import FenceCommand

import PlaceLineCommand

import cv2

import numpy as np

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from PyQt5.QtSvg import *

from PIL import Image

import MainWindow_UI

import QtImageViewer

from SingletonInstance import SingletonInstane

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\Shapes')

from EngineeringPolylineItem import QEngineeringPolylineItem

from EngineeringLineItem import QEngineeringLineItem

from SymbolSvgItem import SymbolSvgItem

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

from EngineeringTextItem import QEngineeringTextItem

from EngineeringLineNoTextItem import QEngineeringLineNoTextItem

from QEngineeringNoteItem import QEngineeringNoteItem

from QEngineeringSizeTextItem import QEngineeringSizeTextItem

from EngineeringUnknownItem import QEngineeringUnknownItem

from QEngineeringEquipmentItem import QEngineeringEquipmentItem

from EngineeringInstrumentItem import QEngineeringInstrumentItem

from EngineeringSpecBreakItem import QEngineeringSpecBreakItem

from AppDocData import *

import SymbolTreeWidget, SymbolPropertyTableWidget

import SymbolEditorDialog

import ItemTreeWidget

import ItemPropertyTableWidget

from UserInputAttribute import UserInputAttribute

from TextItemFactory import TextItemFactory

class MainWindow(QMainWindow, MainWindow_UI.Ui_MainWindow, SingletonInstane):

    """

    This is MainWindow class

    """

    addMessage = pyqtSignal(Enum, str)

    '''

        @brief      initialize

        @author 

        @date   

        @history    humkyung 2018.04.12 add splitter widget

                    Jeongwoo 2018.04.27 Add Signal/Slot Connection 'noteNoSingleClicked'

                    Jeongwoo 2018.05.09 Initialize Action group

                    Jeongwoo 2018.05.10 Add Signal/Slot Connection 'lineNoSingleClicked'

                                        Add QActionGroup for managing checkable action

                    Jeongwoo 2018.06.27 Add Action [Zoom, Fit Window] and Add new actions into ActionGroup

                    humkyung 2018.08.23 add labelStatus to statusbar

                    Euisung 2018.09.27 add OCR Training , Signal/Slot Connection 'oCRTrainingClicked'

                    Euisung 2018.10.05 add OCR Editor , Signal/Slot Connection 'oCRTrainingEdidorClicked'

                    Euisung 2018.10.22 delete Signal/Slot Connection 'oCRTrainingEdidorClicked'

    '''

    def __init__(self):

        super(self.__class__, self).__init__()

        self.setupUi(self)

        self.labelStatus = QLabel(self.statusbar)

        self.labelStatus.setText('미인식 : ')

        self.labelSymbolStatus = QLabel(self.statusbar)

        self.labelSymbolStatus.setText('심볼 : ')

        self.labelLineStatus = QLabel(self.statusbar)

        self.labelLineStatus.setText('라인 : ')

        self.labelTextStatus = QLabel(self.statusbar)

        self.labelTextStatus.setText('텍스트 : ')

        self.statusbar.addPermanentWidget(self.labelSymbolStatus)

        self.statusbar.addPermanentWidget(self.labelLineStatus)

        self.statusbar.addPermanentWidget(self.labelTextStatus)

        self.statusbar.addPermanentWidget(self.labelStatus) 

        docData = AppDocData.instance()

        project = docData.getCurrentProject()

        _translate = QCoreApplication.translate

        self.setWindowTitle(_translate("Digital P&ID - {}".format(project.name), "Digital P&ID - {}".format(project.name)))

        self.lineComboBox = QComboBox(self.toolBar)

        lineTypes = docData.getLineTypes()

        for lineType in lineTypes:

            self.lineComboBox.addItem(lineType)

        self.lineComboBox.currentIndexChanged.connect(self.onLineTypeChanged)

        self.toolBar.insertWidget(self.actionOCR, self.lineComboBox)

        self.toolBar.insertSeparator(self.actionOCR)

        self.graphicsView = QtImageViewer.QtImageViewer(self)

        self.graphicsView.setParent(self.centralwidget)

        self.graphicsView.useDefaultCommand() ##### USE DEFAULT COMMAND

        self.verticalLayout.addWidget(self.graphicsView)

        # Add Custom TreeWidget

        self.dirTreeWidget = SymbolTreeWidget.QSymbolTreeWidget()

        self.dirTreeWidget.header().hide()

        self.symbolTabVerticalLayout.addWidget(self.dirTreeWidget)

        # Add Custom Property TableWidget

        self.propertyTableWidget = SymbolPropertyTableWidget.QSymbolPropertyTableWidget()

        self.symbolTabVerticalLayout.addWidget(self.propertyTableWidget)

        self.dirTreeWidget.singleClicked.connect(self.propertyTableWidget.getClickedSymbol)

        # add splitter widget

        splitter = QSplitter(Qt.Vertical)

        splitter.addWidget(self.dirTreeWidget)

        splitter.addWidget(self.propertyTableWidget)

        self.symbolTabVerticalLayout.addWidget(splitter)

        # up to here

        # Add Custom Result Tree Widget (Symbol Explorer)

        self.resultTreeWidget = ItemTreeWidget.QItemTreeWidget(self.graphicsView)

        self.resultTreeWidget.header().hide()

        self.symbolExplorerVerticalLayout.addWidget(self.resultTreeWidget)

        # Add Empty Widget

        self.resultPropertyTableWidget = ItemPropertyTableWidget.QItemPropertyTableWidget(self)

        self.symbolExplorerVerticalLayout.addWidget(self.resultPropertyTableWidget)

        self.resultTreeWidget.singleClicked.connect(self.resultPropertyTableWidget.onSymbolClicked)

        self.resultTreeWidget.noteNoSingleClicked.connect(self.resultPropertyTableWidget.onNoteClicked)

        self.resultTreeWidget.lineNoSingleClicked.connect(self.resultPropertyTableWidget.onLineNoClicked)

        self.resultTreeWidget.drawingClicked.connect(self.resultPropertyTableWidget.onDrawingClicked)

        # add splitter widget

        splitter = QSplitter(Qt.Vertical)

        splitter.addWidget(self.resultTreeWidget)

        splitter.addWidget(self.resultPropertyTableWidget)

        self.symbolExplorerVerticalLayout.addWidget(splitter)

        # up to here

        # Initialize Action group

        self.actionGroup = QActionGroup(self)

        self.actionGroup.addAction(self.actionRecognition)

        self.actionGroup.addAction(self.actionLineRecognition)

        self.actionGroup.addAction(self.actionLine)

        self.actionGroup.addAction(self.actionGenerateOutput)

        self.actionGroup.addAction(self.actionOCR)

        self.actionGroup.addAction(self.actionZoom)

        self.actionGroup.addAction(self.actionFitWindow)

        self.actionGroup.addAction(self.actionSave)

        self.actionGroup.triggered.connect(self.actionGroupTriggered)

        # connect signals and slots

        self.actionClose.triggered.connect(self.close)

        self.actionOpen.triggered.connect(self.onOpenImageDrawing)

        self.actionLine.triggered.connect(self.onPlaceLine)

        self.actionRecognition.triggered.connect(self.recognize)

        self.actionLineRecognition.triggered.connect(self.recognizeLine)

        self.actionArea.triggered.connect(self.areaConfiguration)

        self.actionConfiguration.triggered.connect(self.configuration)

        self.actionOCR.triggered.connect(self.onAreaOcr)

        self.actionGenerateOutput.triggered.connect(self.generateOutput)

        self.pushButtonCreateSymbol.clicked.connect(self.onCreateSymbolClicked)

        self.pushButtonClearLog.clicked.connect(self.onClearLog)

        self.actionHMB_DATA.triggered.connect(self.onHMBData)

        self.actionItem_Data_List.triggered.connect(self.showItemDataList)

        self.actionCodeTable.triggered.connect(self.onShowCodeTable)

        self.actionImage_Drawing.triggered.connect(self.onViewImageDrawing)

        self.actionViewText.triggered.connect(self.onViewText)

        self.actionViewSymbol.triggered.connect(self.onViewSymbol)

        self.actionViewLine.triggered.connect(self.onViewLine)

        self.actionViewUnknown.triggered.connect(self.onViewUnknown)

        self.actionRotate.triggered.connect(self.onRotate)

        self.actionZoom.triggered.connect(self.onAreaZoom)

        self.actionFitWindow.triggered.connect(self.fitWindow)

        self.actionpdf_to_image.triggered.connect(self.onConvertPDFToImage)

        self.graphicsView.scene.changed.connect(lambda: self.resultTreeWidget.sceneChanged(self.graphicsView.scene.items()))

        self.graphicsView.scene.changed.connect(self.onSceneChanged)

        self.graphicsView.scene.selectionChanged.connect(self.onSelectionChanged)

        self.actionInitialize.triggered.connect(self.onInitializeScene)

        self.resultPropertyTableWidget.cellDoubleClicked.connect(self.resultPropertyTableWidget.cellDoubleClickedEvent)

        self.resultPropertyTableWidget.cellClicked.connect(self.cellClickedEvent)

        self.actionSave.triggered.connect(self.actionSaveCliked)

        self.addMessage.connect(self.onAddMessage)

        self.actionFindReplaceText.triggered.connect(self.findReplaceTextClicked)

        self.pushButtonDetectSymbol.clicked.connect(self.onShowDetectSymbol)

        self.actionOCR_Training.triggered.connect(self.oCRTrainingClicked)

        #self.actionOCR_Training_Editor.triggered.connect(self.oCRTrainingEdidorClicked)

        # removedItems

        self.removedItems = {}

        self.removedItems['LINE'] = []

        self.removedItems['EQUIP'] = []

        self.removedItems['INST'] = []

        self.removedItems['NOTE'] = []

        self.delimiter = '"'

        self.resizeDocks({self.dockWidget}, {self.dockWidgetObjectExplorer.sizeHint().width()}, Qt.Horizontal)

        # load stylesheet file list

        stylesheet_name = QtWidgets.qApp.stylesheet_name

        files = [os.path.splitext(file)[0] for file in os.listdir(os.path.dirname(os.path.realpath(__file__))) if os.path.splitext(file)[1] == '.qss']

        for file in files:

            action = self.menuTheme.addAction(file)

            action.setCheckable(True)

            action.setChecked(True) if stylesheet_name == file else action.setChecked(False)

            action.triggered.connect(partial(self.load_stylesheet, file))

        # up to here

    def load_stylesheet(self, file):

        """

        @brief  load stylesheets

        @author humkyung

        @date   2018.10.29

        """

        QtWidgets.qApp.loadStyleSheet(os.path.join(os.path.dirname(os.path.realpath(__file__)), file))

        app_doc_data = AppDocData.instance()

        configs = [Config('app', 'stylesheet', file)]

        app_doc_data.saveAppConfigs(configs)

        for action in self.menuTheme.actions():

            if action.text() == file: continue

            action.setChecked(False)

    def onShowDetectSymbol(self):

        from DetectSymbolDialog import QDetectSymbolDialog

        dlgDetectSymbol = QDetectSymbolDialog(self)

        dlgDetectSymbol.show()

        dlgDetectSymbol.exec_()

    '''

        @brief      OCR Editor

        @author     euisung

        @date       2018.10.05

        @history    2018.10.16 euisung      no more used, Integrated with oCRTrainingClicked

    '''

    def oCRTrainingEdidorClicked(self):

        from TrainingEditorDialog import QTrainingEditorDialog

        try:

            dialog = QTrainingEditorDialog(self)

            dialog.exec_()

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

        return

    '''

        @brief      OCR Training

        @author     euisung

        @date       2018.09.27

        @history    euisung 2018.10.16 TrainingListDialog -> TrainingImageListDialog

    '''

    def oCRTrainingClicked(self):

        from TrainingImageListDialog import QTrainingImageListDialog

        #import ctypes

        #if not ctypes.windll.shell32.IsUserAnAdmin():

        #    reply = QMessageBox.information(self, 'Error', '관리자 권한으로 실행해 주세요.')

        #    #return

        try:

            dialog = QTrainingImageListDialog(self)

            dialog.exec_()

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

        return

    '''

        @brief      show unknownitem's count

        @author     kyouho

        @date       2018.08.27

    '''

    def findReplaceTextClicked(self):

        if not self.graphicsView.hasImage():

            self.showImageSelectionMessageBox()

            return

        from TextItemEditDialog import QTextItemEditDialog

        self.dlgTextItemEdit = QTextItemEditDialog(self)

        self.dlgTextItemEdit.show()

        self.dlgTextItemEdit.exec_()

    '''

        @brief      show unknownitem's count

        @author     humkyung

        @date       2018.08.23

        @history    humkyung 2018.08.30 display count of symbol, line, text

    '''

    def onSceneChanged(self):

        items = [item for item in self.graphicsView.scene.items() if type(item) is QEngineeringUnknownItem]

        if len(items) > 0:

            self.labelStatus.setText("<font color='red'>미인식 : {}</font>".format(len(items)))

        else:

            self.labelStatus.setText("<font color='black'>미인식 : {}</font>".format(len(items)))

        items = [item for item in self.graphicsView.scene.items() if issubclass(type(item), SymbolSvgItem)]

        self.labelSymbolStatus.setText("<font color='blue'>심볼 : {}</font>".format(len(items)))

        items = [item for item in self.graphicsView.scene.items() if type(item) is QEngineeringLineItem]

        self.labelLineStatus.setText("<font color='blue'>라인 : {}</font>".format(len(items)))

        items = [item for item in self.graphicsView.scene.items() if issubclass(type(item), QEngineeringTextItem)]

        self.labelTextStatus.setText("<font color='blue'>텍스트 : {}</font>".format(len(items)))

    '''

        @brief      action save click event

        @author     kyouho

        @date       2018.08.09

    '''

    def actionSaveCliked(self):

        self.saveToXml(True)

    '''

        @brief      save items to xml

        @author     kyouho

        @date       2018.07.31

    '''

    def saveToXml(self, alert = True):

        import XmlGenerator as xg

        from AppDocData import AppDocData

        docData = AppDocData.instance()

        if docData.imgName is None:

            self.showImageSelectionMessageBox()

            return

        result = xg.writeXmlOnScene(docData.imgName, docData.imgWidth, docData.imgHeight, self.graphicsView.scene)

        if len(self.removedItems['LINE']):

            docData.deleteLineDataList_LineNo(self.removedItems['LINE'])

            self.removedItems['LINE'] = []

        if len(self.removedItems['EQUIP']):

            docData.deleteEquipDataList(self.removedItems['EQUIP'])

            self.removedItems['EQUIP'] = []

        if len(self.removedItems['INST']):

            docData.deleteInstDataList(self.removedItems['INST'])

            self.removedItems['INST'] = []

        if len(self.removedItems['NOTE']):

            docData.deleteNoteDataList(self.removedItems['NOTE'])

            self.removedItems['NOTE'] = []

        if alert:

            resultStr = '[저장 결과]'

            for item in result.items():

                itemName = str(item[0])

                itemSuccessCount = str(item[1][0])

                itemFailUidList = item[1][1]

                resultStr += "\r\n" + itemName + " Save Count : " + itemSuccessCount

                if len(itemFailUidList) > 0:

                    resultStr += "\r\n" + itemName + " Error List(UID)"

                    for uid in itemFailUidList:

                        resultStr += "\r\n" + uid

            QMessageBox.about(self.graphicsView, "알림", resultStr)

    '''

        @brief      refresh resultPropertyTableWidget

        @author     kyouho

        @date       2018.07.19

    '''

    def refreshResultPropertyTableWidget(self):

        items = self.graphicsView.scene.selectedItems()

        if len(items) == 1:

            self.resultPropertyTableWidget.onSymbolClicked(items[0])

    '''

        @brief      resultPropertyTableWidget Cell Click Event

        @author     kyouho

        @date       2018.08.23

    '''

    def cellClickedEvent(self, row, column):

        item = self.graphicsView.scene.selectedItems()

        if len(item) != 1:

            return

        item = item[0]

        cell = self.resultPropertyTableWidget.item(row, column)

        for valueCell, uid in self.resultPropertyTableWidget.attrValueList:

            if valueCell == cell and issubclass(type(item), SymbolSvgItem):

                for attr in item.attrs:

                    if attr.attribute == uid and (issubclass(type(attr), SymbolSvgItem) or type(attr) is QEngineeringTextItem):

                        prevItem = item

                        currentItem = attr

                        rect = currentItem.sceneBoundingRect()

                        self.graphicsView.centerOn(rect.center())

                        self.graphicsView.zoomImage(True, QMouseEvent(QEvent.MouseButtonPress, self.graphicsView.mapFromScene(QPointF(rect.left(), rect.top())), Qt.LeftButton, Qt.LeftButton, Qt.NoModifier), 3)

                        prevItem.setSelected(True)

                        currentItem.setHightlight()

                    elif (issubclass(type(attr), SymbolSvgItem) or type(attr) is QEngineeringTextItem):

                        attr.unsetHightlight()

    '''

        @brief  add message listwidget

        @author humkyung

        @date   2018.07.31

    '''

    def onAddMessage(self, messageType, message):

        from AppDocData import MessageType

        try:

            current = QDateTime.currentDateTime()

            item = QListWidgetItem('{}: {}'.format(current.toString('hh:mm:ss'), message))

            if messageType == MessageType.Error:

                item.setBackground(Qt.red)

            self.listWidgetLog.insertItem(0, item)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      clear log

        @author     humkyung

        @date       2018.08.01

    '''

    def onClearLog(self):

        self.listWidgetLog.clear()

    '''

        @brief      rotate selected symbol

        @author     humkyung

        @date       2018.08.15

    '''

    def onRotate(self, action):

        selected = [item for item in self.graphicsView.scene.selectedItems() if issubclass(type(item), SymbolSvgItem)]

        if len(selected) == 1:

            selected[0].rotateSymbol()

    '''

        @brief      Area Zoom

        @author     Jeongwoo

        @date       2018.06.27

        @history    connect command's rejected signal

    '''

    def onAreaZoom(self, action):

        if self.actionZoom.isChecked():

            cmd = AreaZoomCommand.AreaZoomCommand(self.graphicsView)

            cmd.onRejected.connect(self.onCommandRejected)

            self.graphicsView.command = cmd

    '''

        @brief      Fit Window

        @author     Jeongwoo

        @date       2018.06.27

        @history    2018.06.27  Jeongwoo    Chnage method to initialize command [Set None → DefaultCommand]

    '''

    def fitWindow(self, action):

        self.graphicsView.useDefaultCommand()

        self.graphicsView.zoomImageInit()

    def onConvertPDFToImage(self):

        """

        @brief      convert to selected pdf to image

        @author     humkyung 

        @date       2018.07.09

        @history    Euisung 2018.10.11 hide shell

        """

        try: 

            filePath = os.path.join(os.path.dirname(os.path.realpath(__file__)) , 'bin64', 'PDF_TO_IMAGE.exe')

            subprocess.call(filePath, shell = False)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      selection changed

        @author     humkyung

        @date       2018.06.27

        @history    humkung 2018.07.08 call tree widget's findItem

    '''

    def onSelectionChanged(self):

        items = [item for item in self.graphicsView.scene.selectedItems() if issubclass(type(item), SymbolSvgItem) or \

            type(item) is QEngineeringLineItem or type(item) is QEngineeringLineNoTextItem or type(item) is QEngineeringNoteItem]

        if items:

            item = items[-1]

            self.resultTreeWidget.findItem(item)

            self.resultPropertyTableWidget.showItemProperty(item)

        else:

            self.resultPropertyTableWidget.showItemProperty(None)

    '''

        @brief      Initialize scene and ResultTreeWidget

        @author     Jeongwoo

        @date       2018.06.14

        @history    humkyung 2018.08.16 ask to delete recognized items before remove

    '''

    def onInitializeScene(self, action):

        if not self.graphicsView.hasImage():

            self.actionEquipment.setChecked(False)

            self.showImageSelectionMessageBox()

            return

        msg = QMessageBox()

        msg.setIcon(QMessageBox.Critical)

        msg.setText("선택한 인식한 항목들을 삭제하시겠습니까?\n삭제된 항목들은 복구할 수 없습니다.")

        msg.setWindowTitle("항목 삭제")

        msg.setStandardButtons(QMessageBox.Ok|QMessageBox.Cancel)

        if QMessageBox.Ok == msg.exec_():

            items = self.graphicsView.scene.items()

            for item in items:

                if type(item) is not QGraphicsPixmapItem:

                    self.graphicsView.scene.removeItem(item)

                    if type(item) is QEngineeringLineNoTextItem:

                        self.removedItems['LINE'].append(str(item.uid))

                    elif type(item) is QEngineeringInstrumentItem:

                        self.removedItems['INST'].append(str(item.uid))

                    elif type(item) is QEngineeringEquipmentItem:

                        self.removedItems['EQUIP'].append(str(item.uid))

                    elif type(item) is QEngineeringNoteItem:

                        self.removedItems['NOTE'].append(str(item.uid))

            if self.path is not None:

                baseName = os.path.basename(self.path)

                self.resultTreeWidget.setCurrentPID(baseName)

    '''

        @brief      Manage Checkable Action statement

        @author     Jeongwoo

        @date       2018.05.10

        @history    2018.06.27  Jeongwoo    Chnage method to initialize command [Set None → DefaultCommand]

    '''

    def actionGroupTriggered(self, action):

        if self.graphicsView.command is not None:

            self.graphicsView.useDefaultCommand()

        for _action in self.actionGroup.actions():

            _action.setChecked(False)

        action.setChecked(True)

    '''

        @brief      Create Equipment

        @author     Jeongwoo

        @date       18.05.03

        @history    2018.05.04  Jeongwoo    Add Parameter on CreateSymbolCommand

    '''

    def createEquipment(self):

        if not self.graphicsView.hasImage():

            self.actionEquipment.setChecked(False)

            self.showImageSelectionMessageBox()

            return

        if self.actionEquipment.isChecked():

            self.graphicsView.command = CreateSymbolCommand.CreateSymbolCommand(self.graphicsView, self.resultTreeWidget, self.dirTreeWidget)

        else:

            self.graphicsView.useDefaultCommand()

    '''

        @brief      Create Nozzle

        @author     Jeongwoo

        @date       2018.05.03

        @history    2018.05.04  Jeongwoo    Add Parameter on CreateSymbolCommand

    '''

    def createNozzle(self):

        if not self.graphicsView.hasImage():

            self.actionNozzle.setChecked(False)

            self.showImageSelectionMessageBox()

            return

        if self.actionNozzle.isChecked():

            self.graphicsView.command = CreateSymbolCommand.CreateSymbolCommand(self.graphicsView, self.resultTreeWidget, self.dirTreeWidget)

        else:

            self.graphicsView.useDefaultCommand()

    '''

        @brief      Area OCR

        @author     Jeongwoo

        @date       18.04.18

        @history    2018.05.02  Jeongwoo    Change graphicsView.command by actionOCR checked state

                                            Show MessageBox when imageviewer doesn't have image

    '''

    def onAreaOcr(self):

        if not self.graphicsView.hasImage():

            self.actionOCR.setChecked(False)

            self.showImageSelectionMessageBox()

            return

        if self.actionOCR.isChecked():

            cmd = AreaOcrCommand.AreaOcrCommand(self.graphicsView)

            cmd.onSuccess.connect(self.onRecognizeText)

            cmd.onRejected.connect(self.onCommandRejected)

            self.graphicsView.command = cmd

        else:

            self.graphicsView.useDefaultCommand()

    '''

        @brief      show text recognition dialog

        @author     humkyung

        @date       2018.08.08

    '''

    def onRecognizeText(self, x, y, width, height):

        from OcrResultDialog import QOcrResultDialog

        try:

            image = self.graphicsView.image().copy(x, y, width, height)

            dialog = QOcrResultDialog(self.graphicsView, image, QRectF(x, y, width, height))

            (isAccept, textInfoList) = dialog.showDialog()

            if isAccept:

                if textInfoList is not None and len(textInfoList) > 0:

                    docData = AppDocData.instance()

                    configs = docData.getConfigs('Line No', 'Delimiter')

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

                    lineNoconfigs = docData.getConfigs('Line No', 'Configuration')

                    for textInfo in textInfoList:

                        x = textInfo.getX()

                        y = textInfo.getY()

                        angle = textInfo.getAngle()

                        text = textInfo.getText()

                        width = textInfo.getW()

                        height = textInfo.getH()

                        item = TextItemFactory.instance().createTextItem(text)

                        if item is not None:

                            item.loc = (x, y)

                            item.size = (width, height)

                            item.angle = angle

                            item.setPlainText(text)

                            item.setDefaultTextColor(Qt.blue)

                            item.addTextItemToScene(self.graphicsView.scene)

                            item.transfer.onRemoved.connect(self.itemRemoved)

                        else:

                            message = 'error occured({}) in {}:{}'.format('텍스트 생성에 실패했습니다.', sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

                            self.addMessage.emit(MessageType.Normal, message)

                else:

                    QMessageBox.about(self.graphicsView, "알림", "텍스트 검출 실패")

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

    '''

        @brief  area configuration

    '''

    def areaConfiguration(self):

        from ConfigurationAreaDialog import QConfigurationAreaDialog

        self.dlgConfigurationArea = QConfigurationAreaDialog(self)

        self.dlgConfigurationArea.show()

        self.dlgConfigurationArea.exec_()

    '''

        @brief  configuration

    '''

    def configuration(self):

        from ConfigurationDialog import QConfigurationDialog

        self.dlgConfiguration = QConfigurationDialog(self)

        self.dlgConfiguration.show()

        self.dlgConfiguration.exec_()

    '''

        @brief  show nominal diameter dialog 

        @author humkyung

        @date   2018.06.28

    '''

    def onShowCodeTable(self):

        from CodeTableDialog import QCodeTableDialog

        dlg = QCodeTableDialog(self)

        dlg.exec_()

    '''

        @brief  show HMB data

        @author humkyung

        @date   2018.07.11

    '''

    def onHMBData(self):

        from HMBDialog import QHMBDialog

        dlg = QHMBDialog(self)

        dlg.show()

        dlg.exec_()

    '''

        @brief  show line data list 

        @author humkyung

        @date   2018.05.03

    '''

    def showItemDataList(self):

        from ItemDataExportDialog import QItemDataExportDialog

        self.dlgLineDataList = QItemDataExportDialog(self)

        self.dlgLineDataList.exec_()

    '''

        @brief  Show Image Selection Guide MessageBox

        @author Jeongwoo

        @date   2018.05.02

    '''

    def showImageSelectionMessageBox(self):

        QMessageBox.about(self.graphicsView, "알림", "이미지를 선택하신 후 시도해주세요.")

    '''

        @brief  change selected lines' type by selected line type

        @author humkyung

        @date   2018.06.27

    '''

    def onLineTypeChanged(self, param):

        lineType = self.lineComboBox.itemText(param)

        selected = [item for item in self.graphicsView.scene.selectedItems() if type(item) is QEngineeringLineItem]

        if selected:

            for item in selected:

                item.lineType = lineType

    '''

        @brief      Open image drawing file and then display it

        @author     humkyung

        @date       2018.??.??

        @history    18.04.23 Jeongwoo    Add AppDocData.instance().setCurrentPidSource

                    18.04.23 Jeongwoo    Add Store Set Current Pid Image on AppDocData

                    18.05.02 Jeongwoo    Add useDefaultCommand()

                    humkyung 2018.05.24 load recognition result file if exists

                    Jeongwoo 2018.05.29 load data on xml if xml file exist

                    humkyung 2018.08.22 clear scene before loading xml file

    '''

    def onOpenImageDrawing(self, MainWindow):

        from Drawing import Drawing

        try:

            appDocData = AppDocData.instance()

            project = appDocData.getCurrentProject()

            for item in self.graphicsView.scene.items():

                self.graphicsView.scene.removeItem(item)

            self.path = self.graphicsView.loadImageFromFile(project.getDrawingFilePath())

            if os.path.isfile(self.path):

                appDocData.clear()

                self.graphicsView.useDefaultCommand()

                appDocData.setImgFilePath(self.path)

                appDocData.activeDrawing = Drawing(appDocData.imgName)

                appDocData.setCurrentPidSource(Image.open(self.path))

                self.resultTreeWidget.setCurrentPID(appDocData.activeDrawing.name)

                self.progress = QProgressDialog("잠시만 기다려주세요", "Cancel", 0, 100, self)

                self.progress.setWindowModality(Qt.WindowModal)

                self.progress.setAutoReset(True)

                self.progress.setAutoClose(True)

                self.progress.setMinimum(0)

                self.progress.resize(600,100)

                self.progress.setWindowTitle("파일 읽는 중...")

                self.progress.show()

                ## Load data on xml

                try:

                    path = os.path.join(appDocData.getCurrentProject().getTempPath(), appDocData.imgName + '.xml')

                    if os.path.isfile(path):

                        self.loadRecognitionResultFromXml(path)

                        self.checkAttribute()

                finally:

                    self.progress.setValue(self.progress.maximum())

                    self.progress.hide()

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

        return self.path

    '''

        @brief  visible/invisible image drawing

        @author humkyung

        @date   2018.06.25

    '''

    def onViewImageDrawing(self, isChecked):

        for item in self.graphicsView.scene.items():

            if type(item) is QGraphicsPixmapItem:

                item.setVisible(isChecked)

                break

    '''

        @brief  visible/invisible Text 

        @author humkyung

        @date   2018.06.28

    '''

    def onViewText(self, isChecked):

        selected = [item for item in self.graphicsView.scene.items() if issubclass(type(item), QEngineeringTextItem)]

        for item in selected:

            item.setVisible(isChecked)

    '''

        @brief  visible/invisible Symbol 

        @author humkyung

        @date   2018.06.28

    '''

    def onViewSymbol(self, isChecked):

        selected = [item for item in self.graphicsView.scene.items() if issubclass(type(item), SymbolSvgItem)]

        for item in selected:

            item.setVisible(isChecked)

    '''

        @brief  visible/invisible Line

        @author humkyung

        @date   2018.06.28

    '''

    def onViewLine(self, isChecked):

        selected = [item for item in self.graphicsView.scene.items() if type(item) is QEngineeringLineItem]

        for item in selected:

            item.setVisible(isChecked)

    '''

        @brief  visible/invisible Unknown 

        @author humkyung

        @date   2018.06.28

    '''

    def onViewUnknown(self, isChecked):

        selected = [item for item in self.graphicsView.scene.items() if type(item) is QEngineeringUnknownItem]

        for item in selected:

            item.setVisible(isChecked)

    '''

        @brief  create a symbol

        @history    2018.05.02  Jeongwoo    Change return value of QSymbolEditorDialog (Single variable → Tuple)

                                            Add SymbolSvgItem

                    2018.05.03  Jeongwoo    Change method to draw Svg Item on Scene (svg.addSvgItemToScene)

                                            Change method to make svg and image path

                    2018.06.08  Jeongwoo    Add Paramter on SymbolSvgItem.buildItem()

    '''

    def onCreateSymbolClicked(self):

        cmd = FenceCommand.FenceCommand(self.graphicsView)

        cmd.onSuccess.connect(self.onAreaSelected)

        self.graphicsView.command = cmd

        QApplication.setOverrideCursor(Qt.CrossCursor)

    '''

        @brief      show SymbolEditorDialog with image selected by user

        @author     humkyung

        @date       2018.07.20

    '''

    def onAreaSelected(self, x, y, width, height):

        try:

            image = self.graphicsView.image()

            if image is not None:

                symbolEditorDialog = SymbolEditorDialog.QSymbolEditorDialog(self, image.copy(x, y, width, height), AppDocData.instance().getCurrentProject())

                (isAccepted, isImmediateInsert, offsetX, offsetY, newSym) = symbolEditorDialog.showDialog()

                self.dirTreeWidget.initDirTreeWidget()

                if isAccepted:

                    if isImmediateInsert:

                        svgPath = newSym.getSvgFileFullPath()

                        img = cv2.imread(newSym.getImageFileFullPath(), 1)

                        w, h = (0, 0)

                        if len(img.shape[::-1]) == 2:

                            w, h = img.shape[::-1]

                        else:

                            _chan, w, h = img.shape[::-1]

                        svg = SymbolSvgItem(svgPath)

                        svg.buildItem(newSym.getName(), newSym.getType(), 0, [offsetX, offsetY], [w, h], [float(x) for x in newSym.getOriginalPoint().split(',')], [(float(x.split(',')[0]), float(x.split(',')[1])) for x in newSym.getConnectionPoint().split('/')], newSym.getBaseSymbol(), newSym.getAdditionalSymbol(), newSym.getHasInstrumentLabel)

                        svg.transfer.onRemoved.connect(self.resultTreeWidget.itemRemoved)

                        svg.addSvgItemToScene(self.graphicsView.scene)

                        for connector in svg.connectors:

                            self.graphicsView.scene.addItem(connector)

        finally:

            self.graphicsView.useDefaultCommand()

            QApplication.restoreOverrideCursor()

    '''

        @brief      create a line

        @author     humkyung

        @history    Jeongwoo 2018.05.10 Change method for Checkable action

    '''

    def onPlaceLine(self):        

        if not self.graphicsView.hasImage():

            self.actionLine.setChecked(False)

            self.showImageSelectionMessageBox()

            return

        self.actionLine.setChecked(True)

        if not hasattr(self.actionLine, 'tag'):

            self.actionLine.tag = PlaceLineCommand.PlaceLineCommand(self.graphicsView)

            self.actionLine.tag.onSuccess.connect(self.onLineCreated)

            self.actionLine.tag.onRejected.connect(self.onCommandRejected)

        self.graphicsView.command = self.actionLine.tag

    '''

        @brief      add created lines to scene

        @author     humkyung

        @date       2018.07.23

    '''

    def onLineCreated(self):

        from EngineeringConnectorItem import QEngineeringConnectorItem

        try:

            count = len(self.actionLine.tag._polyline._vertices)

            if count > 1:

                items = []

                lineType = self.lineComboBox.currentText()

                for index in range(count - 1):

                    start = self.actionLine.tag._polyline._vertices[index]

                    end  = self.actionLine.tag._polyline._vertices[index + 1]

                    lineItem = QEngineeringLineItem(vertices=[start, end])

                    lineItem.transfer.onRemoved.connect(self.itemRemoved)

                    lineItem.lineType = lineType

                    if items:

                        lineItem.connectIfPossible(items[-1], 5)

                    else:

                        pt = lineItem.startPoint()

                        selected = self.graphicsView.scene.itemAt(QPointF(pt[0], pt[1]), QTransform())

                        if selected is not None and type(selected) is QEngineeringConnectorItem:

                            lineItem.connectIfPossible(selected.parent, 5)

                    items.append(lineItem)

                    self.graphicsView.scene.addItem(lineItem)

                pt = items[-1].endPoint()

                selected = self.graphicsView.scene.itemAt(QPointF(pt[0], pt[1]), QTransform())

                if selected is not None and type(selected) is QEngineeringConnectorItem:

                    items[-1].connectIfPossible(selected.parent, 5)

        finally:

            self.graphicsView.scene.removeItem(self.actionLine.tag._polyline)

            self.actionLine.tag.reset()

    '''

        @brief      refresh scene

        @author     humkyung

        @date       2018.07.23

    '''

    def onCommandRejected(self, cmd):

        try:

            if type(cmd) is PlaceLineCommand.PlaceLineCommand:

                self.graphicsView.scene.removeItem(self.actionLine.tag._polyline)

                self.graphicsView.scene.update()

                self.actionLine.tag.reset()

                self.actionLine.setChecked(False)

            elif type(cmd) is AreaZoomCommand.AreaZoomCommand:

                self.actionZoom.setChecked(False)

            elif type(cmd) is AreaOcrCommand.AreaOcrCommand:

                self.actionOCR.setChecked(False)

        finally:

            self.graphicsView.useDefaultCommand()

    '''

        @brief      restore to default command when user press Escape key

        @author     humkyung 

        @date       2018.08.09

    '''

    def keyPressEvent(self, event):

        try:

            if event.key() == Qt.Key_Escape:

                checked = self.actionGroup.checkedAction()

                if checked:

                    checked.setChecked(False)

                    self.graphicsView.useDefaultCommand()

            QMainWindow.keyPressEvent(self, event)

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

    '''

        @brief      recognize symbol and text

        @author     humkyung

        @date       2018.04.??

        @history    2018.04.16  humkyung    execute line no tracing

                    2018.05.02  Jeongwoo    Show MessageBox when imageviewer doesn't have image

                    2018.05.25  Jeongwoo    Add parameter on QRecognitionDialog.__init__() and Move thread to QRecognitionDialog

                                            Remove codes below if self.dlg.isAccepted == True

                    2018.05.29  Jeongwoo    Remove connects and comments

    '''

    def recognize(self, MainWindow):

        from RecognitionDialog import QRecognitionDialog

        if not self.graphicsView.hasImage():

            self.showImageSelectionMessageBox()

            return

        try:

            self.removedItems['LINE'] = []

            self.removedItems['EQUIP'] = []

            self.removedItems['INST'] = []

            self.removedItems['NOTE'] = []

            appDocData = AppDocData.instance()

            self.resultTreeWidget.setCurrentPID(appDocData.activeDrawing.name)

            ## up to here

            self.onClearLog()

            self.dlg = QRecognitionDialog(self, self.path)

            self.dlg.exec_()

            if self.dlg.isAccepted == True:

                '''DO NOTHING'''

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

    '''

        @brief      remove item from tree widget and then remove from scene

        @date       2018.05.25

        @author     Jeongwoo

    '''

    def itemRemoved(self, item):

        try:

            self.resultTreeWidget.itemRemoved(item)

            if type(item) is QEngineeringLineNoTextItem:

                self.removedItems['LINE'].append(str(item.uid))

            elif type(item) is QEngineeringInstrumentItem:

                self.removedItems['INST'].append(str(item.uid))

            elif type(item) is QEngineeringEquipmentItem:

                self.removedItems['EQUIP'].append(str(item.uid))

            elif type(item) is QEngineeringNoteItem:

                self.removedItems['NOTE'].append(str(item.uid))

            if item.scene() is not None: item.scene().removeItem(item)

        except Exception as ex:

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            self.addMessage.emit(MessageType.Error, message)

    '''

        @brief      recognize line

        @author     humkyung

        @date       2018.04.19

        @history    Jeongwoo 2018.04.26 Variable name changed (texts → lineNos)

                                        TextItem type changed (QEngineeringTextItem → QEngineeringLineNoTextItem)

                    humkyung 2018.04.26 remove small objects before recognizing line

                    Jeongwoo 2018.05.02 Show MessageBox when imageviewer doesn't have image

                    Jeongwoo 2018.05.25 Move codes about LineDetector

                    humkyung 2018.06.17 show progress dialog

    '''

    def recognizeLine(self, MainWindow):

        from LineNoTracer import LineNoTracer

        from ConnectAttrDialog import QConnectAttrDialog

        if not self.graphicsView.hasImage():

            self.showImageSelectionMessageBox()

            return

        try:

            self.dlgConnectAttr = QConnectAttrDialog(self, self.graphicsView)

            self.dlgConnectAttr.exec_()

            self.resultTreeWidget.InitLineNoItems()

            # construct line no item

            docData = AppDocData.instance()

            for lineno in docData.lineNos:

                item = self.resultTreeWidget.addTreeItem(self.resultTreeWidget.root, lineno)