diff --git a/common/celestial_data_service.py b/common/celestial_data_service.py index a9a7db02e174aa168be68e339f635041f4a1e5ef..51a78549bf2069cf29cd7685bd72d1c77b1a1e6a 100644 --- a/common/celestial_data_service.py +++ b/common/celestial_data_service.py @@ -346,6 +346,18 @@ def get_init_pos_vels(): return init_pos_vels +def init_bodies_pos_vels(bodies): + # 获取模拟的初始位置和速度 + init_pos_vels = get_init_pos_vels() + + for body in bodies: + pos_vels = init_pos_vels.get(type(body).__name__.lower(), None) + if pos_vels is None: + continue + body.init_position = pos_vels['pos'] + body.init_velocity = pos_vels['vel'] + + if __name__ == '__main__': from astropy.time import Time # 时间 diff --git a/sim_scenes/fiction/sirius_is_coming.py b/sim_scenes/fiction/sirius_is_coming.py index 44ee6b2b47d4ab8f09e21ac66adbf83424b589aa..c57c43aab0da614f3f8314a279d635807f7bd6b1 100644 --- a/sim_scenes/fiction/sirius_is_coming.py +++ b/sim_scenes/fiction/sirius_is_coming.py @@ -1,6 +1,6 @@ # -*- coding:utf-8 -*- # title :天狼星进入太阳系场景模拟 -# description :天狼星进入太阳系场景模拟 +# description :天狼星把土星带走了 # author :Python超人 # date :2023-02-11 # link :https://gitcode.net/pythoncr/ @@ -8,44 +8,61 @@ # ============================================================================== from bodies import Sirius, Sun, Mercury, Venus, Earth, \ Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Moon, Asteroids +from common.celestial_data_service import init_bodies_pos_vels from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_MONTH, AU -from sim_scenes.func import mayavi_run, ursina_run +from sim_scenes.func import ursina_run, create_sphere_sky +from ursina import camera, application + +from simulators.ursina.ursina_event import UrsinaEvent if __name__ == '__main__': - # 八大行星：木星(♃)、土星(♄)、天王星(♅)、海王星(♆)、地球(⊕)、金星(♀)、火星(♂)、水星(☿) - # 排列顺序 - # 1、体积：(以地球为1)木星 ：土星 ：天王星 ：海王星 ：地球 ：金星 ：火星 ：水星 = 1330：745：65：60：1：0.86：0.15：0.056 - # 2、质量：(以地球为1)木星 ：土星 ：天王星 ：海王星 ：地球 ：金星 ：火星 ：水星 = 318：95：14.53：17.15：1：0.8：0.11：0.0553 - # 3、离太阳从近到远的顺序：水星、金星、地球、火星、木星、土星、天王星、海王星 - # ===================================================================== - # 以下展示的效果为太阳系真实的距离 # 由于宇宙空间尺度非常大，如果按照实际的天体大小，则无法看到天体，因此需要对天体的尺寸进行放大 sun = Sun(name="太阳", init_velocity=[0, 2, 0], - size_scale=0.8e2) # 太阳放大 80 倍，距离保持不变 + size_scale=0.8e2) # 太阳放大 80 倍，距离保持不变 sirius = Sirius(name="天狼星", init_position=[-10 * AU, 4 * AU, 4 * AU], init_velocity=[20, -10, 0], - size_scale=0.8e2) # 天狼星放大 80 倍，距离保持不变 + size_scale=0.8e2) # 天狼星放大 80 倍，距离保持不变 bodies = [ sirius, sun, - Mercury(name="水星", size_scale=4e3), # 水星放大 4000 倍，距离保持不变 - Venus(name="金星", size_scale=4e3), # 金星放大 4000 倍，距离保持不变 - Earth(name="地球", size_scale=4e3), # 地球放大 4000 倍，距离保持不变 - Mars(name="火星", size_scale=4e3), # 火星放大 4000 倍，距离保持不变 + Mercury(name="水星", size_scale=4e3), # 水星放大 4000 倍，距离保持不变 + Venus(name="金星", size_scale=4e3), # 金星放大 4000 倍，距离保持不变 + Earth(name="地球", size_scale=4e3), # 地球放大 4000 倍，距离保持不变 + Mars(name="火星", size_scale=4e3), # 火星放大 4000 倍，距离保持不变 # Asteroids(name="小行星群", size_scale=1e2, # parent=sun), # 小行星群模拟(仅 ursina 模拟器支持) - Jupiter(name="木星", size_scale=0.8e3), # 木星放大 800 倍，距离保持不变 - Saturn(name="土星", size_scale=0.8e3), # 土星放大 800 倍，距离保持不变 - Uranus(name="天王星", size_scale=0.8e3), # 天王星放大 800 倍，距离保持不变 - Neptune(name="海王星", size_scale=1e3), # 海王星放大 1000 倍，距离保持不变 - Pluto(name="冥王星", size_scale=10e3), # 冥王星放大 10000 倍，距离保持不变(从太阳系的行星中排除) + Jupiter(name="木星", size_scale=0.8e3), # 木星放大 800 倍，距离保持不变 + Saturn(name="土星", size_scale=0.8e3), # 土星放大 800 倍，距离保持不变 + Uranus(name="天王星", size_scale=0.8e3), # 天王星放大 800 倍，距离保持不变 + Neptune(name="海王星", size_scale=1e3), # 海王星放大 1000 倍，距离保持不变 + Pluto(name="冥王星", size_scale=10e3), # 冥王星放大 10000 倍，距离保持不变(从太阳系的行星中排除) ] + init_bodies_pos_vels(bodies) + + def on_ready(): + """ + 事件绑定后，模拟器运行前会触发 + @return: + """ + # 创建天空 + # camera.clip_plane_near = 0.1 + camera.clip_plane_far = 1000000 + create_sphere_sky(scale=200000) + application.time_scale = 5 + + # 运行前会触发 on_ready + UrsinaEvent.on_ready_subscription(on_ready) # 使用 ursina 查看的运行效果 # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- - ursina_run(bodies, SECONDS_PER_MONTH, position=(0, 2 * AU, -11 * AU), - bg_music="sounds/interstellar.mp3") + ursina_run(bodies, SECONDS_PER_MONTH, + position=(0, 2 * AU, -11 * AU), + cosmic_bg='', + show_grid=False, + # bg_music="sounds/interstellar.mp3" + bg_music='sounds/no_glory.mp3', + ) diff --git a/sim_scenes/tri_bodies/two_way_foil.py b/sim_scenes/tri_bodies/two_way_foil.py index a8151a77015f163266376e13c9e1b2b458fd835c..b072f4b3a9c598d70feea73f159d46b1bd13019c 100644 --- a/sim_scenes/tri_bodies/two_way_foil.py +++ b/sim_scenes/tri_bodies/two_way_foil.py @@ -11,7 +11,7 @@ import time from ursina import camera, application from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto -from common.celestial_data_service import get_init_pos_vels +from common.celestial_data_service import get_init_pos_vels, init_bodies_pos_vels from common.consts import SECONDS_PER_WEEK, AU from objs import QuadObj, CircleObj, Obj from sim_scenes.func import camera_look_at, two_bodies_colliding @@ -44,15 +44,16 @@ class TwoWayFoilSim: ] # endregion - # 获取模拟的初始位置和速度 - init_pos_vels = get_init_pos_vels() - - for body in self.bodies: - pos_vels = init_pos_vels.get(type(body).__name__.lower(), None) - if pos_vels is None: - continue - body.init_position = pos_vels['pos'] - body.init_velocity = pos_vels['vel'] + init_bodies_pos_vels(self.bodies) + # # 获取模拟的初始位置和速度 + # init_pos_vels = get_init_pos_vels() + # + # for body in self.bodies: + # pos_vels = init_pos_vels.get(type(body).__name__.lower(), None) + # if pos_vels is None: + # continue + # body.init_position = pos_vels['pos'] + # body.init_velocity = pos_vels['vel'] def build_two_way_foil(self): """ diff --git a/sim_scenes/tri_bodies/two_way_foil_02.py b/sim_scenes/tri_bodies/two_way_foil_02.py index 0cbe44e5e0e43d00f40bd4e142c3b0ac6b39b0c1..96cd888e3d1a44cfdbfc33951f85bf2fa60b67ed 100644 --- a/sim_scenes/tri_bodies/two_way_foil_02.py +++ b/sim_scenes/tri_bodies/two_way_foil_02.py @@ -11,7 +11,7 @@ import time from ursina import camera, application from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto -from common.celestial_data_service import get_init_pos_vels +from common.celestial_data_service import get_init_pos_vels, init_bodies_pos_vels from common.consts import SECONDS_PER_WEEK, AU from objs import QuadObj, CircleObj, Obj from sim_scenes.func import camera_look_at, two_bodies_colliding @@ -44,15 +44,16 @@ class TwoWayFoilSim: ] # endregion - # 获取模拟的初始位置和速度 - init_pos_vels = get_init_pos_vels() - - for body in self.bodies: - pos_vels = init_pos_vels.get(type(body).__name__.lower(), None) - if pos_vels is None: - continue - body.init_position = pos_vels['pos'] - body.init_velocity = pos_vels['vel'] + init_bodies_pos_vels(self.bodies) + # # 获取模拟的初始位置和速度 + # init_pos_vels = get_init_pos_vels() + # + # for body in self.bodies: + # pos_vels = init_pos_vels.get(type(body).__name__.lower(), None) + # if pos_vels is None: + # continue + # body.init_position = pos_vels['pos'] + # body.init_velocity = pos_vels['vel'] def build_two_way_foil(self): """