HYTOS

# coding: utf-8

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import ConnectAttr_UI

import sys

import os

import cv2

from AppDocData import *

from LineDetector import LineDetector

from EngineeringLineItem import QEngineeringLineItem

from EngineeringLineNoTextItem import QEngineeringLineNoTextItem

from QEngineeringFlowArrowItem import QEngineeringFlowArrowItem

from SymbolSvgItem import SymbolSvgItem

from EngineeringTextItem import QEngineeringTextItem

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

'''

'''

class Worker(QThread):

    """ This is Worker class """

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

    from QtImageViewer import QtImageViewer

    import sys

    '''

    '''

    finished = pyqtSignal()

    intReady = pyqtSignal(int)

    displayMessage = pyqtSignal(str)

    updateProgress = pyqtSignal(int)

    def __init__(self, graphicsView, update_line_type, update_flow_mark):

        QThread.__init__(self)

        self.graphicsView = graphicsView

        self._update_line_type = update_line_type

        self._update_flow_mark = update_flow_mark

    '''

        @brief  execute connecting attributes

        @author humkyung

        @date   2018.06.17

    '''

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

        from LineNoTracer import connectAttrImpl

        try:

            connectAttrImpl(self, self._update_line_type, self._update_flow_mark)

        except Exception as ex:

            from App import App 

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

            App.mainWnd().addMessage.emit(MessageType.Error, message)

class QConnectAttrDialog(QDialog):

    """ This is connect attr dialog class """

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

        import ConnectAttr_UI

        QDialog.__init__(self, parent)

        self.parent = parent

        self.graphicsView = graphicsView

        self.ui = ConnectAttr_UI.Ui_ConnectAttr()

        self.ui.setupUi(self)

        self.ui.pushButtonStart.setFocus()

        self.ui.buttonBox.setEnabled(True)

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

        self.isRunned = False

        self.ui.pushButtonStart.clicked.connect(self.connStart)

    def connStart(self):

        '''

            @brief  connection start

            @author euisung

            @date   2019.02.25

        ''' 

        self.ui.buttonBox.setEnabled(False)

        self.ui.pushButtonStart.setEnabled(False)

        self.ui.progressBar.setValue(0)

        self.startThread()

        self.isRunned = True

    '''

        @brief      QListWidget Row Inserted Listener

                    Whenever row inserted, scroll to bottom

        @author     Jeongwoo

        @date       18.04.12

    '''

    def rowInserted(self, item):

        self.ui.listWidget.scrollToBottom()

    '''

        @brief  add item to list widget

        @author humkyung

        @date   2018.06.17

    '''

    def addListItem(self, msg):

        self.ui.listWidget.addItem(msg)

    '''

        @brief      update progressbar with given value

        @author     humkyung

        @date       2018.06.17

        @history    humkyung 2018.06.27 reset value if maxValue is -1

    '''

    def updateProgress(self, maxValue):

        if maxValue != -1:

            self.ui.progressBar.setMaximum(maxValue)

            self.ui.progressBar.setValue(self.ui.progressBar.value() + 1)

        else:

            self.ui.progressBar.setValue(0)

    '''

        @brief  start thread

        @author humkyung

        @date   2018.06.17

    '''

    def startThread(self):

        import timeit

        from LineNoTracer import connectAttrImpl

        try:

            self.ui.buttonBox.setDisabled(True)

            # 1 - create Worker and Thread inside the Form

            self.obj = Worker(self.graphicsView, self.ui.checkBoxUpdateLineType.isChecked(), self.ui.checkBoxUpdateFlowMark.isChecked())

            # 2 - Connect Worker Signals to the Thread slots

            self.obj.displayMessage.connect(self.addListItem)

            self.obj.updateProgress.connect(self.updateProgress)

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

            self.obj.finished.connect(self.dlgExit)

            # 4 - Start the thread

            self.obj.start()

            self.tmStart = timeit.default_timer()

        except Exception as ex:

            from App import App 

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

            App.mainWnd().addMessage.emit(MessageType.Error, message)

        finally:

            self.ui.buttonBox.setDisabled(False)

    '''

        @brief      set buttonbox's enabled flag

        @author     humkyung

        @date       2018.06.17

        @history    humkyung 2018.06.17 add flow mark for pipe run

                    humkyung 2018.08.16 don't update line type according to connected items

    '''

    def dlgExit(self):

        import timeit

        try:

            self.ui.progressBar.setValue(self.ui.progressBar.maximum())

            self.ui.buttonBox.setEnabled(True)

        except Exception as ex:

            from App import App

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

            App.mainWnd().addMessage.emit(MessageType.Error, message)

        finally:

            self.tmStop = timeit.default_timer()    

            seconds = self.tmStop - self.tmStart

            self.ui.listWidget.addItem("\nRunning Time : {} min".format(str(round(seconds/60, 1))) + "\n")