From e20aab13e5299f2d79e39c5c2417206d7c266fe3 Mon Sep 17 00:00:00 2001 From: ZuiLiuaB <137588937+ZuiLiuaB@users.noreply.github.com> Date: Sat, 7 Dec 2024 12:55:54 +0800 Subject: [PATCH] Main --- work.py | 145 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 145 insertions(+) create mode 100644 work.py diff --git a/work.py b/work.py new file mode 100644 index 0000000..ac0691a --- /dev/null +++ b/work.py @@ -0,0 +1,145 @@ +import tkinter as tk +from tkinter import filedialog, messagebox +import matplotlib.pyplot as plt +from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg +import numpy as np +from scipy.interpolate import make_interp_spline, BSpline +from PIL import Image +import io + +def generate_and_display_shape(): + try: + min_vertices = int(min_entry.get()) + max_vertices = int(max_entry.get()) + line_width = float(line_width_entry.get()) + scatter = float(scatter_entry.get()) + + if min_vertices < 3 or max_vertices < 3 or min_vertices > max_vertices: + messagebox.showerror("无效输入", "最小和最大顶点数必须至少为3，并且最小值不应超过最大值。") + return + + # 清除之前的图形 + ax.clear() + + # 生成随机顶点数量 + num_vertices = np.random.randint(min_vertices, max_vertices + 1) + + # 生成随机角度 + angles = np.sort(np.random.rand(num_vertices) * 2 * np.pi) + + # 添加散布 + angles += np.random.normal(0, scatter, num_vertices) + angles = np.sort(angles % (2 * np.pi)) + + # 计算每个顶点的坐标 + x = np.cos(angles) + y = np.sin(angles) + + # 将第一个顶点添加到最后以闭合图形 + x = np.append(x, x[0]) + y = np.append(y, y[0]) + + # 使用B样条插值平滑曲线 + t = np.linspace(0, 1, num_vertices + 1) + t_smooth = np.linspace(0, 1, 300) + + spl_x = make_interp_spline(t, x, k=3) + spl_y = make_interp_spline(t, y, k=3) + + x_smooth = spl_x(t_smooth) + y_smooth = spl_y(t_smooth) + + # 创建图形 + global current_line + if current_line: + current_line.remove() + current_line, = ax.plot(x_smooth, y_smooth, linestyle='-', color='gray', linewidth=line_width) # 显示轨迹 + ax.axis('equal') # 确保比例相等 + ax.axis('off') # 隐藏坐标轴 + canvas.draw() + except ValueError: + messagebox.showerror("无效输入", "请输入有效的整数作为最小和最大顶点数，以及浮点数作为线的粗细和散布。") + +def adjust_line_width(): + try: + line_width = float(line_width_entry.get()) + if current_line: + current_line.set_linewidth(line_width) + canvas.draw() + except ValueError: + messagebox.showerror("无效输入", "请输入有效的浮点数作为线的粗细。") + +def save_shape(): + # 获取当前图形 + buf = io.BytesIO() + plt.savefig(buf, format='png', transparent=True) + buf.seek(0) + image = Image.open(buf).convert("RGBA") + + # 保存图像 + file_path = filedialog.asksaveasfilename(defaultextension=".png", filetypes=[("PNG 文件", "*.png"), ("所有文件", "*.*")]) + if file_path: + image.save(file_path) + messagebox.showinfo("保存成功", f"图像已保存到 {file_path}") + +# 创建主窗口 +root = tk.Tk() +root.title("随机封闭形状生成器-醉流ab") +root.geometry("800x700") + +# 创建Matplotlib图形 +fig, ax = plt.subplots(figsize=(5, 5), dpi=100) +canvas = FigureCanvasTkAgg(fig, master=root) +canvas_widget = canvas.get_tk_widget() +canvas_widget.pack(side=tk.TOP, fill=tk.BOTH, expand=True) + +# 创建输入框和按钮的框架 +input_frame = tk.Frame(root) +input_frame.pack(side=tk.TOP, padx=20, pady=20) + +# 线的粗细输入框和标签 +line_width_label = tk.Label(input_frame, text="线的粗细:") +line_width_label.grid(row=0, column=0, padx=5, pady=5) +line_width_entry = tk.Entry(input_frame) +line_width_entry.insert(0, "1.0") +line_width_entry.grid(row=0, column=1, padx=5, pady=5) + +# 散布输入框和标签 +scatter_label = tk.Label(input_frame, text="散布:") +scatter_label.grid(row=0, column=2, padx=5, pady=5) +scatter_entry = tk.Entry(input_frame) +scatter_entry.insert(0, "0.1") +scatter_entry.grid(row=0, column=3, padx=5, pady=5) + +# 调整按钮 +adjust_button = tk.Button(input_frame, text="调整", command=adjust_line_width, width=10, height=2) +adjust_button.grid(row=0, column=4, padx=5, pady=5) + +# 最小顶点数输入框和标签 +min_label = tk.Label(input_frame, text="最小顶点数 (>3):") +min_label.grid(row=1, column=0, padx=5, pady=5) +min_entry = tk.Entry(input_frame) +min_entry.insert(0, "10") +min_entry.grid(row=1, column=1, padx=5, pady=5) + +# 最大顶点数输入框和标签 +max_label = tk.Label(input_frame, text="最大顶点数:") +max_label.grid(row=1, column=2, padx=5, pady=5) +max_entry = tk.Entry(input_frame) +max_entry.insert(0, "15") +max_entry.grid(row=1, column=3, padx=5, pady=5) + +# 刷新按钮 +refresh_button = tk.Button(input_frame, text="刷新", command=generate_and_display_shape, width=10, height=2) +refresh_button.grid(row=1, column=4, padx=5, pady=5) + +# 保存按钮 +save_button = tk.Button(input_frame, text="保存", command=save_shape, width=10, height=2) +save_button.grid(row=1, column=5, padx=5, pady=5) + +# 初始化图形 +current_line = None +generate_and_display_shape() + +# 运行主循环 +root.mainloop()