HYTOS

import OcrResultDialog

    onSuccess = pyqtSignal(float, float, float, float)

                            rect = selectionBBox.toAlignedRect()

                            self.onSuccess.emit(rect.left(), rect.top(), rect.width(), rect.height())

    <Compile Include="OcrResultDialog.py" />

            cmd = AreaOcrCommand.AreaOcrCommand(self.graphicsView)

            cmd.onSuccess.connect(self.onRecognizeText)

            self.graphicsView.command = cmd

        @brief      show text recognition dialog

        @author     humkyung

        @date       2018.08.08

    '''

    def onRecognizeText(self, x, y, width, height):

        from OcrResultDialog import QOcrResultDialog

        try:

            image = self.graphicsView.image().copy(x, y, width, height)

            dialog = QOcrResultDialog(self.graphicsView, image, QRectF(x, y, width, height))

            (isAccept, textInfoList) = dialog.showDialog()

            if isAccept:

                if textInfoList is not None and len(textInfoList) > 0:

                    docData = AppDocData.instance()

                    configs = docData.getConfigs('Line No', 'Delimiter')

                    delimiter = configs[0].value if 1 == len(configs) else '-'

                    lineNoconfigs = docData.getConfigs('Line No', 'Configuration')

                    for textInfo in textInfoList:

                        x = textInfo.getX()

                        y = textInfo.getY()

                        angle = textInfo.getAngle()

                        text = textInfo.getText()

                        width = textInfo.getW()

                        height = textInfo.getH()

                        item = TextItemFactory.instance().createTextItem(text, delimiter, lineNoconfigs)

                        item.loc = (x, y)

                        item.size = (width, height)

                        item.angle = angle

                        item.setPlainText(text)

                        item.setDefaultTextColor(Qt.blue)

                        item.addTextItemToScene(self.imageViewer.scene)

                else:

                    QMessageBox.about(self.graphicsView, "알림", "텍스트 검출 실패")

        except Exception as ex:

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

            self.addMessage(MessageType.Error, message)

    '''

# coding: utf-8

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import UI_OcrResultDialog

import QtImageViewer

import tesseract_ocr_module as TOCR

from PIL import Image

import io

import numpy as np

import cv2

import math

from App import App

class QOcrResultDialog(QDialog):

    def __init__(self, parent, qimage, boundingBox, isModify = False, text = None):

        QDialog.__init__(self, parent)

        self.textInfoList = None

        self.isModify = isModify

        self.image = qimage

        self.originImageWidth = qimage.width()

        self.originImageHeight = qimage.height()

        self.boundingBox = boundingBox

        self.angle = 0 # angle is degree

        self.ui = UI_OcrResultDialog.Ui_Dialog()

        self.ui.setupUi(self)

        self.imgW = qimage.width()

        self.imgH = qimage.height()

        self.image = self.image.scaled(self.imgW, self.imgH)

        self.graphicsView = QtImageViewer.QtImageViewer(App.mainWnd())

        self.graphicsView.useDefaultCommand() ##### USE DEFAULT COMMAND

        self.graphicsView.setImage(self.image)

        self.ui.tVerticalLayout_2.addWidget(self.graphicsView)

        self.ui.counterClockPushButton_2.clicked.connect(lambda : self.rotateImage(True))

        self.ui.clockPushButton_2.clicked.connect(lambda : self.rotateImage(False))

        self.ui.redetectPushButton_2.clicked.connect(self.detectText)

        if self.isModify == False:

            self.detectText()

        else:

            self.ui.detectResultTextEdit_2.setPlainText(text)

        self.isAccepted = False

    def rotateImage(self, isCounterClock):

        for item in self.graphicsView.scene.items():

            self.graphicsView.scene.removeItem(item)

        self.graphicsView.clearImage()

        transform = QTransform()

        if isCounterClock:

            '''CounterClock'''

            self.angle = (self.angle - 90) % 360

            transform.rotate(-90)

        else:

            '''Clock'''

            self.angle = (self.angle - 270) % 360

            transform.rotate(90)

        print(str(360 - self.angle))

        self.image = self.image.transformed(transform)

        self.graphicsView.setImage(self.image)

    '''

        @history 2018.04.26 Jeongwoo Add Rectangle with modified Coords

                 2018.06.20 Jeongwoo Remove test code

    '''

    def detectText(self):

        try:

            buffer = QBuffer()

            buffer.open(QBuffer.ReadWrite)

            self.image.save(buffer, "PNG")

            pyImage = Image.open(io.BytesIO(buffer.data()))

            img = np.array(pyImage)

            ## exclude area bigger than one characeter size

            #minx = None

            #miny = None

            #maxx = None

            #maxy = None

            #img = cv2.threshold(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY), 127, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)[1]

            #img = cv2.erode(img, np.ones((2,2), np.uint8))

            #width, height = img.shape[::-1]

            #imgContour = np.ones(img.shape, np.uint8)*255

            ##cv2.rectangle(img, (0,0), (width - 1, height - 1), 255, 1)

            #imgNot = np.ones(img.shape, np.uint8)

            #cv2.bitwise_not(img, imgNot)

            #_, contours, hierarchy = cv2.findContours(imgNot, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

            #holes = [contours[i] for i in range(len(contours)) if hierarchy[0][i][3] >= 0]

            #for contour in contours:

            #    [x, y, w, h] = cv2.boundingRect(contour)

            #    # remove too big one

            #    if (w > 60 or h > 60): continue

            #    ratio = w/h

            #    if ratio > 4: continue

            #    area = cv2.contourArea(contour, True)

            #    if area > 0:

            #            cv2.drawContours(imgContour, [contour], -1, (255,255,255), -1)

            #    else:

            #        cv2.drawContours(imgContour, [contour], -1, (0,0,0), -1)

            #    if minx is None or x < minx: minx = x

            #    if miny is None or y < miny: miny = y

            #    if maxx is None or (x + w) > maxx: maxx = (x + w)

            #    if maxy is None or (y + h) > maxy: maxy = (y + h)

            #cv2.imshow('img', imgContour)

            #cv2.waitKey(0)

            #cv2.destroyAllWindows()

            # up to here

            self.textInfoList = TOCR.getTextInfo(img, (round(self.boundingBox.x()), round(self.boundingBox.y())))

            #self.textInfoList = TOCR.getTextInfo(img, (int(self.boundingBox.x()), int(self.boundingBox.y())))

            if self.textInfoList is not None and len(self.textInfoList) > 0:

                self.ui.detectResultTextEdit_2.setText(self.getPlainText(self.textInfoList))

                for textInfo in self.textInfoList:

                    self.graphicsView.scene.addRect(textInfo.getX()-int(self.boundingBox.x()), textInfo.getY()-int(self.boundingBox.y()), textInfo.getW(), textInfo.getH(), QPen(Qt.red, 1, Qt.SolidLine))

            else:

                self.ui.detectResultTextEdit_2.setText("Not Found")

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

    def getPlainText(self, textInfoList):

        text = ''

        for index in range(len(textInfoList)):

            textInfo = textInfoList[index]

            if index != 0:

                text = text + '\n'

            text = text + textInfo.getText()

        return text

    '''

        @brief      OK Button Clicked. Remake TextInfo object

        @author     Jeongwoo

        @date       18.04.19

        @history    18.04.20    Jeongwoo    Calculate Start Point Coordinates by rotated angle

                    18.04.26    Jeongwoo    Scene.itemAt(textX - boundBox.x(), textY - boundBox.y())

    '''

    def accept(self):

        self.isAccepted = True

        text = self.ui.detectResultTextEdit_2.toPlainText()

        splitText = text.split('\n')

        if self.textInfoList is not None and len(self.textInfoList) > 0:

            for index in range(len(splitText)):

                textInfo = self.textInfoList[index]

                item = self.graphicsView.scene.itemAt(QPointF(float(textInfo.getX() - int(self.boundingBox.x())), float(textInfo.getY()-int(self.boundingBox.y()))), QTransform())

                if item is not None:

                    ## Transform rectangle for calculate start point

                    imgTransform = QTransform()

                    if self.angle == 90 or self.angle == 270:

                        imgTransform.translate(self.image.height()*0.5, self.image.width()*0.5)

                    elif self.angle == 0 or self.angle == 360:

                        imgTransform.translate(self.image.width()*0.5, self.image.height()*0.5)

                    imgTransform.rotate(-abs(self.angle))

                    imgTransform.translate(-self.image.width()*0.5, -self.image.height()*0.5)

                    rect = QRect(textInfo.getX() - int(self.boundingBox.x()), textInfo.getY() - int(self.boundingBox.y()), textInfo.getW(), textInfo.getH())

                    rect = imgTransform.mapRect(rect)

                    ## up to here

                    textInfo.setX(rect.x() + int(self.boundingBox.x()))

                    textInfo.setY(rect.y() + int(self.boundingBox.y()))

                    textInfo.setText(splitText[index])

                    radian = round(math.radians(abs(self.angle)), 2)

                    textInfo.setAngle(radian) # 360 degree == 6.28319 radian

                    if radian == 1.57 or radian == 4.71:

                        width = textInfo.getW()

                        height = textInfo.getH()

                        textInfo.setW(height) ## SWAP

                        textInfo.setH(width) ## SWAP

            QDialog.accept(self)

        else:

            QMessageBox.about(self.ui.ocrDialogButtonBox, "알림", "텍스트 검출을 하신 후 다시 시도해주세요.")

    def reject(self):

        self.isAccepted = False

        self.textInfoList = None

        QDialog.reject(self)

    '''

        @brief  Display this QDialog

    '''

    def showDialog(self):

        self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)

        #self.show()

        self.exec_()

        return (self.isAccepted, self.textInfoList)

12) OcrResultDialog.py

        recognizeTextAction = QAction("텍스트 인식")

        recognizeTextAction.triggered.connect(lambda: self.onRecognizeTextEvent())

        menu.addAction(recognizeTextAction)

            App.mainWnd().onAreaSelected(round(rect.left()), round(rect.top()), round(rect.width()), round(rect.height()))

            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(MessageType.Error, message)

    '''

        @brief      show text recognition dialog

        @author     humkyung

        @date       2018.08.08

    '''

    def onRecognizeTextEvent(self):

        from App import App

        try:

            selected = [item for item in self.scene().items() if item.isSelected() and type(item) is QEngineeringUnknownItem]

            rect = QRectF()

            for item in selected:

                rect = rect.united(item.sceneBoundingRect())

            App.mainWnd().onRecognizeText(round(rect.left()), round(rect.top()), round(rect.width()), round(rect.height()))

        except Exception as ex:

from OcrResultDialog import QOcrResultDialog

import OcrResultDialog

import OcrResultDialog

import OcrResultDialog

from OcrResultDialog import QOcrResultDialog

            dialog = QOcrResultDialog(None, AppDocData.instance().getCurrentPidSource().getQImageOnRect(QRect(self.loc[0], self.loc[1], self.size[0]+1, self.size[1]))

import OcrResultDialog