HYTOS

from AppDocData import Config

from AppDocData import Color

import Configuration_UI

class ListView(QListView):

    def __init__(self, *args, **kwargs):

        self.ui.spinBoxShrinkSize.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxShrinkSize.setValue(0)

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

        self.ui.spinBoxMergeSize.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxMergeSize.setValue(10)

        configs = docData.getConfigs('Text Size', 'Min Text Size')

        self.ui.minTextSizeSpinBox.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.minTextSizeSpinBox.setValue(30)

            configs.append(Config('Note No Tag Rule', 'Note No Expression', self.ui.lineEditNoteNoExpression.text()))

            configs.append(Config('OPC Tag No Rule', 'Description', self.ui.lineEditDescription.text()))

            configs.append(Config('OPC Tag No Rule', 'OPC Tag', self.ui.lineEditOPCTag.text()))

            # Add Line Color Option - 2018.07.06 by kyouho

            rbRandomValue = self.ui.radioButtonRandom.isChecked()

            files = appDocData.getDrawingFileList()

            for file in files:

                drawing = [drawing for drawing in drawings if drawing[1] == file]

                    drawings.append([None, file, None])

                item.setIcon(0, QIcon(':newPrefix/image.png'))

            self.treeWidgetDrawingList.root.setText(0, 'P&ID Drawings({})'.format(self.treeWidgetDrawingList.root.childCount()))

        self.graphicsView.setImage(self.image)

    '''

    '''

    def detectText(self):

        try:

            #cv2.destroyAllWindows()

            # up to here

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

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

        @author     humkyung

        @date       2018.07.24

        @history    change parameter updateProgressSignal to worker

    '''

    @staticmethod

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

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

            # up to here

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

                for result in resultTextInfo:

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

            if onlyTextArea:

                return

            # parse texts in area except Drawing area

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

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

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

    '''

    def recognizeTextInArea(self, imgSrc, area, angle=0):

        try:

            img = imgSrc[round(area[1]):round(area[1] + area[3]), round(area[0]):round(area[0] + area[2])]

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

import pytesseract

import TrainingImageList_UI

from TrainingEditorDialog import QTrainingEditorDialog

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

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

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

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

class QTrainingImageListDialog(QDialog):

    trainingDataNumber = 0

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

            boundaryOcrData = None

            if not isBoxFile:

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

## Tesseract path

'''

    @history    Jeongwoo 2018.06.14 Tesseract path changed

'''

#pytesseract.pytesseract.tesseract_cmd = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'Tesseract-OCR', 'tesseract.exe')

pytesseract.pytesseract.tesseract_cmd = os.path.join(os.getenv('ALLUSERSPROFILE'), 'Digital PID', 'Tesseract-OCR', 'tesseract.exe')

DEFAULT_CONF = """

    --psm 6 -c tessedit_char_whitelist=abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-~.,/!@#$%&*(){}[]<>:;+=?\\"

"""

"""

    @history    2018.04.26  Jeongwoo    Make TextInfo object with Calculated Coords (with BoundBox Coords)

                2018.04.30  Jeongwoo    Add QRect.setHeight() in if-statement [(lineRect is not None and currentRect is not None) and lineRect.intersects(currentRect)]

                2018.06.20  Jeongwoo    Remove variable [lastCharHeight] / Change variable [cey], [ch] / Change method to calculate text line height

                humkyung 2018.10.12 change logic to extract text which first get bounding box and then extract character

                2018.10.19  euisung     OCR lang change depend on project name 

"""

    try:

        os.environ['TESSDATA_PREFIX'] = tesseract_path

        textInfoList = []

        docData = AppDocData.instance()

        configs = docData.getConfigs('Text Size', 'Min Text Size')

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

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