307 lines
11 KiB
Python
307 lines
11 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
from PyQt5.QtWidgets import (
|
|
QApplication, QDialog, QPushButton, QVBoxLayout, QLineEdit, QHBoxLayout,
|
|
QGraphicsView, QGraphicsScene, QGraphicsPixmapItem, QGraphicsItem, QMessageBox,
|
|
QMenuBar, QMenu, QAction
|
|
)
|
|
from PyQt5.QtGui import QPixmap, QImage, QColor, QPainter, QTransform
|
|
from PyQt5.QtCore import pyqtSignal, Qt, QRectF, QPointF
|
|
|
|
|
|
class CustomGraphicsView(QGraphicsView):
|
|
def __init__(self, scene, parent=None):
|
|
super().__init__(scene, parent)
|
|
self.zoom_step = 1.1
|
|
self.setTransformationAnchor(QGraphicsView.AnchorUnderMouse)
|
|
self.setResizeAnchor(QGraphicsView.AnchorUnderMouse)
|
|
self.setViewportUpdateMode(QGraphicsView.FullViewportUpdate)
|
|
self.setFocusPolicy(Qt.StrongFocus) # 确保视图可以接收焦点
|
|
|
|
def wheelEvent(self, event):
|
|
modifiers = QApplication.keyboardModifiers()
|
|
# 修改为无需按下 Alt 键即可缩放视图
|
|
if modifiers == Qt.NoModifier:
|
|
delta = event.angleDelta().y()
|
|
if delta > 0:
|
|
factor = self.zoom_step
|
|
else:
|
|
factor = 1 / self.zoom_step
|
|
self.scale(factor, factor)
|
|
event.accept()
|
|
else:
|
|
super().wheelEvent(event)
|
|
|
|
|
|
class CanvasWindow(QDialog):
|
|
save_signal = pyqtSignal()
|
|
close_signal = pyqtSignal()
|
|
|
|
def __init__(self, input_image, canvas_width=512, canvas_height=512):
|
|
super().__init__()
|
|
|
|
self.setWindowTitle("Snap Canvas")
|
|
self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)
|
|
self.input_image = input_image
|
|
self.canvas_width = canvas_width
|
|
self.canvas_height = canvas_height
|
|
self.modified_image = None
|
|
|
|
self.top_left_x = 0
|
|
self.top_left_y = 0
|
|
self.scale_factor = 1.0 # 初始缩放倍数
|
|
|
|
self.scene = QGraphicsScene()
|
|
# 使用自定义的视图
|
|
self.view = CustomGraphicsView(self.scene)
|
|
self.view.setStyleSheet("border: 1px solid black; background-color: #404040;") # 设置深灰色背景区分画布
|
|
|
|
# 设置滚动条策略为需要时显示
|
|
self.view.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded)
|
|
self.view.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)
|
|
|
|
# 启用拖动画布
|
|
self.view.setDragMode(QGraphicsView.ScrollHandDrag)
|
|
|
|
# 创建白色画布
|
|
self.canvas = QImage(self.canvas_width, self.canvas_height, QImage.Format_RGB888)
|
|
self.canvas.fill(QColor('white'))
|
|
|
|
# 添加画布到场景
|
|
self.canvas_pixmap_item = QGraphicsPixmapItem(QPixmap.fromImage(self.canvas))
|
|
self.scene.addItem(self.canvas_pixmap_item)
|
|
|
|
# 添加可调整大小和可旋转的输入图像
|
|
original_pixmap = QPixmap.fromImage(self.input_image)
|
|
self.input_pixmap_item = ResizablePixmapItem(original_pixmap)
|
|
self.input_pixmap_item.setPos(self.canvas_width / 2, self.canvas_height / 2) # 将图像放置在画布中央
|
|
self.scene.addItem(self.input_pixmap_item)
|
|
|
|
# 设置旋转锚点为图像中心
|
|
self.input_pixmap_item.setTransformOriginPoint(self.input_pixmap_item.boundingRect().center())
|
|
|
|
# 创建菜单栏
|
|
self.menu_bar = QMenuBar(self)
|
|
help_menu = QMenu("使用说明(Help)", self)
|
|
self.menu_bar.addMenu(help_menu)
|
|
|
|
# 创建“使用方法”动作
|
|
usage_action = QAction("使用方法", self)
|
|
usage_action.triggered.connect(self.show_usage)
|
|
help_menu.addAction(usage_action)
|
|
|
|
# 创建按钮和输入框
|
|
save_button = QPushButton("保存并关闭", self)
|
|
save_button.clicked.connect(self.save_and_close)
|
|
|
|
rotate_label = QLineEdit(self)
|
|
rotate_label.setPlaceholderText("旋转角度 (°)")
|
|
rotate_label.setFixedWidth(100)
|
|
self.rotate_input = rotate_label
|
|
|
|
rotate_button = QPushButton("旋转", self)
|
|
rotate_button.clicked.connect(self.rotate_image)
|
|
|
|
scale_label = QLineEdit(self)
|
|
scale_label.setPlaceholderText("缩放倍数")
|
|
scale_label.setFixedWidth(100)
|
|
self.scale_input = scale_label
|
|
|
|
scale_button = QPushButton("缩放", self)
|
|
scale_button.clicked.connect(self.scale_image)
|
|
|
|
self.canvas_width_input = QLineEdit(self)
|
|
self.canvas_width_input.setPlaceholderText("画布宽度")
|
|
self.canvas_width_input.setText(str(self.canvas_width))
|
|
self.canvas_height_input = QLineEdit(self)
|
|
self.canvas_height_input.setPlaceholderText("画布高度")
|
|
self.canvas_height_input.setText(str(self.canvas_height))
|
|
|
|
size_button = QPushButton("设置画布大小", self)
|
|
size_button.clicked.connect(self.set_canvas_size)
|
|
|
|
# 布局设置
|
|
hbox = QHBoxLayout()
|
|
hbox.addWidget(self.canvas_width_input)
|
|
hbox.addWidget(self.canvas_height_input)
|
|
hbox.addWidget(size_button)
|
|
hbox.addWidget(self.rotate_input)
|
|
hbox.addWidget(rotate_button)
|
|
hbox.addWidget(self.scale_input)
|
|
hbox.addWidget(scale_button)
|
|
|
|
vbox = QVBoxLayout()
|
|
vbox.setMenuBar(self.menu_bar) # 将菜单栏添加到布局中
|
|
vbox.addWidget(self.view)
|
|
vbox.addLayout(hbox)
|
|
vbox.addWidget(save_button)
|
|
|
|
self.setLayout(vbox)
|
|
|
|
self.scene.setSceneRect(0, 0, self.canvas_width, self.canvas_height)
|
|
self.view.setSceneRect(0, 0, self.canvas_width, self.canvas_height)
|
|
|
|
# 设置窗口最小大小
|
|
self.setMinimumSize(600, 400)
|
|
|
|
def show_usage(self):
|
|
usage_text = (
|
|
"使用方法：\n"
|
|
"1. 滚动鼠标滚轮可缩放视图。\n"
|
|
"2. 按住 Ctrl 键并滚动鼠标滚轮可缩放图像。\n"
|
|
"3. 在输入框中输入旋转角度（度数），点击“旋转”按钮可旋转图像。\n"
|
|
"4. 在输入框中输入缩放倍数，点击“缩放”按钮可缩放图像。\n"
|
|
"5. 拖动图像可移动其位置。\n"
|
|
"6. 使用上方输入框可调整画布大小。\n"
|
|
"7. 点击“保存并关闭”按钮可保存当前图像。\n"
|
|
)
|
|
QMessageBox.information(self, "使用方法", usage_text)
|
|
|
|
def save_and_close(self):
|
|
self.save_image()
|
|
self.accept()
|
|
|
|
def save_image(self):
|
|
# 创建一个与画布大小相同的图像
|
|
image = QImage(self.canvas_width, self.canvas_height, QImage.Format_RGB888)
|
|
image.fill(QColor('white'))
|
|
|
|
painter = QPainter(image)
|
|
target_rect = QRectF(0, 0, self.canvas_width, self.canvas_height)
|
|
|
|
# 渲染场景
|
|
self.scene.render(painter, target_rect, self.scene.sceneRect(), Qt.IgnoreAspectRatio)
|
|
|
|
painter.end()
|
|
|
|
self.modified_image = image
|
|
|
|
# 获取图像左上角的坐标
|
|
top_left_point = self.input_pixmap_item.mapToScene(0, 0)
|
|
self.top_left_x = int(top_left_point.x())
|
|
self.top_left_y = int(top_left_point.y())
|
|
|
|
# 获取缩放倍数
|
|
self.scale_factor = self.input_pixmap_item.current_scale
|
|
|
|
def closeEvent(self, event):
|
|
self.close_signal.emit()
|
|
event.accept()
|
|
|
|
def set_canvas_size(self):
|
|
try:
|
|
width = int(self.canvas_width_input.text())
|
|
height = int(self.canvas_height_input.text())
|
|
except ValueError:
|
|
QMessageBox.warning(self, "输入错误", "画布宽度和高度必须是整数。")
|
|
return
|
|
|
|
# 更新画布宽度和高度
|
|
self.canvas_width = width
|
|
self.canvas_height = height
|
|
|
|
# 更新场景和视图的画布大小
|
|
self.canvas = QImage(self.canvas_width, self.canvas_height, QImage.Format_RGB888)
|
|
self.canvas.fill(QColor('white'))
|
|
self.canvas_pixmap_item.setPixmap(QPixmap.fromImage(self.canvas))
|
|
|
|
self.scene.setSceneRect(0, 0, self.canvas_width, self.canvas_height)
|
|
self.view.setSceneRect(0, 0, self.canvas_width, self.canvas_height)
|
|
|
|
def rotate_image(self):
|
|
try:
|
|
angle = float(self.rotate_input.text())
|
|
except ValueError:
|
|
QMessageBox.warning(self, "输入错误", "旋转角度必须是数字。")
|
|
return
|
|
self.input_pixmap_item.rotate_pixmap(angle)
|
|
|
|
def scale_image(self):
|
|
try:
|
|
scale_factor = float(self.scale_input.text())
|
|
if scale_factor <= 0:
|
|
raise ValueError
|
|
except ValueError:
|
|
QMessageBox.warning(self, "输入错误", "缩放倍数必须是正数。")
|
|
return
|
|
self.input_pixmap_item.scale_pixmap(scale_factor)
|
|
|
|
def get_modified_image(self):
|
|
return self.modified_image
|
|
|
|
def get_top_left_x(self):
|
|
return self.top_left_x
|
|
|
|
def get_top_left_y(self):
|
|
return self.top_left_y
|
|
|
|
def get_scale_factor(self):
|
|
return self.scale_factor
|
|
|
|
|
|
class ResizablePixmapItem(QGraphicsPixmapItem):
|
|
def __init__(self, pixmap):
|
|
super().__init__(pixmap)
|
|
self.setFlags(
|
|
QGraphicsItem.ItemIsSelectable |
|
|
QGraphicsItem.ItemIsMovable
|
|
)
|
|
self.setTransformationMode(Qt.SmoothTransformation)
|
|
self.setAcceptHoverEvents(True)
|
|
self.setAcceptDrops(False)
|
|
self.setTransformOriginPoint(self.boundingRect().center())
|
|
self.current_scale = 1.0
|
|
self.current_rotation = 0.0
|
|
self.min_scale = 0.1
|
|
self.max_scale = 10.0
|
|
|
|
def scale_pixmap(self, scale_factor):
|
|
if self.min_scale <= scale_factor <= self.max_scale:
|
|
self.current_scale = scale_factor
|
|
self.update_transform()
|
|
else:
|
|
QMessageBox.warning(None, "无效的缩放", f"缩放倍数必须在 {self.min_scale} 和 {self.max_scale} 之间。")
|
|
|
|
def rotate_pixmap(self, angle):
|
|
self.current_rotation = angle % 360
|
|
self.update_transform()
|
|
|
|
def update_transform(self):
|
|
transform = QTransform()
|
|
center = self.boundingRect().center()
|
|
transform.translate(center.x(), center.y())
|
|
transform.rotate(self.current_rotation)
|
|
transform.scale(self.current_scale, self.current_scale)
|
|
transform.translate(-center.x(), -center.y())
|
|
self.setTransform(transform)
|
|
|
|
def wheelEvent(self, event):
|
|
if QApplication.keyboardModifiers() == Qt.ControlModifier:
|
|
delta = event.delta() # 修改这里，使用 delta()
|
|
if delta > 0:
|
|
factor = 1.1
|
|
else:
|
|
factor = 0.9
|
|
new_scale = self.current_scale * factor
|
|
self.scale_pixmap(new_scale)
|
|
event.accept()
|
|
else:
|
|
event.ignore() # 确保事件能够传递给视图进行缩放
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
# 示例输入图像，可以替换为您自己的图像路径
|
|
input_image_path = "your_image.png"
|
|
if not os.path.exists(input_image_path):
|
|
# 创建一个示例图像
|
|
input_image = QImage(200, 200, QImage.Format_RGB888)
|
|
input_image.fill(QColor('blue'))
|
|
else:
|
|
input_image = QImage(input_image_path)
|
|
|
|
window = CanvasWindow(input_image)
|
|
window.show()
|
|
sys.exit(app.exec_())
|