HYTOS

from SingletonInstance import SingletonInstane

import re

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

    from PyQt5.QtWidgets import *

except ImportError:

    try:

        from PyQt4.QtCore import *

        from PyQt4.QtGui import *

    except ImportError:

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

from AppDocData import AppDocData

import sys, os

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

from QEngineeringTextItem import QEngineeringTextItem

from QEngineeringLineNoTextItem import QEngineeringLineNoTextItem

from QEngineeringNoteItem import QEngineeringNoteItem

from QEngineeringSizeTextItem import QEngineeringSizeTextItem

from QEngineeringTagNoTextItem import QEngineeringTagNoTextItem

class TextItemFactory(SingletonInstane):

    def __init__(self):

        self.delimiter = '"'

    '''

        @history    2018.04.27  Jeongwoo    Add condition on elif-statement (delimiter != '' and lineNoConfig != '')

                    humkyung 2018.05.02 create engineering size text item if condition matches

                    humkyung 2018.06.19 set line no's color

                    kyouho 2018.07.04 edit for isLineNo method (add parameter)

    '''             

    def createTextItem(self, text, delimiter = '', lineNoConfig = '', linePropertyData = None, tagSeqNoPattern = None):

        import random

        item = None

        try:

            docData = AppDocData.instance()

            configs = docData.getConfigs('Size', 'Delimiter')

            sizeDelimiter = configs[0].value.upper() if 1 == len(configs) else None

            result = (False,)

            if (delimiter != '' and lineNoConfig != ''):

                result = self.isLineNo(text, delimiter, lineNoConfig, linePropertyData, tagSeqNoPattern)

            if self.isNoteNoText(text):

                item = QEngineeringNoteItem()

                item.setToolTip(text)

            elif (delimiter != '' and lineNoConfig != '') and result[0]:

                item = QEngineeringLineNoTextItem(text)

                item.setToolTip(text)

                # get color option

                configs = docData.getConfigs('Line Color', 'Visible Option')

                colorProperty = ''

                if configs:

                    visibleOption = configs[0].value

                else:

                    visibleOption = 'Random'

                configs = docData.getConfigs('Color Property', 'State')

                if configs:

                    colorProperty = configs[0].value

                # set color

                if visibleOption == 'Random':

                    colorCount = len(docData.colors)

                    if colorCount:

                        colorIdx = random.randrange(0, colorCount)

                        rgb = docData.colors[colorIdx]

                        item.setColor(QColor(rgb.red, rgb.green, rgb.blue).name())

                else:

                    configs = lineNoConfig[0].value.split(self.delimiter)

                    values = result[1]

                    index = configs.index(colorProperty)

                    if index >= 0:

                        value = values[index]

                        if colorProperty == 'Nominal Diameter':

                            configs = docData.getConfigs('Line No', 'Size Unit')

                            if configs and configs[0].value == "Inch":

                                value = docData.convertInchToMetric(value)

                        configs = docData.getConfigs(colorProperty, value)

                        if configs:

                            data = configs[0].value

                            rgb = data.split(',')

                            item.setColor(QColor(int(rgb[0]), int(rgb[1]), int(rgb[2])).name())

                    # up to here

            elif self.isSizeText(text, sizeDelimiter):

                item = QEngineeringSizeTextItem()

                item.setToolTip(text)

            elif self.isTagNoText(text):

                item = QEngineeringTagNoTextItem()

                item.setToolTip(text)

            else:

                item = QEngineeringTextItem()

                item.setToolTip(text)

        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 item

    '''

        @brief  check if given text is possible to be line no

        @author humkyung

        @date   2018.04.20

        @history 2018.04.26 Jeongwoo Moved here from QEngineeringTextItem

                 2018.04.30 Jeongwoo lineNoConfig None Check

                 kyouho 2018.07.04 edit logic

    '''

    def isLineNo(self, text, delimiter, lineNoConfig, linePropertyData, tagSeqNoPattern):

        try:

            res = []

            if lineNoConfig is None or len(lineNoConfig) == 0:

                return (False,)

            tokens = text.split(delimiter)

            patternText = lineNoConfig[0].value.replace(self.delimiter, '').split(self.delimiter)

            #Line Number에 Delimiter가 있을 수 있음

            if lineNoConfig[0].value != 'None' and len(tokens) >= len(patternText):

                configs = lineNoConfig[0].value.split(self.delimiter)

                loopText = text.replace(' ','')

                for propertyName in configs:

                    isTable = True

                    loopList = []

                    if propertyName == 'Tag Seq No':

                        isTable = False

                    elif propertyName == delimiter:

                        loopList.append(delimiter)

                    else:

                        loopList = linePropertyData[propertyName.replace(' ','').replace('&','n')]

                    if isTable:

                        find = False

                        for loop in loopList:

                            result = self.isStartWithWord(loopText, loop)

                            if result[0]:

                                loopText = result[1]

                                res.append(result[2])

                                find = True

                                break

                        if not find:

                            return (False,)

                    else:

                        result = self.isTagSeqNo(loopText, tagSeqNoPattern)

                        if result[0]:

                            loopText = result[1]

                            res.append(result[2])

                        else:

                            return (False,)

                if loopText == '':

                    return (True, res)

                else:

                    return (False,)

            return (False,)

        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      Check Text startwith word

        @author     kyouho

        @date       2018.07.04

    '''

    def isStartWithWord(self, text, word):

        sliceText = text[0:len(word)]

        if sliceText == word:

            return (True, text[len(sliceText):len(text)], word)

        else:

            return (False,)

    '''

        @brief      Check Text startwith number

        @author     kyouho

        @date       2018.07.05

    '''

    def isStartWithNumber(self, text):

        sliceText = text[0:1]

        num = ''

        if self.isNumber(sliceText):

            text = text[1:len(text)]

            num += text 

            while len(text) > 0:

                sliceText = text[0:1]

                if self.isNumber(sliceText):

                    text = text[1:len(text)]

                    num += text

                else:

                    break

            return (True, text, num)

        else:

            return (False,)

    '''

        @brief      Check Number

        @author     kyouho

        @date       2018.07.05

        @history    kyouho 2018.07.09   add Refular Expression

    '''

    def isNumber(self, num):

        p = re.compile('(^[0-9]+$)')

        result = p.match(num)

        if result:

            return True

        else:

            return False

    '''

        @brief      Check Number

        @author     kyouho

        @date       2018.07.05

    '''

    def isTagSeqNo(self, text, pattern):

        try:

            if pattern is not None:

                attr = text

                result = eval(pattern)

                if self.isNumber(result):

                    sliceText = text[0:len(result)]

                    return (True, text[len(sliceText):len(text)], sliceText)

                else:

                    return (False,)

            else:

                return (False,)

        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      Check if given text is Note No Text (ex : NOTE 1, NOTE 2, ...)

        @author     Jeongwoo

        @date       2018.04.26

        @history    2018.05.16  Jeongwoo    Modify Validator flag

    '''

    def isNoteNoText(self, text):

        if QRegExpValidator(QRegExp("NOTE\s\d+")).validate(text, 0)[0] == QValidator.Acceptable :

            return True

        else:

            return False

    '''

        @brief  check if given text is size text

        @author humkyung

        @date   2018.05.02

    '''

    def isSizeText(self, text, delimiter=None):

        docData = AppDocData.instance()

        sizeDataList = docData.getNomialPipeSizeData()

        tokens = text.upper().split(delimiter) if delimiter is not None else [text]

        for token in tokens:

            matches = [sizeData for sizeData in sizeDataList if sizeData.sizeValue() == token]

            if not matches: return False

        return True

    '''

        @brief  check if given text is tag no

        @author humkyung

        @date   2018.05.03

    '''

    def isTagNoText(self, text):

        docData = AppDocData.instance()

        dataList = docData.getEquipmentDataList()

        matches = [data for data in dataList if data.tagNo == text]

        return (len(matches) > 0)