diff --git a/sim_scenes/solar_system/earth_moon_reality.py b/sim_scenes/solar_system/earth_moon_reality.py index 12196f2ce9725136a9ff10db8d2cbc2a4af995fa..12675b43edc865f10b96d945fee8062736121ea1 100644 --- a/sim_scenes/solar_system/earth_moon_reality.py +++ b/sim_scenes/solar_system/earth_moon_reality.py @@ -14,8 +14,9 @@ from bodies import Sun, Mercury, Venus, Earth, Mars, Asteroids, Jupiter, Saturn, from common.celestial_data_service import get_body_posvel, recalc_moon_position, calc_solar_acceleration, \ set_solar_system_celestial_position, set_earth_rotation from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, AU -from sim_scenes.func import ursina_run, camera_look_at +from sim_scenes.func import ursina_run, camera_look_at, create_text_panel from simulators.ursina.entities.body_timer import TimeData +from simulators.ursina.entities.entity_utils import clear_trails from simulators.ursina.ui.control_ui import ControlUI from simulators.ursina.ursina_config import UrsinaConfig from simulators.ursina.ursina_event import UrsinaEvent @@ -28,9 +29,7 @@ class SolarSystemRealitySim: @param debug_mode: 是否为调试模式 """ - self.show_asteroids = False self.clock_position_center = False - self.show_earth_clouds = False self.debug_mode = False self.recalc_moon_pos = True @@ -44,15 +43,15 @@ class SolarSystemRealitySim: # 地月缩放比例 # 为了更好的展示效果，需要对月球的位置重新计算（使得地月距离放大，月球相对地球方向不变），重新计算位置后，地球和月球可以放大1000倍以上 - if self.recalc_moon_pos: # 重新计算月球位置 - self.earth_size_scale = 10e3 if self.debug_mode else 1e3 - self.sun_size_scale = 0.04e2 if self.debug_mode else 0.4e2 - self.moon_size_scale = 2e3 - else: - # 不重新计算，则地月的距离相对整个太阳系会非常近，因此，月球只放大了30倍 - self.earth_size_scale = 2.5e1 - self.moon_size_scale = 5e1 - self.sun_size_scale = 1e1 + # if self.recalc_moon_pos: # 重新计算月球位置 + # self.earth_size_scale = 10e3 if self.debug_mode else 1e3 + # self.sun_size_scale = 0.04e2 if self.debug_mode else 0.4e2 + # self.moon_size_scale = 2e3 + # else: + # # 不重新计算，则地月的距离相对整个太阳系会非常近，因此，月球只放大了30倍 + # self.earth_size_scale = 2.5e1 + # self.moon_size_scale = 5e1 + # self.sun_size_scale = 1e1 self.earth_size_scale = 1 self.moon_size_scale = 1 @@ -70,20 +69,22 @@ class SolarSystemRealitySim: self.earth_camera.camera_init_val = 0 self.moon = Moon(name="月球", size_scale=self.moon_size_scale, rotation_speed=0.4065) # 月球 - + self.sun.show_trail = False + self.earth.trail_scale_factor = 0.2 + self.moon.trail_scale_factor = 0.3 # 所有天体 self.bodies = [self.sun, self.earth, self.earth_camera, self.moon] - def init_earth(self): - """ - 初始化地球 - @return: - """ - # 让地球显示自转轴线 - self.earth.rotate_axis_color = (255, 255, 50) - # 如果为调试模式，则太阳光对地球无效，方便查看 - if self.debug_mode: - self.earth.set_light_disable(True) + # def init_earth(self): + # """ + # 初始化地球 + # @return: + # """ + # # 让地球显示自转轴线 + # self.earth.rotate_axis_color = (255, 255, 50) + # # 如果为调试模式，则太阳光对地球无效，方便查看 + # if self.debug_mode: + # self.earth.set_light_disable(True) def show_clock(self, dt): """ @@ -119,10 +120,15 @@ class SolarSystemRealitySim: """ # 运行前触发 + self.text_panel = create_text_panel() + self.text_panel.text = "太阳缩放：1.0\n地球缩放：1.0\n月球缩放：1.0" + # camera.rotation_z = -20 # if self.debug_mode: # camera.fov = 30 # 调试时，拉近摄像机距离 - # camera.fov = 1 + # camera.fov = 40 + camera.clip_plane_near = 10 + # camera.clip_plane_far = 1000000 camera.parent = self.earth_camera.planet # camera.update = self.camera_update @@ -138,6 +144,7 @@ class SolarSystemRealitySim: # 设置后，可以调整鼠标键盘的控制速度 application.time_scale = 2 + # # def camera_update(self): # camera.x = -50 # 100 @@ -147,24 +154,83 @@ class SolarSystemRealitySim: # # 摄像机看向地球 # camera_look_at(self.earth) + def body_show(self, body): + body.planet.enabled = True + body.show_trail = True + + def body_hide(self, body): + body.planet.enabled = False + body.show_trail = False + # clear_trails(body.planet) + + def body_scale(self, body, value): + body.planet.init_scale *= value + if hasattr(body.planet.main_entity, "trail_scale"): + body.planet.main_entity.trail_scale *= value def set_camera_pos(self, time_data: TimeData): - # if time_data.total_days > 120: - # self.earth_camera.camera_init_val = +3000000 - # elif time_data.total_days > 90: - # self.earth_camera.camera_init_val = +1200000 - # elif time_data.total_days > 90: - # self.earth_camera.camera_init_val += 600000 - # elif time_data.total_days > 60: - # self.earth_camera.camera_init_val += 100000 + if time_data.total_days > 120: + self.earth_camera.camera_init_val += 300000 + elif time_data.total_days > 90: + self.earth_camera.camera_init_val += 60000 + elif time_data.total_days > 90: + self.earth_camera.camera_init_val += 20000 + elif time_data.total_days > 30: + self.earth_camera.camera_init_val += 18000 + elif time_data.total_days > 10: + self.earth_camera.camera_init_val += 4000 + elif time_data.total_days > 2: + self.earth_camera.camera_init_val += 500 + + # if UrsinaConfig.trail_factor < 50: + # self.body_scale(1.002) + if time_data.total_days < 5: + self.earth.show_trail = False + self.moon.show_trail = False + if 30 > time_data.total_days > 5: + self.earth.show_trail = True + self.moon.show_trail = True + self.body_scale(self.moon, 1.0025) + self.body_scale(self.earth, 1.002) + elif 60 > time_data.total_days > 30: + # self.body_hide(self.moon) + self.body_scale(self.moon, 1.002) + self.body_scale(self.earth, 1.002) + self.body_scale(self.sun, 1.002) + elif 150 > time_data.total_days > 60: + self.earth.planet.init_scale = 0.01 + self.earth.planet.main_entity.trail_scale = 0.03 + # self.body_scale(self.earth, 0.02) + # self.body_hide(self.earth) + # self.earth.planet.init_scale = 0.01 + # self.body_show(self.moon) + # self.body_scale(1.0015) + # self.sun_scale(1.0015) + # else: + # self.body_scale(1.002) + # self.sun_scale(1.0005) + + print("%s,%s" % (round(self.moon.planet.body_scale, 1), time_data.total_days)) # camera.x = -300 # 100 # camera.z = 200 # camera.y += self.earth_camera.camera_init_val * UrsinaConfig.SCALE_FACTOR - camera.x = -60 # 100 + # camera.x = -80 # 100 # camera.z = -10 - camera.y = 50 + dis_au = round(camera.y / UrsinaConfig.SCALE_FACTOR / AU, 2) + + if dis_au < 400: + if self.earth_camera.camera_init_val > 0: + camera.y += self.earth_camera.camera_init_val * UrsinaConfig.SCALE_FACTOR + else: + camera.y = 80 + + self.text_panel.text = "太阳大小缩放：%.1f\n地球大小缩放：%.1f\n月球大小缩放：%.1f\n摄像机距地球：%.2f天文单位" % \ + (self.sun.planet.body_scale, + self.earth.planet.body_scale, + self.moon.planet.body_scale, + dis_au) # UrsinaConfig.trail_factor = 3 * math.sqrt(camera.y / 250) pass @@ -183,7 +249,7 @@ class SolarSystemRealitySim: # 调整摄像机的位置 self.set_camera_pos(time_data) # 摄像机看向地球 - camera_look_at(self.earth) + camera_look_at(self.sun) # 显示时钟 self.show_clock(dt) diff --git a/simulators/ursina/entities/planet.py b/simulators/ursina/entities/planet.py index 35ab447a3c0eb2a492db912dd5d72b8c0d3e9eea..8b17a0bed91858a992ea8fec4cabff11e925dab4 100644 --- a/simulators/ursina/entities/planet.py +++ b/simulators/ursina/entities/planet.py @@ -31,6 +31,15 @@ class Planet(Entity): clear_trails(self) self.body.reset() + @property + def init_scale(self): + return self.__init_scale + + @init_scale.setter + def init_scale(self, val): + self.body_scale *= val / self.__init_scale + self.__init_scale = val + def __init__(self, body_view: BodyView): self.body_view = body_view if hasattr(self.body, "rotation_speed"): @@ -45,7 +54,9 @@ class Planet(Entity): else: scale = self.body.size_scale * UrsinaConfig.SCALE_FACTOR - self.init_scale = scale + self.__init_scale = scale + self.body_scale = 1.0 + if hasattr(body_view, "texture"): texture = load_texture(body_view.texture) # color.white