diff --git a/sim_scenes/solar_system/speed_of_light.py b/sim_scenes/solar_system/speed_of_light.py index 0a013c99ca7467cd4ec42fcde5f3849b1339d54f..4984b052f09d251c79c66d39a04f0f63acfbbc17 100644 --- a/sim_scenes/solar_system/speed_of_light.py +++ b/sim_scenes/solar_system/speed_of_light.py @@ -11,7 +11,7 @@ from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_YEAR, A from sim_scenes.func import mayavi_run, ursina_run from simulators.ursina.ursina_config import UrsinaConfig from simulators.ursina.ursina_event import UrsinaEvent -from ursina import Text, Panel, color, camera +from ursina import Text, Panel, color, camera, Vec3 def create_bodies(): @@ -26,15 +26,15 @@ def create_bodies(): sun = Sun(name="太阳", size_scale=0.6e2) # 太阳放大 60 倍，距离保持不变 bodies = [ sun, - Mercury(name="水星", size_scale=3e3), # 水星放大 3000 倍，距离保持不变 - Venus(name="金星", size_scale=4e3), # 金星放大 4000 倍，距离保持不变 - Earth(name="地球", size_scale=4e3), # 地球放大 4000 倍，距离保持不变 - Mars(name="火星", size_scale=4e3), # 火星放大 4000 倍，距离保持不变 - Asteroids(name="小行星群", size_scale=3.2e2, - parent=sun), # 小行星群模拟(仅 ursina 模拟器支持) - Jupiter(name="木星", size_scale=0.7e3), # 木星放大 600 倍，距离保持不变 - Saturn(name="土星", size_scale=0.7e3), # 土星放大 600 倍，距离保持不变 - Uranus(name="天王星", size_scale=0.7e3), # 天王星放大 600 倍，距离保持不变 + Mercury(name="水星", size_scale=2e3), # 水星放大 2000 倍，距离保持不变 + Venus(name="金星", size_scale=2e3), # 金星放大 2000 倍，距离保持不变 + Earth(name="地球", size_scale=2e3), # 地球放大 2000 倍，距离保持不变 + Mars(name="火星", size_scale=2e3), # 火星放大 2000 倍，距离保持不变 + # Asteroids(name="小行星群", size_scale=3.2e2, + # parent=sun), # 小行星群模拟(仅 ursina 模拟器支持) + Jupiter(name="木星", size_scale=0.6e3), # 木星放大 600 倍，距离保持不变 + Saturn(name="土星", size_scale=0.6e3), # 土星放大 600 倍，距离保持不变 + Uranus(name="天王星", size_scale=0.7e3), # 天王星放大 700 倍，距离保持不变 Neptune(name="海王星", size_scale=1e3), # 海王星放大 1000 倍，距离保持不变 Pluto(name="冥王星", size_scale=10e3), # 冥王星放大 10000 倍，距离保持不变(从太阳系的行星中排除) ] @@ -46,13 +46,14 @@ def create_bodies(): return bodies -def create_light(): +def create_light(size_scale=1e4): """ 用天体模拟一个光子 @return: """ - return Body(name='光速', mass=0, size_scale=1e4, color=(255, 255, 0), - init_position=[AU / 2, 0, 0], + return Body(name='光速', mass=0, size_scale=size_scale, color=(255, 255, 0), + # init_position=[AU / 3, 0, 0], + init_position=[AU / 12, 0, 0], init_velocity=[0, 0, 299792.458]).set_light_disable(True) # 1光速=299792.458 千米/秒(km/秒) @@ -87,6 +88,10 @@ arrived_bodies = [] text_panel = None arrived_info = "" +CAMERA_FOLLOW_LIGHT = None # 不跟随光 +CAMERA_FOLLOW_LIGHT = 'ForwardView' # 向前看 +# CAMERA_FOLLOW_LIGHT = 'SideView' # 侧面看 + def on_reset(): global arrived_info @@ -100,6 +105,14 @@ def on_ready(): global text_panel text_panel = create_text_panel() + if CAMERA_FOLLOW_LIGHT == "SideView": + camera.parent = light_body.planet + camera.rotation_y = -85 + elif CAMERA_FOLLOW_LIGHT == "ForwardView": + camera.parent = light_body.planet + light_body.planet.enabled = False + camera.rotation_y = -15 + def on_timer_changed(time_text, time_data): global arrived_info @@ -125,16 +138,31 @@ UrsinaEvent.on_reset_subscription(on_reset) UrsinaEvent.on_ready_subscription(on_ready) +if CAMERA_FOLLOW_LIGHT == "SideView": + position = (2 * AU, 0, -AU / 8) + show_trail = True + light_size_scale = 1e3 +elif CAMERA_FOLLOW_LIGHT == "ForwardView": + position = (0, AU / 10, -AU) + show_trail = False + light_size_scale = 1e2 +else: + position = (0, 2 * AU, -11 * AU) + show_trail = True + light_size_scale = 5e3 + # 创建太阳系天体（忽略质量，引力无效，初速度全部为0） bodies = create_bodies() # 创建一个以光速前进的天体（模拟一个光子，质量为0才能达到光速） -light_body = create_light() +light_body = create_light(size_scale=light_size_scale) bodies.append(light_body) # 使用 ursina 查看的运行效果 # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- -ursina_run(bodies, 60, position=(0, 2 * AU, -11 * AU), - show_trail=True, show_timer=True, +ursina_run(bodies, 60, + position=position, + show_trail=show_trail, show_timer=True, + # view_closely=True, bg_music="sounds/interstellar.mp3")