HYTOS

OCR_OPTION_NOT_EXEC = 0

OCR_OPTION_ALL_FIND = 1

OCR_OPTION_HALF_AND_HALF = 2

from Area import Area

class SymbolBase:

    def __init__(self, sName, sType, threshold = None, minMatchCount = 0

                 , isDetectOnOrigin = False, rotationCount = 4, ocrOption = OCR_OPTION_NOT_EXEC, isContainChild = 0

                 , originalPoint = None, connectionPoint = None, baseSymbol = None, additionalSymbol = None

                 , isExceptDetect = 0, hasInstrumentLabel = 0, uid = None, width = None, height = None, iType = -2, detectFlip = None, text_area=None):

        self.uid = uid ## Auto increased Unique Id

        self.sName = sName

        self.sType = sType

        self.iType = iType

        self.threshold = threshold

        self.minMatchCount = minMatchCount

        self.isDetectOnOrigin = isDetectOnOrigin

        self.rotationCount = rotationCount

        self.ocrOption = ocrOption

        self.isContainChild = isContainChild

        self.originalPoint = originalPoint

        self.connectionPoint = connectionPoint

        self.baseSymbol = baseSymbol

        self.additionalSymbol = additionalSymbol

        self.isExceptDetect = isExceptDetect

        self.hasInstrumentLabel = hasInstrumentLabel

        self.detectFlip = detectFlip

        self._owner = None

        self.width = width

        self.height = height

        self.text_area = None

        if type(text_area) is str and text_area:

            text_area = text_area.split('/')

            self.text_area = []

            for area_str in text_area:

                area = Area('Text Area')

                area.parse(area_str)

                self.text_area.append(area)

        else:

            self.text_area = text_area

        self._pixmap = None

    @property

    def pixmap(self):

        return self._pixmap

    @pixmap.setter

    def pixmap(self, value):

        self._pixmap = value

    def setUid(self, uid):

        self.uid = uid

    def getUid(self):

        return self.uid

    def setName(self, sName):

        self.sName = sName

    def getName(self):

        return self.sName

    def setType(self, type):

        self.sType = type

    def getType(self):

        return self.sType

    '''

        @brief      getter of owner

        @author     humkyung

        @date       2018.07.20

    '''

    @property

    def owner(self):

        return self._owner

    '''

        @brief      setter of owner

        @author     humkyung

        @date       2018.07.20

    '''

    @owner.setter

    def owner(self, value):

        self._owner = value

    def getPath(self):

        import os

        from AppDocData import AppDocData

        return os.path.join(AppDocData.instance().getCurrentProject().getImageFilePath(), self.sType, self.sName + ".png")

    def setThreshold(self, threshold):

        self.threshold = threshold

    def getThreshold(self):

        return self.threshold

    def setMinMatchCount(self, minMatchCount):

        self.minMatchCount = minMatchCount

    def getMinMatchCount(self):

        return self.minMatchCount

    def setIsDetectOnOrigin(self, isDetectOnOrigin):

        self.isDetectOnOrigin = isDetectOnOrigin

    def getIsDetectOnOrigin(self):

        return self.isDetectOnOrigin

    def setRotationCount(self, rotationCount):

        self.rotationCount = rotationCount

    def getRotationCount(self):

        return self.rotationCount

    def setOcrOption(self, ocrOption):

        self.ocrOption = ocrOption

    def getOcrOption(self):

        return self.ocrOption

    def setIsContainChild(self, isContainChild):

        self.isContainChild = isContainChild

    def getIsContainChild(self):

        return self.isContainChild

    def setOriginalPoint(self, originalPoint):

        self.originalPoint = originalPoint

    def getOriginalPoint(self):

        return self.originalPoint

    def parse_connection_pts(self, origin: list) -> list:

        """

        parse connection points and add origin point

        :param origin:

        :return: connector points which added by origin

        """

        res = []

        for param in self.connectionPoint.split('/'):

            if not param:

                continue

            tokens = param.split(',')

            res.append(

                ('AUTO', origin[0] + float(tokens[0]), origin[1] + float(tokens[1]), '0') if len(tokens) == 2 else

                (tokens[0], origin[0] + float(tokens[1]), origin[1] + float(tokens[2]), '0') if len(tokens) == 3 else

                (tokens[0], origin[0] + float(tokens[1]), origin[1] + float(tokens[2]), tokens[3]))

        return res

    def setConnectionPoint(self, connectionPoint):

        self.connectionPoint = connectionPoint

    def getConnectionPoint(self):

        return self.connectionPoint

    def setBaseSymbol(self, baseSymbol):

        self.baseSymbol = baseSymbol

    def getBaseSymbol(self):

        return self.baseSymbol

    def setAdditionalSymbol(self, additionalSymbol):

        self.additionalSymbol = additionalSymbol

    def getAdditionalSymbol(self):

        return self.additionalSymbol

    def setIsExceptDetect(self, isExceptDetect):

        self.isExceptDetect = isExceptDetect

    def getIsExceptDetect(self):

        return self.isExceptDetect

    def getDetectFlip(self):

        return self.detectFlip

    def getText_area(self):

        return self.text_area

    @property

    def text_area_str(self):

        return '/'.join([area.toString() for area in self.text_area]) if self.text_area else ''

    '''

        @Auther     Yechecol

        @Date       2018.07.03

        @History    hasInstrumentLabel add

    '''

    def setHasInstrumentLabel(self, hasInstrumentLabel):

        self.hasInstrumentLabel = hasInstrumentLabel

    def getHasInstrumentLabel(self):

        return self.hasInstrumentLabel

    '''

        @brief      Get Image File Full Path

        @author     Jeongwoo

        @date       2018.05.03

    '''

    def getImageFileFullPath(self):

        import os

        from AppDocData import AppDocData

        project = AppDocData.instance().getCurrentProject()

        if project is not None:

            return os.path.join(project.getImageFilePath(), self.getType(), self.getName() + ".png")

        return None

    '''

        @brief      Get Svg File Full Path

        @author     Jeongwoo

        @date       2018.05.03

    '''

    def getSvgFileFullPath(self):

        from AppDocData import AppDocData

        project = AppDocData.instance().getCurrentProject()

        if project is not None:

            return project.getSvgFilePath() + "/" + self.getType() + "/" + self.getName() + ".svg"

        return None

class imgLine():

    def __init__(self, start, end):

        self.start = start

        self.end = end

    def tostring(self):

        return str(self.start.real) + ',' + str(self.start.imag) + ' ' + str(self.end.real) + ',' + str(self.end.imag)

    def getStart(self):

        return self.start

    def getEnd(self):

        return self.end