HYTOS

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

""" This is library module """

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from PyQt5.QtSvg import *

class LibraryItemWidget(QWidget):

    COLUMN = 3

    def __init__(self, parent=None):

        super().__init__(parent)

        self.setAcceptDrops(True)

        self.layout = QGridLayout()

        self.setLayout(self.layout)

    def remove_library(self, item):

        matches = [idx for idx in range(self.layout.count()) if self.layout.itemAt(idx).widget() is item]

        if matches:

            layout_item = self.layout.takeAt(matches[0])

            self.layout.removeWidget(item)

            item.deleteLater()

            del item

            self.layout.removeItem(layout_item)

            del layout_item

            self.layout.update()

    def dragEnterEvent(self, event: QDragEnterEvent) -> None:

        if issubclass(type(event.source()), QTreeWidget) and event.mimeData().hasFormat('text/plain'):

            # check if symbol is already registered

            for idx in range(self.layout.count()):

                widget = self.layout.itemAt(idx).widget()

                id = f"{event.mimeData().tag.getType()}.{event.mimeData().tag.getName()}"

                if widget.id == id:

                    event.ignore()

                    return

            # up to here

            event.accept()

        else:

            event.ignore()

    def dropEvent(self, event: QDropEvent) -> None:

        if hasattr(event.mimeData(), 'tag'):

            item = LibraryItem(event.mimeData().tag)

            row = int((self.layout.count()) / LibraryItemWidget.COLUMN)

            col = (self.layout.count()) % LibraryItemWidget.COLUMN

            self.layout.addWidget(item, row, col)

            event.acceptProposedAction()

class LibraryItem(QFrame):

    def __init__(self, symbol):

        QFrame.__init__(self)

        self.id = f"{symbol.getType()}.{symbol.getName()}"

        self._symbol = symbol

        # An Icon and a label below

        icon = QLabel()

        icon.resize(symbol.pixmap.width(), symbol.pixmap.height())

        icon.setPixmap(symbol.pixmap)

        layout = QGridLayout()

        layout.addWidget(icon, 0, 0, Qt.AlignHCenter)

        title = QLabel(symbol.getName())

        font = title.font()

        font.setPixelSize(10)

        title.setFont(font)

        layout.addWidget(title, 1, 0, Qt.AlignTop | Qt.AlignHCenter)

        self.setLayout(layout)

        self.setMaximumSize(96, 96)

    @property

    def symbol(self):

        return self._symbol

    def keyPressEvent(self, event: QKeyEvent) -> None:

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

            self.parentWidget().remove_library(self)

            return

        super(LibraryItem, self).keyPressEvent(event)

    """

    Drag and drop management

    """

    def enterEvent(self, event):

        self.setFrameStyle(QFrame.Panel | QFrame.Sunken)

        self.setCursor(Qt.OpenHandCursor)

        self.setFocus()

    def leaveEvent(self, event):

        self.setFrameStyle(QFrame.NoFrame)

        self.setCursor(Qt.ArrowCursor)

    def mousePressEvent(self, mouseEvent):

        self.setCursor(Qt.ClosedHandCursor)

    def mouseReleaseEvent(self, mouseEvent):

        self.setCursor(Qt.OpenHandCursor)

    def mouseMoveEvent(self, event):

        """Drag-n-Drop on left click"""

        if event.buttons() != Qt.LeftButton:

            return

        mime = QMimeData()

        mime.setText(self._symbol.getName())

        drag = QDrag(self)

        drag.setMimeData(mime)

        originalPoint = self._symbol.getOriginalPoint()

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

        drag.setPixmap(self._symbol.pixmap)

        drag.exec(Qt.CopyAction)