feat: 新增 plot_wave_save_gif/plot_wave_save_mp4 开关

- 所有 input.txt 新增 plot_wave_save_gif / plot_wave_save_mp4 参数
- dynamics.py 将参数传入 plot_wave()
- plot_wave.py 根据 save_gif/save_mp4 标志条件保存文件
- 默认均为 0（不输出文件，只显示动画窗口）

dynamics.py

@@ -507,7 +507,11 @@ def run_case(config_path, runtime_base, input_dir="input", output_dir="output",
             try:
                 import plot_wave as pw
                 print("[run] 正在绘制波形与能量图…")
-                pw.plot_wave(str(output_dir_abs))
+                pw.plot_wave(
+                    str(output_dir_abs),
+                    save_gif=int(config.get("plot_wave_save_gif", 0)),
+                    save_mp4=int(config.get("plot_wave_save_mp4", 0)),
+                )
             except Exception as e:
                 print(f"[run] 绘制波形图失败: {e}")
 

examples/case01/input/input.txt

@@ -8,7 +8,9 @@
 step_simulate:  1   # 运行物理模拟 → output/trajectory.txt
 step_sample:    1   # 抽帧 → output/display.txt
 step_plot:      1   # 绘制轨迹/能量图 → output/trajectory_plots.png
-step_plot_wave: 0   # 绘制波形能量动画 → output/wave_animation.gif
+step_plot_wave: 0   # 绘制波形能量动画
+plot_wave_save_gif: 0  # 输出波形 GIF（需 step_plot_wave=1）
+plot_wave_save_mp4: 0  # 输出波形 MP4（需 step_plot_wave=1）
 step_animation: 1   # 自动播放 VisPy 3D 动画窗口（需安装 vispy）
 force_calc: 0       # 强制重新计算：1=跳过缓存强算，0=自动使用已有输出
 

examples/case02/input/input.txt

@@ -8,7 +8,9 @@
 step_simulate:  1   # 运行物理模拟 → output/trajectory.txt
 step_sample:    1   # 抽帧 → output/display.txt
 step_plot:      1   # 绘制轨迹/能量图 → output/trajectory_plots.png
-step_plot_wave: 0   # 绘制波形能量动画 → output/wave_animation.gif
+step_plot_wave: 0   # 绘制波形能量动画
+plot_wave_save_gif: 0  # 输出波形 GIF（需 step_plot_wave=1）
+plot_wave_save_mp4: 0  # 输出波形 MP4（需 step_plot_wave=1）
 step_animation: 1   # 自动播放 VisPy 3D 动画窗口（需安装 vispy）
 force_calc: 0       # 强制重新计算：1=跳过缓存强算，0=自动使用已有输出
 

examples/case03/input/input.txt

@@ -8,7 +8,9 @@
 step_simulate:  1   # 运行物理模拟 → output/trajectory.txt
 step_sample:    1   # 抽帧 → output/display.txt
 step_plot:      0   # 绘制轨迹/能量图 → output/trajectory_plots.png
-step_plot_wave: 0   # 绘制波形能量动画 → output/wave_animation.gif
+step_plot_wave: 0   # 绘制波形能量动画
+plot_wave_save_gif: 0  # 输出波形 GIF（需 step_plot_wave=1）
+plot_wave_save_mp4: 0  # 输出波形 MP4（需 step_plot_wave=1）
 step_animation: 1   # 自动播放 VisPy 3D 动画窗口（需安装 vispy）
 force_calc: 0       # 强制重新计算：1=跳过缓存强算，0=自动使用已有输出
 

examples/case04/input/input.txt

@@ -8,7 +8,9 @@
 step_simulate:  1   # 运行物理模拟 → output/trajectory.txt
 step_sample:    1   # 抽帧 → output/display.txt
 step_plot:      0   # 绘制轨迹/能量图 → output/trajectory_plots.png
-step_plot_wave: 0   # 绘制波形能量动画 → output/wave_animation.gif
+step_plot_wave: 0   # 绘制波形能量动画
+plot_wave_save_gif: 0  # 输出波形 GIF（需 step_plot_wave=1）
+plot_wave_save_mp4: 0  # 输出波形 MP4（需 step_plot_wave=1）
 step_animation: 1   # 自动播放 VisPy 3D 动画窗口（需安装 vispy）
 force_calc: 0       # 强制重新计算：1=跳过缓存强算，0=自动使用已有输出
 

examples/case05/input/input.txt

@@ -9,7 +9,9 @@ step_simulate:  0   # 运行物理模拟 → output/trajectory.txt
 step_sample:    0   # 抽帧 → output/display.txt
 step_plot:      0   # 绘制轨迹/能量图 → output/trajectory_plots.png
 step_animation: 0   # 自动播放 VisPy 3D 动画窗口（需安装 vispy）
-step_plot_wave: 1   # 绘制波形图
+step_plot_wave: 1   # 绘制波形能量动画
+plot_wave_save_gif: 0  # 输出波形 GIF（需 step_plot_wave=1）
+plot_wave_save_mp4: 0  # 输出波形 MP4（需 step_plot_wave=1）
 force_calc: 0       # 强制重新计算：1=跳过缓存强算，0=自动使用已有输出
 
 # ── 计算引擎 ──────────────────────────────────

examples/case06/input/input.txt

@@ -8,7 +8,9 @@
 step_simulate:  1   # 运行物理模拟 → output/trajectory.txt
 step_sample:    1   # 抽帧 → output/display.txt
 step_plot:      0   # 绘制轨迹/能量图 → output/trajectory_plots.png
-step_plot_wave: 0   # 绘制波形能量动画 → output/wave_animation.gif
+step_plot_wave: 0   # 绘制波形能量动画
+plot_wave_save_gif: 0  # 输出波形 GIF（需 step_plot_wave=1）
+plot_wave_save_mp4: 0  # 输出波形 MP4（需 step_plot_wave=1）
 step_animation: 1   # 自动播放 VisPy 3D 动画窗口（需安装 vispy）
 force_calc: 0       # 强制重新计算：1=跳过缓存强算，0=自动使用已有输出
 

plot_wave.py

@@ -83,8 +83,14 @@ def compute_energy(x, y, z, vx, vy, vz, masses, mass_arr,
     return ek_sys, us_sys, ug_sys, ugr_sys
 
 
-def plot_wave(output_dir):
+def plot_wave(output_dir, save_gif=False, save_mp4=False):
-    """主绘图函数：读取 display.txt 并生成波形+能量动画。"""
+    """主绘图函数：读取 display.txt 并生成波形+能量动画。
+
+    Args:
+        output_dir: 输出目录（含 display.txt）
+        save_gif:   是否保存 GIF
+        save_mp4:   是否保存 MP4
+    """
     data = load_disp_data(output_dir)
 
     n_frames = int(data["n_frames"])
@@ -234,12 +240,17 @@ def plot_wave(output_dir):
 
     ani = FuncAnimation(fig, update, frames=n_frames, interval=50, blit=True)
 
-    # ── 先输出 GIF（自动循环）──
+    # ── 输出 GIF ──
+    if save_gif:
         gif_path = os.path.join(output_dir, "wave_animation.gif")
         ani.save(gif_path, writer="pillow", fps=min(20, max(1, n_frames // 5)))
         print(f"[plot_wave] GIF 已保存: {gif_path}")
 
-    # ── 再输出 MP4（需要 ffmpeg）──
+    # ── 输出 MP4（需要 ffmpeg）──
+    gif_path = None
+    if save_gif:
+        gif_path = os.path.join(output_dir, "wave_animation.gif")
+    if save_mp4:
         try:
             import matplotlib.animation as manim
             import matplotlib.pyplot as _plt