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

        #self.itemDoubleClicked.connect(self.itemDoubleClickEvent)

    '''

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

    '''

    def showSymbolEditorDialog(self, sym):

        try:

            path = sym.getSvgFileFullPath()

            image = QImage(path, "SVG")

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

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

            self.initSymbolTreeWidget()

        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 getSymbolByItemUID(self, item):

        uid = item.text(1)

        return AppDocData.instance().getSymbolByQuery("uid", uid)

    def itemDoubleClickEvent(self, item, columnNo):

        sym = self.getSymbolByItemUID(item)

        if sym is not None:

            self.showSymbolEditorDialog(sym)

    '''

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

    '''

    def initSymbolTreeWidget(self):

        project = AppDocData.instance().getCurrentProject()

        if project is not None:

            self.clear()

            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:

                if symbolCategory == 'Stream Line':

                    continue

                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[2]])

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

                    symbolList = AppDocData.instance().getSymbolListByUID(symbolType[0])

                    for symbol in symbolList:

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

                        symbolItem = QTreeWidgetItem(category, [symbol.sName, symbol.uid])

                        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)

    '''

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

                mime.setText(symData.getUid())

                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)