Python超人-宇宙模拟器

sim_scenes/tri_bodies/water_drop_03.py

+# -*- coding:utf-8 -*-
+# title           :三体水滴场景模拟
+# description     :三体水滴场景模拟
+# author          :Python超人
+# date            :2023-10-23
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.8
+# ==============================================================================
+
+
+from bodies import Sun, Earth, Moon
+from objs import CoreValagaClas, SciFiBomber, WaterDrop
+from common.consts import AU, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH
+from sim_scenes.func import ursina_run, camera_look_at, create_main_entity, two_bodies_colliding, create_sphere_sky
+from simulators.ursina.entities.body_timer import TimeData
+from simulators.ursina.entities.entity_utils import create_directional_light
+from simulators.ursina.ursina_config import UrsinaConfig
+from simulators.ursina.ursina_event import UrsinaEvent
+from ursina import camera, application
+
+if __name__ == '__main__':
+    """
+    三体水滴场景模拟
+    """
+    OFFSETTING = 0
+    WATER_SPEED = 400
+    # TODO: 可以抵消月球带动地球的力，保持地球在原地
+    # OFFSETTING = 0.01265
+    sun = Sun(name="太阳", size_scale=6e1, init_position=[0, 0, -3 * AU]).set_ignore_gravity(True)
+    earth_size_scale = 2.5e2
+    resolution = 100
+    earth = Earth(init_position=[0, -2500000, 0],
+                  texture="earth-huge.jpg",
+                  # rotate_angle=0,
+                  rotation_speed=0,
+                  init_velocity=[OFFSETTING, 0, 0],
+                  size_scale=earth_size_scale).set_ignore_gravity(True).set_resolution(resolution)  # 地球放大 5 倍，距离保持不变
+    # 创建云层（texture纹理图使用了透明云层的图片，云层的 size_scale 要稍微比地球大一点）
+    clouds = Earth(name="地球云层", texture="transparent_clouds.png",
+                   # rotate_angle=0,
+                   rotation_speed=0,
+                   init_position=[0, -2500000, 0],
+                   size_scale=earth_size_scale * 1.01,
+                   parent=earth).set_ignore_gravity(True).set_resolution(resolution)
+
+    water_drop = WaterDrop(init_position=[AU / 300, 0, AU / 100],
+                           texture="drops_bright.png",
+                           # trail_color=[200, 200, 255],
+                           init_velocity=[-WATER_SPEED, 0, 0],
+                           # size_scale=4e4,
+                           size_scale=1e3
+                           ).set_ignore_gravity(True).set_light_disable(True)
+    # moon = Moon(init_position=[0, 0, 363104],  # 距地距离约: 363104 至 405696 km
+    #             init_velocity=[-1.03, 0, 0], size_scale=2e1)  # 月球放大 10 倍，距离保持不变
+    # moon.set_light_disable(True)
+    d = 100000
+    num_x = 10
+    num_y = 10
+    num_z = 10
+
+    d = 100000
+    num_x = 2
+    num_y = 2
+    num_z = 2
+
+    x_offset = - (num_x) / 2 * d
+    y_offset = - (num_y) / 2 * (d / 2)
+    z_offset = - (num_z) / 2 * d
+    ship_list = []
+    for x in range(num_x):
+        for y in range(num_y):
+            for z in range(num_z):
+                core_valaga_clas = CoreValagaClas(size_scale=15, name=f'宇宙飞船:{num_x}:{num_y}:{num_z}',
+                                                  init_position=[x_offset + (x + 0.5) * d,
+                                                                 y_offset + (2 * y - 0.5) * (d / 2),
+                                                                 z_offset + (z + 0.8) * d - 20000]).set_ignore_gravity(
+                    True)
+                ship_list.append(core_valaga_clas)
+
+    for x in range(num_x):
+        for y in range(num_y):
+            for z in range(num_z):
+                sci_fi_bomber = SciFiBomber(size_scale=3.5, name=f'飞船:{num_x}:{num_y}:{num_z}',
+                                            init_position=[x_offset + (x + 0.5) * d,
+                                                           y_offset + (2 * y - 1.5) * (d / 2),
+                                                           z_offset + (z + 0.8) * d - 20000]).set_ignore_gravity(True)
+                ship_list.append(sci_fi_bomber)
+
+    # earth.rotation_speed /= 6  # 地球的转速降低50倍
+
+    bodies = [sun, earth, clouds, water_drop] + ship_list
+
+    import math
+
+
+    def calc_acceleration(current_pos, to_pos, desired_velocity):
+        """
+        Calculate the acceleration vector components (x, y, z) for an object to move from its current position to a specified position,
+        considering the desired velocity.
+        @param current_pos: Current position of the object (in a three-dimensional coordinate system)
+        @param to_pos: Specified position (in a three-dimensional coordinate system)
+        @param desired_velocity: Desired velocity of the object
+        @return: Acceleration vector components (x, y, z)
+        """
+        acceleration = [0, 0, 0]
+        # Calculate the differences in each dimension
+        diff_x = to_pos[0] - current_pos[0]
+        diff_y = to_pos[1] - current_pos[1]
+        diff_z = to_pos[2] - current_pos[2]
+        # Calculate the total distance
+        total_distance = math.sqrt(diff_x ** 2 + diff_y ** 2 + diff_z ** 2)
+        # Calculate the time required to reach the specified position
+        time = total_distance / desired_velocity
+        # Calculate the acceleration required to reach the desired velocity in the calculated time
+        acceleration_magnitude = desired_velocity / time
+        # Calculate the acceleration in each dimension
+        acceleration[0] = diff_x / time
+        acceleration[1] = diff_y / time
+        acceleration[2] = diff_z / time
+        # Adjust the acceleration using the acceleration magnitude
+        acceleration[0] *= acceleration_magnitude
+        acceleration[1] *= acceleration_magnitude
+        acceleration[2] *= acceleration_magnitude
+        return acceleration
+
+
+    def calc_velocity(current_pos, to_pos, velocity_fact):
+        """
+        Calculate the velocity vector components (x, y, z) for an object to move from its current position to a specified position,
+        considering the velocity factor.
+        @param current_pos: Current position of the object (in a three-dimensional coordinate system)
+        @param to_pos: Specified position (in a three-dimensional coordinate system)
+        @param velocity_fact: Velocity factor
+        @return: Velocity vector components (x, y, z)
+        """
+        velocity = [0, 0, 0]
+        # Calculate the differences in each dimension
+        diff_x = to_pos[0] - current_pos[0]
+        diff_y = to_pos[1] - current_pos[1]
+        diff_z = to_pos[2] - current_pos[2]
+        # Calculate the total distance
+        total_distance = math.sqrt(diff_x ** 2 + diff_y ** 2 + diff_z ** 2)
+        # Calculate the proportions of velocity in each dimension
+        velocity[0] = diff_x / total_distance
+        velocity[1] = diff_y / total_distance
+        velocity[2] = diff_z / total_distance
+        # Adjust the velocity using the velocity factor
+        velocity[0] *= velocity_fact
+        velocity[1] *= velocity_fact
+        velocity[2] *= velocity_fact
+        return velocity
+
+
+    target_ship_index = 0
+
+
+    def on_timer_changed(time_data: TimeData):
+        global target_ship_index
+
+        if target_ship_index > len(ship_list) - 1:
+            return
+
+        target = ship_list[target_ship_index]
+        if time_data.total_days > 0.:
+            velocity = calc_velocity(water_drop.position, target.position, 10)
+            water_drop.velocity = velocity
+
+            water_drop.look_at(target, rotation_x=None, rotation_y=None, rotation_z=None)
+
+        camera_look_at(water_drop, rotation_z=0)
+
+        if two_bodies_colliding(water_drop, target):
+            # 将石头隐藏、设置引力无效后，展示爆炸效果
+            print("目标已经击毁", target)
+            water_drop.explode(target, scale=1e1, fps=0.5)
+            water_drop.planet.enabled = True
+            target_ship_index += 1
+
+
+    def on_ready():
+        camera.clip_plane_near = 0.001
+        camera.clip_plane_far = 2100
+        create_sphere_sky(scale=2000)
+        application.time_scale = 0.05
+
+        UrsinaConfig.trail_type = 'line'
+        UrsinaConfig.trail_length = 10
+        for body in bodies:
+            if isinstance(body, CoreValagaClas):
+                body.planet.rotation_x = 0  # -10
+            elif isinstance(body, SciFiBomber):
+                body.planet.rotation_x = -90
+
+        water_drop.planet.main_entity.rotation_z = 90
+
+        water_drop.init_position = (0, 0, 2000)
+        water_drop.init_velocity = [0, 0, 0]
+
+        create_main_entity(water_drop, rotation_y=90)
+
+
+    # 订阅事件后，上面2个函数功能才会起作用
+    # 运行前会触发 on_ready
+    UrsinaEvent.on_ready_subscription(on_ready)
+    # 运行中，每时每刻都会触发 on_timer_changed
+    UrsinaEvent.on_timer_changed_subscription(on_timer_changed)
+
+    # 使用 ursina 查看的运行效果
+    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
+    # position = 左-右+、上+下-、前+后-
+    ursina_run(bodies,
+               SECONDS_PER_DAY / 24,
+               # SECONDS_PER_WEEK * 4,
+               # position=(0, 0, -220000),
+               position=(0, 0, -20000),
+               # position=(0, 0, 0),
+               show_grid=False,
+               cosmic_bg="",
+               # gravity_works=False,
+               # save_cube_map=True,
+               show_timer=True,
+               timer_enabled=True,
+               show_camera_info=False,
+               show_control_info=False,
+               view_closely=True,
+               show_trail=True
+               )