HYTOS

    @brief  remove small objects from given image

    @author humkyung

    @date   2018.04.26

    @history    2018.05.25  Jeongwoo    Moved from MainWindow

'''

def removeSmallObjects(image):

    try:

        docData = AppDocData.instance()

        configs = docData.getConfigs('Small Object Size', 'Min Area')

        minArea = int(configs[0].value) if 1 == len(configs) else 20

        configs = docData.getConfigs('Small Object Size', 'Max Area')

        maxArea = int(configs[0].value) if 1 == len(configs) else 50

        #path = os.path.join(AppDocData.instance().getCurrentProject().getTempPath(), 'before_contours.png')

        #cv2.imwrite(path, image)

        _,contours,_ = cv2.findContours(image, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE);

        selectedContours=[]

        for contour in contours:

            #if cv2.isContourConvex(contour):

                #approx = cv2.approxPolyDP(contour, 0.2*cv2.arcLength(contour, True), True)

            area = cv2.contourArea(contour)

            if area > minArea and area < maxArea: selectedContours.append(contour)

        contourImage = cv2.drawContours(image, selectedContours, -1, (255,255,255), -1); # draw contour with white color

        #path = os.path.join(AppDocData.instance().getCurrentProject().getTempPath(), 'contours.png')

        #cv2.imwrite(path, contourImage)

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

    return contourImage

'''

    @history    2018.05.25  Jeongwoo    Moved from MainWindow

'''

def recognizeLine(listWidget, graphicsView):

    from QEngineeringFlowArrowItem import QEngineeringFlowArrowItem

    from QEngineeringLineNoTextItem import QEngineeringLineNoTextItem

    from QEngineeringTextItem import QEngineeringTextItem

    from LineDetector import LineDetector

    try:

        #remove already existing line and flow arrow item

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

        for item in items:

            graphicsView.scene.removeItem(item)

        #up to here

        # detect line

        connectedLines = []

        area = AppDocData.instance().getArea('Drawing')

        area.img = removeSmallObjects(area.img)

        detector = LineDetector(area.img)

        symbols = []

        for item in graphicsView.scene.items():

            if issubclass(type(item), SymbolSvgItem):

                symbols.append(item)

                res = detector.detectConnectedLine(item, round(area.x), round(area.y))

                if res is not None:

                    for line in res: connectedLines.append(line)

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

        for symbol in symbols:

            symbol.connectAttribute(texts)

        lineNos = [item for item in graphicsView.scene.items() if type(item) is QEngineeringLineNoTextItem]

        if len(connectedLines) > 1:

            detector.mergeLines(connectedLines, toler=20)

            # connect line to symbol

            try:

                for line in connectedLines:

                    matches = [symbol for symbol in symbols if symbol.isConnectable(line, (round(area.x), round(area.y)), toler=20)]

                    for symbol in matches:

                        detector.connectLineToSymbol(line, (round(area.x), round(area.y)), symbol)

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

            # up to here

            # connect line to line

            try:

                for line in connectedLines[:]:

                    matches = [it for it in connectedLines if (line != it) and (not detector.isParallel(line, it))]

                    for match in matches:

                        detector.connectLineToLine(match, line)

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

            # up to here

        lines = []

        for pts in connectedLines:

            processLine = QEngineeringLineItem()

            lines.append(processLine)

            for pt in pts:

                processLine._pol.append(QPointF(pt[0] + round(area.x), pt[1] + round(area.y)))

            processLine.buildItem()

            graphicsView.scene.addItem(processLine)

    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:

        listWidget.addItem('Finish Line searching')

'''

                humkyung 2018.05.26 add parameters(graphicsView, isSymbolTextChecked, isLineChecked)

def executeRecognition(path, listWidget, graphicsView, isSymbolTextChecked, isLineChecked):

            if not os.path.isfile(mainRes):

                item = QListWidgetItem('{} file is not found'.format(os.path.basename(mainRes)))

                item.setBackground(Qt.red)

                listWidget.addItem(item)

                continue

            #load original src & symbol

            src = cv2.imread(mainRes, 1)

            #gray scale

            if len(src.shape) == 3:

                srcGray = cvtGrayImage(src)

            else:

                srcGray = src.copy()

            srcGray = cv2.threshold(srcGray, 127, 255, cv2.THRESH_BINARY)[1]

            area = docData.getArea('Drawing')

                #TODO: 영역을 설정한 값과 이미지 좌표계를 차이를 보정

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

            if isSymbolTextChecked:

                initMainSrc(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, docData.imgWidth, docData.imgHeight = src.shape[::-1]

                drawFoundSymbolsOnCanvas(mainRes, docData.imgWidth, docData.imgHeight, listWidget)

                docData.imgName = 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)

                ####area = AppDocData.instance().getArea('Drawing')

                ####(_tempOcrSrc, tInfoList) = OCR.removeTextFromNpArray(area.img if area is not None else srcGray, area.x if area is not None else 0, area.y if area is not None else 0)

                #####(srcGray, tInfoList) = OCR.removeTextFromNpArray(area.img if area is not None else srcGray)

                ####if area is not None:

                ####    srcGray[int(area.y):int(area.y+area.height), int(area.x):int(area.x+area.width)] = _tempOcrSrc

                ####else:

                ####    srcGray = _tempOcrSrc

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

                global MIN_TEXT_SIZE

                for textInfo in textInfoList:

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

                    if textInfo.getW() >= MIN_TEXT_SIZE or textInfo.getH() >= MIN_TEXT_SIZE:

                        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)

                removedSymbolImgPath = os.path.join(project.getTempPath(), os.path.basename(path))

                cv2.imwrite(removedSymbolImgPath, srcGray)

                area = AppDocData.instance().getArea('Drawing')

                if area is not None:

                    area.img = srcGray[round(area.y+1):round(area.y+area.height), round(area.x+1):round(area.x+area.width)]

                cv2.imwrite(os.path.join(project.getTempPath(), "rect_" + os.path.basename(path)), src)

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

            if isLineChecked:

                recognizeLine(listWidget, graphicsView)

from QGraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

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

            self.xmlPath = executeRecognition(self.path, self.listWidget, self.graphicsView, self.isSymbolTextChecked, self.isLineChecked)

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

            self.startThread()

        self.obj.graphicsView = self.graphicsView

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

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

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

            self.xmlPath = self.obj.xmlPath

            self.loadRecognitionResult(self.xmlPath[0])

            self.ui.buttonBox.setEnabled(True)