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,36 +240,41 @@ def plot_wave(output_dir):
 
     ani = FuncAnimation(fig, update, frames=n_frames, interval=50, blit=True)
 
-    # ── 先输出 GIF（自动循环）──
+    # ── 输出 GIF ──
-    gif_path = os.path.join(output_dir, "wave_animation.gif")
+    if save_gif:
-    ani.save(gif_path, writer="pillow", fps=min(20, max(1, n_frames // 5)))
+        gif_path = os.path.join(output_dir, "wave_animation.gif")
-    print(f"[plot_wave] GIF 已保存: {gif_path}")
+        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）──
-    try:
+    gif_path = None
-        import matplotlib.animation as manim
+    if save_gif:
-        import matplotlib.pyplot as _plt
+        gif_path = os.path.join(output_dir, "wave_animation.gif")
-
+    if save_mp4:
-        # 尝试通过 imageio_ffmpeg 定位 ffmpeg
-        ffmpeg_path = None
         try:
-            import imageio_ffmpeg
+            import matplotlib.animation as manim
-            ffmpeg_path = imageio_ffmpeg.get_ffmpeg_exe()
+            import matplotlib.pyplot as _plt
-        except Exception:
-            pass
-        if ffmpeg_path and os.path.exists(ffmpeg_path):
-            _plt.rcParams['animation.ffmpeg_path'] = ffmpeg_path
 
-        ffps = min(20, max(1, n_frames // 5))
+            # 尝试通过 imageio_ffmpeg 定位 ffmpeg
-        writer = manim.FFMpegWriter(fps=ffps, codec="libx264",
+            ffmpeg_path = None
-                                     extra_args=["-pix_fmt", "yuv420p"])
+            try:
-        mp4_path = os.path.join(output_dir, "wave_animation.mp4")
+                import imageio_ffmpeg
-        ani.save(mp4_path, writer=writer)
+                ffmpeg_path = imageio_ffmpeg.get_ffmpeg_exe()
-        print(f"[plot_wave] MP4 已保存: {mp4_path}")
+            except Exception:
-    except FileNotFoundError:
+                pass
-        print("[plot_wave] 警告: 未找到 ffmpeg，跳过 MP4 输出")
+            if ffmpeg_path and os.path.exists(ffmpeg_path):
-    except Exception as e:
+                _plt.rcParams['animation.ffmpeg_path'] = ffmpeg_path
-        print(f"[plot_wave] 警告: MP4 输出失败 ({e})，跳过")
+
+            ffps = min(20, max(1, n_frames // 5))
+            writer = manim.FFMpegWriter(fps=ffps, codec="libx264",
+                                         extra_args=["-pix_fmt", "yuv420p"])
+            mp4_path = os.path.join(output_dir, "wave_animation.mp4")
+            ani.save(mp4_path, writer=writer)
+            print(f"[plot_wave] MP4 已保存: {mp4_path}")
+        except FileNotFoundError:
+            print("[plot_wave] 警告: 未找到 ffmpeg，跳过 MP4 输出")
+        except Exception as e:
+            print(f"[plot_wave] 警告: MP4 输出失败 ({e})，跳过")
 
     # ── 最后显示动画窗口 ──
     plt.show()