Python超人-宇宙模拟器

sim_scenes/featured/two_way_foil.py → sim_scenes/featured/two_way_foil_sim.py

@@ -2,24 +2,33 @@
 # title           :三体二向箔场景模拟
 # description     :三体二向箔场景模拟
 # author          :Python超人
-# date            :2023-10-27
+# date            :2024-01-20
 # link            :https://gitcode.net/pythoncr/
 # python_version  :3.9
 # ==============================================================================
-import time
-
-from ursina import camera, application, lerp
 
+from ursina import camera, application, lerp, Vec3
 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 common.image_utils import resize_image
+from common.celestial_data_service import init_bodies_reality_pos_vels, conv_to_astropy_time, \
+    set_solar_system_celestial_position, init_bodies_pos_vels
+from common.consts import SECONDS_PER_YEAR, AU
+from common.func import calculate_distance
+from bodies import Earth
 from objs import QuadObj, CircleObj, Obj
-from sim_scenes.func import camera_look_at, two_bodies_colliding, camera_move_control, get_run_speed_factor
+from sim_scenes.func import create_text_panel, camera_look_at, get_run_speed_factor, \
+    camera_move_update, camera_move_to_target_update, camera_move_control, two_bodies_colliding
 from sim_scenes.func import ursina_run, create_sphere_sky
+from sim_scenes.solar_system.halley_comet_lib import HalleyCometSimBase, HalleyCometParams, \
+    create_halley_comet, create_orbit_line
 from sim_scenes.universe_sim_scenes import UniverseSimScenes
-from simulators.ursina.ursina_event import UrsinaEvent
 from simulators.func import ext_fun_for_method
+from simulators.ursina.entities.body_timer import TimeData
+from simulators.ursina.entities.entity_utils import get_value_direction_vectors
+from simulators.ursina.ui.control_ui import ControlUI
+from simulators.ursina.ursina_config import UrsinaConfig
+from simulators.ursina.ursina_event import UrsinaEvent
+from simulators.ursina.ursina_mesh import create_label
+import time
 
 
 class TwoWayFoilSim(UniverseSimScenes):
@@ -27,33 +36,36 @@ class TwoWayFoilSim(UniverseSimScenes):
     三体二向箔场景模拟
     """
 
-    def __init__(self):
-        self.current_stage = self.stage_01
-
     def build_solar_system(self):
         # region 构建太阳系
-        self.sun, self.mercury, self.venus, self.earth, self.mars, self.jupiter, \
-        self.saturn, self.uranus, self.neptune, self.pluto = \
-            Sun(size_scale=0.8e2, texture="sun_light.jpg"), \
-            Mercury(size_scale=2e3, distance_scale=1.5), \
-            Venus(size_scale=2e3, distance_scale=1.5), \
-            Earth(size_scale=2e3, distance_scale=1.5, rotate_angle=0), \
-            Mars(size_scale=2e3, distance_scale=1.4), \
-            Jupiter(size_scale=0.3e3, distance_scale=0.72), \
-            Saturn(size_scale=0.3e3, distance_scale=0.52), \
-            Uranus(size_scale=0.8e3, distance_scale=0.36), \
-            Neptune(size_scale=0.8e3, distance_scale=0.27), \
-            Pluto(size_scale=10e3, distance_scale=0.23)
-
+        show_trail = True
+
+        self.sun = Sun(size_scale=0.6e2, show_trail=show_trail,
+                       # color=(255, 250, 245),
+                       # texture="",
+                       texture="sun_light.jpg"
+                       )
+        self.mercury = Mercury(size_scale=2e3, distance_scale=1.5, show_trail=show_trail)
+        self.venus = Venus(size_scale=2e3, distance_scale=1.5, show_trail=show_trail)
+        self.earth = Earth(size_scale=2e3, distance_scale=1.5, rotate_angle=0, show_trail=show_trail)
+        self.mars = Mars(size_scale=2e3, distance_scale=1.4, show_trail=show_trail)
+        self.jupiter = Jupiter(size_scale=0.3e3, distance_scale=0.72, show_trail=show_trail)
+        self.saturn = Saturn(size_scale=0.3e3, distance_scale=0.52, show_trail=show_trail)
+        self.uranus = Uranus(size_scale=0.8e3, distance_scale=0.33, show_trail=show_trail)
+        self.neptune = Neptune(size_scale=0.8e3, distance_scale=0.25, show_trail=show_trail)
+        self.pluto = Pluto(size_scale=1e4, distance_scale=0.23, show_trail=show_trail)
         self.bodies = [
-            self.sun, self.mercury, self.venus, self.earth, self.mars, self.jupiter,
-            self.saturn, self.uranus, self.neptune, self.pluto
+            self.sun,  # 太阳
+            self.mercury,  # 水星
+            self.venus,  # 金星
+            self.earth,  # 地球
+            self.mars,  # 火星
+            self.jupiter,  # 木星
+            self.saturn,  # 土星
+            self.uranus,  # 天王星
+            self.neptune,  # 海王星
+            self.pluto,  # 冥王星
         ]
-        # endregion
-        # self.sun.glows = (2, 1.005, 0.01)
-        # # glows = (glow_num:10, glow_scale:1.03 glow_alpha:0.1~1)
-        # self.sun.glows = (0, 1.005, 0.1, 160)
-        self.sun.glows = (2, 1.005, 0.01)
 
         self.two_way_foil_dt_factor = 1
         self.sun.two_way_foil_dt = 0.003 * self.two_way_foil_dt_factor
@@ -62,6 +74,37 @@ class TwoWayFoilSim(UniverseSimScenes):
 
         init_bodies_pos_vels(self.bodies)
 
+    def build(self):
+        """
+        构建太阳系系统以及哈雷彗星
+        @return:
+        """
+        self.build_solar_system()
+        self.build_two_way_foil()
+
+    def init_settings(self):
+        """
+        初始化设置
+        @return:
+        """
+        from ursina import color
+        # 创建天空
+        create_sphere_sky(scale=50000, rotation_x=200, rotation_y=None, rotation_z=45)
+        # UrsinaConfig.trail_type = "curve_line"
+        # UrsinaConfig.trail_length = 300
+        UrsinaConfig.trail_type = "line"
+        # UrsinaConfig.trail_length = 152  # 尾巴数量刚刚好
+        UrsinaConfig.trail_length = 130
+        UrsinaConfig.trail_thickness_factor = 3
+
+        # UrsinaConfig.trail_length = 180
+        UrsinaConfig.trail_factor = 3
+
+        # camera.clip_plane_near = 0.1
+        camera.clip_plane_far = 51000
+        # camera.fov = 60
+        self.current_stage = self.stage_01
+
     def build_two_way_foil(self):
         """
         创建二向箔（一个原始的方形二向箔，一个不断扩展的圆形二向箔）
@@ -70,10 +113,10 @@ class TwoWayFoilSim(UniverseSimScenes):
         # 原始的方形二向箔
         self.two_way_foil = QuadObj(texture='two_way_foil.png',
                                     # size_scale=4e7,
-                                    size_scale=1e7,
+                                    size_scale=1e6,
                                     init_velocity=[0, -50, 60],
                                     init_position=[0, 5 * AU, -6 * AU]) \
-            .set_light_disable(True).set_ignore_gravity(True)
+            .set_light_disable(True)  # .set_ignore_gravity(False)
         # 不断扩展的圆形二向箔
         self.two_way_foil_circle = CircleObj(texture="two_way_foil_circle.png",
                                              size_scale=self.two_way_foil.size_scale * 2,
@@ -83,70 +126,51 @@ class TwoWayFoilSim(UniverseSimScenes):
         self.bodies.append(self.two_way_foil)
         self.bodies.append(self.two_way_foil_circle)
 
-    def build(self):
-        self.build_solar_system()
-        self.build_two_way_foil()
+    def s_f(self, value=1):
+        if value == 0:
+            return 0
+        return get_run_speed_factor() * value
 
-    def on_ready(self):
+    def camera_move(self, time_data):
         """
-        事件绑定后，模拟器运行前会触发
+        摄像机移动控制
+        @param dt:
         @return:
         """
-        # 创建天空
-        # camera.clip_plane_near = 0.1
-        camera.clip_plane_far = 1000000
-        create_sphere_sky(scale=200000)
-        application.time_scale = 5
-        # 圆形二向箔初始化（一开始不显示）
-        self.two_way_foil_circle.planet.rotation_x = 90
-        self.two_way_foil_circle.planet.enabled = False
-
-        def change_two_way_foil():
-            # 长方形膜状物，长八点五厘米，宽五点二厘米，比一张信用卡略大一些，极薄，看不出任何厚度。
-            # 封装状态下晶莹剔透、无色透明。待机模式下由于封装力场逐渐蒸发，会发出白光，使其表面呈纯白色，看上去就是一张纸条
-            self.two_way_foil.planet.scale_x = 8.5
-            self.two_way_foil.planet.scale_y = 5.2
-
-        glow_circle = self.sun.planet.glow_circle
-        fixed_star_ratio = glow_circle.scale_x / self.sun.planet.scale_x
-
-        def glow_circle_update():
-            glow_circle.enabled = self.sun.planet.enabled
-            glow_circle.position = self.sun.planet.position
-            glow_circle.scale_x = fixed_star_ratio * self.sun.planet.scale_y
-            if glow_circle.scale_x < 5:
-                glow_circle.update = lambda :None
-                glow_circle.enabled = False
-            glow_circle.scale_z = fixed_star_ratio * self.sun.planet.scale_x
-            # glow_circle.scale_z = fixed_star_ratio * fixed_star.scale_z
-            glow_circle.look_at(camera)
-
-        glow_circle.update = glow_circle_update
+        # 摄像机移动控制数据
+        camera_move_infos = [
+            # 条件：年份
+            # 移动的信息：
+            # 按坐标系方向移动 x:右+左-, y:升+降-, z:前+(接近太阳)后-(远离太阳)
+            # 以摄像机视角移动 f:前 b:后 l:左 r:右 u:上 d:下
+            (0, {"f": 5.2}),
+            # (1983, {"to": {"ct_id": 1, "t": 10}}),
+            (10, {}),
+            (12, {"f": 0.2}),
+            (15, {"f": 0.3}),
+            (18, {"f": 0.4}),
+            # (30, {"f": 0.5}),
+            # (40, {"f": 0.6}),
+            (160, {"b": 2}),
+            (1988, {"y": -3, "z": -12}),
+            (1989, {"z": -8, "f": -5}),
+            (1993, {"z": -8, "f": -3}),
+            (1995, {"z": -8}),
+            (2000, {"z": -8, "y": -0.2}),
+            (2013, {}),
+            (2048, {"f": 3}),
+            (2062, {"y": -3}),
+            (2063, {"y": -10, "z": 2}),
+            (2181, {}),
+            # (2082, {"exit": True})
+        ]
 
-        ext_fun_for_method(self.two_way_foil.planet, after_run_fun=change_two_way_foil)
+        def cond_cb(ps):
+            return ps["next_cond"] > time_data.total_days >= ps["cond"]
 
-    def gen_pixcel_image(self, planet):
-        """
-        将纹理图改为像素图片
-        @param planet:
-        @return:
-        """
-        import os
-        if planet.texture is None:
-            return
-        in_img = str(planet.texture.path)
-        if "_pixcel" in in_img:
-            return
-        out_img = os.path.split(in_img)
-        out_file = out_img[1].split('.')
-        out_file = f'{out_file[0]}_pixcel.{out_file[1]}'
-        out_img = os.path.abspath(os.path.join(out_img[0], "temp", out_file))
-        if not os.path.exists(out_img):
-            resize_image(in_img, out_img, 50, 25)
-        from ursina import Texture
-        planet.texture = Texture(out_img)
-
-        return out_img
+        camera_move_control(camera_move_infos,
+                            cond_cb=cond_cb,
+                            value_conv=self.s_f, smooth=1)
 
     def flatten_animation(self, body):
         """
@@ -273,6 +297,35 @@ class TwoWayFoilSim(UniverseSimScenes):
         # 圆形二向箔不断旋转的效果
         self.two_way_foil_circle.planet.rotation_z += 0.4
 
+    def on_ready(self):
+        """
+        事件绑定后，模拟器运行前会触发
+        @return:
+        """
+        # 初始化设置
+        self.init_settings()
+
+        # self.set_window_size((int(1920 * r), int(1080 * r)), False)
+        # 最大分辨率的高度-1，保证不全屏
+        # self.set_window_size((1920, 1079), False)
+        application.time_scale = 0.00001
+        # 显示网格以及坐标线
+        # self.show_grid_axises()
+        self.two_way_foil.planet.alpha = 0.95
+        # 圆形二向箔初始化（一开始不显示）
+        self.two_way_foil_circle.planet.rotation_x = 90
+        self.two_way_foil_circle.planet.enabled = False
+        self.two_way_foil.scale_factor = 0.6
+
+        def change_two_way_foil():
+            # 长方形膜状物，长八点五厘米，宽五点二厘米，比一张信用卡略大一些，极薄，看不出任何厚度。
+            # 封装状态下晶莹剔透、无色透明。待机模式下由于封装力场逐渐蒸发，会发出白光，使其表面呈纯白色，看上去就是一张纸条
+            self.two_way_foil.planet.scale_x = 8.5 * self.two_way_foil.scale_factor
+            self.two_way_foil.planet.scale_y = 5.2 * self.two_way_foil.scale_factor
+            # self.two_way_foil.scale_factor -= 1e-4
+
+        ext_fun_for_method(self.two_way_foil.planet, after_run_fun=change_two_way_foil)
+
     def on_timer_changed(self, time_data):
         """
 
@@ -281,75 +334,38 @@ class TwoWayFoilSim(UniverseSimScenes):
         """
         # 原始方形二向箔飞行的翻转效果
         if self.two_way_foil.planet.enabled:
-            self.two_way_foil.planet.rotation_x += 0.1
-            self.two_way_foil.planet.rotation_y += 1
-        # # 摄像机始终看向二向箔
-        # camera_look_at(self.two_way_foil)
-
-        target = self.get_target()
-        if target is not None:
-            if target is self.sun:
-                # 摄像机始终看向二向箔
-                camera_look_at(self.two_way_foil)
-            else:
-                # 摄像机始终看向二向箔
-                camera_look_at(self.two_way_foil)
-                camera.position += camera.forward*10
+            self.two_way_foil.planet.rotation_x += 0.01
+            self.two_way_foil.planet.rotation_y += 0.5
+        # dt = time_data.get_datetime(self.start_time)
+        # # 摄像机看向哈雷彗星
+        camera_look_at(self.two_way_foil, rotation_z=0)
+
+        # camera.look_at_2d(self.halley_comet.planet.position, axis='y')
+
+        time_total_hours = time_data.total_hours
+
+        # if not hasattr(self, "last_total_hours"):
+        #     self.last_total_hours = time_total_hours
+        #     self.update_halley_comet_info(dt)
+        # print("time_total_hours", time_total_hours)
+        # # 摄像机移动每10小时更新一次
+        # self.interval_run(fun=self.camera_move, fun_args=[dt],
+        #                   total_times=time_total_hours,
+        #                   interval=20)
         self.camera_move(time_data)
-                # camera.position = \
-                #     lerp(camera.position, target.planet.position, 0.001)
-
         self.current_stage()
 
-    def s_f(self, value=1):
-        if value == 0:
-            return 0
-        return get_run_speed_factor() * value
-
-    def camera_move(self, time_data):
-        """
-        摄像机移动控制
-        @param dt:
-        @return:
-        """
-        # 摄像机移动控制数据
-        camera_move_infos = [
-            # 条件：年份
-            # 移动的信息：
-            # 按坐标系方向移动 x:右+左-, y:升+降-, z:前+(接近太阳)后-(远离太阳)
-            # 以摄像机视角移动 f:前 b:后 l:左 r:右 u:上 d:下
-            (1, {"l": 2}),
-            (10, {"f": 50}),
-            # (40, {"f": 80}),
-            # (1983, {"to": {"ct_id": 1, "t": 10}}),
-            # (2, {"x": 2, "y": -2, "z": -15}),
-            # (3, {"y": -6, "z": -12}),
-            # (4, {"y": -3, "z": -12}),
-            # (5, {"z": -8, "f": -5}),
-            # (6, {"z": -8, "f": -3}),
-            # (7, {"z": -8}),
-            # (8, {"z": -8, "y": -0.2}),
-            # (9, {}),
-            # (100, {"f": 10}),
-            # (200, {"f": 10}),
-            # (500, {"f": 10}),
-            # (2062, {"y": -3}),
-            # (2063, {"y": -10, "z": 2}),
-            (2181, {}),
-            # (2082, {"exit": True})
-        ]
-
-        camera_move_control(camera_move_infos,
-                            cond_cb=lambda ps: ps["next_cond"] > time_data.total_days >= ps["cond"],
-                            value_conv=self.s_f, smooth=None)
-
 
 if __name__ == '__main__':
     """
-    三体二向箔场景模拟
+
     """
-    sim = TwoWayFoilSim()
+
+    params = TwoWayFoilSim()
+
+    sim = TwoWayFoilSim(params)
     sim.build()
+
     # 订阅事件后，上面2个函数功能才会起作用
     # 运行中，每时每刻都会触发 on_timer_changed
     UrsinaEvent.on_timer_changed_subscription(sim.on_timer_changed)
@@ -359,13 +375,18 @@ if __name__ == '__main__':
     # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
     # position = 左-右+、上+下-、前+后-
     ursina_run(sim.bodies,
-               SECONDS_PER_WEEK,
-               position=(0, 2 * AU, -11 * AU),
-               # position=(0, 5 * AU, -5 * AU),
+               SECONDS_PER_YEAR / 48 * get_run_speed_factor(),
+               # position=(0, 2 * AU, -11 * AU),
+               # position=(0, 0.5 * AU, -5 * AU),
+               # position=(2 * AU, -5 * AU, -20 * AU),
+               position=(2 * AU, 2 * AU, -10 * AU),
                cosmic_bg='',
-               bg_music='sounds/no_glory.mp3',
+               # show_trail=True,
+               # bg_music='sounds/no_glory.mp3',
                show_camera_info=False,
+               # video_recoder=True,
                show_control_info=False,
                timer_enabled=True,
                show_timer=True,
-               show_grid=False)
+               show_grid=False
+               )