Python超人-宇宙模拟器

sim_lab/wormhole_sim.py

+# -*- 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)