Python超人-宇宙模拟器

objs/halley_comet.py

+# -*- coding:utf-8 -*-
+# title           :水滴
+# description     :水滴
+# author          :Python超人
+# date            :2023-08-02
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.8
+# ==============================================================================
+from objs.obj import Obj
+
+
+class HalleComet(Obj):
+    """
+    哈雷彗星
+    来源：https://www.cgmodel.com/model/500318.html
+    """
+
+    def __init__(self, name="哈雷彗星", mass=5.97237e24,
+                 init_position=[0, 0, 0],
+                 init_velocity=[0, 0, 0],
+                 texture="drops.png", size_scale=1.0, distance_scale=1.0,
+                 ignore_mass=False, density=1e3, color=(7, 0, 162),
+                 trail_color=None, show_name=False,
+                 model="drops.obj", rotation=(0, 0, 0),
+                 parent=None, gravity_only_for=[]):
+        params = {
+            "name": name,
+            "mass": mass,
+            "init_position": init_position,
+            "init_velocity": init_velocity,
+            "density": density,
+            "color": color,
+            "texture": texture,
+            "size_scale": size_scale,
+            "distance_scale": distance_scale,
+            "ignore_mass": ignore_mass,
+            "trail_color": trail_color,
+            "show_name": show_name,
+            "parent": parent,
+            "rotation": rotation,
+            "gravity_only_for": gravity_only_for,
+            "model": model
+        }
+        super().__init__(**params)
+
+
+from ursina.prefabs.primitives import Shader
+
+matcap_shader = Shader(name='matcap_shader', language=Shader.GLSL, vertex='''#version 140
+uniform mat4 p3d_ModelViewProjectionMatrix;
+uniform mat4 p3d_ModelMatrix;
+uniform mat4 p3d_ModelViewMatrix;
+uniform mat3 p3d_NormalMatrix;
+in vec4 p3d_Vertex;
+in vec3 p3d_Normal;
+
+out vec3 eye;
+out vec3 view_normal;
+// reflect alternative:
+// r = e - 2. * dot( n, e ) * n;
+
+void main() {
+    gl_Position = p3d_ModelViewProjectionMatrix * p3d_Vertex;
+
+    eye = normalize(vec3(p3d_ModelViewMatrix * vec4(p3d_Vertex.xyz, 1.0)));
+    view_normal = normalize( p3d_NormalMatrix * p3d_Normal );
+}
+''',
+                       fragment='''
+#version 130
+uniform sampler2D p3d_Texture0;
+uniform vec4 p3d_ColorScale;
+
+in vec3 eye;
+in vec3 view_normal;
+out vec4 fragColor;
+
+void main() {
+
+    vec3 r = reflect( eye, view_normal );
+    float m = 2. * sqrt( pow( r.x, 2. ) + pow( r.y, 2. ) + pow( r.z + 1., 2. ) );
+    vec2 vN = r.xy / m + .5;
+
+    vec3 base = texture2D( p3d_Texture0, vN ).rgb;
+    // vec3 base = texture2D( p3d_Texture0, uv ).rgb;
+    fragColor = vec4( base, 1. ) * p3d_ColorScale;
+}
+
+''',
+                       )
+
+if __name__ == '__main__':
+    shader = matcap_shader
+
+    halle_comet = HalleComet(
+        # texture="drops_normal.png"
+        # texture="drops_uvw.png"
+        texture="drops.png"
+    )
+    halle_comet.init_velocity = [0, 0, -10]
+    print(halle_comet)
+
+
+    def on_ready():
+        halle_comet.planet.rotation_x = 90
+        halle_comet.planet.shader = shader,
+        halle_comet.planet.init_scale = halle_comet.planet.scale_x * 2
+        # halle_comet.planet.init_update = halle_comet.planet.update
+
+    def on_timer_changed(time_data):
+        # halle_comet.planet.init_update = body.planet.update
+        halle_comet.planet.scale_z = halle_comet.planet.init_scale
+
+
+    halle_comet.show_demo(size_scale=1000000, on_ready_fun=on_ready, on_timer_changed_fun=on_timer_changed)

objs/obj.py

@@ -482,7 +482,7 @@ class Obj(metaclass=ABCMeta):
 
         return body_data
 
-    def show_demo(self, size_scale=10000, on_ready_fun=None):
+    def show_demo(self, size_scale=10000, on_ready_fun=None, on_timer_changed_fun=None):
         from sim_scenes.func import ursina_run
         from sim_scenes.func import camera_look_at
         from simulators.ursina.ursina_event import UrsinaEvent
@@ -496,8 +496,16 @@ class Obj(metaclass=ABCMeta):
             if callable(on_ready_fun):
                 on_ready_fun()
 
+        def on_timer_changed(time_data):
+            if callable(on_timer_changed_fun):
+                on_timer_changed_fun(time_data)
+
         # 运行前会触发 on_ready
-        UrsinaEvent.on_ready_subscription(on_ready)
+        if on_ready_fun is not None:
+            UrsinaEvent.on_ready_subscription(on_ready)
+
+        if on_timer_changed_fun is not None:
+            UrsinaEvent.on_timer_changed_subscription(on_timer_changed)
 
         ursina_run(bodies=[self], position=(0, 3000000, -9000000))
 

sim_scenes/solar_system/halley_comet.py

+# -*- coding:utf-8 -*-
+# title           :哈雷彗星场景模拟
+# description     :哈雷彗星场景模拟
+# author          :Python超人
+# date            :2023-10-25
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.9
+# ==============================================================================
+import time
+
+from ursina import camera, application
+
+from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto
+from common.celestial_data_service import get_init_pos_vels, init_bodies_pos_vels
+from common.consts import SECONDS_PER_WEEK, AU
+from objs import QuadObj, CircleObj, Obj
+from sim_scenes.func import camera_look_at, two_bodies_colliding
+from sim_scenes.func import ursina_run, create_sphere_sky
+from simulators.ursina.ursina_event import UrsinaEvent
+
+
+class HalleyCometSim:
+    """
+    哈雷彗星场景模拟
+    """
+
+    def __init__(self):
+        pass
+
+    def build_solar_system(self):
+        # region 构建太阳系
+        self.sun = Sun(size_scale=0.8e2)
+        self.bodies = [
+            self.sun,  # 太阳放大 80 倍
+            Mercury(size_scale=4e3, distance_scale=1.3),  # 水星放大 4000 倍，距离放大 1.3 倍
+            Venus(size_scale=4e3, distance_scale=1.3),  # 金星放大 4000 倍，距离放大 1.3 倍
+            Earth(size_scale=4e3, distance_scale=1.3, rotate_angle=0),  # 地球放大 4000 倍，距离放大 1.3 倍
+            Mars(size_scale=4e3, distance_scale=1.2),  # 火星放大 4000 倍，距离放大 1.2 倍
+            Jupiter(size_scale=0.68e3, distance_scale=0.72),  # 木星放大 680 倍，距离缩小到真实距离的 0.72
+            Saturn(size_scale=0.68e3, distance_scale=0.52),  # 土星放大 680 倍，距离缩小到真实距离的 0.52
+            Uranus(size_scale=1.5e3, distance_scale=0.36),  # 天王星放大 1500 倍，距离缩小到真实距离的 0.36
+            Neptune(size_scale=1.5e3, distance_scale=0.27),  # 海王星放大 1500 倍，距离缩小到真实距离的 0.27
+            Pluto(size_scale=10e3, distance_scale=0.23),  # 冥王星放大 10000 倍，距离缩小到真实距离的 0.23(从太阳系的行星中排除)
+        ]
+        # endregion
+
+        init_bodies_pos_vels(self.bodies)
+
+    def build_halley_comet(self):
+        """
+        创建哈雷彗星
+        @return:
+        """
+        self.halley_comet = QuadObj(texture='two_way_foil.png',
+                                    # size_scale=4e7,
+                                    size_scale=1e7,
+                                    init_velocity=[20, -6, 150],
+                                    init_position=[0, 0.5 * AU, -10 * AU]) \
+            .set_light_disable(True)
+
+        self.bodies.append(self.halley_comet)
+
+
+    def build(self):
+        self.build_solar_system()
+        self.build_halley_comet()
+
+    def on_ready(self):
+        """
+        事件绑定后，模拟器运行前会触发
+        @return:
+        """
+        # 创建天空
+        # camera.clip_plane_near = 0.1
+        camera.clip_plane_far = 1000000
+        create_sphere_sky(scale=200000)
+        application.time_scale = 5
+
+
+
+    def on_timer_changed(self, time_data):
+        """
+
+        @param time_data:
+        @return:
+        """
+        # 原始方形二向箔飞行的翻转效果
+        if self.halley_comet.planet.enabled:
+            self.halley_comet.planet.rotation_x += 0.1
+            self.halley_comet.planet.rotation_y += 1
+        # 摄像机始终看向二向箔
+        camera_look_at(self.halley_comet)
+
+
+
+if __name__ == '__main__':
+    """
+    三体二向箔场景模拟
+    """
+    sim = HalleyCometSim()
+    sim.build()
+    # 订阅事件后，上面2个函数功能才会起作用
+    # 运行中，每时每刻都会触发 on_timer_changed
+    UrsinaEvent.on_timer_changed_subscription(sim.on_timer_changed)
+    # 运行前会触发 on_ready
+    UrsinaEvent.on_ready_subscription(sim.on_ready)
+    # 使用 ursina 查看的运行效果
+    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
+    # position = 左-右+、上+下-、前+后-
+    ursina_run(sim.bodies,
+               SECONDS_PER_WEEK,
+               # position=(0, 2 * AU, -11 * AU),
+               position=(0, 0.5 * AU, -5 * AU),
+               cosmic_bg='',
+               bg_music='sounds/no_glory.mp3',
+               show_camera_info=False,
+               show_control_info=False,
+               timer_enabled=True,
+               show_grid=False)