Python超人-宇宙模拟器

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 = 左-右+、上+下-、前+后-

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)