80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
from PySide6.QtWidgets import QApplication, QWidget, QMessageBox
|
|
from PySide6.QtGui import QPainter, QPen, QColor
|
|
from PySide6.QtCore import Qt, QPoint
|
|
|
|
|
|
class SnapLineWidget(QWidget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.points = []
|
|
self.selected_line = None
|
|
self.snapping_range = 20 # Range in pixels for snapping
|
|
|
|
def mousePressEvent(self, event):
|
|
if event.button() == Qt.LeftButton:
|
|
self.points.append(event.pos())
|
|
self.update()
|
|
|
|
if event.button() == Qt.RightButton:
|
|
for i in range(len(self.points) - 1):
|
|
if self.is_point_on_line(event.pos(), self.points[i], self.points[i + 1]):
|
|
self.selected_line = i
|
|
break
|
|
else:
|
|
self.selected_line = None
|
|
|
|
def mouseMoveEvent(self, event):
|
|
if event.buttons() & Qt.LeftButton:
|
|
if self.selected_line is not None:
|
|
self.points[self.selected_line] = event.pos()
|
|
else:
|
|
self.points[-1] = event.pos()
|
|
self.update()
|
|
|
|
def mouseDoubleClickEvent(self, event):
|
|
pass
|
|
|
|
def distance(self, p1, p2):
|
|
return ((p1.x() - p2.x()) ** 2 + (p1.y() - p2.y()) ** 2) ** 0.5
|
|
|
|
def is_point_on_line(self, p, p1, p2):
|
|
distance1 = self.distance(p, p1)
|
|
distance2 = self.distance(p, p2)
|
|
total_distance = self.distance(p1, p2)
|
|
return abs(distance1 + distance2 - total_distance) < 0.1
|
|
|
|
def paintEvent(self, event):
|
|
painter = QPainter(self)
|
|
|
|
# Set the background color
|
|
painter.fillRect(self.rect(), QColor('white'))
|
|
|
|
pen = QPen(Qt.black)
|
|
pen.setWidth(1)
|
|
painter.setPen(pen)
|
|
|
|
for i in range(len(self.points) - 1):
|
|
painter.drawLine(self.points[i], self.points[i + 1])
|
|
|
|
# Draw points
|
|
for point in self.points:
|
|
painter.drawEllipse(point, 3, 3)
|
|
|
|
if self.selected_line is not None:
|
|
p1 = self.points[self.selected_line]
|
|
p2 = self.points[self.selected_line + 1]
|
|
painter.setPen(QPen(Qt.red, 2))
|
|
painter.drawLine(p1, p2)
|
|
|
|
|
|
# Example usage
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
app = QApplication(sys.argv)
|
|
window = SnapLineWidget()
|
|
window.setWindowTitle("Snap Line Widget")
|
|
window.resize(800, 600)
|
|
window.show()
|
|
sys.exit(app.exec())
|