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 AppDocData

from LineDetector import LineDetector

from EngineeringLineItem import QEngineeringLineItem

from QEngineeringLineNoTextItem import QEngineeringLineNoTextItem

from QEngineeringFlowArrowItem import QEngineeringFlowArrowItem

from SymbolSvgItem import SymbolSvgItem

from QEngineeringTextItem import QEngineeringTextItem

from QGraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

'''

'''

class Worker(QObject):

    from PyQt5.QtCore import QThread

    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)

    '''

        @brief  execute connecting attributes

        @author humkyung

        @date   2018.06.17

    '''

    #pyqtSlot()

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

        from LineNoTracer import connectAttrImpl

        try:

            connectAttrImpl(self)

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

'''

'''

class QConnectAttrDialog(QDialog):

    '''

    '''

    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.buttonBox.setEnabled(True)

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

        self.isAccepted = False

        self.ui.buttonBox.setEnabled(False)

        self.ui.progressBar.setValue(0)

        self.startThread()

    '''

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

    def accept(self):

        self.isAccepted = True

        QDialog.accept(self)

    '''

        @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

        self.ui.buttonBox.setDisabled(True)

        # 1 - create Worker and Thread inside the Form

        self.obj = Worker()  # no parent!

        self.obj.graphicsView = self.graphicsView

        self.thread = QThread()  # no parent!

        # 2 - Move the Worker object to the Thread object

        self.obj.moveToThread(self.thread)

        # 3 - Connect Worker Signals to the Thread slots

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

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

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

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

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

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

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

        # 6 - Start the thread

        self.thread.start()

        self.tmStart = timeit.default_timer()

    '''

        @brief      set buttonbox's enabled flag

        @author     humkyung

        @date       2018.06.17

        @history    humkyung 2018.06.17 add flow mark for pipe run

    '''

    def dlgExit(self):

        import timeit

        try:

            # update line type

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

            for line in lines:

                line.updateLineType()

            # up to here

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

            self.ui.buttonBox.setEnabled(True)

        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:

            self.tmStop = timeit.default_timer()    

            seconds = self.tmStop - self.tmStart

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