HYTOS

# coding: utf-8

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import Recognition_UI

import sys

import os

import cv2

from AppDocData import AppDocData

from LineDetector import LineDetector

from QEngineeringLineItem import QEngineeringLineItem

from QEngineeringLineNoTextItem import QEngineeringLineNoTextItem

from QEngineeringFlowArrowItem import QEngineeringFlowArrowItem

from SymbolSvgItem import SymbolSvgItem

from QEngineeringTextItem import QEngineeringTextItem

from QGraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

'''

    @history    2018.05.25  Jeongwoo    Add pyqtSignal(recognizeLine, loadRecognitionResult)

'''

class Worker(QObject):

    from PyQt5.QtCore import QThread

    from PyQt5.QtWidgets import QApplication, QLabel, QWidget, QGridLayout, QListWidget

    from QtImageViewer import QtImageViewer

    import sys

    import timeit

    '''

        @history    2018.05.30  Jeongwoo    Remove parameter on recognizeLine signal

    '''

    finished = pyqtSignal()

    intReady = pyqtSignal(int)

    recognizeLine = pyqtSignal(QListWidget, QtImageViewer)

    loadRecognitionResult = pyqtSignal(list, list, list)

    '''

        @history    2018.05.25  Jeongwoo    Add if-statements by isSymbolTextChecked and isLineChecked variable

                    2018.05.28  Jeongwoo    Add Parameter 'xmlPath[0]'

                    2018.05.30  Jeongwoo    Remove import recognizeLine and self.xmlPath and remove parameter on recognizeLine.emit() (xmlPath)

    '''

    #pyqtSlot()

    def procCounter(self): # A slot takes no params

        from DTI_PID import executeRecognition

        import timeit

        try:

            start = timeit.default_timer()

            executeRecognition(self.loadRecognitionResult, self.path, self.listWidget, self.isSymbolTextChecked)

            if self.isLineChecked:

                self.recognizeLine.emit(self.listWidget, self.graphicsView)

            self.finished.emit()

        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))

        finally:

            stop = timeit.default_timer()    

            seconds = stop - start

            self.listWidget.addItem("\nRunning Time : " + str(seconds / 60) + "min\n")

'''

    @history    2018.05.25  Jeongwoo    Add pyqtSignal(svgItemClicked, itemRemoved)

'''

class QRecognitionDialog(QDialog):

    svgItemClicked = pyqtSignal(SymbolSvgItem)

    itemRemoved = pyqtSignal(QGraphicsItem)

    '''

        @history    2018.05.25  Jeongwoo    Add parameter and initialize / Connect recognizeButton signal and slot

                    2018.05.29  Jeongwoo    Chnage parameter(graphicsView → parent) and Get graphicsView from parent

    '''

    def __init__(self, parent, path): #Parent is MainWindow

        QDialog.__init__(self, parent)

        self.parent = parent

        self.graphicsView = parent.graphicsView

        self.path = path

        self.xmlPath = None

        self.ui = Recognition_UI.Ui_Recognition()

        self.ui.setupUi(self)

        self.ui.listWidget.model().rowsInserted.connect(self.rowInserted) ## connect to func rowInserted(self, item)

        self.ui.recognizeButton.clicked.connect(self.recognizeButtonClicked)

        self.isAccepted = False

    '''

        @brief      QListWidget Row Inserted Listener

                    Whenever row inserted, scroll to bottom

        @author     Jeongwoo

        @date       18.04.12

        @history    .

    '''

    def rowInserted(self, item):

        self.ui.listWidget.scrollToBottom()

    def accept(self):

        self.isAccepted = True

        QDialog.accept(self)

    def recognizeButtonClicked(self, event):

        if self.ui.symbolTextCheckBox.isChecked() or self.ui.lineCheckBox.isChecked():

            self.startThread()

    '''

        @brief  start thread

        @author humkyung

        @date   2018.04.??

        @history    2018.05.25  Jeongwoo    Moved from MainWindow

                    2018.05.28  Jeongwoo    Add connects (self.loadRecognitionResult, recognizeLine)

    '''

    def startThread(self):

        from DTI_PID import recognizeLine

        self.ui.buttonBox.setDisabled(True)

        # 1 - create Worker and Thread inside the Form

        self.obj = Worker()  # no parent!

        self.obj.path = self.path

        self.obj.listWidget = self.ui.listWidget

        self.obj.graphicsView = self.graphicsView

        self.obj.isSymbolTextChecked = self.ui.symbolTextCheckBox.isChecked()

        self.obj.isLineChecked = self.ui.lineCheckBox.isChecked()

        self.thread = QThread()  # no parent!

        # 2 - Connect Worker`s Signals to Form method slots to post data.

        #self.obj.intReady.connect(self.onIntReady)

        # 3 - Move the Worker object to the Thread object

        self.obj.moveToThread(self.thread)

        # 4 - Connect Worker Signals to the Thread slots

        self.obj.finished.connect(self.thread.quit)

        self.obj.loadRecognitionResult.connect(self.loadRecognitionResult)

        self.obj.recognizeLine.connect(recognizeLine)

        # 5 - Connect Thread started signal to Worker operational slot method

        self.thread.started.connect(self.obj.procCounter)

        # * - Thread finished signal will close the app if you want!

        self.thread.finished.connect(self.dlgExit)

        # 6 - Start the thread

        self.thread.start()

    '''

        @brief set buttonbox's enabled flag

        @history    2018.05.25  Jeongwoo    Moved from MainWindow

                    2018.05.30  Jeongwoo    Remove self.xmlPath variable and Add writeXmlOnScene

    '''

    def dlgExit(self):

        import XmlGenerator as xg

        try:

            #self.xmlPath = self.obj.xmlPath

            self.ui.buttonBox.setEnabled(True)

            docData = AppDocData.instance()

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

        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))

    '''

        @history    2018.05.29  Jeongwoo    Call parent's method

    '''

    def loadRecognitionResult(self, symbolList, textInfoList, noteTextInfoList):

        self.parent.loadRecognitionResultByList(symbolList, textInfoList, noteTextInfoList)