265 lines
9.0 KiB
Python
265 lines
9.0 KiB
Python
import os
|
||
from pathlib import Path
|
||
import json
|
||
import math
|
||
from PIL import Image
|
||
|
||
|
||
class ForegroundImage:
|
||
"""前景图对象，存储图片信息和变换参数"""
|
||
|
||
def __init__(self, file_path):
|
||
self.file_path = file_path
|
||
# 四个顶点坐标 (左上, 右上, 左下, 右下)
|
||
self.vertices = [
|
||
(0.0, 0.0), # top-left
|
||
(100.0, 0.0), # top-right
|
||
(0.0, 100.0), # bottom-left
|
||
(100.0, 100.0) # bottom-right
|
||
]
|
||
self.angle = 0.0 # 旋转角度
|
||
self.mask_shape = "rect" # 遗罩形状
|
||
self.locked = False # 是否锁定
|
||
self.visible = True # 是否可见
|
||
self.z_index = 0 # 图层顺序
|
||
self.hidden = False # 是否隐藏（用于批量处理模式）
|
||
|
||
# 添加属性x, y, w, h的getter和setter方法，使其与vertices保持同步
|
||
@property
|
||
def x(self):
|
||
return self.vertices[0][0] # 左上角x坐标作为x属性
|
||
|
||
@x.setter
|
||
def x(self, value):
|
||
# 更新所有顶点的x坐标以保持宽度不变
|
||
dx = value - self.vertices[0][0]
|
||
for i in range(len(self.vertices)):
|
||
x, y = self.vertices[i]
|
||
self.vertices[i] = (x + dx, y)
|
||
|
||
@property
|
||
def y(self):
|
||
return self.vertices[0][1] # 左上角y坐标作为y属性
|
||
|
||
@y.setter
|
||
def y(self, value):
|
||
# 更新所有顶点的y坐标以保持高度不变
|
||
dy = value - self.vertices[0][1]
|
||
for i in range(len(self.vertices)):
|
||
x, y = self.vertices[i]
|
||
self.vertices[i] = (x, y + dy)
|
||
|
||
@property
|
||
def w(self):
|
||
# 宽度为右上角和左上角x坐标的差值
|
||
return self.vertices[1][0] - self.vertices[0][0]
|
||
|
||
@w.setter
|
||
def w(self, value):
|
||
# 保持左上角不变，调整右上角和右下角的x坐标
|
||
if value > 0:
|
||
old_w = self.vertices[1][0] - self.vertices[0][0]
|
||
if old_w != 0:
|
||
x0, y0 = self.vertices[0] # 左上角
|
||
x1, y1 = self.vertices[1] # 右上角
|
||
x2, y2 = self.vertices[2] # 左下角
|
||
x3, y3 = self.vertices[3] # 右下角
|
||
|
||
# 调整右上角和右下角的x坐标
|
||
self.vertices[1] = (x0 + value, y1) # 右上角
|
||
self.vertices[3] = (x0 + value, y3) # 右下角
|
||
|
||
@property
|
||
def h(self):
|
||
# 高度为左下角和左上角y坐标的差值
|
||
return self.vertices[2][1] - self.vertices[0][1]
|
||
|
||
@h.setter
|
||
def h(self, value):
|
||
# 保持左上角不变，调整左下角和右下角的y坐标
|
||
if value > 0:
|
||
old_h = self.vertices[2][1] - self.vertices[0][1]
|
||
if old_h != 0:
|
||
x0, y0 = self.vertices[0] # 左上角
|
||
x1, y1 = self.vertices[1] # 右上角
|
||
x2, y2 = self.vertices[2] # 左下角
|
||
x3, y3 = self.vertices[3] # 右下角
|
||
|
||
# 调整左下角和右下角的y坐标
|
||
self.vertices[2] = (x2, y0 + value) # 左下角
|
||
self.vertices[3] = (x3, y0 + value) # 右下角
|
||
|
||
def to_dict(self):
|
||
"""转换为字典，用于序列化"""
|
||
return {
|
||
"file_path": self.file_path,
|
||
"vertices": self.vertices,
|
||
"angle": self.angle,
|
||
"mask_shape": self.mask_shape,
|
||
"locked": self.locked,
|
||
"visible": self.visible,
|
||
"z_index": self.z_index,
|
||
"hidden": self.hidden
|
||
}
|
||
|
||
@classmethod
|
||
def from_dict(cls, data):
|
||
"""从字典创建对象"""
|
||
fg = cls(data["file_path"])
|
||
fg.vertices = data["vertices"]
|
||
fg.angle = data["angle"]
|
||
fg.mask_shape = data.get("mask_shape", "rect")
|
||
fg.locked = data.get("locked", False)
|
||
fg.visible = data.get("visible", True)
|
||
fg.z_index = data.get("z_index", 0)
|
||
fg.hidden = data.get("hidden", False)
|
||
return fg
|
||
|
||
|
||
class Template:
|
||
"""模板对象，存储尺寸和背景信息"""
|
||
|
||
def __init__(self, ratio=(16, 9), width_px=1200):
|
||
self.ratio = ratio
|
||
self.width_px = width_px
|
||
self.height_px = int(width_px * ratio[1] / ratio[0])
|
||
self.bg_image = None
|
||
self.bg_color = "#FFFFFF" # 默认白色背景
|
||
|
||
def set_custom_size(self, width, height):
|
||
"""设置自定义尺寸"""
|
||
self.width_px = width
|
||
self.height_px = height
|
||
# 根据实际尺寸设置比例
|
||
gcd = math.gcd(width, height)
|
||
self.ratio = (width // gcd, height // gcd)
|
||
|
||
def load_background(self, file_path):
|
||
"""加载背景图"""
|
||
try:
|
||
self.bg_image = Image.open(file_path).convert("RGBA")
|
||
return True
|
||
except Exception as e:
|
||
print(f"加载背景图失败: {e}")
|
||
return False
|
||
|
||
def clear_background(self):
|
||
"""清除背景图"""
|
||
self.bg_image = None
|
||
|
||
def to_dict(self):
|
||
"""转换为字典，用于序列化"""
|
||
return {
|
||
"ratio": self.ratio,
|
||
"width_px": self.width_px,
|
||
"height_px": self.height_px,
|
||
"bg_color": self.bg_color
|
||
}
|
||
|
||
@classmethod
|
||
def from_dict(cls, data):
|
||
"""从字典创建对象"""
|
||
template = cls(data["ratio"], data["width_px"])
|
||
template.height_px = data["height_px"]
|
||
template.bg_color = data.get("bg_color", "#FFFFFF")
|
||
return template
|
||
|
||
|
||
class Project:
|
||
"""项目对象，管理模板和前景图"""
|
||
|
||
def __init__(self):
|
||
self.template = Template()
|
||
self.foregrounds = []
|
||
self.canvas_zoom = 1.0
|
||
self.file_path = None # 项目文件路径
|
||
|
||
def add_foreground(self, file_path, reference_fg=None):
|
||
"""添加前景图"""
|
||
fg = ForegroundImage(file_path)
|
||
|
||
# 如果提供了参考图片，则复制其属性
|
||
if reference_fg is not None:
|
||
fg.vertices = [tuple(v) for v in reference_fg.vertices] # 复制顶点
|
||
fg.angle = reference_fg.angle
|
||
fg.mask_shape = reference_fg.mask_shape
|
||
fg.locked = reference_fg.locked
|
||
fg.visible = reference_fg.visible
|
||
else:
|
||
# 尝试获取图片原始尺寸并设置初始大小
|
||
try:
|
||
with Image.open(file_path) as img:
|
||
w, h = img.size
|
||
# 按比例缩小到模板的2/3大小
|
||
scale = min(self.template.width_px * 2/3 / w, self.template.height_px * 2/3 / h)
|
||
# 计算初始位置（居中）
|
||
initial_w = w * scale
|
||
initial_h = h * scale
|
||
initial_x = (self.template.width_px - initial_w) / 2
|
||
initial_y = (self.template.height_px - initial_h) / 2
|
||
|
||
# 设置顶点坐标
|
||
fg.vertices = [
|
||
(initial_x, initial_y), # top-left
|
||
(initial_x + initial_w, initial_y), # top-right
|
||
(initial_x, initial_y + initial_h), # bottom-left
|
||
(initial_x + initial_w, initial_y + initial_h) # bottom-right
|
||
]
|
||
except Exception as e:
|
||
print(f"获取图片尺寸失败: {e}")
|
||
|
||
# 设置图层顺序
|
||
fg.z_index = len(self.foregrounds)
|
||
self.foregrounds.append(fg)
|
||
return fg
|
||
|
||
def remove_foreground(self, fg):
|
||
"""移除前景图"""
|
||
if fg in self.foregrounds:
|
||
self.foregrounds.remove(fg)
|
||
# 更新图层顺序
|
||
self.update_z_indices()
|
||
|
||
def update_z_indices(self):
|
||
"""更新所有前景图的图层顺序"""
|
||
for i, fg in enumerate(sorted(self.foregrounds, key=lambda x: x.z_index)):
|
||
fg.z_index = i
|
||
|
||
def save(self, file_path=None):
|
||
"""保存项目到文件"""
|
||
if file_path is None:
|
||
file_path = self.file_path
|
||
|
||
if file_path is None:
|
||
return False
|
||
|
||
data = {
|
||
"template": self.template.to_dict(),
|
||
"foregrounds": [fg.to_dict() for fg in self.foregrounds],
|
||
"canvas_zoom": self.canvas_zoom
|
||
}
|
||
|
||
try:
|
||
with open(file_path, "w", encoding="utf-8") as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
self.file_path = file_path
|
||
return True
|
||
except Exception as e:
|
||
print(f"保存项目失败: {e}")
|
||
return False
|
||
|
||
def load(self, file_path):
|
||
"""从文件加载项目"""
|
||
try:
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
|
||
self.template = Template.from_dict(data["template"])
|
||
self.foregrounds = [ForegroundImage.from_dict(fg_data) for fg_data in data["foregrounds"]]
|
||
self.canvas_zoom = data.get("canvas_zoom", 1.0)
|
||
self.file_path = file_path
|
||
return True
|
||
except Exception as e:
|
||
print(f"加载项目失败: {e}")
|
||
return False
|