HYTOS

import QtImageViewer

def detectSymbolsOnPid(mainRes, targetSymbols, listWidget):

        detectSymbolOnPid(mainRes, detailTarget, listWidget)

'''

    @brief detect symbol on PID

'''

def detectSymbolOnPid(mainRes, targetSymbol, listWidget):

    listWidget.addItem('current symbol : ' + symbolPath)

    listWidget.addItem('finish symbol(count) / AvgMpCount : ' + symbolPath + '(' + str(foundSymbolCount) + ') / avgMpCount : ' + str(avgMpCount))

def drawFoundSymbols(symbol, listWidget):

        listWidget.addItem("tesseract result : \n" + ocrData)

def drawFoundSymbolsOnCanvas(mainRes, width, height, listWidget):

        drawFoundSymbols(symbol, listWidget)

def main(path, listWidget):

    try:

        srcList = []

        srcList.append(path)

        initTargetSymbolDataList()

        for mainRes in srcList:

            #Init src

            src = []

            srcGray = []

            ocrCompletedSrc = []

            searchedSymbolList = []

            textInfoList = []

            start = timeit.default_timer()

            initMainSrc(mainRes)

            listWidget.addItem("Start recognition : " + mainRes)

            #threadLock = threading.Lock()

            pool = futures.ThreadPoolExecutor(max_workers = THREAD_MAX_WORKER)

            for targetItem in targetSymbolList:

                if type(targetItem).__name__ == 'list':

                    #detectSymbolsOnPid(mainRes, target)

                    pool.submit(detectSymbolsOnPid, mainRes, targetItem, listWidget)

                else:

                    #detectSymbolOnPid(mainRes, target)

                    pool.submit(detectSymbolOnPid, mainRes, targetItem, listWidget)

            pool.shutdown(wait = True)

            chan, srcWidth, srcHeight = src.shape[::-1]

            drawFoundSymbolsOnCanvas(mainRes, srcWidth, srcHeight, listWidget)

            srcName = os.path.splitext(os.path.basename(mainRes))[0]

            pool = futures.ThreadPoolExecutor(max_workers = THREAD_MAX_WORKER)

            for sym in searchedSymbolList:

                #threadLock.acquire()

                pool.submit(removeDetectedSymbol, sym)

                pool.submit(drawRectOnSrc, sym)

                #drawRectOnSrc()

                #threadLock.release()

            pool.shutdown(wait = True)

            (srcGray, tInfoList) = OCR.removeTextFromNpArray(srcGray)

            #srcGray = TOCR.removeTextFromNpArray(srcGray, TOCR.FLAG_IMAGE_TO_DATA)

            for textInfo in textInfoList:

                #if not checkTextInSymbol((textInfo.getX(), textInfo.getY())):

                removeText(srcGray, textInfo.getText(), textInfo.getX(), textInfo.getY(), textInfo.getW(), textInfo.getH())

            ## Remove Noise

            kernel1 = np.ones((2, 2), np.uint8)

            srcGray = cv2.dilate(srcGray, kernel1)

            #kernel2 = np.ones((4, 4), np.uint8)

            srcGray = cv2.erode(srcGray, kernel1)

            for sym in searchedSymbolList:

                drawTempLine(sym)

            removedSymbolImgPath = os.path.dirname(os.path.realpath(__file__)) + "\\res\\Result\\result_fslo_"+srcName+"_600dpi.png"

            cv2.imwrite(removedSymbolImgPath, srcGray)

            cv2.imwrite(os.path.dirname(os.path.realpath(__file__)) + "\\res\\Result\\result_fslo_rect_"+srcName+"_600dpi.png", src)

            ####### Potrace

            imgLineList = potrace.passpotrace(removedSymbolImgPath)

            xg.writeXml(srcName, srcWidth, srcHeight, searchedSymbolList, textInfoList, imgLineList)

            listWidget.addItem("Searched symbol count : " + str(len(searchedSymbolList)))

            stop = timeit.default_timer()    

            seconds = stop - start

            f = open(os.path.dirname(os.path.realpath(__file__)) + "\\res\\Result\\result_"+srcName+"_600dpi.txt", 'a+')

            data = "Running Time : " + str(seconds / 60) + "min\n"

            f.write(data)

            f.close()

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

    except Exception as ex:

        msg = QMessageBox()

        msg.setIcon(QMessageBox.Critical)

        msg.setText(ex)

        msg.exec_()

class Worker(QObject):

    from PyQt5.QtCore import QThread

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

    import sys

    finished = pyqtSignal()

    intReady = pyqtSignal(int)

    #pyqtSlot()

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

        main(self.path, self.listWidget)

        self.finished.emit()

        self.graphicsView = QtImageViewer.QtImageViewer()

        self.graphicsView.setParent(self.centralwidget)

        self.verticalLayout.addWidget(self.graphicsView)

        # connect signals and slots

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

        @brief Open image drawing file and then display it

        MainWindow.path = MainWindow.graphicsView.loadImageFromFile()

        return MainWindow.path

    '''

        @brief recognize symbol and text

    '''

    def recognize(self, MainWindow):

        import Recognition_UI

        try:

            self.dlg = QDialog(self)

            self.ui = Recognition_UI.Ui_Recognition()

            self.ui.setupUi(self.dlg)

            self.startThread()

            self.dlg.show()

            self.dlg.exec_()

        except Exception as ex:

            print('에러가 발생했습니다.\n', ex)

    '''

        @brief start thread

    '''

    def startThread(self):

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

        # 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

    '''

    def dlgExit(self):

        self.ui.buttonBox.setEnabled(True)

    sys.exit(app.exec_())

    <bool>false</bool>

            options = QFileDialog.Options()

            options |= QFileDialog.DontUseNativeDialog

                fileName = QFileDialog.getOpenFileName(self, "Open image file", os.getcwd(), "Image files(*.png *.jpg)", options=options)

                fileName, dummy = QFileDialog.getOpenFileName(self, "Open image file", os.getcwd(), "Image files(*.png *.jpg)", options=options)

        return fileName

<?xml version="1.0" encoding="UTF-8"?>

<ui version="4.0">

 <class>Recognition</class>

 <widget class="QDialog" name="Recognition">

  <property name="windowModality">

   <enum>Qt::WindowModal</enum>

  </property>

  <property name="geometry">

   <rect>

    <x>0</x>

    <y>0</y>

    <width>845</width>

    <height>300</height>

   </rect>

  </property>

  <property name="font">

   <font>

    <family>맑은 고딕</family>

   </font>

  </property>

  <property name="windowTitle">

   <string>설계정보 인식 중...</string>

  </property>

  <layout class="QGridLayout" name="gridLayout">

   <item row="1" column="0">

    <widget class="QDialogButtonBox" name="buttonBox">

     <property name="enabled">

      <bool>false</bool>

     </property>

     <property name="orientation">

      <enum>Qt::Horizontal</enum>

     </property>

     <property name="standardButtons">

      <set>QDialogButtonBox::Cancel|QDialogButtonBox::Ok</set>

     </property>

    </widget>

   </item>

   <item row="0" column="0">

    <widget class="QListWidget" name="listWidget"/>

   </item>

  </layout>

 </widget>

 <resources/>

 <connections>

  <connection>

   <sender>buttonBox</sender>

   <signal>accepted()</signal>

   <receiver>Recognition</receiver>

   <slot>accept()</slot>

   <hints>

    <hint type="sourcelabel">

     <x>248</x>

     <y>254</y>

    </hint>

    <hint type="destinationlabel">

     <x>157</x>

     <y>274</y>

    </hint>

   </hints>

  </connection>

  <connection>

   <sender>buttonBox</sender>

   <signal>rejected()</signal>

   <receiver>Recognition</receiver>

   <slot>reject()</slot>

   <hints>

    <hint type="sourcelabel">

     <x>316</x>

     <y>260</y>

    </hint>

    <hint type="destinationlabel">

     <x>286</x>

     <y>274</y>

    </hint>

   </hints>

  </connection>

 </connections>

</ui>