admin
e62e536cee
- 新增 compute_accel_conservative / accel_conservative： 保守力加速度（弹簧+重力+原子间引力），不含阻尼，供蛙跳专用 - 重写 leapfrog_step / leapfrog_full： - 无阻尼：纯辛积分器，每步 1 次力计算（原 Velocity-Verlet 需 2 次） - 有阻尼：半隐式处理 v(t+dt/2)=[v(t-dt/2)*(1-α)+a_c*dt]/(1+α)，无条件稳定 - 主循环加初始化反向半步 v(-dt/2)=v(0)-0.5*a_c(0)*dt - 修复 C/C++ number of frames 字段写采样帧数而非总积分步数的 bug - Python 引擎：新增 display.npz 二进制格式，draw.py/plot_wave.py 优先读取 - 编译参数统一为 -O3 -march=native -ffast-math
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""
|
|
Case runner for Dynamics case06 — 1D atomic chain (transverse wave).
|
|
|
|
This script keeps program and data separated:
|
|
- program: ../../dynamics.py
|
|
- input: ./input
|
|
- output: ./output
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import importlib.util
|
|
from pathlib import Path
|
|
|
|
|
|
CASE_DIR = Path(__file__).resolve().parent
|
|
DYNAMICS_PATH = Path("..") / ".." / "dynamics.py"
|
|
INPUT_DIR = Path("input")
|
|
OUTPUT_DIR = Path("output")
|
|
CONFIG_FILE = INPUT_DIR / "input.txt"
|
|
|
|
|
|
def load_dynamics_module(module_path: Path):
|
|
spec = importlib.util.spec_from_file_location("dynamics_module", module_path)
|
|
if spec is None or spec.loader is None:
|
|
raise ImportError(f"无法加载 dynamics.py: {module_path}")
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="运行 Dynamics 示例案例 case06")
|
|
parser.add_argument("--no-plot", action="store_true", help="跳过 matplotlib 绘图")
|
|
args = parser.parse_args()
|
|
|
|
dynamics_path = (CASE_DIR / DYNAMICS_PATH).resolve()
|
|
input_dir = (CASE_DIR / INPUT_DIR).resolve()
|
|
output_dir = (CASE_DIR / OUTPUT_DIR).resolve()
|
|
config_path = (CASE_DIR / CONFIG_FILE).resolve()
|
|
|
|
module = load_dynamics_module(dynamics_path)
|
|
module.run_case(
|
|
config_path=config_path,
|
|
runtime_base=CASE_DIR,
|
|
input_dir=input_dir,
|
|
output_dir=output_dir,
|
|
no_plot=args.no_plot,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|