HYTOS

# coding: utf-8

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

from PyQt5.QtGui import *

from QtImageViewer import QtImageViewer

import os

import sqlite3

import sys

import SG_DbHelper

import symbol, SymbolBase

import potrace

import UI_SymbolEditor

from AppDocData import AppDocData

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

import CropCommand, HandCommand, ZoomCommand, PenCommand, EraserCommand, AreaEraserCommand, OriginalPointCommand, ConnectionPointCommand

class QSymbolEditorDialog(QDialog):

    FILE_NUMBER = 0

    def __init__(self, parent, image, project):

        QDialog.__init__(self, parent)

        self.image = image

        self.project = project

        self.ui = UI_SymbolEditor.Ui_Dialog()

        self.ui.setupUi(self)

        self.setupImageViewer()

        self.setupTools()

        self.initForms()

        self.initContents()

        self.isAccepted = False

        self.dbHelper = SG_DbHelper.SG_DbHelper(self.project.getPath())

    '''

        @brief  Set up QtImageViewer and QImage

    '''

    def setupImageViewer(self):

        x = self.ui.imageViewContainer.x()

        y = self.ui.imageViewContainer.y()

        width = self.ui.imageViewContainer.frameGeometry().width()

        height = self.ui.imageViewContainer.frameGeometry().height()

        self.ui.imageView = QtImageViewer()

        self.ui.imageView.setGeometry(QtCore.QRect(0, y, height, height))

        self.ui.imageView.aspectRatioMode = QtCore.Qt.KeepAspectRatio

        self.ui.imageView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)

        self.ui.imageView.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)

        self.ui.imageView.canZoom = True

        self.ui.imageView.canPan = True

        image = self.image.copy()

        self.imgW = image.width()

        self.imgH = image.height()

        image = image.scaled(self.imgW, self.imgH)

        self.ui.imageView.setImage(image)

        self.ui.imageViewerContainerLayout.addWidget(self.ui.imageView)

    '''

        @brief  Set up Hand, Crop, ETC Tools

    '''

    def setupTools(self):

        self.ui.handButton.clicked.connect(self.handToolClickEvent)

        self.ui.penButton.clicked.connect(self.penToolClickEvent)

        self.ui.penWidthSpinBox.valueChanged.connect(self.penWidthChangedEvent)

        self.ui.eraserButton.clicked.connect(self.eraserToolClickEvent)

        self.ui.eraserSpinBox.valueChanged.connect(self.eraserWidthChangedEvent)

        self.ui.areaEraserButton.clicked.connect(self.areaEraserToolClickEvent)

        self.ui.zoomButton.clicked.connect(self.zoomToolClickEvent)

        self.ui.initZoomButton.clicked.connect(self.zoomInitToolClickEvent)

        self.ui.guidelineCheckbox.stateChanged.connect(self.guidelineStateChangedEvent)

        #self.ui.additionalSymbolListWidget.keyPressEvent.connect(self.additionalSymbolListKeyPressEvent)

        #self.ui.connectionPointList.keyPressEvent.connect(self.additionalSymbolListKeyPressEvent)

    '''

        @brief  Init Forms with type and default values

    '''

    def initForms(self):

        self.ui.idLineEdit.setValidator(QRegExpValidator(QtCore.QRegExp("^[1-9]\d+$")))

        self.ui.thresholdLineEdit.setValidator(QRegExpValidator(QtCore.QRegExp("^[0-9]\d+$"))) # ([0-1]{1}[.])?[0-9]+

        self.ui.minMatchPointLineEdit.setValidator(QRegExpValidator(QtCore.QRegExp("^[0-9]\d+$")))

        self.initIsOriginDetectComboBoxItems()

        self.initOcrOptionComboBoxItems()

        self.initDefaultSymbolDirectionComboBoxItems()

        self.ui.addAdditionalSymbolButton.clicked.connect(self.addAdditionalSymbol)

        self.ui.addOriginalPointButton.clicked.connect(self.addOriginalPoint)

        self.ui.addConnectionPointButton.clicked.connect(self.addConnectionPoint)

        self.initBaseSymbolComboBoxItems()

        self.initAdditionalSymbolComboBoxItems()

    '''

        @brief  Set data on forms, For modifying symbol

    '''

    def initContents(self):

        self.ui.targetDBLineEdit.setText(self.project.getPath()+'/db/ITI_PID.db')

        #self.ui.isOriginDetectComboBox

        #self.ui.ocrOptionComboBox

        #self.ui.baseSymbolComboBox

        #self.initBaseSymbolComboBoxItems()

        #self.ui.defaultSymbolDirectionComboBox

        #self.ui.additionalSymbolComboBox

    '''

        @brief  Init ComboBox Items For DB Field [IsDetectOrigin]

    '''

    def initIsOriginDetectComboBoxItems(self):

        self.ui.isOriginDetectComboBox.addItem("원본 도면", 0)

        self.ui.isOriginDetectComboBox.addItem("텍스트 제거 도면", 1)

    '''

        @brief  Init ComboBox Items For DB Field [OcrOption]

    '''

    def initOcrOptionComboBoxItems(self):

        self.ui.ocrOptionComboBox.addItem("OCR 미적용", 0)

        self.ui.ocrOptionComboBox.addItem("일반 심볼", 1)

        self.ui.ocrOptionComboBox.addItem("Instrument 계통", 2)

    '''

        @brief  Init ComboBox Items For Direction of DB Field [additionalSymbol]

    '''

    def initDefaultSymbolDirectionComboBoxItems(self):

        self.ui.defaultSymbolDirectionComboBox.addItem("UP", 0)

        self.ui.defaultSymbolDirectionComboBox.addItem("DOWN", 2)

        self.ui.defaultSymbolDirectionComboBox.addItem("LEFT", 3)

        self.ui.defaultSymbolDirectionComboBox.addItem("RIGHT", 1)

    '''

        @brief  Init ComboBox Items For DB Field [baseSymbol]

    '''

    def initBaseSymbolComboBoxItems(self):

        symbolNameList = AppDocData.instance().getSymbolNameList()

        symbolNameList.sort()

        symbolNameList.insert(0, "None")

        for name in symbolNameList:

            self.ui.baseSymbolComboBox.addItem(name)

    '''

        @brief  Init ComboBox Items For symbolName of DB Field [additionalSymbol]

    '''

    def initAdditionalSymbolComboBoxItems(self):

        symbolNameList = AppDocData.instance().getSymbolNameList()

        symbolNameList.sort()

        symbolNameList.insert(0, "None")

        for name in symbolNameList:

            self.ui.additionalSymbolComboBox.addItem(name)

    '''

        @brief  remove ConnectionPoint Circles (Using for loop)

    '''

    def removeConnectionPointCircles(self, circlePointList):

        for circlePoint in circlePointList:

            self.removeConnectionPointCircle(circlePoint)

    '''

        @brief  remove each ConnectionPoint Circle

    '''

    def removeConnectionPointCircle(self, circlePoint):

        self.ui.imageView.scene.removeItem(self.ui.imageView.scene.itemAt(QtCore.QPointF(int(circlePoint.x()), int(circlePoint.y())), QTransform()))

    '''

        @brief  Display this QDialog

    '''

    def showDialog(self):

        self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowContextHelpButtonHint)

        self.show()

        self.exec_()

        return self.isAccepted

    '''

        @brief  Using input [Name], make file name

                If the [Name] exists, add number like [FILE_NAME(1), FILE_NAME(2), ...]

                Recursive function

    '''

    def makeFileName(self, originName, newName):

        imageFolderDir = self.project.getPath() + "\\image"

        imageDir = imageFolderDir + "\\" + newName + ".png"

        svgFolderDir = self.project.getPath() + "\\svg"

        svgDir = svgFolderDir + "\\" + newName + ".svg"

        if (os.path.exists(imageDir)) or (self.dbHelper.isExistFileName(imageDir) or (os.path.exists(svgDir))):

            self.FILE_NUMBER = self.FILE_NUMBER + 1

            imgName = originName + "({})".format(self.FILE_NUMBER)

            return self.makeFileName(originName, imgName)

        else:

            return newName

    '''

        @brief  Make symbol object for saving on DB

    '''

    def makeSymbolData(self):

        symId = self.ui.idLineEdit.text()

        name = self.ui.nameLineEdit.text()

        type = '' #Empty

        self.FILE_NUMBER = 0

        fileName = self.makeFileName(name, name)

        path = self.project.getPath() + "\\image\\" + fileName + ".png"

        threshold = self.ui.thresholdLineEdit.text()

        minMatchPoint = self.ui.minMatchPointLineEdit.text()

        isDetectOnOrigin = self.ui.isOriginDetectComboBox.currentData()

        rotationCount = self.ui.rotationCountSpinBox.value()

        ocrOption = self.ui.ocrOptionComboBox.currentData()

        isContainChild = 1 if self.ui.isContainChildCheckBox.isChecked() else 0

        originalPoint = self.ui.originalPointLineEdit.text()

        connectionPoint = self.makeConnectionPointListString()

        baseSymbol = self.ui.baseSymbolComboBox.currentText()

        additionalSymbol = self.makeAdditionalSymbolListString()

        convertedThreshold = int(threshold) / 100.0

        newSym = symbol.SymbolBase(int(symId), fileName, type, path, convertedThreshold, int(minMatchPoint), isDetectOnOrigin

                                   , rotationCount, ocrOption, isContainChild, originalPoint, connectionPoint, baseSymbol, additionalSymbol)

        return newSym

    '''

        @brief  Make AdditionalSymbol String

                [AdditionalSymbolString = DIRECTION,SYMBOL_NAME/DIRECTION,SYMBOL_NAME/...]

    '''

    def makeAdditionalSymbolListString(self):

        ret = ""

        listItems = []

        for index in range(self.ui.additionalSymbolListWidget.count()):

            listItems.append(self.ui.additionalSymbolListWidget.item(index))

        if listItems is not None:

            for index in range(len(listItems)):

                item = listItems[index]

                text = item.text()

                if index != 0:

                    ret = ret + "/"

                ret = ret + text

        return ret

    '''

        @brief  Make ConnectionPoint String

                [ConnectionPointString = x1,y1/x2,y2/...]

    '''

    def makeConnectionPointListString(self):

        ret = ""

        listItems = []

        for index in range(self.ui.connectionPointList.count()):

            listItems.append(self.ui.connectionPointList.item(index))

        if listItems is not None:

            for index in range(len(listItems)):

                item = listItems[index]

                text = item.text()

                if index != 0:

                    ret = ret + "/"

                ret = ret + text

        return ret

    '''

        @brief  Called when Save Button Clicked

                Validation Check → Make Symbol Data → Insert Symbol Data into DB → Save png and svg files

    '''

    def accept(self):

        print("save")

        isValid, exceptionMsg = self.isValidSymbolInfo()

        if isValid:

            print("valid symbol info")

            isAdded, fileName, imagePath = self.dbHelper.insertSymbol(self.makeSymbolData())

            if isAdded:

                try:

                    image = self.ui.imageView.image()

                    if image is not None:

                        image.save(imagePath, 'PNG')

                        potrace.convertImageToSvg(imagePath, self.project.getPath()+"/svg/" + fileName + ".svg")

                        self.isAccepted = True

                        QDialog.accept(self)

                except:

                    self.resetInsertSymbol(imagePath, fileName)

                    self.isAccepted = False

                    QMessageBox.about(self.ui.buttonBox, "알림", "심볼 저장 과정 중 문제가 발생했습니다.")

            else:

                QMessageBox.about(self.ui.buttonBox, "알림", "심볼 저장 과정 중 문제가 발생했습니다.")

        else:

            print("invalid symbol info")

            QMessageBox.about(self.ui.buttonBox, "알림", exceptionMsg)

    '''

        @brief  Called When Cancel Button Clicked

    '''

    def cancel(self):

        print("cancel")

        self.isAccepted = Flase

        QDialog.cancel(self)

    '''

        @brief  Called When error occured while saving png and svg files

                Delete png, svg files and record from DB

    '''

    def resetInsertSymbol(self, imagePath, fileName):

        if os.path.exists(imagePath):

            os.remove(imagePath)

        svgPath = self.project.getPath()+"/svg/" + fileName + ".svg"

        if os.path.exists(svgPath):

            os.remove(svgPath)

        AppDocData.instance().deleteSymbol(imagePath)

    def keyPressEvent(self, event):

        if event.key() == QtCore.Qt.Key_Delete:

            if self.ui.connectionPointList.hasFocus():

                selectedItems = self.ui.connectionPointList.selectedItems()

                if selectedItems is not None:

                    for item in selectedItems:

                        text = item.text()

                        x = int(text.split(",")[0])

                        y = int(text.split(",")[1])

                        self.removeConnectionPointCircle(QtCore.QPointF(x, y))

                        self.ui.connectionPointList.takeItem(self.ui.connectionPointList.row(item))

            elif self.ui.additionalSymbolListWidget.hasFocus():

                selectedItems = self.ui.additionalSymbolListWidget.selectedItems()

                if selectedItems is not None:

                    for item in selectedItems:

                        self.ui.additionalSymbolListWidget.takeItem(self.ui.additionalSymbolListWidget.row(item))

    '''

        @brief  Hand Tool Button Clicked

    '''

    def handToolClickEvent(self, event):

        print("hand tool clicked")

        self.ui.imageView.command = HandCommand.HandCommand(self.ui.imageView)

    '''

        @brief  Zoom Init Tool Button Clicked

    '''

    def zoomInitToolClickEvent(self, event):

        print("zoom init tool clicked")

        self.ui.imageView.command = None

        self.ui.imageView.zoomImageInit()

    '''

        @brief  Zoom Tool Button Clicked

    '''

    def zoomToolClickEvent(self, event):

        print("zoom tool clicked")

        self.ui.imageView.command = ZoomCommand.ZoomCommand(self.ui.imageView)

    '''

        @brief  Pen Tool Button Clicked

    '''

    def penToolClickEvent(self, event):

        print("Pen")

        width = self.ui.toolWidget.findChild(QSpinBox, 'penWidthSpinBox').value()

        self.ui.imageView.command = PenCommand.PenCommand(self.ui.imageView)

        self.ui.imageView.command.width = width

    '''

        @brief  Pen Width Value Changed

    '''

    def penWidthChangedEvent(self, value):

        print("Pen Width " + str(value))

        if self.ui.imageView.command is not None and type(self.ui.imageView.command) is PenCommand.PenCommand:

            self.ui.imageView.command.width = value

    '''

        @brief  Eraser Tool Button Clicked

    '''

    def eraserToolClickEvent(self, event):

        print("eraser")

        width = self.ui.toolWidget.findChild(QSpinBox, 'eraserSpinBox').value()

        self.ui.imageView.command = EraserCommand.EraserCommand(self.ui.imageView)

        self.ui.imageView.command.width = width

    '''

        @brief  Eraser Width Value Changed

    '''

    def eraserWidthChangedEvent(self, value):

        print("eraser " + str(value))

        if self.ui.imageView.command is not None and type(self.ui.imageView.command) is EraserCommand.EraserCommand:

            self.ui.imageView.command.width = value

    '''

        @brief  Area Eraser Tool Button Clicked

    '''

    def areaEraserToolClickEvent(self, event):

        print("area eraser")

        self.ui.imageView.command = AreaEraserCommand.AreaEraserCommand(self.ui.imageView)

    '''

        @brief  Guideline Check State Changed

    '''

    def guidelineStateChangedEvent(self, value):

        if self.ui.guidelineCheckbox.isChecked():

            self.ui.imageView.showGuideline(True)

        else:

            self.ui.imageView.showGuideline(False)

    '''

        @brief  Add AdditionalSymbol String on ListWidget

    '''

    def addAdditionalSymbol(self, event):

        print("addAdditionalSymbol")

        additionalSymbolIndex = self.ui.additionalSymbolComboBox.currentIndex()

        if additionalSymbolIndex != 0:

            print("Symbol Selected")

            direction = self.ui.defaultSymbolDirectionComboBox.currentText()

            symbolName = self.ui.additionalSymbolComboBox.currentText()

            text = "{},{}".format(direction, symbolName)

            if self.isAlreadyAdded(text):

                QMessageBox.about(self.ui.buttonBox, "알림", "이미 추가된 아이템입니다.")

            else:

                self.ui.additionalSymbolListWidget.addItem(text)

    '''

        @brief  Check the text is already added

    '''

    def isAlreadyAdded(self, text):

        for index in range(self.ui.additionalSymbolListWidget.count()):

            item = self.ui.additionalSymbolListWidget.item(index)

            if item.text() == text:

                return True

        return False

    '''

        @brief  Original Point Tool Button Clicked

    '''

    def addOriginalPoint(self, event):

        print("addOriginalPoint")

        self.ui.imageView.command = OriginalPointCommand.OriginalPointCommand(self.ui.imageView, self.ui.originalPointLineEdit)

    '''

        @brief  Connection Point Tool Button Clicked

    '''

    def addConnectionPoint(self, event):

        print("addConnectionPoint")

        self.ui.imageView.command = ConnectionPointCommand.ConnectionPointCommand(self.ui.imageView, self.ui.connectionPointLineEdit, self.ui.connectionPointList)

    '''

        @brief  Validation Check

        @return (isValid, errorMsg)

    '''

    def isValidSymbolInfo(self):

        print("isValid")

        EXCEPTION_MSG_FORMAT = "{} 입력을 확인해주세요."

        EXCEPTION_MSG_DUPLICATED_FORMAT = "이미 저장된 {} 값입니다."

        infoTitle = ""

        idText = self.ui.idLineEdit.text()

        id = int(idText) if idText else -1

        if (id == -1 or id < 100):

            infoTitle = self.ui.idLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        if self.dbHelper.isExistData('symId', id):

            infoTitle = self.ui.idLabel.text()

            return (False, EXCEPTION_MSG_DUPLICATED_FORMAT.format(infoTitle))

        if not self.ui.nameLineEdit.text():

            infoTitle = self.ui.nameLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        thresholdText = self.ui.thresholdLineEdit.text()

        threshold = float(thresholdText) if thresholdText else -1

        if not(threshold >= 0 and threshold <= 100):

            infoTitle = self.ui.thresholdLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        minMatchPointText = self.ui.minMatchPointLineEdit.text()

        minMatchPoint = float(minMatchPointText) if minMatchPointText else -1

        if not(minMatchPoint >= 0):

            infoTitle = self.ui.minMatchPointLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        if self.ui.baseSymbolComboBox.currentIndex() == 0: #default value(None) index

            infoTitle = self.ui.baseSymbolLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        #Additional Symbol is Nullable

        if not self.ui.originalPointLineEdit.text() or self.ui.imageView.isOriginalPointSelected == False:

            infoTitle = self.ui.originalPointLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        if not (self.ui.connectionPointList.count() > 0):

            infoTitle = self.ui.connectionPointLabel.text()

            return (False, EXCEPTION_MSG_FORMAT.format(infoTitle))

        return True, None