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

    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)

    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)

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

    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

    def initIsOriginDetectComboBoxItems(self):

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

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

    def initOcrOptionComboBoxItems(self):

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

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

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

    def initDefaultSymbolDirectionComboBoxItems(self):

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

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

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

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

    def initBaseSymbolComboBoxItems(self):

        symbolNameList = AppDocData.instance().getSymbolNameList()

        symbolNameList.insert(0, "None")

        for name in symbolNameList:

            self.ui.baseSymbolComboBox.addItem(name)

    def initAdditionalSymbolComboBoxItems(self):

        symbolNameList = AppDocData.instance().getSymbolNameList()

        symbolNameList.insert(0, "None")

        for name in symbolNameList:

            self.ui.additionalSymbolComboBox.addItem(name)

    def removeConnectionPointCircles(self, circlePointList):

        for circlePoint in circlePointList:

            self.removeConnectionPointCircle(circlePoint)

    def removeConnectionPointCircle(self, circlePoint):

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

    def showDialog(self):

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

        self.show()

        self.exec_()

        return self.isAccepted

    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

    def makeSymbolData(self):

        symId = self.ui.idLineEdit.text()

        name = self.ui.nameLineEdit.text()

        type = self.ui.typeLineEdit.text()

        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

    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

    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

    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:

                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

            else:

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

            QDialog.accept(self)

        else:

            print("invalid symbol info")

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

    def cancel(self):

        print("cancel")

        self.isAccepted = Flase

        QDialog.cancel(self)

    def keyPressEvent(self, event):

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

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

    def handToolClickEvent(self, event):

        print("hand tool clicked")

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

    def zoomInitToolClickEvent(self, event):

        print("zoom init tool clicked")

        self.ui.imageView.command = None

        self.ui.imageView.zoomImageInit()

    def zoomToolClickEvent(self, event):

        print("zoom tool clicked")

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

    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

    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

    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

    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

    def areaEraserToolClickEvent(self, event):

        print("area eraser")

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

    def guidelineStateChangedEvent(self, value):

        if self.ui.guidelineCheckbox.isChecked():

            self.ui.imageView.showGuideline(True)

        else:

            self.ui.imageView.showGuideline(False)

    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)

    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

    def addOriginalPoint(self, event):

        print("addOriginalPoint")

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

    def addConnectionPoint(self, event):

        print("addConnectionPoint")

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

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

        #if not self.ui.typeLineEdit.text():

        #    infoTitle = self.ui.typeLabel.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