diff --git a/sim_scenes/tri_bodies/tri_bodies_04.py b/sim_scenes/tri_bodies/tri_bodies_04.py new file mode 100644 index 0000000000000000000000000000000000000000..e8b419fe58ae7deb946701d61f86969e4210e984 --- /dev/null +++ b/sim_scenes/tri_bodies/tri_bodies_04.py @@ -0,0 +1,66 @@ +# -*- coding:utf-8 -*- +# title :三体场景模拟02 +# description :三体场景模拟(3个太阳、1个地球) +# author :Python超人 +# date :2023-02-11 +# link :https://gitcode.net/pythoncr/ +# python_version :3.8 +# ============================================================================== +from bodies import Sun, Earth +from common.consts import SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_YEAR, AU +from sim_scenes.func import mayavi_run, mpl_run, ursina_run + +if __name__ == '__main__': + """ + 注释: 3个太阳 + 可以修改影响效果的参数为: + 1、三个方向的初始位置 init_position[x, y, z] + 2、三个方向的初始速度 init_velocity[x, y, z] + 3、天体质量 mass + """ + # 代码案例如下: + SIZE_SCALE = 5e1 # 生成的太阳放大 50 倍 + RUN_DIAMETER = 1.392e6 # 真实太阳的直径 + velocity_rate = 22 + bodies = [ + Sun(name="太阳1", mass=1e30, color=(111, 140, 255), rotation_speed=0.1, + init_position=[-100 * RUN_DIAMETER, 0, 0], + init_velocity=[0.347113 * velocity_rate, 0.532727 * velocity_rate, 0], + size_scale=SIZE_SCALE, texture="fixed_star.png"), + Sun(name="太阳2", mass=1e30, color=(250, 195, 47), rotation_speed=0.1, + init_position=[100 * RUN_DIAMETER, 0, 0], + init_velocity=[0.347113 * velocity_rate, 0.532727 * velocity_rate, 0], + size_scale=SIZE_SCALE, texture="fixed_star.png"), + Sun(name="太阳3", mass=1e30, color=(198, 29, 3), rotation_speed=0.1, + init_position=[0, 0, 0], + init_velocity=[-0.694226 * velocity_rate, -1.065454 * velocity_rate, 0], + size_scale=SIZE_SCALE, texture="fixed_star.png"), + ] + + # 按照以上代码案例格式生成代码,要求 init_position 、init_velocity、mass 生成后,要保证在万有引力的作用下,能正常运行,不会碰撞 + """ +太阳1: + 初始位置:(100000, 200000, 300000) km + 初始速度:(-1.0, -2.0, 3.0) km/s + 质量:2.5 x 10^30 kg + +太阳2: + 初始位置:(-150000, 250000, 350000) km + 初始速度:(2.0, -3.0, -1.0) km/s + 质量:2.0 x 10^30 kg + +太阳3: + 初始位置:(200000, -300000, 400000) km + 初始速度:(-3.0, 1.0, -2.0) km/s + 质量:2.8 x 10^30 kg + """ + # 使用 mayavi 查看的运行效果 + # mayavi_run(bodies, SECONDS_PER_WEEK, view_azimuth=0) + + # 使用 matplotlib 查看运行效果 + # mpl_run(bodies, SECONDS_PER_WEEK) + + # 使用 ursina 查看的运行效果 + # 常用快捷键: P:运行和暂停 O:重新开始 I:显示天体轨迹 + # position = 左-右+、上+下-、前+后- + ursina_run(bodies, SECONDS_PER_WEEK, position=(0 * AU, 0, -3 * AU), show_trail=True)