Python超人-宇宙模拟器

sim_scenes/funny/black_holes_control_stars.py

 # -*- coding:utf-8 -*-
-# title           :三体场景模拟02
-# description     :三体场景模拟（3个太阳、1个地球）
+# title           :黑洞模拟
+# description     :黑洞模拟
 # author          :Python超人
 # date            :2023-07-26
 # link            :https://gitcode.net/pythoncr/

sim_scenes/funny/black_holes_control_stars_01.py

+# -*- coding:utf-8 -*-
+# title           :黑洞模拟
+# description     :黑洞模拟
+# author          :Python超人
+# date            :2023-10-27
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.8
+# ==============================================================================
+from bodies import Sun, Earth
+from common.consts import SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_YEAR, AU
+from sim_scenes.func import mayavi_run, mpl_run, ursina_run, create_sphere_sky
+
+from bodies import FixedStar, Body
+from simulators.ursina.entities.body_timer import TimeData
+from simulators.ursina.ursina_event import UrsinaEvent
+
+
+class BlackHole(FixedStar):
+    def __init__(self, name="黑洞", mass=1.9891e35,
+                 init_position=[0, 0, 0],
+                 init_velocity=[0, 0, 0],
+                 color=(0, 0, 0),
+                 # texture="cosmic3.jpg",  # 虫洞
+                 texture="",  # 黑洞
+                 size_scale=1.0, distance_scale=1.0,
+                 rotation_speed=0.6130, ignore_mass=False, trail_color=None, show_name=False):
+        params = {
+            "name": name,
+            "mass": mass,
+            "init_position": init_position,
+            "init_velocity": init_velocity,
+            "density": 1.408e20,
+            "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.glows = (0, 1.015, 0.08, (255, 255, 255))
+
+
+if __name__ == '__main__':
+    """
+    注释： 3个太阳
+    可以修改影响效果的参数为： 
+    1、三个方向的初始位置 init_position[x, y, z]
+    2、三个方向的初始速度 init_velocity[x, y, z]
+    3、天体质量 mass    
+    """
+    # 代码案例如下：
+    SIZE_SCALE = 5e1  # 生成的太阳放大 50 倍
+    RUN_DIAMETER = 1.392e6  # 真实太阳的直径
+    velocity_rate = 11000
+    bodies = [
+        Sun(name="太阳1", mass=1e30, color=(111, 140, 255), rotation_speed=0.1,
+            init_position=[-100 * RUN_DIAMETER, 0, 0],
+            init_velocity=[0.347113 * velocity_rate, 0.532727 * velocity_rate, 0],
+            size_scale=SIZE_SCALE, texture="fixed_star.png"),
+        BlackHole(init_position=[-100 * RUN_DIAMETER, 100 * RUN_DIAMETER, 0], size_scale=SIZE_SCALE/5)
+    ]
+
+
+    def on_ready():
+        from ursina import camera, application
+        # 创建天空
+        # camera.clip_plane_near = 0.1
+        camera.clip_plane_far = 1000000
+        create_sphere_sky(scale=200000)
+        application.time_scale = 3
+
+
+    def on_timer_changed(time_data: TimeData):
+        pass
+
+
+    # 订阅事件后，上面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,
+               position=(-1.5 * AU, 0, -5 * AU),
+               show_grid=False,
+               show_camera_info=False,
+               show_control_info=False,
+               cosmic_bg='',
+               show_trail=True)