HYTOS

    """

    This is EngineeringPolylineItem Class

    """

            self._path = None

        self._path = QPainterPath()

        self._path.addPolygon(self._pol)

        self.setPath(self._path)

import sys

import os

import http.client

import urllib

import base64

import json

import cv2

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.image as mpimg

from PIL import Image

from io import BytesIO

import TextInfo as ti

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

except:

    from PyQt4.QtCore import *

    from PyQt4.QtGui import *

EXCEPTION_LIST = ['0', 'o', 'O'

                  , '-0', '-o', '-O', '—0', '—o', '—O']

STANDARD_SIZE = 30

def getRotatedAngleByOrientation(orientation):

    angle = 0

    if orientation == "Up":

        angle = 0

    elif orientation == "Left":

        angle = 90

    elif orientation == "Down":

        angle = 180

    elif orientation == "Right":

        angle = 270

    else:   # general case

        angle = 0

    return angle

'''

    @history    18.04.25    Jeongwoo    Calculate Left-Top Coord by QRect and QTransform

'''

def getCoordOnRotatedImage(rAngle, x, y, originImageWidth, originImageHeight, otw, oth):

    rx = None

    ry = None

    transform = QTransform()

    if rAngle == 90 or rAngle == 270:

        transform.translate(originImageHeight*0.5, originImageWidth*0.5)

    elif rAngle == 0 or rAngle == 180:

        transform.translate(originImageWidth*0.5, originImageHeight*0.5)

    transform.rotate(-abs(rAngle))

    transform.translate(-originImageWidth*0.5, -originImageHeight*0.5)

    rect = QRect(x, y, otw, oth)

    rect = transform.mapRect(rect)

    rx = rect.x()

    ry = rect.y()

    rwidth = rect.width()

    rheight = rect.height()

    return (rx, ry, rwidth, rheight)

def getRollbackRotateCodeByOrientation(orientation):

    rotatedAngle = None

    if orientation == "Left":

        rotatedAngle = cv2.ROTATE_90_COUNTERCLOCKWISE

    elif orientation == "Right":

        rotatedAngle = cv2.ROTATE_90_CLOCKWISE

    else:

        rotatedAngle = cv2.ROTATE_180

    return rotatedAngle

def getRotateCodeByOrientation(orientation):

    rotatedAngle = None

    if orientation == "Left":

        rotatedAngle = cv2.ROTATE_90_CLOCKWISE

    elif orientation == "Right":

        rotatedAngle = cv2.ROTATE_90_COUNTERCLOCKWISE

    else:

        rotatedAngle = cv2.ROTATE_180

    return rotatedAngle

####################################

###############################################

#### Update or verify the following values.  ###

###############################################

# Replace the subscription_key string value with your valid subscription key.

subscription_key = '0c331f82837b4adeb5cbdb80a6c03d68' # master

'''

[NAME]

CognitiveService_ImagePnID

[KEY 1]

0c331f82837b4adeb5cbdb80a6c03d68

[KEY 2]

642218f51d364b00bb6d750d2575ca3f

'''

# Replace or verify the region.

#

# You must use the same region in your REST API call as you used to obtain your

# subscription keys.

# For example, if you obtained your subscription keys from the westus region,

# replace

# "westcentralus" in the URI below with "westus".

#

# NOTE: Free trial subscription keys are generated in the westcentralus region,

# so if you are using

# a free trial subscription key, you should not need to change this region.

uri_base = 'eastasia.api.cognitive.microsoft.com' # southeast asia for master

headers = {

    # Request headers.

    #'Content-Type': 'application/json',

    'Content-Type': 'application/octet-stream',

    'Ocp-Apim-Subscription-Key': subscription_key,

}

#params = urllib.urlencode({

params = urllib.parse.urlencode({

    # Request parameters.  The language setting "unk" means automatically

    # detect the language.

    'language': 'en',

    #'language': 'unk',

    'detectOrientation ': 'false',

})

# The URL of a JPEG image containing text.

#body = "{'url':'D:/Visual Studio Project/PyOpenCv/PyOpenCv/res/symbol1.png'}"

#body =

#"{'url':'https://upload.wikimedia.org/wikipedia/commons/thumb/a/af/Atomist_quote_from_Democritus.png/338px-Atomist_quote_from_Democritus.png'}"

#body = "{'url':'http://cfile9.uf.tistory.com/image/99AFE1505A322DBD114994'}"

'''

    @history    18.04.25    Jeongwoo    Init TextInfo Object with getCoordOnRotatedImage()

'''

def detectAndRemoveText(image, offsetX, offsetY):

    tmp = image.copy()

    img = np.array(tmp)

    tInfo = []

    try:

        tmp = image.copy()

        bin_img = BytesIO()

        tmp.save(bin_img, format='PNG')

        tmp.close()

        img_data = bin_img.getvalue()

        bin_img.close()

        #region REST API with raw file

        # Execute the REST API call and get the response.

        conn = http.client.HTTPSConnection(uri_base)

        conn.request("POST", "/vision/v1.0/ocr?%s" % params, img_data, headers)

        response = conn.getresponse()

        data = response.read()

        #endregion

        #region REST API with url

        ## Execute the REST API call and get the response.

        #conn = httplib.HTTPSConnection(uri_base)

        #conn.request("POST", "/vision/v1.0/ocr?%s" % params, body, headers)

        #response = conn.getresponse()

        #data = response.read()

        #endregion

        # 'data' contains the JSON data.  The following formats the JSON data

        # for display.

        parsed = json.loads(data)

        #print ("Response:")

        #print (json.dumps(parsed, sort_keys=True, indent=2))

        conn.close()

        #chan, w, h = img.shape[::-1]

        w, h = img.shape[::-1]

        orientation = parsed['orientation']

        #if orientation != "Up":

        #    img = cv2.rotate(img, getRotateCodeByOrientation(orientation))

        count = 0

        print(str(parsed['orientation']))

        print(str(parsed['textAngle']))

        for cs in parsed['regions']:

            for re in cs['lines']:

                strArr = re['boundingBox'].split(',')

                (rx, ry, rwidth, rheight) = getCoordOnRotatedImage(getRotatedAngleByOrientation(orientation), int(strArr[0]), int(strArr[1]), w if orientation == 'Up' or orientation == 'Down' else h, h if orientation == 'Up' or orientation == 'Down' else w, int(strArr[2]), int(strArr[3]))

                text = ""

                for words in re['words']:

                    print(words)

                    text = text + " " + words['text']

                #print(text)

                if not(text in EXCEPTION_LIST and height < STANDARD_SIZE):

                    tInfo.append(ti.TextInfo(text.strip(), rx + int(offsetX), ry + int(offsetY), rwidth, rheight, getRotatedAngleByOrientation(orientation)))

                    count = count + 1

        print("OCR Result count : " + str(count))

        #if orientation != "Up":

        #    img = cv2.rotate(img, getRollbackRotateCodeByOrientation(orientation))

        return (img, tInfo)

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

#def removeText(imagePath):

#    image = Image.open(imagePath)

#    ret = np.array(image)

#    textInfoList = []

#    for i in range(1):

#        (ret, tInfo) = detectAndRemoveText(Image.fromarray(ret))

#        #ret = cv2.rotate(ret, cv2.ROTATE_90_CLOCKWISE)

#        textInfoList.extend(tInfo)

#    image.close()

#    return (ret, textInfoList)

def removeTextFromNpArray(img, offsetX, offsetY):

    ret = img.copy()

    textInfoList = []

    imgW, imgH = ret.shape[::-1]

    for i in range(1):

        (ret, tInfo) = detectAndRemoveText(Image.fromarray(ret), offsetX, offsetY)

        #ret = cv2.rotate(ret, cv2.ROTATE_90_CLOCKWISE)

        textInfoList.extend(tInfo)

    return (ret, textInfoList)

## NOT USED

def removeText(img, text, x, y, imgW, imgH, width, height, angle):

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

    w, h = roi.shape[::-1]

    if w > 0 and h > 0:

        temp = roi.copy()

        tempBin = cv2.bitwise_not(temp)

        img[y:y+height, x:x+width] = cv2.bitwise_xor(roi, tempBin)

    return img

def getTextInfoFromNpArray(image):

    try:

        tmp = Image.fromarray(image)

        bin_img = BytesIO()

        tmp.save(bin_img, format='PNG')

        tmp.close()

        img_data = bin_img.getvalue()

        bin_img.close()

        #region REST API with raw file

        # Execute the REST API call and get the response.

        conn = http.client.HTTPSConnection(uri_base)

        conn.request("POST", "/vision/v1.0/ocr?%s" % params, img_data, headers)

        response = conn.getresponse()

        data = response.read()

        #endregion

        #region REST API with url

        ## Execute the REST API call and get the response.

        #conn = httplib.HTTPSConnection(uri_base)

        #conn.request("POST", "/vision/v1.0/ocr?%s" % params, body, headers)

        #response = conn.getresponse()

        #data = response.read()

        #endregion

        # 'data' contains the JSON data.  The following formats the JSON data

        # for display.

        parsed = json.loads(data)

        #print ("Response:")

        #print (json.dumps(parsed, sort_keys=True, indent=2))

        conn.close()

        chan, w, h = image.shape[::-1]

        orientation = parsed['orientation']

        if orientation != "Up":

            image = cv2.rotate(image, getRotateCodeByOrientation(orientation))

        count = 0

        textInfoList = []

        for cs in parsed['regions']:

            for re in cs['lines']:

                for words in re['words']:

                    print(words)

                    text = words['text']

                    strArr = words['boundingBox'].split(',')

                    pt = (int(strArr[0]), int(strArr[1]))

                    width = int(strArr[2])

                    height = int(strArr[3])

                    textInfoList.append((text, pt, width, height))

        if orientation != "Up":

            image = cv2.rotate(image, getRollbackRotateCodeByOrientation(orientation))

        return textInfoList

    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 None

if __name__ == '__main__':

    from AppDocData import AppDocData

    im = cv2.imread('D:/Visual Studio Project/DTIPID/DTIPID/DTI_PID/DTI_PID/res/UY1-K-2000_P1_300dpi.png', 1)

    #im = cv2.imread('D:/TestSrc/ConvertPNG/10077467_PFD-Diesel Hydrotreating_171120_P1_1100DPI.png', 1)

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

    if area is not None:

        im = im[int(area.y):int(area.y+area.height), int(area.x):int(area.x+area.width)]

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

    (_img, textInfoList) = removeTextFromNpArray(im, 0, 0)

    for tinfo in textInfoList:

        cv2.putText(im, tinfo.getText(), (tinfo.getX(), tinfo.getY()+tinfo.getH()), 2, 1.0, (0, 0, 0))

        cv2.rectangle(im, (tinfo.getX(), tinfo.getY()), (tinfo.getX()+tinfo.getW(), tinfo.getY()+tinfo.getH()), (0, 0, 255), 2)

    cv2.imwrite(os.path.dirname(os.path.realpath(__file__)) + "/res/Result/Standalone_OCR_Test.png", im)