HYTOS

from AppDocData import AppDocData

    import cv2

from PIL import PngImagePlugin, JpegImagePlugin

from PIL import Image

from PIL.ImageQt import ImageQt

        path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

        path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

            path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

        path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

            path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

        path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

        return os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID', 'Explode.svg')

import os.path

import AbstractCommand

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

    from PyQt5.QtWidgets import *

except ImportError:

    try:

        from PyQt4.QtCore import Qt, QPoint, QPointF, QRectF, pyqtSignal, QT_VERSION_STR, QBuffer, QRect, QRegExp

        from PyQt4.QtGui import QGraphicsView, QGraphicsScene, QImage, QPixmap, QPainterPath, QFileDialog, QPainter, QColor, QPen, QBrush, QCursor, QTransform, QFont, QRegExpValidator, QValidator

    except ImportError:

        raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.")

import OcrResultDialog

import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.realpath(__file__))) + '\\Shapes')

from EngineeringTextItem import QEngineeringTextItem

from EngineeringLineNoTextItem import QEngineeringLineNoTextItem

from EngineeringNoteItem import QEngineeringNoteItem

from TextItemFactory import TextItemFactory

from AppDocData import AppDocData

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

class AreaOcrCommand(AbstractCommand.AbstractCommand):

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

    onRejected = pyqtSignal(AbstractCommand.AbstractCommand)

    def __init__(self, imageViewer):

        super(AreaOcrCommand, self).__init__(imageViewer)

        self.name = 'AreaOcr' 

        self.imageViewer.setCursor(QCursor(Qt.CrossCursor))

        self._rubberBand = QRubberBand(QRubberBand.Rectangle, self.imageViewer)

        self._origin = QPoint()

        self.isLeftClicked = False

    '''

        @brief      pan image by left click and drag

        @author     Jeongwoo

        @date       18.04.19

        @history    18.04.20    Jeongwoo    Transform text box

                    humkyung 2018.04.20 set item's location with selection box

                    Jeongwoo 2018.04.25 Add if state with QEngineeringNoteItem

                    Jeongwoo 2018.04.26 Change method to create TextItem with TextItemFactory

                    Jeongwoo 2018.05.02 Add Checking imageViewer has image if-statement, when mouseReleaseEvent happen

                    humkyung 2018.08.29 use rubberband for selection text area

    '''

    def execute(self, param):

        event = param[1]

        scenePos = param[2]

        if 'mousePressEvent' == param[0]:

            if event.button() == Qt.LeftButton:

                self.isLeftClicked = True

                self._origin = event.pos()

                self._rubberBand.setGeometry(QRect(self._origin, QSize()))

                self._rubberBand.show()

            self.imageViewer.leftMouseButtonPressed.emit(scenePos.x(), scenePos.y())

        elif 'mouseMoveEvent' == param[0] and event.buttons() == Qt.LeftButton:

            if self._rubberBand.isVisible():

                self._rubberBand.setGeometry(QRect(self._origin, event.pos()).normalized())

        elif 'mouseReleaseEvent' == param[0]:

            QGraphicsView.mouseReleaseEvent(self.imageViewer, event)

            try:

                if event.button() == Qt.LeftButton:

                    self.isLeftClicked = False

                if self.imageViewer.canZoom and event.button() == Qt.LeftButton:

                    self._rubberBand.hide()

                    topLeft = self.imageViewer.mapToScene(self._rubberBand.geometry().topLeft())

                    bottomRight = self.imageViewer.mapToScene(self._rubberBand.geometry().bottomRight())

                    rect = QRectF(topLeft, bottomRight)

                    if rect.isValid():

                        if self.imageViewer.hasImage():

                            rect = rect.toAlignedRect()

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

                        else:

                            QMessageBox.about(self.imageViewer, self.tr('Notice'), self.tr('Please check the image.'))

                elif event.button() == Qt.RightButton:

                    if self.isLeftClicked == False:

                        self.onRejected.emit(self)

            finally:

                pass

        self.isTreated = False

    def undo(self):

        pass

    def redo(self):

        pass

import os.path

import AbstractCommand

try:

    from PyQt5.QtCore import Qt, QPoint, QPointF, QRectF, pyqtSignal, QT_VERSION_STR, QBuffer, QRect, QRegExp

    from PyQt5.QtGui import QImage, QPixmap, QPainterPath, QPainter, QColor, QPen, QBrush, QCursor, QTransform, QFont, QRegExpValidator, QValidator

    from PyQt5.QtWidgets import QGraphicsView, QGraphicsScene, QFileDialog, QMessageBox

except ImportError:

    try:

        from PyQt4.QtCore import Qt, QPoint, QPointF, QRectF, pyqtSignal, QT_VERSION_STR, QBuffer, QRect, QRegExp

        from PyQt4.QtGui import QGraphicsView, QGraphicsScene, QImage, QPixmap, QPainterPath, QFileDialog, QPainter, QColor, QPen, QBrush, QCursor, QTransform, QFont, QRegExpValidator, QValidator

    except ImportError:

        raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.")

import sys

from AppDocData import AppDocData

from TextDetector import TextDetector

import numpy as np

from PIL import Image

import io

import cv2

class RemoveTextCommand(AbstractCommand.AbstractCommand):

    '''

        @history    2018.05.09  Jeongwoo    Draw Rect on ±1 area

                    2018.06.11  Jeongwoo    Change method to set image (setImage → setPixmap/updateViewer)

    '''

    def __init__(self, imageViewer):

        super(RemoveTextCommand, self).__init__(imageViewer)

        self.name = 'RemoveText' 

        #self.imageViewer.setCursor(QCursor(Qt.ArrowCursor))

        image = self.imageViewer.image()

        buffer = QBuffer()

        buffer.open(QBuffer.ReadWrite)

        image.save(buffer, "PNG")

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

        dst = cv2.cvtColor(np.array(pyImage), cv2.COLOR_BGR2GRAY)

        textDetector = TextDetector()

        self.textInfoList = textDetector.detectTextAreas(dst, (0, 0))

        pixmap = self.imageViewer.pixmap()

        ADJUST = 1

        for textInfo in self.textInfoList:

            painter = QPainter()

            painter.begin(pixmap)

            painter.setPen(QColor(255, 255, 255))

            painter.setBrush(QColor(255, 255, 255))

            painter.drawRect(QRect(textInfo.getX()-ADJUST, textInfo.getY()-ADJUST, textInfo.getW()+ADJUST, textInfo.getH()+ADJUST))

            painter.end()

        pixmapHandle = self.imageViewer.getPixmapHandle()

        if pixmapHandle is not None:

            pixmapHandle.setPixmap(pixmap)

            self.imageViewer.setSceneRect(QRectF(pixmap.rect()))

            self.imageViewer.updateViewer()

    def execute(self, param):

        event = param[1]

        scenePos = param[2]

        self.isTreated = False

    def undo(self):

        pass

    def redo(self):

        pass

        tessdata_path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID', 'Tesseract-OCR', 'tessdata')

from datetime import datetime

        self.log_path = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID', 'id2.log')

        logging.basicConfig(filename=self.log_path , filemode='a', level=logging.CRITICAL)

        self.logger.critical(datetime.now().strftime('%Y-%m-%d %H:%M:%S'))

import AreaOcrCommand

from TrainingImageListDialog import QTrainingImageListDialog

# coding: utf-8

"""

    This is ocr result dialog module

"""

from PIL import Image

import io

import numpy as np

import math

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import OcrResultDialog_UI

import QtImageViewer

import tesseract_ocr_module as TOCR

from App import App

from AppDocData import *

class QOcrResultDialog(QDialog):

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

        QDialog.__init__(self, parent)

        self.textInfoList = []

        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 = OcrResultDialog_UI.Ui_Dialog()

        self.ui.setupUi(self)

        appDocData = AppDocData.instance()

        configs = appDocData.getAppConfigs('app', 'mode')

        if configs and 1 == len(configs) and 'advanced' == configs[0].value:

            pass

        else:

            self.ui.pushButtonMakeTrainingImage.setVisible(False)

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

        self.ui.pushButtonMakeTrainingImage.clicked.connect(self.pushButtonMakeTrainingImageClicked)

        if self.isModify == False:

            self.detectText()

        else:

            self.ui.detectResultTextEdit.setPlainText(text)

        self.isAccepted = False

    '''

        @brief      Make OCR Training Image

        @author     euisung

        @date       2018.10.16

        @history    euisung     2018.11.02       add notice push

    '''

    def pushButtonMakeTrainingImageClicked(self):

        import uuid

        uid = str(uuid.uuid4()) + '.png'

        appDocData = AppDocData.instance()

        project = appDocData.getCurrentProject()

        trainingImgPath = os.path.join(project.getTrainingFilePath(), uid)

        self.image.save(trainingImgPath)

        QMessageBox.about(self, self.tr("INFO"), self.tr('Successfully saved.'))

        QDialog.reject(self)

    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)

        self.textInfoList = []

    '''

        @history 2018.04.26 Jeongwoo    Add Rectangle with modified Coords

                 2018.06.20 Jeongwoo    Remove test code

                 2018.11.08 euisung     add white char list check process on db

                 2018.11.22 euisung     OCR lang apply fixed

    '''

    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)

            #self.image.save('c:\\temp\\a.png')

            docData = AppDocData.instance()

            # get ocr data of area which has the text

            pt = self.boundingBox.center()

            areas = [area for area in docData.getAreaList() if area.contains((pt.x(), pt.y()))]

            ocr_data = sorted(areas, key=lambda attr: attr.area)[0].OCRData if areas else 'eng'

            # up to here

            whiteCharList = docData.getConfigs('Text Recognition', 'White Character List')

            if len(whiteCharList) is 0:

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

            else:

                self.textInfoList = TOCR.getTextInfo(img, (round(self.boundingBox.x()), round(self.boundingBox.y())), 0, language=ocr_data, conf = whiteCharList[0].value)

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

                self.ui.detectResultTextEdit.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.setText("Not Found")

        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)

    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

        try:

            text = self.ui.detectResultTextEdit.toPlainText()

            if text == '' or text == 'Not Found':

                QMessageBox.about(self.ui.ocrDialogButtonBox, 'Notice', 'Please try again after recognition or type.')

                return

            splitText = text.split('\n')

            if not len(self.textInfoList) > 0:

                import cv2, sys

                from TextInfo import TextInfo

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

                buffer = QBuffer()

                buffer.open(QBuffer.ReadWrite)

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

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

                img = np.array(pyImage)

                img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

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

                cv2.bitwise_not(img, imgNot)

                imgNot = cv2.dilate(imgNot, np.ones((8,8), np.uint8))

                image, contours, hierarchy = cv2.findContours(imgNot, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

                minX, minY, maxX, maxY = sys.maxsize, sys.maxsize, 0 ,0

                if len(contours) is 0:

                    minX, minY, maxX, maxY = self.boundingBox.x(), self.boundingBox.y(), self.boundingBox.x() + self.image.width(), self.boundingBox.y() + self.image.height()

                else:

                    minX, minY, maxX, maxY = sys.maxsize, sys.maxsize, 0, 0

                    for cnt in contours:

                        x, y, w, h = cv2.boundingRect(cnt)

                        minX = min(x ,minX)

                        minY = min(y, minY)

                        maxX = max(x + w, maxX)

                        maxY = max(y + h, maxY)

                    minX, minY, maxX, maxY = minX + self.boundingBox.x(), minY + self.boundingBox.y(), maxX + self.boundingBox.x(), maxY + self.boundingBox.y()

                self.textInfoList.append(TextInfo(text, minX, minY, maxX - minX, maxY - minY, 0))

            if 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

                self.textInfoList = self.textInfoList[:len(splitText)]

                QDialog.accept(self)

        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)

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

        return (self.isAccepted, self.textInfoList)

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file '.\UI\OcrResultDialog.ui'

#

# Created by: PyQt5 UI code generator 5.11.3

#

# WARNING! All changes made in this file will be lost!

from PyQt5 import QtCore, QtGui, QtWidgets

class Ui_Dialog(object):

    def setupUi(self, Dialog):

        Dialog.setObjectName("Dialog")

        Dialog.resize(1080, 650)

        Dialog.setMinimumSize(QtCore.QSize(1080, 650))

        font = QtGui.QFont()

        font.setFamily("맑은 고딕")

        font.setBold(True)

        font.setWeight(75)

        Dialog.setFont(font)

        self.gridLayout_2 = QtWidgets.QGridLayout(Dialog)

        self.gridLayout_2.setObjectName("gridLayout_2")

        self.splitter = QtWidgets.QSplitter(Dialog)

        self.splitter.setOrientation(QtCore.Qt.Vertical)

        self.splitter.setObjectName("splitter")

        self.topWidget = QtWidgets.QWidget(self.splitter)

        self.topWidget.setMinimumSize(QtCore.QSize(0, 150))

        self.topWidget.setObjectName("topWidget")

        self.verticalLayout_2 = QtWidgets.QVBoxLayout(self.topWidget)

        self.verticalLayout_2.setContentsMargins(0, 0, 0, 0)

        self.verticalLayout_2.setObjectName("verticalLayout_2")

        self.tVerticalLayout_2 = QtWidgets.QVBoxLayout()

        self.tVerticalLayout_2.setObjectName("tVerticalLayout_2")

        self.horizontalLayout = QtWidgets.QHBoxLayout()

        self.horizontalLayout.setSizeConstraint(QtWidgets.QLayout.SetFixedSize)

        self.horizontalLayout.setContentsMargins(8, 8, 8, 8)

        self.horizontalLayout.setObjectName("horizontalLayout")

        self.targetTextLabel_2 = QtWidgets.QLabel(self.topWidget)

        self.targetTextLabel_2.setMinimumSize(QtCore.QSize(0, 60))

        self.targetTextLabel_2.setMaximumSize(QtCore.QSize(16777215, 60))

        font = QtGui.QFont()

        font.setBold(True)

        font.setWeight(75)

        self.targetTextLabel_2.setFont(font)

        self.targetTextLabel_2.setObjectName("targetTextLabel_2")

        self.horizontalLayout.addWidget(self.targetTextLabel_2)

        self.counterClockPushButton_2 = QtWidgets.QPushButton(self.topWidget)

        self.counterClockPushButton_2.setMinimumSize(QtCore.QSize(0, 60))

        self.counterClockPushButton_2.setMaximumSize(QtCore.QSize(60, 60))

        self.counterClockPushButton_2.setText("")

        icon = QtGui.QIcon()

        icon.addPixmap(QtGui.QPixmap(":/newPrefix/Rotate_Minus.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.counterClockPushButton_2.setIcon(icon)

        self.counterClockPushButton_2.setIconSize(QtCore.QSize(32, 32))

        self.counterClockPushButton_2.setObjectName("counterClockPushButton_2")

        self.horizontalLayout.addWidget(self.counterClockPushButton_2)

        self.clockPushButton_2 = QtWidgets.QPushButton(self.topWidget)

        self.clockPushButton_2.setMinimumSize(QtCore.QSize(0, 60))

        self.clockPushButton_2.setMaximumSize(QtCore.QSize(60, 60))

        self.clockPushButton_2.setText("")

        icon1 = QtGui.QIcon()

        icon1.addPixmap(QtGui.QPixmap(":/newPrefix/Rotate_Plus.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.clockPushButton_2.setIcon(icon1)

        self.clockPushButton_2.setIconSize(QtCore.QSize(32, 32))

        self.clockPushButton_2.setObjectName("clockPushButton_2")

        self.horizontalLayout.addWidget(self.clockPushButton_2)

        self.redetectPushButton_2 = QtWidgets.QPushButton(self.topWidget)

        self.redetectPushButton_2.setMinimumSize(QtCore.QSize(0, 60))

        self.redetectPushButton_2.setMaximumSize(QtCore.QSize(60, 60))

        self.redetectPushButton_2.setText("")

        icon2 = QtGui.QIcon()

        icon2.addPixmap(QtGui.QPixmap(":/newPrefix/OCR.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.redetectPushButton_2.setIcon(icon2)

        self.redetectPushButton_2.setIconSize(QtCore.QSize(32, 32))

        self.redetectPushButton_2.setObjectName("redetectPushButton_2")

        self.horizontalLayout.addWidget(self.redetectPushButton_2)

        self.tVerticalLayout_2.addLayout(self.horizontalLayout)

        self.horizontalLayoutGraphicsView = QtWidgets.QHBoxLayout()

        self.horizontalLayoutGraphicsView.setObjectName("horizontalLayoutGraphicsView")

        self.tVerticalLayout_2.addLayout(self.horizontalLayoutGraphicsView)

        self.verticalLayout_2.addLayout(self.tVerticalLayout_2)

        self.bottomWidget = QtWidgets.QWidget(self.splitter)

        self.bottomWidget.setMinimumSize(QtCore.QSize(0, 150))

        self.bottomWidget.setObjectName("bottomWidget")

        self.verticalLayout_4 = QtWidgets.QVBoxLayout(self.bottomWidget)

        self.verticalLayout_4.setContentsMargins(0, 0, 0, 0)

        self.verticalLayout_4.setObjectName("verticalLayout_4")

        self.detectResultVerticalLayout_2 = QtWidgets.QVBoxLayout()

        self.detectResultVerticalLayout_2.setContentsMargins(8, 8, 8, 8)

        self.detectResultVerticalLayout_2.setObjectName("detectResultVerticalLayout_2")

        self.detectResultLabel_2 = QtWidgets.QLabel(self.bottomWidget)

        font = QtGui.QFont()

        font.setBold(True)

        font.setWeight(75)

        self.detectResultLabel_2.setFont(font)

        self.detectResultLabel_2.setObjectName("detectResultLabel_2")

        self.detectResultVerticalLayout_2.addWidget(self.detectResultLabel_2)

        self.detectResultTextEdit = QtWidgets.QTextEdit(self.bottomWidget)

        font = QtGui.QFont()

        font.setFamily("Consolas")

        font.setPointSize(15)

        self.detectResultTextEdit.setFont(font)

        self.detectResultTextEdit.setObjectName("detectResultTextEdit")

        self.detectResultVerticalLayout_2.addWidget(self.detectResultTextEdit)

        self.verticalLayout_4.addLayout(self.detectResultVerticalLayout_2)

        self.gridLayout_2.addWidget(self.splitter, 1, 0, 1, 1)

        self.horizontalLayout_2 = QtWidgets.QHBoxLayout()

        self.horizontalLayout_2.setObjectName("horizontalLayout_2")

        self.pushButtonMakeTrainingImage = QtWidgets.QPushButton(Dialog)

        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Fixed)

        sizePolicy.setHorizontalStretch(0)

        sizePolicy.setVerticalStretch(0)

        sizePolicy.setHeightForWidth(self.pushButtonMakeTrainingImage.sizePolicy().hasHeightForWidth())

        self.pushButtonMakeTrainingImage.setSizePolicy(sizePolicy)

        self.pushButtonMakeTrainingImage.setObjectName("pushButtonMakeTrainingImage")

        self.horizontalLayout_2.addWidget(self.pushButtonMakeTrainingImage)

        self.ocrDialogButtonBox = QtWidgets.QDialogButtonBox(Dialog)

        self.ocrDialogButtonBox.setOrientation(QtCore.Qt.Horizontal)

        self.ocrDialogButtonBox.setStandardButtons(QtWidgets.QDialogButtonBox.Cancel|QtWidgets.QDialogButtonBox.Ok)

        self.ocrDialogButtonBox.setObjectName("ocrDialogButtonBox")

        self.horizontalLayout_2.addWidget(self.ocrDialogButtonBox)

        self.gridLayout_2.addLayout(self.horizontalLayout_2, 3, 0, 1, 1)

        self.retranslateUi(Dialog)

        self.ocrDialogButtonBox.accepted.connect(Dialog.accept)

        self.ocrDialogButtonBox.rejected.connect(Dialog.reject)

        QtCore.QMetaObject.connectSlotsByName(Dialog)

    def retranslateUi(self, Dialog):

        _translate = QtCore.QCoreApplication.translate

        Dialog.setWindowTitle(_translate("Dialog", "OCR"))

        self.targetTextLabel_2.setText(_translate("Dialog", "Recognition Object"))

        self.counterClockPushButton_2.setToolTip(_translate("Dialog", "반시계 방향 회전"))

        self.clockPushButton_2.setToolTip(_translate("Dialog", "시계 방향 회전"))

        self.detectResultLabel_2.setText(_translate("Dialog", "Recognition Result"))

        self.pushButtonMakeTrainingImage.setText(_translate("Dialog", "Save OCR Training Image"))

import MainWindow_rc

if __name__ == "__main__":

    import sys

    app = QtWidgets.QApplication(sys.argv)

    Dialog = QtWidgets.QDialog()

    ui = Ui_Dialog()

    ui.setupUi(Dialog)

    Dialog.show()

    sys.exit(app.exec_())

import cv2

from OcrResultDialog import QOcrResultDialog

import OcrResultDialog

import OcrResultDialog

import OcrResultDialog

import OcrResultDialog

import OcrResultDialog

import CropCommand, HandCommand, ZoomCommand, PenCommand, EraserCommand, AreaEraserCommand, OriginalPointCommand, ConnectionPointCommand, AreaZoomCommand, FitImageCommand, RemoveTextCommand, RotateImageCommand, FlipImageCommand

# coding: utf-8

"""

    This is text detector module

"""

import sys

import os

import cv2

import numpy as np

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from PyQt5.QtSvg import *

from AppDocData import *

import TextInfo as ti

import tesseract_ocr_module as TOCR

MIN_TEXT_SIZE = 10

THREAD_MAX_WORKER = os.cpu_count()

class TextDetector:

    '''

        @brief  constructor

        @author humkyung

        @date   2018.07.11

    '''

    def __init__(self):

        self.textInfoList = []

        self.otherTextInfoList = []

        self.titleBlockTextInfoList = []

    '''

        @brief  detect text areas

        @author humkyung

        @date   2018.06.16

    '''

    def detectTextAreas(self, img, offset):

        tInfoList = []

        try:

            tInfoList = self.getTextAreaInfo(img, offset[0], offset[1])

        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 tInfoList

    '''

        @brief      Get Text Area info by contour

        @author     Jeongwoo

        @date       2018.06.05

        @history    2018.06.08  Jeongwoo    Add angle

                    humkyung 2018.06.18 fixed logic to detect text area

    '''

    def getTextAreaInfo(self, imgGray, offsetX, offsetY):

        from AppDocData import AppDocData

        appDocData = AppDocData.instance()

        project = appDocData.getCurrentProject()

        configs = appDocData.getConfigs('Text Size', 'Max Text Size')

        maxTextSize = int(configs[0].value) if 1 == len(configs) else 100

        minSize = 5

        contourImg = np.ones(imgGray.shape, np.uint8) * 255

        binaryImg,mask = cv2.threshold(imgGray, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)

        image, contours, hierarchy = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)

        for contour in contours:

            # remove too big one or horizontal/vertical line

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

            area = cv2.contourArea(contour, True)

            # skip one which size is greater than max size or less then minimum size

            if area >= 0:

                if (w > maxTextSize or h > maxTextSize) or (w <= minSize and h <= minSize): continue

            if area >= 0:

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

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

            else:

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

        path = os.path.join(project.getTempPath(), 'OCR_{}.png'.format(appDocData.imgName))

        cv2.imwrite(path, contourImg)

        rects = []

        configs = appDocData.getConfigs('Text Recognition', 'Expand Size')

        expandSize = int(configs[0].value) if 1 == len(configs) else 10

        configs = appDocData.getConfigs('Text Recognition', 'Shrink Size')

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

        eroded = cv2.erode(contourImg, np.ones((expandSize,expandSize), np.uint8))

        #path = os.path.join(project.getTempPath(), 'ERODED_OCR_{}.png'.format(appDocData.imgName))

        #cv2.imwrite(path, eroded)

        eroded = cv2.bitwise_not(eroded)

        image, contours, hierarchy = cv2.findContours(eroded, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

        for contour in contours:

            area = cv2.contourArea(contour, True)

            if area < 0:

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

                img = contourImg[y:y+h, x:x+w]

                img = cv2.bitwise_not(img)

                horizontal,max_width = 0,0

                vertical,max_height = 0,0

                _, _contours, _ = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

                for xx in _contours:

                    [_x, _y, _w, _h] = cv2.boundingRect(xx)

                    cv2.rectangle(img, (_x, _y), (_x+_w, _y+_h), 255, 1)

                    max_width = _x if _x > max_width else max_width

                    max_height = _y if _y > max_height else max_height

                    if (_w < _h) or (_w > maxTextSize and _h < maxTextSize): # width is greater than height

                        horizontal += 1 + (_w*_h)/(w*h)

                    else:

                        vertical += 1 + (_w*_h)/(w*h)

                if (w < 10 and h < 10) or (max_width > maxTextSize and max_height > maxTextSize): continue; # skip too small or big one

                """

                if w > maxTextSize:

                    horizontal = 1

                elif h > maxTextSize:

                    vertical = 1

                else:

                    if shrinkSize > 0:

                        img = cv2.erode(img, np.ones((shrinkSize,shrinkSize), np.uint8))

                    _, _contours, _ = cv2.findContours(img, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)

                    for xx in _contours:

                        [_x, _y, _w, _h] = cv2.boundingRect(xx)

                        cv2.rectangle(img, (_x, _y), (_x+_w, _y+_h), 255, 1)

                        if (_w < _h) or (_w > maxTextSize and _h < maxTextSize): # width is greater than height

                            horizontal += 1 + (_w*_h)/(w*h)

                        else:

                            vertical += 1 + (_w*_h)/(w*h)

                """

                """

                if horizontal > vertical:

                    filePath = os.path.join(project.getTempPath(), "Tile", "H-{}-{}-{}-{}.png".format(x,y,w,h))

                else:

                    filePath = os.path.join(project.getTempPath(), "Tile", "V-{}-{}-{}-{}.png".format(x,y,w,h))

                cv2.imwrite(filePath, img)

                """

                rects.append([0 if horizontal > vertical else 90, QRect(x, y, w, h)])

        configs = appDocData.getConfigs('Text Recognition', 'Merge Size')

        mergeSize = int(configs[0].value) if 1 == len(configs) else 10

        # merge rectangles

        intersected = True

        while intersected:

            intersected = False

            for rect in rects[:]:   # clone rects

                if 0 == rect[0]:

                    rectExpand = rect[1].adjusted(-mergeSize, 0, mergeSize, 0)

                else:

                    rectExpand = rect[1].adjusted(0, -mergeSize, 0, mergeSize)

                matches = [x for x in rects if (x[0] == rect[0]) and rectExpand.intersects(x[1])]

                if len(matches) > 1:

                    united = matches[0]

                    for _rect in matches:

                        united[1] = united[1].united(_rect[1])

                        if _rect in rects: rects.remove(_rect)

                    rects.append(united)

                    intersected = True

                    break

        list = []

        for rect in rects:

            angle = rect[0]

            list.append(ti.TextInfo('', round(offsetX) + rect[1].x(), round(offsetY) + rect[1].y(), rect[1].width(), rect[1].height(), angle))

            x = rect[1].x()

            y = rect[1].y()

            w = rect[1].width()

            h = rect[1].height()

            img = contourImg[y:y+h, x:x+w]

            ## DEBUG

            #if angle == 0:

            #    filePath = os.path.join(project.getTempPath(), "Tile", "H-{}-{}-{}-{}.png".format(x,y,w,h))

            #else:

            #    filePath = os.path.join(project.getTempPath(), "Tile", "V-{}-{}-{}-{}.png".format(x,y,w,h))

            #cv2.imwrite(filePath, img)

            ## up to here

        return list

    '''

        @brief      recognize text of given text info

        @author     humkyung

        @date       2018.07.24

        @history    change parameter updateProgressSignal to worker

                    2018.11.08 euisung     add white char list check process on db

    '''

    @staticmethod

    def recognizeTextFromImage(tInfo, imgOCR, offset, searchedSymbolList, worker, listWidget, maxProgressValue):

        import re

        res = []

        appDocData = AppDocData.instance()

        try:

            x = tInfo.getX() - round(offset[0])

            y = tInfo.getY() - round(offset[1])

            img = imgOCR[y:y+tInfo.getH(), x:x+tInfo.getW()]

            # set angle 0 if symbol contains the text area is instrumentation

            category = None

            contains = [symbol for symbol in searchedSymbolList if symbol.contains(tInfo)]

            if contains:

                _type = contains[0].getType()

                category = appDocData.getSymbolCategoryByType(_type)

                if 'Instrumentation' == category: tInfo.setAngle(0)

            # up to here

            whiteCharList = appDocData.getConfigs('Text Recognition', 'White Character List')

            if len(whiteCharList) is 0:

                resultTextInfo = TOCR.getTextInfo(img, (x, y), tInfo.getAngle(), language=appDocData.OCRData)

            else:

                resultTextInfo = TOCR.getTextInfo(img, (x, y), tInfo.getAngle(), language=appDocData.OCRData, conf = whiteCharList[0].value)

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

                for result in resultTextInfo:

                    result.setX(result.getX() + round(offset[0]))

                    result.setY(result.getY() + round(offset[1]))

                    if 'Instrumentation' == category:

                        text = re.sub('[^a-zA-Z0-9]+', '', result.getText())

                        result.setText(text)

                res.extend(resultTextInfo)

                if listWidget is not None:

                    item = QListWidgetItem('{},{},{} is recognized'.format(resultTextInfo[0].getX(), resultTextInfo[0].getY(), resultTextInfo[0].getText()))

                    listWidget.addItem(item)

            else:

                pass

            if worker is not None: worker.updateProgress.emit(maxProgressValue, resultTextInfo[0].getText() if resultTextInfo is not None and 1 == len(resultTextInfo) else None)

        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)

            worker.displayLog.emit(MessageType.Error, message)

        return res

    '''

        @brief      read image drawing and then remove text

        @author     jwkim

        @date       

        @history    humkyung 2018.04.06 check if file exists

                    Jeongwoo 2018.05.09 Use Tesseract OCR after Azure OCR (Azure OCR : Getting text area)

                    Jeongwoo 2018.05.25 Add condition on if-statement

                    Jeongwoo 2018.06.05 Get text area data list by config.type

                    Jeongwoo 2018.06.08 Add angle Parameter on TOCR.getTextInfo

                    humkyung 2018.06.16 update proessbar while recognizing text

                    humkyung 2018.07.03 remove white space and replace given oldStr with newStr

                    humkyung 2018.07.07 change method name to recognizeText

                    euisung  2018.11.08 add white char list check process on db

                    euisung  2018.11.12 add title block properties

    '''

    def recognizeText(self, imgSrc, offset, tInfoList, searchedSymbolList, worker, listWidget, maxProgressValue, onlyTextArea = False):

        import concurrent.futures as futures

        from Area import Area

        try:

            self.otherTextInfoList = []

            self.titleBlockTextInfoList = []

            self.textInfoList = []

            appDocData = AppDocData.instance()

            project = appDocData.getCurrentProject()

            path = os.path.join(project.getTempPath(), 'OCR_{}.png'.format(appDocData.imgName))

            if os.path.isfile(path):

                imgOCR = cv2.imread(path, 1)

                imgOCR = cv2.threshold(cv2.cvtColor(imgOCR, cv2.COLOR_BGR2GRAY), 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)[1]

                pool = futures.ThreadPoolExecutor(max_workers = THREAD_MAX_WORKER)

                for tInfo in tInfoList:

                    future = pool.submit(TextDetector.recognizeTextFromImage, tInfo, imgOCR, offset, searchedSymbolList, worker, listWidget, maxProgressValue)

                    data = future.result()

                    if data: self.textInfoList.extend(data)

                pool.shutdown(wait = True)

                if onlyTextArea:

                    return

                # parse texts in area except Drawing area

                whiteCharList = appDocData.getConfigs('Text Recognition', 'White Character List')

                for area in appDocData.getAreaList():

                    if area.name == 'Drawing': continue

                    if area.name == 'Unit':

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

                        if len(whiteCharList) is 0:

                            texts = TOCR.getTextInfo(img, (area.x, area.y), 0, language='eng')

                        else:

                            texts = TOCR.getTextInfo(img, (area.x, area.y), 0, language='eng', conf = whiteCharList[0].value)

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

                            appDocData.activeDrawing.setAttr('Unit', texts[0].getText())

                            self.otherTextInfoList.append([area.name, texts])

                    else:

                        if area is not None and hasattr(area, 'img') and area.img is not None:

                            if len(whiteCharList) is 0:

                                texts = TOCR.getTextInfo(area.img, (area.x, area.y), 0, language='eng')

                            else:

                                texts = TOCR.getTextInfo(area.img, (area.x, area.y), 0, language='eng', conf=whiteCharList[0].value)

                            self.otherTextInfoList.append([area.name, texts])

                titleBlockProps = appDocData.getTitleBlockProperties()

                for titleBlockProp in titleBlockProps:

                    area = Area(titleBlockProp[0])

                    area.parse(titleBlockProp[2])

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

                    if len(whiteCharList) is 0:

                        texts = TOCR.getTextInfo(img, (area.x, area.y), 0, language=appDocData.OCRData)

                    else:

                        texts = TOCR.getTextInfo(img, (area.x, area.y), 0, language='eng', conf=whiteCharList[0].value)

                    self.titleBlockTextInfoList.append([area.name, texts])

                if worker is not None: worker.updateProgress.emit(maxProgressValue, None)

        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)

            worker.displayLog.emit(MessageType.Error, message)

    '''

        @brief      remove text from image

        @author     humkyung

        @date       2018.07.24

    '''

    def removeTextFromImage(self, imgSrc, offset):

        appDocData = AppDocData.instance()

        project = appDocData.getCurrentProject()

        path = os.path.join(project.getTempPath(), 'OCR_{}.png'.format(appDocData.imgName))

        if os.path.isfile(path):

            imgOCR = cv2.imread(path)

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

            # remove recognized text from image

            for text in self.textInfoList:

                x = round(text.getX() - offset[0])

                y = round(text.getY() - offset[1])

                width = round(text.getW())

                height = round(text.getH())

                self.removeText(imgSrc, (round(text.getX()), round(text.getY())), imgOCR[y:y+height, x:x+width])

            # up to here

    '''

        @brief  remove text from image by using ocr image

        @author

    '''

    def removeText(self, img, pt, imgOCR):

        try:

            x = round(pt[0])

            y = round(pt[1])

            width, height = imgOCR.shape[::-1]

            temp = img[y:y+height, x:x+width]

            imgOCR = cv2.erode(imgOCR, np.ones((3,3), np.uint8))

            mask = cv2.bitwise_or(temp, imgOCR)

            imgXOR = cv2.bitwise_xor(temp, mask)

            img[y:y+height, x:x+width] = cv2.bitwise_not(imgXOR)

        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 img

import sys

import os, time, subprocess, ctypes, datetime, copy

from ctypes import wintypes

_GetShortPathNameW = ctypes.windll.kernel32.GetShortPathNameW

_GetShortPathNameW.argtypes = [wintypes.LPCWSTR, wintypes.LPWSTR, wintypes.DWORD]

_GetShortPathNameW.restype = wintypes.DWORD

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

#from PyQt5.QtChart import *

from AppDocData import AppDocData, Source

import pytesseract

import TrainingImageList_UI

from TrainingEditorDialog import QTrainingEditorDialog

import tesseract_ocr_module as TOCR

import numpy as np

import pyqtgraph as pg

dataPath = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID')

tesseractPath = os.path.join(dataPath, 'Tesseract-OCR', 'tessdata')

pytesseract.pytesseract.tesseract_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'tesseract.exe')

tesseract_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'tesseract.exe')

unicharset_extractor_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'unicharset_extractor.exe')

set_unicharset_properties_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'set_unicharset_properties.exe')

shapeclustering_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'shapeclustering.exe')

mftraining_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'mftraining.exe')

cntraining_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'cntraining.exe')

combine_tessdata_cmd = os.path.join(dataPath, 'Tesseract-OCR', 'combine_tessdata.exe')

# for reset chart

defaultCharList = [['0',0],['1',0],['2',0],['3',0],['4',0],['5',0],['6',0],['7',0],['8',0],['9',0],['a',0],['b',0],['c',0],['d',0],['e',0],['f',0],['g',0],['h',0],['i',0],['j',0],['k',0],['l',0],['m',0],['n',0],['o',0],['p',0],['q',0],['r',0],['s',0],['t',0],['u',0],['w',0],['x',0],['y',0],['z',0],['A',0],['B',0],['C',0],['D',0],['E',0],['F',0],['G',0],['H',0],['I',0],['J',0],['K',0],['L',0],['M',0],['N',0],['O',0],['P',0],['Q',0],['R',0],['S',0],['T',0],['U',0],['V',0],['W',0],['X',0],['Y',0],['Z',0]]

#// for remove noise

#noisePassList = ['.', '\"', '\'', ',', '`', '-', '+']

class EmittingStream(QObject):

    """

    This is EmittingStream class

    """

    textWritten = pyqtSignal(str)

    def write(self, text):

        self.textWritten.emit(str(text))

class QTrainingImageListDialog(QDialog):

    """

    This is training image list dialog class

    """

    TRAINING_DATA_COUNT = 5

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.setWindowFlag(Qt.WindowMinMaxButtonsHint)

        self.ui = TrainingImageList_UI.Ui_TraingingImageListDialog()

        self.ui.setupUi(self)

        self.ui.progressBar.setValue(0)

        self.charList = []

        # for List 

        self.ui.tableWidgetList.setSelectionMode(QAbstractItemView.SingleSelection) 

        self.ui.tableWidgetList.setColumnCount(4)

        ## column header 명 설정

        self.ui.tableWidgetList.setHorizontalHeaderLabels([self.tr('No.'), self.tr('Image List'), self.tr('Modified Date'), self.tr('Box Contents')])

        self.ui.tableWidgetList.horizontalHeaderItem(1).setToolTip(self.tr('Image Name')) # header tooltip

        self.ui.tableWidgetList.horizontalHeaderItem(2).setToolTip(self.tr('Edit Status')) # header tooltip

        self.ui.tableWidgetList.horizontalHeaderItem(3).setToolTip(self.tr('Box Contents')) # header tooltip

        self.ui.tableWidgetList.horizontalHeaderItem(1).setSizeHint(QSize(30, 30))

        self.ui.tableWidgetList.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeToContents)

        self.ui.tableWidgetList.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)

        self.ui.tableWidgetList.horizontalHeader().setSectionResizeMode(2, QHeaderView.Stretch)

        self.ui.tableWidgetList.horizontalHeader().setSectionResizeMode(3, QHeaderView.Stretch)

        self.ui.tableWidgetList.resizeColumnsToContents()

        self.ui.tableWidgetList.setEditTriggers(QAbstractItemView.NoEditTriggers)

        self.listUpdate()

        # connect signals and slots

        self.ui.tableWidgetList.cellDoubleClicked.connect(self.listCellDoubleClicked)

        self.ui.pushButtonBox.clicked.connect(self.pushButtonBoxClicked)

        self.ui.pushButtonMakeTrainingData.clicked.connect(self.makeTrainingDataClicked)

        self.ui.pushButtonImageDelete.clicked.connect(self.pushButtonImageDeleteClicked)

        self.ui.pushButtonBoxDelete.clicked.connect(self.pushButtonBoxDeleteClicked)

        self.ui.pushButtonClose.clicked.connect(self.pushButtonCloseClicked)

        self.ui.pushButtonDeleteBatchChar.clicked.connect(self.pushButtonDeleteBatchCharClicked)

        # delete character in box

        self.ui.pushButtonDeleteBatchChar.setEnabled(False)

        self.ui.pushButtonDeleteBatchChar.setHidden(True)

        self.ui.lineEditTargetChar.setEnabled(False)

        self.ui.lineEditTargetChar.setHidden(True)

        # Install the custom output stream

        #sys.stdout = EmittingStream(textWritten=self.normalOutputWritten)

        os.environ['TESSDATA_PREFIX'] = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID', 'Tesseract-OCR', 'tessdata')

        appDocData = AppDocData.instance()

        project = appDocData.getCurrentProject()

        dataList = appDocData.getTrainingFileList()

        self.charList = []

        for charCounts in defaultCharList:

            charCount = [charCounts[0], charCounts[1]]

            self.charList.append(charCount)

        for data in dataList:

            if data.find('.boxS') is not -1:

                boxPath = os.path.join(project.getTrainingFilePath(), data)

                fw = open(boxPath, 'r', encoding='utf8')

                boxContent = fw.read()

                lines = boxContent.split('\n')

                for line in lines:

                    if not line: continue

                    char,min_x,min_y,max_x,max_y = line.split(' ')

                    matches = [_char for _char in self.charList if _char[0] == char]

                    if matches:

                        matches[0][1] = matches[0][1] + 1

                    else:

                        self.charList.append([char, 1])

                fw.close()

        self.makeChart()

    def __del__(self):

        # Restore sys.stdout

        #sys.stdout = sys.__stdout__

        pass

    def normalOutputWritten(self, text):

        """Append text to the QTextEdit."""

        # Maybe QTextEdit.append() works as well, but this is how I do it:

        cursor = self.ui.textEditConsole.textCursor()

        cursor.movePosition(QTextCursor.End)

        cursor.insertText(text)

        self.ui.textEditConsole.setTextCursor(cursor)

        self.ui.textEditConsole.ensureCursorVisible()

    '''

        @brief      close dialog by button click

        @author     euisung

        @date       2018.10.18

    '''

    def pushButtonCloseClicked(self):

        QDialog.reject(self)

    '''

        @brief      update image list

        @author     euisung

        @date       2018.10.18

    '''

    def listUpdate(self):

        appDocData = AppDocData.instance()

        project = appDocData.getCurrentProject()

        dataList = appDocData.getTrainingFileList()

        imgCount = 0

        for data in dataList:

            if data.find('.png') is not -1:

                imgCount += 1

        self.ui.tableWidgetList.setRowCount(imgCount)

        row = 0

        for data in dataList:

            if data.find('.png') is not -1:

                self.ui.tableWidgetList.setItem(row, 0, QTableWidgetItem(str(row + 1)))

                self.ui.tableWidgetList.setItem(row, 1, QTableWidgetItem(data))

                allDataList = appDocData.getTrainingFileList()

                for adata in allDataList:

                    boxName = data.replace('.png', '.boxS')

                    if adata.find(boxName) is not -1:

                        boxPath = os.path.join(project.getTrainingFilePath(), boxName)

                        modifiedTime = str(datetime.datetime.strptime(time.ctime(os.path.getmtime(boxPath)), "%a %b %d %H:%M:%S %Y"))

                        self.ui.tableWidgetList.setItem(row, 2, QTableWidgetItem(modifiedTime))

                        fw = open(boxPath, 'r', encoding='utf8')

                        boxContent = fw.read()

                        fw.close()

                        boxContent = boxContent.split('\n')

                        boxContent = boxContent[:-1]

                        boxchars = ''

                        for char in boxContent:

                            boxchars += char[0]

                        boxContent = QTableWidgetItem(boxchars)

                        boxContent.setFont(QFont('Consolas', 9, QFont.Bold))

                        self.ui.tableWidgetList.setItem(row, 3, QTableWidgetItem(boxContent))

                        break

                    else:

                        self.ui.tableWidgetList.setItem(row, 2, QTableWidgetItem(''))

                        self.ui.tableWidgetList.setItem(row, 3, QTableWidgetItem(''))

                row += 1

    '''

        @brief      delete training box only by button click

        @author     euisung

        @date       2018.10.18

    '''

    def pushButtonBoxDeleteClicked(self):

        try:

            row = self.ui.tableWidgetList.selectedIndexes()[0].row()

            col = self.ui.tableWidgetList.selectedIndexes()[0].column()

            drawingName = self.ui.tableWidgetList.item(row, 1).text()

            boxDate = self.ui.tableWidgetList.item(row, 2).text()

            if boxDate == '':

                return

        except Exception as ex:

            pass

        try:

            reply = QMessageBox.question(self, self.tr('Continue?'), self.tr('Are you sure you want to delete the box job? '), QMessageBox.Yes, QMessageBox.Cancel)

            if reply == QMessageBox.Yes:

                appDocData = AppDocData.instance()

                project = appDocData.getCurrentProject()

                drawingPath = os.path.join(project.getTrainingFilePath(), drawingName)

                boxName = drawingPath.replace('.png', '.boxS')

                if os.path.isfile(boxName):

                    os.remove(boxName)

                self.ui.tableWidgetList.item(row, 2).setText('')

                self.ui.tableWidgetList.item(row, 3).setText('')

                #self.listUpdate()

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

    '''

        @brief      delete training image and box by button click

        @author     euisung

        @date       2018.10.18

    '''

    def pushButtonImageDeleteClicked(self):

        try:

            row = self.ui.tableWidgetList.selectedIndexes()[0].row()

            col = self.ui.tableWidgetList.selectedIndexes()[0].column()

            drawingName = self.ui.tableWidgetList.item(row, 1).text()

        except Exception as ex:

            pass

        try:

            reply = QMessageBox.question(self, self.tr('Continue?'), self.tr('Are you sure you want to delete selected image? '), QMessageBox.Yes, QMessageBox.Cancel)

            if reply == QMessageBox.Yes:

                appDocData = AppDocData.instance()

                project = appDocData.getCurrentProject()

                drawingPath = os.path.join(project.getTrainingFilePath(), drawingName)

                if os.path.isfile(drawingPath):

                    os.remove(drawingPath)

                boxName = drawingPath.replace('.png', '.boxS')

                if os.path.isfile(boxName):

                    os.remove(boxName)

                self.ui.tableWidgetList.removeRow(row)

                #self.listUpdate()

        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 dataReady(self):

        cursor = self.ui.textEditConsole.textCursor()

        cursor.movePosition(cursor.End)

        cursor.insertText(str(self.process.readAll()))

        self.ui.textEditConsole.ensureCursorVisible()