HYTOS

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 *

import os

import sys

import SymbolBase

import symbol

import SymbolEditorDialog

import QSymbolDisplayDialog

class QSymbolTreeWidget(QTreeWidget):

    #Add signal

    singleClicked = pyqtSignal(SymbolBase.SymbolBase)

    TREE_DATA_ROLE = Qt.UserRole

    def __init__(self):

        QTreeWidget.__init__(self)

        self.setDragEnabled(True)   # enable drag

        self.initDirTreeWidget()

        self.isDoubleClicked = False

        self.itemDoubleClicked.connect(self.itemDoubleClickEvent)

        self.setContextMenuPolicy(Qt.CustomContextMenu)

        self.customContextMenuRequested.connect(self.openContextMenu)

        self.currentItemChanged.connect(self.onCurrentItemChanged)

    '''

        @brief      Show Context Menu

        @author     Jeongwoo

        @date       18.04.??

        @history    Jeongwoo 2018.04.23 Symbol object Null Check when show context menu

                    humkyung 2018.08.14 add menu for symbol type

    '''

    def openContextMenu(self, position):

        indexes = self.selectedIndexes()

        itemPosition = self.mapTo(self, position)

        item = self.itemAt(itemPosition)

        data = item.data(0, self.TREE_DATA_ROLE)

        if data is not None and type(data) is symbol.SymbolBase:

            sym = self.getSymbolByItemName(item, 0)

            text = item.text(0)

            if len(indexes) > 0:

                level = 0

                index = indexes[0]

                while index.parent().isValid():

                    index = index.parent()

                    level += 1

            if sym is not None:

                menu = QMenu()

                editSymbolAction = QAction(self.tr("Edit Symbol"))

                editSymbolAction.triggered.connect(lambda: self.editSymbolActionClickEvent(item, 0))

                menu.addAction(editSymbolAction)

                displaySymbolAction = QAction(self.tr("Display Symbol"))

                displaySymbolAction.triggered.connect(lambda: self.displaySymbolActionClickEvent(sym.getType(), text))

                menu.addAction(displaySymbolAction)

                deleteSymbolAction = QAction(self.tr("Delete Symbol"))

                deleteSymbolAction.triggered.connect(lambda: self.deleteSymbolActionClickEvent(sym.getType(), text))

                menu.addAction(deleteSymbolAction)

                menu.exec_(self.viewport().mapToGlobal(position))

        else:

            menu = QMenu()

            editAttrAction = QAction(self.tr("Edit Attribute"))

            editAttrAction.triggered.connect(lambda: self.onEditAttrClicked(item, 0))

            menu.addAction(editAttrAction)

            menu.exec_(self.viewport().mapToGlobal(position))

    def editSymbolActionClickEvent(self, item, columNo):

        self.showSymbolEditorDialog(item, columNo)

    '''

        @brief      popup attribute editor dialog

        @author     humkyung

        @date       2018.08.13

    '''

    def onEditAttrClicked(self, item, columnNo):

        from SymbolAttrEditorDialog import QSymbolAttrEditorDialog

        try:

                data = item.data(0, self.TREE_DATA_ROLE)

                dlg = QSymbolAttrEditorDialog(self, data)

                dlg.exec_()

        except Exception as ex:

            from App import App

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    def displaySymbolActionClickEvent(self, itemType, itemName):

        project = AppDocData.instance().getCurrentProject()

        image = QImage(os.path.join(project.getImageFilePath(), itemType, itemName, "PNG")) #itemName includes ".png"

        dialog = QSymbolDisplayDialog.QSymbolDisplayDialog(image)

        dialog.showDialog()

    def deleteSymbolActionClickEvent(self, itemType, itemName):

        msg = QMessageBox()

        msg.setIcon(QMessageBox.Critical)

        msg.setText(self.tr('Are you sure you want to delete selected symbol?\nData can not be restored!'))

        msg.setWindowTitle(self.tr('Delete symbol'))

        msg.setStandardButtons(QMessageBox.Ok|QMessageBox.Cancel)

        result = msg.exec_()

        self.handleDeleteSymbolAction(result, itemType, itemName)

    '''

        @history    2018.05.03  Jeongwoo    Modify file path with ".png" and ".svg"

                                            Use project object when making svgPath

    '''

    def handleDeleteSymbolAction(self, result, itemType, itemName):

        if result == QMessageBox.Ok:

            project = AppDocData.instance().getCurrentProject()

            imagePath = os.path.join(project.getImageFilePath(), itemType, itemName  + '.png') # itemName DOESN'T includes ".png"

            if os.path.exists(imagePath):

                os.remove(imagePath)

            svgPath = os.path.join(project.getSvgFilePath(), itemType, itemName + '.svg')

            if os.path.exists(svgPath):

                os.remove(svgPath)

            AppDocData.instance().deleteSymbol(itemName)

            self.initDirTreeWidget()

        else:

            pass

    '''

        @history    2018.05.02  Jeongwoo    Change return value of QSymbolEditorDialog (Single variable → Tuple)

    '''

    def initDirTreeWidget(self):

        project = AppDocData.instance().getCurrentProject()

        if project is not None:

            self.clear()

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

            self.makeChildDir()

            self.loadSymbolInfo()

            self.expandAll()

    '''

        @brief      Load Symbol Info and add TreeItem with DB

        @author     Jeongwoo

        @date       18.04.20

        @history    Jeongwoo 2018.05.03 Get Svg File Path by SymbolBase.getSvgFileFullPath()

                    humkyung 2018.07.30 sort child items

    '''

    def loadSymbolInfo(self):

        try:

            symbolCategoryList = AppDocData.instance().getSymbolCategoryList()

            for symbolCategory in symbolCategoryList:

                parent = QTreeWidgetItem(self, [symbolCategory])

                parent.setData(0, self.TREE_DATA_ROLE, symbolCategory)

                symbolTypeList = AppDocData.instance().getSymbolTypeListByCategory(symbolCategory)

                for symbolType in symbolTypeList:

                    category = QTreeWidgetItem(parent, [symbolType[1]])

                    category.setData(0, self.TREE_DATA_ROLE, symbolType)

                    symbolList = AppDocData.instance().getSymbolListByType('type', symbolType[1])

                    for symbol in symbolList:

                        symbolItem = QTreeWidgetItem(category, [symbol.getName()])

                        symbolItem.setData(0, self.TREE_DATA_ROLE, symbol)

                        svgPath = symbol.getSvgFileFullPath()

                        icon = QIcon(svgPath)

                        symbolItem.setIcon(0, icon)

                        symbolItem.svgFilePath = svgPath # save svg file path

                    parent.sortChildren(0, Qt.AscendingOrder)

        except Exception as ex:

            from App import App

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    '''

        @brief      Make Directory

        @author     Jeongwoo

        @date       18.04.??

        @history    18.04.12    Jeongwoo    Add output, temp Directory

    '''

    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)

        outputDir = project.getOutputPath()

        if not os.path.exists(outputDir):

            os.makedirs(outputDir)

        tempDir = project.getTempPath()

        if not os.path.exists(tempDir):

            os.makedirs(tempDir)

    '''

        @history    2019.07.11  yeonjin    심벌 편집 시 png 파일이 아닌 svg 파일을 불러 오도록 수정

    '''

    def showSymbolEditorDialog(self, item, columnNo):

        try:

            sym = self.getSymbolByItemName(item, columnNo)

            if sym is not None:

                #path = sym.getPath()

                #image = QImage(path, "PNG")

                path = sym.getSvgFileFullPath()

                image = QImage(path, "SVG")

                symbolEditorDialog = SymbolEditorDialog.QSymbolEditorDialog(self, image, AppDocData.instance().getCurrentProject(), sym)

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

                self.initDirTreeWidget()

            else:

                QMessageBox.about(self, self.tr('Error'), self.tr('Error occurs during loading symbol data.'))

        except Exception as ex:

            from App import App

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    def itemDoubleClickEvent(self, item, columnNo):

        self.isDoubleClicked = True

        sym = self.getSymbolByItemName(item, columnNo)

        itemName = item.text(columnNo)

        if sym is not None:

            self.showSymbolEditorDialog(item, columnNo)

        self.isDoubleClicked = False

    '''

        @brief      Get Symbol data by symbol name

        @author     Jeongwoo

        @date       18.04.20

    '''

    def getSymbolByItemName(self, item, columnNo):

        itemName = item.text(columnNo)

        name = itemName

        sym = AppDocData.instance().getSymbolByQuery("name", name)

        return sym

    '''

        @breif      show symbol's property when selection changed

        @author     humkyung

        @date       2018.07.30

    '''

    def onCurrentItemChanged(self, current, previous):

        item = self.currentItem()

        if item is not None:

            data = item.data(0, self.TREE_DATA_ROLE)

            if data is not None and type(data) is symbol.SymbolBase:

                self.singleClicked.emit(data)

    '''

        @brief  start drag

        @author humkyung

        @date   2018.04.17

        @history    18.04.20    Jeongwoo    Change Path in QPixmap

                    18.06.21    Jeongwoo    Casting string to float and int

    '''

    def startDrag(self, dropAction):

        try:

            items = self.selectedItems()

            if items and hasattr(items[0], 'svgFilePath'):

                symData = items[0].data(0, self.TREE_DATA_ROLE)

                pixmap = QPixmap(items[0].svgFilePath)

                mime = QMimeData()

                mime.setText(symData.getName())

                drag = QDrag(self)

                drag.setMimeData(mime) 

                originalPoint = symData.getOriginalPoint()

                drag.setHotSpot(QPoint(int(float(originalPoint.split(",")[0])), int(float(originalPoint.split(",")[1]))))

                drag.setPixmap(pixmap)        

                drag.exec(Qt.CopyAction)

        except Exception as ex:

            from App import App

            message = 'error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)