Использование библиотеки PyQt5

Создание блок-схемы на основе кода

500

Нужно сделать блок-схему как в файле, используя мой код: import sys import os from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QLabel, QLineEdit, QPushButton, QHBoxLayout) from PyQt5.QtCore import Qt, QTimer, QPointF, QLibraryInfo from PyQt5.QtGui import QPainter, QPen, QColor, QPolygonF, QFont, QTransform, QMouseEvent def configure_qt_plugins(): qt_plugin_path = os.path.join(os.path.dirname(QLibraryInfo.location(QLibraryInfo.PrefixPath)), 'plugins') os.environ['QT_QPA_PLATFORM_PLUGIN_PATH'] = qt_plugin_path possible_paths = [ os.path.join(sys.prefix, 'Lib', 'site-packages', 'PyQt5', 'Qt5', 'plugins'), os.path.join(sys.prefix, 'Library', 'plugins'), 'C:\\Program Files\\Python38\\Lib\\site-packages\\PyQt5\\Qt5\\plugins', '/usr/lib/x86_64-linux-gnu/qt5/plugins', '/usr/local/lib/qt5/plugins' ] for path in possible_paths: if os.path.exists(path): os.environ['QT_QPA_PLATFORM_PLUGIN_PATH'] = path break os.environ['QT_DEBUG_PLUGINS'] = '1' configure_qt_plugins() class BezierCurveWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle("Кривая Безье 4-го порядка") self.setGeometry(100, 100, 800, 600) self.central_widget = QWidget() self.setCentralWidget(self.central_widget) self.layout = QVBoxLayout() self.central_widget.setLayout(self.layout) # Поля для ввода контрольных точек self.point_inputs = [] for i in range(5): hbox = QHBoxLayout() label = QLabel(f"Точка P{i + 1}:") hbox.addWidget(label) input_x = QLineEdit() input_x.setPlaceholderText("x") input_x.setFixedWidth(60) hbox.addWidget(input_x) input_y = QLineEdit() input_y.setPlaceholderText("y") input_y.setFixedWidth(60) hbox.addWidget(input_y) self.point_inputs.append((input_x, input_y)) self.layout.addLayout(hbox) # Панель управления control_panel = QHBoxLayout() # Кнопка очистки экрана self.clear_button = QPushButton("Очистить экран") self.clear_button.clicked.connect(self.clear_screen) control_panel.addWidget(self.clear_button) self.layout.addLayout(control_panel) # Создаем canvas с математической системой координат self.canvas = BezierCanvas(self, self.point_inputs) self.layout.addWidget(self.canvas) # Устанавливаем точки по умолчанию self.default_points = [ QPointF(-200, 100), QPointF(-100, -100), QPointF(0, 0), QPointF(100, 100), QPointF(200, -100) ] self.canvas.set_control_points(self.default_points.copy()) self.update_point_inputs() # Кнопки управления кривой curve_panel = QHBoxLayout() self.draw_button = QPushButton("Построить кривую") self.draw_button.clicked.connect(self.update_curve) curve_panel.addWidget(self.draw_button) self.animate_button = QPushButton("Анимировать") self.animate_button.clicked.connect(self.toggle_animation) curve_panel.addWidget(self.animate_button) # Кнопки масштабирования self.zoom_in_button = QPushButton("+") self.zoom_in_button.clicked.connect(self.zoom_in) curve_panel.addWidget(self.zoom_in_button) self.zoom_out_button = QPushButton("-") self.zoom_out_button.clicked.connect(self.zoom_out) curve_panel.addWidget(self.zoom_out_button) self.layout.addLayout(curve_panel) # Анимация self.animation_timer = QTimer() self.animation_timer.timeout.connect(self.animate) self.animation_step = 0 self.animation_direction = 1 def clear_screen(self): if self.animation_timer.isActive(): self.animation_timer.stop() self.animate_button.setText("Анимировать") self.animation_step = 0 self.canvas.clear_all() self.canvas.update() def zoom_in(self): self.canvas.zoom(1.2) def zoom_out(self): self.canvas.zoom(0.8) def update_point_inputs(self): for i, (x_input, y_input) in enumerate(self.point_inputs): if i < len(self.canvas.control_points): x_input.setText(str(int(self.canvas.control_points[i].x()))) y_input.setText(str(int(self.canvas.control_points[i].y()))) def update_curve(self): points = [] for x_input, y_input in self.point_inputs: try: x = float(x_input.text()) y = float(y_input.text()) points.append(QPointF(x, y)) except ValueError: pass if len(points) == 5: self.canvas.set_control_points(points) self.canvas.update() def toggle_animation(self): if self.animation_timer.isActive(): self.animation_timer.stop() self.animate_button.setText("Анимировать") else: self.animation_step = 0 self.animation_timer.start(50) self.animate_button.setText("Остановить") def animate(self): self.animation_step += 0.01 * self.animation_direction if self.animation_step >= 1 or self.animation_step <= 0: self.animation_direction *= -1 self.canvas.set_animation_progress(self.animation_step) self.canvas.update() class BezierCanvas(QWidget): def __init__(self, parent=None, point_inputs=None): super().__init__(parent) self.control_points = [] self.animation_progress = 0 self.canvas_offset = QPointF(0, 0) self.setMinimumSize(600, 400) self.point_radius = 8 self.show_coordinates = True self.point_inputs = point_inputs self.center_x = self.width() // 2 self.center_y = self.height() // 2 self.scale_factor = 1.0 self.animation_object_pos = QPointF(0, 0) self.animation_object_direction = QPointF(1, 1) self.animation_object_radius = 10 self.show_lines = True self.show_curve = True self.show_points = True # Переменные для перемещения холста self.dragging_canvas = False self.last_mouse_pos = None # Фиксированный размер сетки self.grid_size = 20 self.grid_color = QColor(220, 220, 220, 150) def clear_all(self): self.animation_progress = 0 self.animation_object_pos = QPointF(0, 0) self.show_lines = False self.show_curve = False self.show_points = False def reset_all(self): self.show_lines = True self.show_curve = True self.show_points = True def zoom(self, factor): self.scale_factor *= factor self.update() def set_control_points(self, points): self.control_points = points.copy() self.reset_all() self.update() def set_animation_progress(self, progress): self.animation_progress = progress self.update_animation_object() self.update() def update_animation_object(self): if len(self.control_points) >= 5: t = self.animation_progress self.animation_object_pos = self.calculate_bezier_point(t) if (t > 0.01) and (t < 0.99): next_pos = self.calculate_bezier_point(t + 0.01) prev_pos = self.calculate_bezier_point(t - 0.01) self.animation_object_direction = next_pos - prev_pos self.animation_object_direction /= (self.animation_object_direction.manhattanLength() + 0.0001) def mousePressEvent(self, event: QMouseEvent): if event.button() == Qt.LeftButton: self.dragging_canvas = True self.last_mouse_pos = event.pos() def mouseMoveEvent(self, event: QMouseEvent): if self.dragging_canvas and self.last_mouse_pos is not None: delta = event.pos() - self.last_mouse_pos self.canvas_offset += delta self.last_mouse_pos = event.pos() self.update() def mouseReleaseEvent(self, event: QMouseEvent): self.dragging_canvas = False self.last_mouse_pos = None def paintEvent(self, event): painter = QPainter(self) painter.setRenderHint(QPainter.Antialiasing) # Применяем масштабирование painter.translate(int(self.center_x + self.canvas_offset.x()), int(self.center_y + self.canvas_offset.y())) painter.scale(self.scale_factor, -self.scale_factor) # Рисуем статичную сетку painter.save() painter.setTransform(QTransform()) painter.setPen(QPen(self.grid_color, 1)) # Вертикальные линии x = (self.center_x + self.canvas_offset.x()) % (self.grid_size * self.scale_factor) while x < self.width(): painter.drawLine(int(x), 0, int(x), self.height()) x += self.grid_size * self.scale_factor # Горизонтальные линии y = (self.center_y + self.canvas_offset.y()) % (self.grid_size * self.scale_factor) while y < self.height(): painter.drawLine(0, int(y), self.width(), int(y)) y += self.grid_size * self.scale_factor painter.restore() # Рисуем оси координат axis_length = 300 painter.setPen(QPen(Qt.black, 2 / self.scale_factor)) painter.drawLine(int(-axis_length), 0, int(axis_length), 0) # Ось X painter.drawLine(0, int(-axis_length), 0, int(axis_length)) # Ось Y # Стрелки осей arrow_size = 8 / self.scale_factor painter.drawLine( int(axis_length - arrow_size), int(-arrow_size), int(axis_len

Артем Ермолин