From 9c7d6f16fbe76622da60199ae34c6a219b2be65f Mon Sep 17 00:00:00 2001 From: march3 Date: Mon, 23 Oct 2023 15:14:10 +0800 Subject: [PATCH] =?UTF-8?q?Python=E8=B6=85=E4=BA=BA-=E5=AE=87=E5=AE=99?= =?UTF-8?q?=E6=A8=A1=E6=8B=9F=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sim_scenes/tri_bodies/water_drop_03.py | 227 +++++++++++++++++++++++++ 1 file changed, 227 insertions(+) create mode 100644 sim_scenes/tri_bodies/water_drop_03.py diff --git a/sim_scenes/tri_bodies/water_drop_03.py b/sim_scenes/tri_bodies/water_drop_03.py new file mode 100644 index 0000000..94a751f --- /dev/null +++ b/sim_scenes/tri_bodies/water_drop_03.py @@ -0,0 +1,227 @@ +# -*- 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 + ) -- GitLab