HYTOS

import sqlite3

import os

import sys

import symbol

import SymbolBase

from sqlite3 import Error

from AppDocData import AppDocData

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

#           NOT USED

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

class SG_DbHelper():

    ROOT_DIR = ""

    DB_PATH = "db"

    DB_NAME = "ITI_PID.db"

    CREATE_SYMBOLS_TABLE_SQL = '''

        CREATE TABLE IF NOT EXISTS 'Symbol' (

            'uid'	INTEGER PRIMARY KEY AUTOINCREMENT,

	        'symId'	INTEGER NOT NULL,

	        'name'	TEXT NOT NULL,

	        'type'	TEXT,

	        'threshold'	NUMERIC NOT NULL DEFAULT 0.4,

	        'minMatchPoint'	INTEGER NOT NULL DEFAULT 0,

	        'isDetectOrigin'	INTEGER NOT NULL DEFAULT 0,

	        'rotationCount'	INTEGER NOT NULL DEFAULT 4,

	        'ocrOption'	INTEGER NOT NULL DEFAULT 0,

	        'isContainChild'	INTEGER NOT NULL DEFAULT 0,

	        'originalPoint'	TEXT,

	        'connectionPoint'	TEXT,

	        'baseSymbol'	TEXT,

	        'additionalSymbol'	TEXT,

            'isExceptDetect'    INTEGER DEFAULT 0

        );

    '''

    CREATE_LINEPROPERTIES_TABLE_SQL = '''

        CREATE TABLE IF NOT EXISTS LineProperties (

            UID TEXT NOT NULL,

            Name TEXT NOT NULL,

            CONSTRAINT LineProperties_PK PRIMARY KEY (UID)

        );

        CREATE UNIQUE INDEX LineProperties_Name_IDX ON LineProperties (Name);

    '''

    #CREATE_SYMBOLS_TABLE_SQL = '''

    #    CREATE TABLE 'Symbol' (

    #        'uid'	INTEGER PRIMARY KEY AUTOINCREMENT,

	   #     'symId'	INTEGER NOT NULL,

	   #     'name'	TEXT NOT NULL,

	   #     'type'	TEXT,

	   #     'path'	TEXT,

	   #     'threshold'	NUMERIC NOT NULL DEFAULT 0.4,

	   #     'minMatchPoint'	INTEGER NOT NULL DEFAULT 0,

	   #     'isDetectOrigin'	INTEGER NOT NULL DEFAULT 0,

	   #     'rotationCount'	INTEGER NOT NULL DEFAULT 4,

	   #     'ocrOption'	INTEGER NOT NULL DEFAULT 0,

	   #     'isContainChild'	INTEGER NOT NULL DEFAULT 0,

	   #     'originalPoint'	TEXT,

	   #     'connectionPoint'	TEXT,

	   #     'baseSymbol'	TEXT,

	   #     'additionalSymbol'	TEXT

    #    );

    #'''

    INSERT_SYMBOL_SQL = '''

        INSERT INTO Symbol(symId, name, type, threshold, minMatchPoint, isDetectOrigin, rotationCount, ocrOption, isContainChild, originalPoint, connectionPoint, baseSymbol, additionalSymbol, isExceptDetect) 

        VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);

    '''

    #INSERT_SYMBOL_SQL = '''

    #    INSERT INTO Symbol(symId, name, type, path, threshold, minMatchPoint, isDetectOrigin, rotationCount, ocrOption, isContainChild, originalPoint, connectionPoint, baseSymbol, additionalSymbol) 

    #    VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);

    #'''

    UPDATE_SYMBOL_SQL = '''

        UPDATE Symbol

        SET

            symId = ?, name = ?, type = ?, threshold = ?, minMatchPoint = ?, isDetectOrigin = ?,

            rotationCount = ?, ocrOption = ?, isContainChild = ?, originalPoint = ?, connectionPoint = ?,

            baseSymbol = ?, additionalSymbol = ?, isExceptDetect = ?

        WHERE uid = ?

    '''

    #UPDATE_SYMBOL_SQL = '''

    #    UPDATE Symbol

    #    SET

    #        symId = ?, name = ?, type = ?, path = ?, threshold = ?, minMatchPoint = ?, isDetectOrigin = ?,

    #        rotationCount = ?, ocrOption = ?, isContainChild = ?, originalPoint = ?, connectionPoint = ?,

    #        baseSymbol = ?, additionalSymbol = ?

    #    WHERE uid = ?

    #'''

    '''

        @history    18.04.20    Jeongwoo    Change dbFullPath (Root dir DB→ Each project DB)

                    18.05.03    Jeongwoo    Remove RootDir parameter

    '''

    def __init__(self):

        print("DB Helper __init__")

        self.dbFullPath = AppDocData.instance().getCurrentProject().getDbFilePath() + "/" + self.DB_NAME

        #if not os.path.isfile(dbFullPath):

        #    print("DB doesn't exist")

        try:

            conn = sqlite3.connect(self.dbFullPath)

            print(sqlite3.version)

            cursor = conn.cursor()

            cursor.execute(self.CREATE_SYMBOLS_TABLE_SQL)

        except Error as e:

            print(e)

        finally:

            conn.close()

    def isExistData(self, fieldName, data):

        rows = None

        try:

            conn = sqlite3.connect(self.dbFullPath)

            cursor = conn.cursor()

            sql = ""

            if isinstance(data, str):

                sql = "SELECT * FROM Symbol WHERE " + fieldName + " = '"+ data +"'"

            else:

                sql = "SELECT * FROM Symbol WHERE " + fieldName + " = "+ str(data) +""

            cursor.execute(sql)

            rows = cursor.fetchall()

        except Error as e:

            print(e)

        finally:

            conn.close()

            if rows is not None and len(rows) > 0:

                return True

            else:

                return False

    '''

        @history    2018.05.02  Jeongwoo    Parameter and WHERE changed (path → name)

    '''

    def isExistFileName(self, name):

        rows = None

        try:

            conn = sqlite3.connect(self.dbFullPath)

            cursor = conn.cursor()

            sql = "SELECT * FROM Symbol WHERE name = '"+ name +"'"

            cursor.execute(sql)

            rows = cursor.fetchall()

        except Error as e:

            print(e)

        finally:

            conn.close()

            if rows is not None and len(rows) > 0:

                return True

            else:

                return False

    '''

        @history    2018.05.02  Jeongwoo    Fix bug : symbol.getIsExceptDetect → symbol.getIsExceptDetect()

    '''

    def insertSymbol(self, symbol):

        isAdded = False

        try:

            conn = sqlite3.connect(self.dbFullPath)

            cursor = conn.cursor()

            query = (symbol.getId(), symbol.getName(), symbol.getType(), symbol.getThreshold()

                           , symbol.getMinMatchCount(), symbol.getIsDetectOnOrigin(), symbol.getRotationCount()

                           , symbol.getOcrOption(), symbol.getIsContainChild()

                           , symbol.getOriginalPoint(), symbol.getConnectionPoint()

                           , symbol.getBaseSymbol(), symbol.getAdditionalSymbol(), symbol.getIsExceptDetect())

            cursor.execute(self.INSERT_SYMBOL_SQL, query)

            conn.commit()

            isAdded = True

        except Error as e:

            print(e)

        finally:

            conn.close()

            return (isAdded, symbol.getType(), symbol.getName(), symbol.getPath())

    def updateSymbol(self, symbol):

        isUpdated = False

        try:

            conn = sqlite3.connect(self.dbFullPath)

            cursor = conn.cursor()

            query = (symbol.getId(), symbol.getName(), symbol.getType(), symbol.getThreshold()

                           , symbol.getMinMatchCount(), symbol.getIsDetectOnOrigin(), symbol.getRotationCount()

                           , symbol.getOcrOption(), symbol.getIsContainChild()

                           , symbol.getOriginalPoint(), symbol.getConnectionPoint()

                           , symbol.getBaseSymbol(), symbol.getAdditionalSymbol(), symbol.getIsExceptDetect(), symbol.getUid())

            cursor.execute(self.UPDATE_SYMBOL_SQL, query)

            conn.commit()

            isUpdated = True

        except Error as e:

            print(e)

        finally:

            conn.close()

            return (isUpdated, symbol.getType(), symbol.getName(), symbol.getPath())

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file 'D:\Anaconda3\output\SymbolGenerator.ui'

#

# Created by: PyQt5 UI code generator 5.6

#

# WARNING! All changes made in this file will be lost!

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

from PyQt5.QtGui import *

from QcImageViewer import QcImageViewer

import cv2

from QtImageViewer import QtImageViewer

import os

import sqlite3

from ProjectDialog import Ui_Dialog

#from SymbolEditor import SymbolEditor

import QSymbolEditorDialog

import sys

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

import CropCommand, HandCommand, ZoomCommand, PenCommand, AreaZoomCommand

from AppDocData import AppDocData

DB_NAME = os.path.dirname(os.path.realpath(__file__)) + "\\Project_db.db"

class Ui_MainWindow(object):

    def __init__(self):

        self.project = None

    def setupUi(self, MainWindow):

        #defaultFilePath = "D:\\Visual Studio Project\\DTIPID\\DTIPID\\DTI_PID\\DTI_PID\\res\\UY1-K-2000_P1_300dpi.png"

        #defaultImg = cv2.imread("D:\\Visual Studio Project\\DTIPID\\DTIPID\\DTI_PID\\DTI_PID\\res\\UY1-K-2000_P1_300dpi.png", 1)

        #_chan, self.imgW, self.imgH = defaultImg.shape[::-1]

        MainWindow.setObjectName("MainWindow")

        MainWindow.showFullScreen()

        mainWindowWidth = MainWindow.frameGeometry().width()

        mainWindowHeight = MainWindow.frameGeometry().height()

        #region central widget setting

        self.centralwidget = QtWidgets.QWidget(MainWindow)

        self.centralwidget.setObjectName("centralwidget")

        self.centralwidget.setGeometry(QtCore.QRect(0, 0, mainWindowWidth, mainWindowHeight - 50))

        #endregion

        centralWidth = self.centralwidget.frameGeometry().width()

        centralHeight = self.centralwidget.frameGeometry().height()

        #region top widget setting

        self.topWidget = QtWidgets.QWidget(self.centralwidget)

        self.topWidget.setObjectName("topWidget")

        self.topWidget.setGeometry(QtCore.QRect(0, 0, mainWindowWidth, 40))

        self.topWidget.setLayout(QHBoxLayout())

        self.initToolMenu(self.topWidget)

        #endregion

        #region contents widget setting

        self.contentsWidget = QtWidgets.QWidget(self.centralwidget)

        self.contentsWidget.setObjectName("contentsWidget")

        self.contentsWidget.setGeometry(QtCore.QRect(0, self.topWidget.y()+self.topWidget.height(), mainWindowWidth, centralHeight - 40))

        contentsWidgetWidth = self.contentsWidget.frameGeometry().width()

        contentsWidgetHeight = self.contentsWidget.frameGeometry().height()

        #endregion

        #region right widget setting

        self.rightWidget = QtWidgets.QWidget(self.contentsWidget)

        self.rightWidget.setObjectName("rightWidget")

        self.rightWidget.setGeometry(QtCore.QRect(contentsWidgetWidth*4//5, 0, contentsWidgetWidth//5, contentsWidgetHeight))

        self.rightWidget.setLayout(QVBoxLayout())

        rightWidgetWidth = self.rightWidget.frameGeometry().width()

        rightWidgetHeight = self.rightWidget.frameGeometry().height()

        margin = 6

        rightLayout = self.rightWidget.layout()

        rightLayout.setContentsMargins(margin, 0, margin, 0)

        self.treeAreaWidget = QtWidgets.QWidget(self.rightWidget)

        self.treeAreaWidget.setObjectName("treeWidget")

        self.treeAreaWidget.setGeometry(QtCore.QRect(0, 0, rightWidgetWidth, rightWidgetHeight//2))

        self.treeAreaWidget.setStyleSheet("background-color:white;border:1px solid black;")

        self.treeAreaWidget.setLayout(QVBoxLayout())

        treeAreaWidgetWidth = self.treeAreaWidget.frameGeometry().width()

        treeAreaWidgetHeight = self.treeAreaWidget.frameGeometry().height()

        self.treeWidget = QtWidgets.QTreeWidget(self.treeAreaWidget)

        self.treeWidget.setGeometry(0, 0, treeAreaWidgetWidth-10, treeAreaWidgetHeight)

        self.treeWidget.header().hide()

        self.treeWidget.expandToDepth(0)

        rightLayout.addWidget(self.treeAreaWidget)

        self.propertyWidget = QtWidgets.QWidget(self.rightWidget)

        self.propertyWidget.setObjectName("propertyWidget")

        self.propertyWidget.setGeometry(QtCore.QRect(0, rightWidgetHeight//2, rightWidgetWidth, rightWidgetHeight//2))

        self.propertyWidget.setStyleSheet("background-color:white;border:1px solid black;")

        rightLayout.addWidget(self.propertyWidget)

        #endregion

        #region image widget setting

        self.imageWidget = QtWidgets.QWidget(self.contentsWidget)

        self.imageWidget.setObjectName("imageWidget")

        self.imageWidget.setGeometry(QtCore.QRect(0, 0, contentsWidgetWidth - rightWidgetWidth, contentsWidgetHeight))

        #endregion

        imageWidgetWidth = self.imageWidget.frameGeometry().width()

        imageWidgetHeight = self.imageWidget.frameGeometry().height()

        #region imageView(ImageViewer) setting

        self.imageView = QtImageViewer()

        self.imageView.resize(imageWidgetWidth, imageWidgetHeight)

        self.imageView.aspectRatioMode = QtCore.Qt.KeepAspectRatio

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

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

        self.imageView.canZoom = True

        self.imageView.canPan = True

        #image = QImage(defaultFilePath)

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

        #self.imageView.setImage(image)

        self.imageView.setParent(self.imageWidget)

        #endregion

        MainWindow.setCentralWidget(self.centralwidget)

        #region init Menu bar

        self.menubar = QtWidgets.QMenuBar(MainWindow)

        self.menubar.setGeometry(QtCore.QRect(0, 0, mainWindowWidth, 21))

        self.menubar.setObjectName("menubar")

        self.menuFile = QtWidgets.QMenu(self.menubar)

        self.menuFile.setObjectName("menuFile")

        MainWindow.setMenuBar(self.menubar)

        self.statusbar = QtWidgets.QStatusBar(MainWindow)

        self.statusbar.setObjectName("statusbar")

        MainWindow.setStatusBar(self.statusbar)

        self.actionSelectProject = QtWidgets.QAction(MainWindow)

        self.actionSelectProject.setObjectName("actionSelectProject")

        self.actionOpenImage = QtWidgets.QAction(MainWindow)

        self.actionOpenImage.setObjectName("actionOpenImage")

        self.actionSave = QtWidgets.QAction(MainWindow)

        self.actionSave.setObjectName("actionSave")

        self.actionExit = QtWidgets.QAction(MainWindow)

        self.actionExit.setObjectName("actionExit")

        self.menuFile.addAction(self.actionSelectProject)

        self.menuFile.addAction(self.actionOpenImage)

        self.menuFile.addAction(self.actionSave)

        self.menuFile.addAction(self.actionExit)

        self.menubar.addAction(self.menuFile.menuAction())

        self.actionSelectProject.triggered.connect(self.actionSelectProjectClicked)

        self.actionOpenImage.triggered.connect(self.actionOpenImageClicked)

        self.actionExit.triggered.connect(qApp.quit)

        self.retranslateUi(MainWindow)

        #endregion

        QtCore.QMetaObject.connectSlotsByName(MainWindow)

        project = self.showProjectSelectionDialog()

        self.initTreeWidget(project)

    def showProjectSelectionDialog(self):

        self.projectDialog = Ui_Dialog()

        project = self.projectDialog.showDialog()

        return project

    def retranslateUi(self, MainWindow):

        _translate = QtCore.QCoreApplication.translate

        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))

        self.menuFile.setTitle(_translate("MainWindow", "File"))

        self.actionSelectProject.setText(_translate("MainWindow", "Select Project"))

        self.actionOpenImage.setText(_translate("MainWindow", "Open Image"))

        self.actionSave.setText(_translate("MainWindow", "Save"))

        self.actionExit.setText(_translate("MainWindow", "Exit"))

    def makeChildDir(self):

        project = AppDocData.instance().getCurrentProject()

        dbDir = project.getDbFilePath()

        if not os.path.exists(dbDir):

            os.makedirs(dbDir)

        imgDir = project.getImageFilePath()

        if not os.path.exists(imgDir):

            os.makedirs(imgDir)

        svgDir = project.getSvgFilePath()

        if not os.path.exists(svgDir):

            os.makedirs(svgDir)

    def loadDirectoryInfo(self, startPath, tree):

        for element in os.listdir(startPath):

            pathInfo = startPath + "/" + element

            parentItem = QTreeWidgetItem(tree, [os.path.basename(element)])

            if os.path.isdir(pathInfo):

                self.loadDirectoryInfo(pathInfo, parentItem)

    def initTreeWidget(self, project):

        if project is not None:

            self.treeWidget.clear()

            self.project = project

            print("Project : " + self.project.getName())

            projectPath = self.project.getPath().replace("\\", "/")

            self.makeChildDir()

            self.loadDirectoryInfo(projectPath, self.treeWidget)

    def actionSelectProjectClicked(self):

        project = self.showProjectSelectionDialog()

        self.initTreeWidget(project)

    def actionOpenImageClicked(self):

        options = QFileDialog.Options()

        options |= QFileDialog.DontUseNativeDialog

        fileName, _ = QFileDialog.getOpenFileName(None, "QFileDialog.getOpenFileName()", "D:\\Visual Studio Project\\DTIPID\\DTIPID\\DTI_PID\\DTI_PID\\res\\Symbols\\UY1_300dpi", "Image Files (*.png)", options=options)

        if fileName:

            print(fileName)

            image = QImage(fileName).copy()

            self.imgW = image.width()

            self.imgH = image.height()

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

            self.imageView.setImage(image)

    def initToolMenu(self, topWidget):

        layout = topWidget.layout()

        handTool = QPushButton()

        handTool.setText("Hand Tool")

        handTool.clicked.connect(self.handToolClickEvent)

        layout.addWidget(handTool)

        cropTool = QPushButton()

        cropTool.setText("Crop Tool")

        cropTool.clicked.connect(self.cropToolClickEvent)

        layout.addWidget(cropTool)

        areaZoomTool = QPushButton()

        areaZoomTool.setText("Area Zoom Tool")

        areaZoomTool.clicked.connect(self.areaZoomToolClickEvent)

        layout.addWidget(areaZoomTool)

        zoomTool = QPushButton()

        zoomTool.setText("Zoom Tool")

        zoomTool.clicked.connect(self.zoomToolClickEvent)

        layout.addWidget(zoomTool)

        zoomInitTool = QPushButton()

        zoomInitTool.setText("Zoom Init Tool")

        zoomInitTool.clicked.connect(self.zoomInitToolClickEvent)

        layout.addWidget(zoomInitTool)

        nextStepTool = QPushButton()

        nextStepTool.setText("Next Step")

        nextStepTool.clicked.connect(self.nextStepToolClickEvent)

        layout.addWidget(nextStepTool)

    def handToolClickEvent(self, event):

        print("hand tool clicked")

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

    def cropToolClickEvent(self, event):

        print("crop tool clicked")

        self.imageView.command = CropCommand.CropCommand(self.imageView)

    def zoomInitToolClickEvent(self, event):

        print("zoom init tool clicked")

        self.imageView.command = None

        self.imageView.zoomImageInit()

    def zoomToolClickEvent(self, event):

        print("zoom tool clicked")

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

    def areaZoomToolClickEvent(self, event):

        print("area zoom tool clicked")

        self.imageView.command = AreaZoomCommand.AreaZoomCommand(self.imageView)

    '''

        @history    2018.05.02  Jeongwoo    Return value changed (Single variable → Tuple)

    '''

    def nextStepToolClickEvent(self, event):

        print("next step tool clicked")

        if self.imageView.hasImage():

            self.symbolEditorDialog = QSymbolEditorDialog.QSymbolEditorDialog(self.imageView, self.imageView.image(), self.project)

            (isAccepted, isImmediateInsert, offsetX, offsetY, newSym) = self.symbolEditorDialog.showDialog()

            if isAccepted:

                self.initTreeWidget(self.project)

        else:

            QMessageBox.about(self.topWidget, "알림", "이미지를 선택한 후 다시 시도해주세요.")

if __name__ == "__main__":

    import sys

    app = QtWidgets.QApplication(sys.argv)

    MainWindow = QtWidgets.QMainWindow()

    ui = Ui_MainWindow()

    ui.setupUi(MainWindow)

    MainWindow.show()

    sys.exit(app.exec_())