diff --git a/sim_lab/solarsystem_planets_size.py b/sim_lab/solarsystem_planets_size.py index 3de89144d85ac4fd343c426b1a94c7893c78313c..f07f16a0084d3abc98570b602eab4e1ef35d1b1e 100644 --- a/sim_lab/solarsystem_planets_size.py +++ b/sim_lab/solarsystem_planets_size.py @@ -11,6 +11,7 @@ import threading from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Moon from common.consts import SECONDS_PER_HOUR, SECONDS_PER_DAY, AU from sim_scenes.func import ursina_run, create_sphere_sky +from sim_scenes.universe_sim_scenes import UniverseSimScenes from simulators.ursina.entities.body_timer import TimeData, AppTimeUtil from simulators.ursina.entities.entity_utils import create_directional_light from simulators.ursina.ursina_event import UrsinaEvent @@ -176,6 +177,8 @@ if __name__ == '__main__': # 运行前会触发 on_ready UrsinaEvent.on_ready_subscription(on_ready) + UniverseSimScenes.set_window_size((1920, 1079), False) + # 使用 ursina 查看的运行效果 # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- diff --git a/sim_scenes/solar_system/solarsystem_planets_size.py b/sim_scenes/solar_system/solarsystem_planets_size.py index f96ddd6c4f7a3e68c594acc91cbec002e7f3da97..5c45ef45cfa90d80d7fa4a45e80e9d588384ede8 100644 --- a/sim_scenes/solar_system/solarsystem_planets_size.py +++ b/sim_scenes/solar_system/solarsystem_planets_size.py @@ -6,40 +6,49 @@ # link :https://gitcode.net/pythoncr/ # python_version :3.8 # ============================================================================== +import threading + from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Moon from common.consts import SECONDS_PER_HOUR, SECONDS_PER_DAY, AU -from sim_scenes.func import mayavi_run, ursina_run +from sim_scenes.func import ursina_run, create_sphere_sky +from sim_scenes.universe_sim_scenes import UniverseSimScenes from simulators.ursina.entities.body_timer import TimeData, AppTimeUtil from simulators.ursina.entities.entity_utils import create_directional_light from simulators.ursina.ursina_event import UrsinaEvent +from simulators.ursina.ursina_config import UrsinaConfig # 可以调整整体放大倍数 ，比例会保持不变 FACTOR = 10 # 地球和月球之间的距离常量，距地距离约: 363104 至 405696 km，平均距离 384000 km E_M_DISTANCE = 405696 * FACTOR -earth1 = Earth("地球1", size_scale=FACTOR, init_position=[0, 0, 0]) +earth1 = Earth("地球1", size_scale=FACTOR, init_position=[0, 0, 0], rotate_angle=-23.44) moon = Moon("月球", size_scale=FACTOR, init_position=[E_M_DISTANCE, 0, 0]) -earth2 = Earth("地球2", texture="earth2.jpg", size_scale=FACTOR) +earth2 = Earth("地球2", texture="earth2.jpg", size_scale=FACTOR, rotate_angle=-23.44) bodies = [ earth1, moon, Mercury(name="水星", size_scale=FACTOR), Venus(name="金星", size_scale=FACTOR), earth2, Mars(name="火星", size_scale=FACTOR), - Jupiter(name="木星", size_scale=FACTOR), + Jupiter(name="木星", texture='jupiter_hd.jpg', size_scale=FACTOR), Saturn(name="土星", size_scale=FACTOR).show_rings(False), Uranus(name="天王星", size_scale=FACTOR), Neptune(name="海王星", size_scale=FACTOR), Pluto(name="冥王星", size_scale=FACTOR) ] -if __name__ == '__main__': - last_diameter = earth1.diameter * FACTOR / 2 - plant_positions = [] +body_rotation_y = [200, 100, 0, 0, 200, 30, 160, 20] + +for body in bodies: + body.rotation_speed = 0 +if __name__ == '__main__': # 使用应用的计时器工具 app_time_util = AppTimeUtil() + key_point_time_util = AppTimeUtil() + last_diameter = earth1.diameter * FACTOR / 2 + plant_positions = [] for i, body in enumerate(bodies): body.rotation_speed /= 10 # 星体的旋转速度减小10倍 body.ignore_mass = True @@ -48,6 +57,8 @@ if __name__ == '__main__': plant_positions.append([(body.diameter * FACTOR / 2) + last_diameter, 0, 0]) last_diameter += body.diameter * FACTOR # print(body) + import ursina + from ursina import camera, time, Vec3, application def on_ready(): @@ -56,17 +67,95 @@ if __name__ == '__main__': create_directional_light(position=(E_M_DISTANCE / 2, E_M_DISTANCE * 20, -E_M_DISTANCE * 100), light_num=3, target=earth1) + application.time_scale = 0.001 + sky = create_sphere_sky(scale=2000) + sky.alpha = 0.5 + # sky.rotation_y = -90 + sky.rotation_x = 200 + # sky.rotation_z = 200 + + for idx, body in enumerate(bodies): + if idx <= len(body_rotation_y) - 1: + body.planet.rotation_y = body_rotation_y[idx] + + + key_points = [(E_M_DISTANCE / 2, 0, -E_M_DISTANCE), + (0, 0, -E_M_DISTANCE / 4), + (100000, 0, -E_M_DISTANCE / 4), + (200000, 0, -E_M_DISTANCE / 4), + (300000, 0, -E_M_DISTANCE / 4), + (400000, 0, -E_M_DISTANCE / 4), + (500000, 0, -E_M_DISTANCE), # 木星 + (1800000, 0, -E_M_DISTANCE), # 土星 + (3000000, 0, -E_M_DISTANCE), # 天王星 + (3800000, 0, -E_M_DISTANCE), # 海王星 + (4000000, 0, -E_M_DISTANCE / 6), # 冥王星 + (4000000, 0, -E_M_DISTANCE / 3), + (3900000, 0, -E_M_DISTANCE / 1.5), + (3600000, 0, -E_M_DISTANCE / 1.2), + (3000000, 0, -E_M_DISTANCE / 1.1), + (E_M_DISTANCE / 2, 0, -E_M_DISTANCE), + ] + ursina_kps = [Vec3(point) * UrsinaConfig.SCALE_FACTOR for point in key_points] + + interval = 5 def on_timer_changed(time_data: TimeData): + # camera_time = app_time_util.get_param("camera_time", 2) + key_point_index = app_time_util.get_param("key_point_index", 0) body_index = app_time_util.get_param("body_index", 2) # 从第三个星球（水星）开始 + + if key_point_index + 1 >= len(ursina_kps): + # camera.position = ursina_kps[-1] + return + if body_index < len(bodies): - last_time = app_time_util.get_param("last_time", 3) # 间隔3秒出现水星 + last_time = app_time_util.get_param("last_time", interval + 3) # 间隔3秒出现水星 # 判断是否第一个到达指定的时间 if app_time_util.is_first_arrival(last_time, time_data): - bodies[body_index].init_position = plant_positions[body_index - 2] + # target_position = plant_positions[body_index - 2] + + def move_planet(planet_body, target_pos): + import numpy as np + + def warp(): + # current_pos = planet_body.init_position + t_pos = np.array(target_pos) * UrsinaConfig.SCALE_FACTOR + for _ in range(200): + current_pos = planet_body.init_position + c_pos = current_pos * UrsinaConfig.SCALE_FACTOR + n_pos = ursina.lerp(Vec3(c_pos[0], c_pos[1], c_pos[2]), + Vec3(t_pos[0], t_pos[1], t_pos[2]), 6e-2) + new_pos = np.array([n_pos[0], n_pos[1], n_pos[2]]) / UrsinaConfig.SCALE_FACTOR + planet_body.init_position = new_pos + time.sleep(0.0001) + + planet_body.init_position = target_pos + + return warp + + f = move_planet(bodies[body_index], plant_positions[body_index - 2]) + # f() + t = threading.Thread(target=f) + t.start() + # time.sleep(0.1) + app_time_util.inc_param("body_index", 1) - app_time_util.inc_param("last_time", 3) # 间隔3秒出现下一个行星 + app_time_util.inc_param("last_time", interval) # 间隔3秒出现下一个行星 + + print("key_point_index", key_point_index) + current_point = ursina_kps[key_point_index] + target_point = ursina_kps[key_point_index + 1] + camera_time = 2 + key_point_index * interval + + if time_data.app_time > camera_time: + dt = (time_data.app_time - camera_time) / interval + if dt <= 1: + camera.position = ursina.lerp(current_point, target_point, dt) + # print(camera.position) + elif key_point_time_util.is_first_arrival(camera_time, time_data): + app_time_util.inc_param("key_point_index", 1) # 运行中，每时每刻都会触发 on_timer_changed @@ -74,10 +163,17 @@ if __name__ == '__main__': # 运行前会触发 on_ready UrsinaEvent.on_ready_subscription(on_ready) + UniverseSimScenes.set_window_size((1920, 1079), False) + # 使用 ursina 查看的运行效果 # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- - ursina_run(bodies, SECONDS_PER_HOUR, - position=(E_M_DISTANCE / 2, 0, -E_M_DISTANCE * 1), - view_closely=True, + ursina_run(bodies, SECONDS_PER_HOUR / 2, + position=key_points[0], + cosmic_bg='', + show_control_info=False, + show_exit_button=False, + show_camera_info=False, + show_grid=False, + # view_closely=True, timer_enabled=True)