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.setIconSize(QSize(32, 32))

    '''

        @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 occurred({}) 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):

        self.clear()

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

        from PyQt5.QtSvg import QSvgRenderer

        try:

            app_doc_data = AppDocData.instance()

            symbolCategoryList = app_doc_data.getSymbolCategoryList()

            for symbolCategory in symbolCategoryList:

                if symbolCategory == 'Stream Line':

                    continue

                parent = QTreeWidgetItem(self, [symbolCategory])

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

                symbolTypeList = app_doc_data.getSymbolTypeListByCategory(symbolCategory)

                for symbolType in symbolTypeList:

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

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

                    symbolList = app_doc_data.getSymbolListByUID(symbolType[0])

                    for symbol in symbolList:

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

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

                        shape = app_doc_data.read_symbol_shape(app_doc_data.activeDrawing.path, symbol.uid)

                        if shape:

                            pixmap = QPixmap(QSize(32, 32))

                            pixmap.loadFromData(shape)

                            icon = QIcon(pixmap)

                            symbolItem.setIcon(0, icon)

                            symbolItem.svgFilePath = None  # reset svg file path

                        else:

                            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 occurred({}) 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  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)

                if items[0].svgFilePath:

                    pixmap = QPixmap(items[0].svgFilePath)

                else:

                    app_doc_data = AppDocData.instance()

                    shape = app_doc_data.read_symbol_shape(app_doc_data.activeDrawing.path, symData.uid)

                    if shape:

                        pixmap = QPixmap()

                        pixmap.loadFromData(shape)

                mime = QMimeData()

                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 occurred({}) in {}:{}'.format(repr(ex), sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

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