Python超人-宇宙模拟器

sim_scenes/earth/earth_satellites.py

@@ -10,98 +10,19 @@ from bodies import Moon, Earth, Body
 from objs import Satellite, Satellite2
 from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH
 from sim_scenes.func import mayavi_run, ursina_run, camera_look_at
-import math
-import random
-
-# def calculate_satellite_velocity(earth: Earth, altitude):
-#     """
-#
-#     @param earth_raduis:
-#     @param altitude:
-#     @return:
-#     """
-#     # 地球引力常数
-#     G = 6.67430e-11
-#     # 地球质量
-#     M = earth.mass  # 5.97e24
-#     # 地球半径
-#     R = earth.raduis + altitude
-#     # 地球表面的线速度
-#     v_surface = math.sqrt(G * M / R)
-#     # 地球卫星轨道速度
-#     v_satellite = math.sqrt(G * M / (R + altitude))
-#     return v_satellite - v_surface
-
-import random
-
-import random
-import math
-
 from simulators.ursina.entities.body_timer import TimeData
 from simulators.ursina.ursina_config import UrsinaConfig
 from simulators.ursina.ursina_event import UrsinaEvent
 
-
-# def get_satellite_position_velocity(earth_mass, earth_position, earth_radius, altitude):
-#     # 万有引力常数
-#     G = 6.6743 * 10 ** (-11)
-#     # 地球质量
-#     M = earth_mass
-#     # 地球半径+海拔高度
-#     R = earth_radius + altitude
-#
-#     # 随机生成卫星的位置，确保在球面上
-#     phi = random.uniform(0, 2 * math.pi)
-#     costheta = random.uniform(-1, 1)
-#     u = random.uniform(0, 1)
-#
-#     theta = math.acos(costheta)
-#     r = R * (u ** (1 / 3))
-#
-#     x = r * math.sin(theta) * math.cos(phi)
-#     y = r * math.sin(theta) * math.sin(phi)
-#     z = r * math.cos(theta)
-#
-#     # 计算速度的方向
-#     # 位置矢量
-#     r_vec = [x, y, z]
-#     # 速度方向和位置矢量垂直，采用向量叉积的性质
-#     v_dir = [-y, x, 0]
-#     # 归一化
-#     v_dir_norm = [v / math.sqrt(x ** 2 + y ** 2) for v in v_dir]
-#
-#     # 计算速度大小
-#     v = math.sqrt(G * M / R)
-#
-#     # 计算速度矢量
-#     v_vec = [v * dir for dir in v_dir_norm]
-#
-#     # 计算卫星的位置和速度
-#     position = [earth_position[0] + x, earth_position[1] + y, earth_position[2] + z]
-#     velocity = v_vec
-#
-#     return tuple(position), [0, 0, 0]
-#
-#
-# def generate_satellite_coordinates(earth_radius, altitude):
-#     theta = random.uniform(0, 2 * math.pi)  # 在0~2π内随机生成一个角度
-#     phi = random.uniform(0, math.pi)  # 在0~π内随机生成一个角度
-#     r = earth_radius + altitude  # 地球半径+海拔高度
-#     x = r * math.sin(phi) * math.cos(theta)  # 计算x坐标
-#     y = r * math.sin(phi) * math.sin(theta)  # 计算y坐标
-#     z = r * math.cos(phi)  # 计算z坐标
-#
-#     return [x, y, z]
-
-
 if __name__ == '__main__':
     """
-    地球、6个月球
+    地球和卫星模拟
     """
     # 地球在中心位置
     earth = Earth(init_position=[0, 0, 0], size_scale=1, texture="earth_hd.jpg", init_velocity=[0, 0, 0])
-    # 北斗卫星高度为2.13-2.15万千米。GPS卫星平均轨道高度2.02万千米。
+
     bodies = [earth]
+    # 卫星位置和初始速度信息
     satellite_infos = [
         {"position": [0, 0, 10002], "velocity": [6.3, 0, 0]},
         {"position": [0, 0, -12000], "velocity": [5.75, 0, 0]},
@@ -111,9 +32,6 @@ if __name__ == '__main__':
         {"position": [0, 0, -10000], "velocity": [0, 6.3, 0]},
     ]
     for i, info in enumerate(satellite_infos):
-        # altitude = random.randint(4000, 10000)
-        # position, velocity = get_satellite_position_velocity(earth.mass, earth.init_position, earth.raduis, altitude)
-
         satellite = Satellite(name=f'卫星{i + 1}', mass=4.4e10, size_scale=2e2, color=(255, 200, 0),
                               init_position=info["position"],
                               init_velocity=info["velocity"])
@@ -127,6 +45,7 @@ if __name__ == '__main__':
         UrsinaConfig.trail_type = "line"
         pass
 
+
     def on_timer_changed(time_data: TimeData):
         for info in satellite_infos:
             info["satellite"].planet.look_at(earth.planet)

sim_scenes/fiction/earth_rocks.py

@@ -6,13 +6,10 @@
 # link            :https://gitcode.net/pythoncr/
 # python_version  :3.8
 # ==============================================================================
-from bodies import Moon, Earth, Body
+from bodies import Earth, Body
 from objs import RockSnow, Rock, create_rock
-from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH
-from sim_scenes.func import mayavi_run, ursina_run, camera_look_at, two_bodies_colliding
-import math
-import random
-
+from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY
+from sim_scenes.func import ursina_run, camera_look_at, two_bodies_colliding
 from simulators.ursina.entities.body_timer import TimeData
 from simulators.ursina.ursina_config import UrsinaConfig
 from simulators.ursina.ursina_event import UrsinaEvent
@@ -22,18 +19,20 @@ if __name__ == '__main__':
     岩石坠落地球
     """
     # 地球在中心位置
-    earth = Earth(init_position=[0, 0, 0], size_scale=1, texture="earth_hd.jpg", init_velocity=[0, 0, 0])
+    earth = Earth(init_position=[0, 0, 0], init_velocity=[0, 0, 0],
+                  rotation_speed=0, size_scale=1, texture="earth_hd.jpg")
 
     bodies = [earth]
+    # 岩石位置和初始速度信息
     infos = [
-        {"position": [0, 0, 10002], "velocity": [2.3, 0, 0]},
-        {"position": [0, 0, -12000], "velocity": [1.75, 0, 0]},
-        {"position": [0, 8000, 0], "velocity": [1.05, 0, 0]},
-        {"position": [0, -12002, 0], "velocity": [1.75, 0, 0]},
-        {"position": [0, 18000, 0], "velocity": [1.05, 0, 0]},
-        {"position": [0, -22002, 0], "velocity": [1.75, 0, 0]},
-        {"position": [0, 0, 8002], "velocity": [0, 1.05, 0]},
-        {"position": [0, 0, -10000], "velocity": [0, 1.3, 0]},
+        {"position": [0, 0, 10002], "velocity": [4.5, 0, 0]},
+        {"position": [0, 0, -12000], "velocity": [3.5, 0, 0]},
+        {"position": [0, 8000, 0], "velocity": [4.5, 0, 0]},
+        {"position": [0, -12002, 0], "velocity": [3.5, 0, 0]},
+        {"position": [0, 18000, 0], "velocity": [3.0, 0, 0]},
+        {"position": [-20000, 0, 0], "velocity": [0, -2.0, 0]},
+        {"position": [0, 0, 18002], "velocity": [0, 2.5, 0]},
+        {"position": [0, 0, -10000], "velocity": [0, 3.3, 0]},
     ]
     for i, info in enumerate(infos):
         rock = create_rock(no=i % 7 + 1, name=f'岩石{i + 1}', mass=4.4e10, size_scale=5e2, color=(255, 200, 0),

sim_scenes/func.py

@@ -59,17 +59,29 @@ def mayavi_run(bodies, dt=SECONDS_PER_WEEK,
     mlab.show()
 
 
-def camera_look_at(body, rotation_x=None, rotation_y=None, rotation_z=None):
+def set_camera_parent(target):
+    from ursina import camera
+    if hasattr(target, "planet"):
+        camera.parent = target.planet.main_entity
+    else:
+        camera.parent = target
+
+
+def camera_look_at(target, rotation_x=None, rotation_y=None, rotation_z=None):
     """
     让摄像机看向指定天体
-    @param body: 天体
+    @param target: 天体
     @param rotation_x: x轴旋转角度（None表示不旋转）
     @param rotation_y: y轴旋转角度（None表示不旋转）
     @param rotation_z: z轴旋转角度（None表示不旋转）
     @return:
     """
     from ursina import camera
-    camera.look_at(body.planet)
+    if hasattr(target, "planet"):
+        camera.look_at(target.planet.main_entity)
+    else:
+        camera.look_at(target)
+
     if rotation_x is not None:
         camera.rotation_x = rotation_x
     if rotation_y is not None:
@@ -199,7 +211,7 @@ def create_solar_system_bodies(ignore_mass=False, init_velocity=None):
         Jupiter(name="木星", size_scale=0.3e3),  # 木星放大 300 倍，距离保持不变
         Saturn(name="土星", size_scale=0.3e3),  # 土星放大 300 倍，距离保持不变
         Uranus(name="天王星", size_scale=0.4e3),  # 天王星放大 400 倍，距离保持不变
-        Neptune(name="海王星", size_scale=1e3),  # 海王星放大 1000 倍，距离保持不变
+        Neptune(name="海王星", size_scale=1e3),  # 海王星放大 800 倍，距离保持不变
         Pluto(name="冥王星", size_scale=10e3),  # 冥王星放大 10000 倍，距离保持不变(从太阳系的行星中排除)
     ]
 

sim_scenes/solar_system/speed_of_light_init.py

@@ -43,8 +43,8 @@ class SpeedOfLightInit:
             self.camera_position = (AU / 5, 0, 0)
             self.show_trail = True
             self.light_size_scale = 2e6
-            self.light_init_position = [AU, 0, 0]
-            self.auto_control_speed = True
+            self.light_init_position = [AU, AU / 20, 0]
+            # self.auto_control_speed = True
         elif self.__camera_follow_light in ["SideViewActualSize"]:
             # 摄像机位置 = 前-后+、上+下-、左-右+、
             # self.camera_position = (0, AU, 0)
@@ -58,8 +58,8 @@ class SpeedOfLightInit:
             self.camera_position = (0, -AU / 25, -AU / 50)
             self.show_trail = True
             self.light_size_scale = 2e6
-            self.light_init_position = [AU / 12, 0, 0]
-            # self.auto_control_speed = True
+            self.light_init_position = [AU / 8, AU / 10, 0]
+            self.auto_control_speed = True
         else:
             # 摄像机位置 = 左-右+、上+下-、前+后-
             self.camera_position = (0, AU, -6 * AU)
@@ -155,7 +155,7 @@ class SpeedOfLightInit:
             camera.rotation_z = -90
             camera.rotation_y = -85
         elif self.__camera_follow_light in ["ForwardView"]:
-            camera.rotation_x = -75
+            camera.rotation_x = -72
         elif self.__camera_follow_light in ["SideViewActualSize"]:
             self.light_ship.planet.rotation_x = 0
             # camera.rotation_z = -90
@@ -196,22 +196,22 @@ class SpeedOfLightInit:
             run_speed_maps = [
                 {"au": 0., "secs": 1},
                 {"au": 0.008, "secs": 1},
-                {"au": "?", "secs": SECONDS_PER_MINUTE * 10},
+                {"au": "?", "secs": SECONDS_PER_MINUTE * 5},
                 {"au": 0.386, "secs": 1},
                 {"au": 0.3865, "secs": 1},  # [00:03:12] 到达 [水星] 0.384 AU
-                {"au": "?", "secs": SECONDS_PER_MINUTE * 10},
+                {"au": "?", "secs": SECONDS_PER_MINUTE * 5},
                 {"au": 0.721, "secs": 1},
-                {"au": 0.723, "secs": 1},  # [00:06:00] 到达 [金星] 0.721 AU
-                {"au": "?", "secs": SECONDS_PER_MINUTE * 10},
+                {"au": 0.724, "secs": 1},  # [00:06:00] 到达 [金星] 0.721 AU
+                {"au": "?", "secs": SECONDS_PER_MINUTE * 5},
                 {"au": 1.0, "secs": 1},
                 {"au": 1.002, "secs": 1},  # [00:08:19] 到达 [地球] 1.0 AU
-                {"au": "?", "secs": SECONDS_PER_MINUTE * 10},
+                {"au": "?", "secs": SECONDS_PER_MINUTE * 5},
                 {"au": 1.52, "secs": 1},
                 {"au": 1.522, "secs": 1},  # [00:12:39] 到达 [火星] 1.52 AU
-                {"au": "?", "secs": SECONDS_PER_MINUTE * 30},
+                {"au": "?", "secs": SECONDS_PER_MINUTE * 25},
                 {"au": 5.19, "secs": 1},
                 {"au": 5.20, "secs": 1},  # [00:43:10] 到达 [木星] 5.19 AU
-                {"au": "?", "secs": SECONDS_PER_MINUTE * 30},
+                {"au": "?", "secs": SECONDS_PER_MINUTE * 25},
                 {"au": 9.5, "secs": 1},
                 {"au": 9.51, "secs": 1},  # [01:19:01] 到达 [土星] 9.5 AU
                 {"au": "?", "secs": SECONDS_PER_MINUTE * 30},

sim_scenes/solar_system/transit_of_venus_mercury.py

@@ -8,15 +8,17 @@
 # ==============================================================================
 from bodies import Sun, Mercury, Venus, Earth
 from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_HOUR, AU
-from sim_scenes.func import mayavi_run, ursina_run, camera_look_at
+from sim_scenes.func import ursina_run, set_camera_parent
 from simulators.ursina.entities.body_timer import TimeData
 from simulators.ursina.ursina_event import UrsinaEvent
 
 if __name__ == '__main__':
     # 水星、金星凌日
+    # 以地球的视角看水星、金星凌日
     earth = Earth(
         name="地球",
-        rotation_speed=0,
+        rotate_angle=0,  # 不倾斜
+        rotation_speed=0,  # 不旋转
         texture="transparent.png"  # 地球纹理透明，不会挡住摄像机视线
     )
     sun = Sun(name="太阳", size_scale=5e1)  # 太阳放大 50 倍
@@ -40,7 +42,7 @@ if __name__ == '__main__':
     def on_ready():
         from ursina import camera
         # 摄像机跟随地球（模拟在地球上看到的效果）
-        camera.parent = earth.planet
+        set_camera_parent(earth)
 
         if hasattr(camera, "sky"):
             # 摄像机跟随地球后，需要对深空背景进行调整，否则看到的是黑色背景
@@ -52,8 +54,7 @@ if __name__ == '__main__':
 
     def on_timer_changed(time_data: TimeData):
         # 时时刻刻的让地球看向太阳（摄像机跟随地球看向太阳）
-        earth.planet.look_at(sun.planet)
-        earth.planet.rotation_z = 0
+        earth.look_at(sun, rotation_z=0)
 
 
     UrsinaEvent.on_ready_subscription(on_ready)

simulators/ursina_simulator.py

@@ -53,6 +53,7 @@ class UrsinaSimulator(Simulator):
         for i, view in enumerate(self.body_views):
             # pos = tuple(body.position)
             # ursina_view = UrsinaView(body)
+            self.bind_event(view.body)
             view.update()
             self.ursina_views.append(view)
             # planets.append(newPlanet)
@@ -60,6 +61,28 @@ class UrsinaSimulator(Simulator):
         self.adjust_system_motion_params()
         UrsinaEvent.on_searching_bodies_subscription(type(self).__name__, self.on_searching_bodies)
 
+    def bind_event(self, body):
+
+        def body_look_at(target, rotation_x=None, rotation_y=None, rotation_z=None):
+            """
+            让 body 看向 target 目标天体
+            @param target: 目标天体
+            @param rotation_x: x轴旋转角度（None表示不旋转）
+            @param rotation_y: y轴旋转角度（None表示不旋转）
+            @param rotation_z: z轴旋转角度（None表示不旋转）
+            @return:
+            """
+            if hasattr(target, "planet") and hasattr(body, "planet"):
+                body.planet.main_entity.look_at(target.planet.main_entity)
+                if rotation_x is not None:
+                    body.planet.rotation_x = rotation_x
+                if rotation_y is not None:
+                    body.planet.rotation_y = rotation_y
+                if rotation_z is not None:
+                    body.planet.rotation_z = rotation_z
+
+        body.look_at = body_look_at
+
     # def get_bodies_max_distance(self, body_views):
     #     max_distance = 0
     #     for b1 in body_views: