- Basic polygon to body

drawing_modules/draw_widget2d.py

@@ -0,0 +1,79 @@
+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())