small_obj_augmentation/DragLabel.py at master · Histamine03/small_obj_augmentation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from PyQt5.QtWidgets import QLabel
from PyQt5.QtGui import QPixmap, QDrag
from PyQt5.QtCore import QByteArray, QBuffer, Qt, QMimeData


# <DraggableLabel 클래스>
class DraggableLabel(QLabel):
    def __init__(self, pixmap, object_info, parent=None):
        super().__init__(parent)
        self.setPixmap(pixmap)
        self.setAcceptDrops(True)
        self.object_info = object_info

    def mousePressEvent(self, event):
        if event.button() == Qt.LeftButton:
# 드래그 시작
            drag = QDrag(self)
            mime_data = QMimeData()

# 이미지를 QByteArray로 변환하여 mime_data에 저장
            byte_array = QByteArray()
            buffer = QBuffer(byte_array)
            buffer.open(QBuffer.WriteOnly)
            self.pixmap().save(buffer, "PNG")
            mime_data.setData("application/x-dnditemdata", byte_array)

            drag.setMimeData(mime_data)
            drag.setPixmap(self.pixmap().scaled(100, 100, Qt.KeepAspectRatio))

# 크기 변환
            scaled_pixmap = self.pixmap().scaled(100, 100, Qt.KeepAspectRatio)
            drag.setPixmap(scaled_pixmap)

            hot_spot = event.pos() - self.rect().topLeft()
            drag.setHotSpot(hot_spot)
            drag.exec_(Qt.CopyAction | Qt.MoveAction)

    # def dragEnterEvent(self, event):
    #     if event.mimeData().hasFormat("application/x-dnditemdata"):
    #         event.acceptProposedAction()

    # def dropEvent(self, event):
    #     if event.mimeData().hasFormat("application/x-dnditemdata"):

    #         # 드롭된 위치의 좌표 값을 얻기
    #         drop_position = event.pos()
    #         print("Dropped at:", drop_position.x(), drop_position.y())

    #         # 드롭된 이미지를 QByteArray에서 QPixmap으로 변환
    #         byte_array = event.mimeData().data("application/x-dnditemdata")
    #         buffer = QBuffer(byte_array)
    #         buffer.open(QBuffer.ReadOnly)
    #         pixmap = QPixmap()
    #         pixmap.loadFromData(buffer.data(), "PNG")

    #         # 드롭된 위치에 라벨을 생성하고 이미지를 설정
    #         label = QLabel(self)
    #         label.setPixmap(pixmap)
    #         label.move(drop_position)
    #         label.show()

    #         event.acceptProposedAction()