Python超人-宇宙模拟器

bodies/__init__.py

@@ -43,3 +43,5 @@ from bodies.fixed_stars.procyon import Procyon  # 南河三
 
 #
 from bodies.color_body import ColorBody
+
+from bodies.universe_body import UniverseBody

bodies/universe_body.py

+# -*- coding:utf-8 -*-
+# title           :宇宙体
+# description     :模拟一个宇宙体
+# author          :Python超人
+# date            :2023-11-17
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.9
+# ==============================================================================
+
+from bodies import Body
+
+
+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)
+        # self.set_resolution(50)
+        self.set_ignore_gravity(True)
+        self.set_light_disable(False)
+
+
+class OursUniverse(UniverseBody):
+    def __init__(self, name="我们的宇宙",
+                 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": 'bg_pan.jpg',
+            "size_scale": size_scale,
+            # "distance_scale": distance_scale,
+
+            # "ignore_mass": ignore_mass,
+            # "trail_color": trail_color,
+            # "show_name": show_name
+        }
+        super().__init__(**params)
+
+
+def create_universe_body(name, texture, 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,
+    }
+    return UniverseBody(**params)

sim_lab/wormhole_sim.py

@@ -7,6 +7,7 @@
 # python_version  :3.9
 # ==============================================================================
 from bodies import Sun, Earth
+from bodies.universe_body import create_universe_body
 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
@@ -15,114 +16,37 @@ 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 = "bg_pan_red.jpg"
+ours_cosmic_bg = "cosmic_pan_1.jpg"
+worms_cosmic_bg = "cosmic_pan_4.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__':
-    """
-  
-    """
-    # 代码案例如下：
+class WormholeSim:
     SIZE_SCALE = 1  # 生成的太阳放大 50 倍
     RUN_DIAMETER = 1.392e6
     velocity_rate = 22
 
-    camera_target = CoreValagaClas(name="摄像机目标", mass=1e30, color=(111, 140, 255),
+    def __init__(self):
+        self.camera_target = CoreValagaClas(name="摄像机目标", mass=1e30, color=(111, 140, 255),
                                             # init_position=[0, 0, 0],
                                             init_position=[0, 0, AU / 10],
                                             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/120],
+                                            size_scale=WormholeSim.SIZE_SCALE * 1e1)  # .set_ignore_gravity(True)
+        self.universes = []
+
+        self.ours_universe = create_universe_body("我们的宇宙", "cosmic_pan_1.jpg",
+                                                  size_scale=WormholeSim.SIZE_SCALE,
+                                                  init_position=[0, 0, -AU / 120],
                                                   init_velocity=[0, 0, 0],
-    ).set_ignore_gravity(True)
-    worm_hole = WormHole(
-        # init_position=[RUN_DIAMETER, RUN_DIAMETER, 0],
-        init_position=[0, 0, AU/120],
+                                                  )
+        self.universes.append(self.ours_universe)
+        self.worm_hole = create_universe_body("虫洞1", "cosmic_pan_4.jpg",
+                                              init_position=[0, 0, AU / 120],
                                               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]
-
+                                              size_scale=WormholeSim.SIZE_SCALE)
+        self.universes.append(self.worm_hole)
+        self.bodies = [self.camera_target] + self.universes
 
-    def on_ready():
+    def on_ready(self):
         """
         事件绑定后，模拟器运行前会触发
         @return:
@@ -135,29 +59,30 @@ if __name__ == '__main__':
         # camera.scale = 1000
         # camera.parent = camera_target.planet
         # camera.rotation_x = 90
+        camera.fov = 120
 
-        worm_hole.planet.rotation_y = 30
-        ours_universe.planet.rotation_y = 130
+        self.worm_hole.planet.rotation_y = 30
+        self.ours_universe.planet.rotation_y = 130
 
-        camera_look_at(camera_target)
+        camera_look_at(self.camera_target)
         camera.collider = "sphere"
-        worm_hole.planet.collider = "sphere"
-        ours_universe.planet.init_scale = SIZE_SCALE * 1000
+        self.worm_hole.planet.collider = "sphere"
+        self.ours_universe.planet.init_scale = self.SIZE_SCALE * 1000
+        self.camera_target.current_universe = self.ours_universe
+        self.camera_target.targe_universe = self.worm_hole
         # camera_position = camera.position
 
-
-    def on_timer_changed(time_data):
-        global camera_position
+    def on_timer_changed(self, time_data):
         # return
         # 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:
+        self.camera_target.planet.look_at(self.camera_target.targe_universe.planet)
+        if camera.intersects(self.camera_target.targe_universe.planet).hit:
             # worm_hole.planet.init_scale = SIZE_SCALE * 1000
-            ours_universe.planet.init_scale = SIZE_SCALE
-            if worm_hole.planet.init_scale < SIZE_SCALE*1000:
-                worm_hole.planet.init_scale *= 1.1
+            self.camera_target.current_universe.planet.init_scale = self.SIZE_SCALE
+            if self.camera_target.targe_universe.planet.init_scale < self.SIZE_SCALE * 1000:
+                self.camera_target.targe_universe.planet.init_scale *= 1.1
             # if ours_universe.planet.init_scale > SIZE_SCALE:
             #     ours_universe.planet.init_scale /= 1.1
             if not hasattr(camera, "init_rotation_y"):
@@ -165,11 +90,11 @@ if __name__ == '__main__':
                 camera.rotation_y_v = 0.01
 
             if abs(camera.init_rotation_y - camera.rotation_y) < 10:
-                camera.rotation_y_v += 0.005
+                camera.rotation_y_v += 0.5
             elif abs(camera.init_rotation_y - camera.rotation_y) < 90:
-                camera.rotation_y_v += 0.01
+                camera.rotation_y_v += 0.3
             elif abs(camera.init_rotation_y - camera.rotation_y) < 120:
-                camera.rotation_y_v -= 0.01
+                camera.rotation_y_v -= 0.1
             elif abs(camera.init_rotation_y - camera.rotation_y) < 180:
                 camera.rotation_y_v -= 0.01
                 if camera.rotation_y_v < 0.1:
@@ -185,27 +110,37 @@ if __name__ == '__main__':
         # camera_target.planet.position = lerp(camera_target.planet.position, worm_hole.planet.position, 1e-2)
         # print(2,camera_target.planet.position)
         # camera.position = (camera.position[0],camera.position[1],camera.position[2]) #
-        camera.position = lerp(camera.position, camera_target.planet.position, 0.3e-2)
+        camera.position = lerp(camera.position, self.camera_target.planet.position, 0.2e-2)
 
         # camera_position = [camera_position[0], camera_position[1], camera_position[2] - 0.01]
         # camera.position = camera_position
 
-
+    def build_events(self):
         # 订阅事件后，上面2个函数功能才会起作用
         # 运行中，每时每刻都会触发 on_timer_changed
-    UrsinaEvent.on_timer_changed_subscription(on_timer_changed)
+        UrsinaEvent.on_timer_changed_subscription(wormhole_sim.on_timer_changed)
         # 运行前会触发 on_ready
-    UrsinaEvent.on_ready_subscription(on_ready)
+        UrsinaEvent.on_ready_subscription(wormhole_sim.on_ready)
 
+    def run(self):
+        self.build_events()
         # 使用 ursina 查看的运行效果
         # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
         # position = 左-右+、上+下-、前+后-
-    ursina_run(bodies, SECONDS_PER_DAY / 24,
+        ursina_run(self.bodies, SECONDS_PER_DAY / 24,
                    # position=(0, AU / 1200, -1.003 * AU),
                    # position=[-AU/100, 0,  0],
-               position=[0, 0, -AU/10],
+                   position=[0, 0, -AU / 10],
                    cosmic_bg='',
                    # gravity_works=False,
                    timer_enabled=True,
                    show_grid=False,
                    show_trail=False)
+
+
+if __name__ == '__main__':
+    """
+  
+    """
+    wormhole_sim = WormholeSim()
+    wormhole_sim.run()