diff --git a/sim_lab/wormhole_sim.py b/sim_lab/wormhole_sim.py new file mode 100644 index 0000000000000000000000000000000000000000..863f4b937d4649083b1443b74f2bc586d7955122 --- /dev/null +++ b/sim_lab/wormhole_sim.py @@ -0,0 +1,184 @@ +# -*- coding:utf-8 -*- +# title :虫洞效果模拟 +# description :虫洞效果模拟 +# author :Python超人 +# date :2023-11-03 +# link :https://gitcode.net/pythoncr/ +# python_version :3.9 +# ============================================================================== +from bodies import Sun, Earth +from common.consts import SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH, SECONDS_PER_YEAR, AU +from objs import Obj, SpaceShip, StarWarsSpeeder, ScifiGunship, SciFiBomber, CoreValagaClas +from sim_scenes.func import mayavi_run, mpl_run, ursina_run, camera_look_at, two_bodies_colliding +from ursina import camera, application, lerp + +from bodies import Body +from simulators.ursina.ursina_event import UrsinaEvent + +ours_cosmic_bg = "bg_pan.jpg" +worms_cosmic_bg = "cosmic3.jpg" + + +class UniverseBody(Body): + def __init__(self, name="虫洞", mass=1.9891e35, + init_position=[0, 0, 0], + init_velocity=[0, 0, 0], + color=(0, 0, 0), + texture=None, + density=1.408e20, + size_scale=1.0, distance_scale=1.0, + rotation_speed=0, ignore_mass=False, trail_color=None, show_name=False): + params = { + "name": name, + "mass": mass, + "init_position": init_position, + "init_velocity": init_velocity, + "density": density, + "color": color, + "texture": texture, + "size_scale": size_scale * 10000, + "distance_scale": distance_scale, + "rotation_speed": rotation_speed, + "ignore_mass": ignore_mass, + "trail_color": trail_color, + "show_name": show_name + } + super().__init__(**params) + + +class WormHole(UniverseBody): + def __init__(self, name="虫洞", mass=1.9891e35, texture=worms_cosmic_bg, size_scale=1.0, + init_position=[0, 0, 0], init_velocity=[0, 0, 0]): + params = { + "name": name, + "mass": mass, + "init_position": init_position, + "init_velocity": init_velocity, + "density": 1.408e20, + # "color": color, + "texture": texture, + "size_scale": size_scale, + # "distance_scale": distance_scale, + # "rotation_speed": rotation_speed, + # "ignore_mass": ignore_mass, + # "trail_color": trail_color, + # "show_name": show_name + } + super().__init__(**params) + + +class OursUniverse(UniverseBody): + def __init__(self, name="我们的宇宙", texture=ours_cosmic_bg, size_scale=1.0, + init_position=[0, 0, 0], init_velocity=[0, 0, 0]): + params = { + "name": name, + "mass": 1.9891e35, + "init_position": init_position, + "init_velocity": init_velocity, + "density": 1.408e20, + # "color": color, + "texture": texture, + "size_scale": size_scale, + # "distance_scale": distance_scale, + + # "ignore_mass": ignore_mass, + # "trail_color": trail_color, + # "show_name": show_name + } + super().__init__(**params) + + +if __name__ == '__main__': + """ + + """ + # 代码案例如下： + SIZE_SCALE = 1e1 # 生成的太阳放大 50 倍 + RUN_DIAMETER = 1.392e6 + velocity_rate = 22 + + camera_target = CoreValagaClas(name="摄像机目标", mass=1e30, color=(111, 140, 255), + init_position=[0, 0, -AU], + init_velocity=[0, 0, 0], + size_scale=SIZE_SCALE * 1e1) # .set_ignore_gravity(True) + ours_universe = OursUniverse( + # size_scale=SIZE_SCALE * 1e3, + size_scale=SIZE_SCALE , + init_position=[0, 0, -AU], + init_velocity=[0, 0, 0], + ).set_ignore_gravity(True) + worm_hole = WormHole( + # init_position=[RUN_DIAMETER, RUN_DIAMETER, 0], + init_position=[0, 0, AU], + init_velocity=[0, 0, 0], + size_scale=SIZE_SCALE) # .set_ignore_gravity(True) + bodies = [ + # camera_target, + ours_universe, + worm_hole + ] + + camera_position = [0, 0, 0] + + + def on_ready(): + """ + 事件绑定后，模拟器运行前会触发 + @return: + """ + # 创建天空 + camera.clip_plane_near = 0.001 + camera.clip_plane_far = 10000000000000 + # create_sphere_sky(scale=200000) + application.time_scale = 0.001 + # camera.scale = 1000 + # camera.parent = camera_target.planet + # camera.rotation_x = 90 + camera_look_at(worm_hole) + camera.collider="sphere" + worm_hole.planet.collider="sphere" + ours_universe.planet.init_scale = SIZE_SCALE*1000 + # camera_position = camera.position + + + def on_timer_changed(time_data): + global camera_position + # pass + # camera_look_at(camera_target) + # camera.rotation_x -= 2 + # camera_target.planet.look_at(worm_hole.planet) + if camera.intersects(worm_hole.planet).hit: + if worm_hole.planet.init_scale < SIZE_SCALE*1000: + worm_hole.planet.init_scale *= 1.1 + if ours_universe.planet.init_scale > SIZE_SCALE: + ours_universe.planet.init_scale /= 1.1 + camera.rotation_y += 0.1 + # camera_look_at(ours_universe) + else: + # if two_bodies_colliding(camera_target, worm_hole): + # camera_target.stop_and_ignore_gravity() + # print(1,camera_target.planet.position) + # camera_target.planet.position = lerp(camera_target.planet.position, worm_hole.planet.position, 1e-2) + # print(2,camera_target.planet.position) + camera.position = lerp(camera.position, worm_hole.planet.position, 1e-2) + + # camera_position = [camera_position[0], camera_position[1], camera_position[2] - 0.01] + # camera.position = camera_position + + + # 订阅事件后，上面2个函数功能才会起作用 + # 运行中，每时每刻都会触发 on_timer_changed + UrsinaEvent.on_timer_changed_subscription(on_timer_changed) + # 运行前会触发 on_ready + UrsinaEvent.on_ready_subscription(on_ready) + + # 使用 ursina 查看的运行效果 + # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 + # position = 左-右+、上+下-、前+后- + ursina_run(bodies, SECONDS_PER_DAY / 24, + # position=(0, AU / 1200, -1.003 * AU), + cosmic_bg='', + # gravity_works=False, + timer_enabled=True, + show_grid=False, + show_trail=False)