Python超人-宇宙模拟器

sim_scenes/featured/transformed_mars_ani.py

+# -*- coding:utf-8 -*-
+# title           :改造后火星的效果
+# description     :改造后火星的效果
+# author          :Python超人
+# date            :2023-12-01
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.9
+# ==============================================================================
+import os
+import sys
+
+from bodies import Earth
+from common.consts import SECONDS_PER_HOUR
+from objs import Satellite2
+from sim_scenes.fiction.transformed_mars_ani_3d import transformed_mars_ani
+from sim_scenes.func import ursina_run
+from simulators.ursina.entities.body_timer import TimeData
+from simulators.ursina.entities.entity_utils import create_directional_light
+from simulators.ursina.ursina_event import UrsinaEvent
+
+if __name__ == '__main__':
+    transformed_mars_ani(
+        transformed_texture="mars.jpg",
+        texture="mars.png",
+        bg_texture="bg_pan.jpg",
+        bg_texture_rotation=True,
+        camera3d=False,
+        append_satellite=True,
+        append_space_station=True,
+        size_scale=1.5
+    )

sim_scenes/fiction/transformed_mars_ani_3d.py

@@ -12,10 +12,13 @@ import sys
 from bodies import Earth
 from common.consts import SECONDS_PER_HOUR
 from objs import Satellite2
+from objs.sci_fi_space_station import ScifiSpaceStation
 from sim_scenes.func import ursina_run, create_sphere_sky
 from simulators.ursina.entities.body_timer import TimeData
 from simulators.ursina.entities.entity_utils import create_directional_light
+from simulators.ursina.ursina_config import UrsinaConfig
 from simulators.ursina.ursina_event import UrsinaEvent
+from ursina import application, camera
 
 
 def create_satellites():
@@ -25,14 +28,15 @@ def create_satellites():
         # {"position": [0, 0, -12000], "velocity": [5.75, 0, 0]},
         # {"position": [0, 8000, 0], "velocity": [7.05, 0, 0]},
         # {"position": [0, -12002, 0], "velocity": [5.75, 0, 0]},
-        {"position": [0, 0, 14000], "velocity": [0, 5, 0]},
+        # {"position": [0, 0, 14000], "velocity": [0, 5, 0]},
+        {"position": [0, 0, 35000], "velocity": [4 / 2, 3 / 2, 0]},
         # {"position": [0, 0, -10000], "velocity": [0, 6.3, 0]},
     ]
     satellites = []
     for i, info in enumerate(satellite_infos):
         # Satellite Satellite2
         satellite = Satellite2(name=f'卫星{i + 1}', mass=4.4e10,
-                               size_scale=2e2, color=(255, 200, 0),
+                               size_scale=1e2, color=(255, 200, 0),
                                init_position=info["position"],
                                init_velocity=info["velocity"])
         # info["satellite"] = satellite
@@ -40,8 +44,22 @@ def create_satellites():
     return satellites
 
 
+def create_space_station(mars_radius):
+    space_station = ScifiSpaceStation(name="空间站",
+                                      size_scale=0.1,
+                                      # init_position=[0.8e4, 0, -2.5e4],
+                                      # init_position=[0.1 * mars_radius, 0, -3.70e4],  # (1.80 * mars.radius, 0, -38000
+                                      init_position=[0.1 * mars_radius, 0, -4.60e4],
+                                      # init_position=[0.1 * mars_radius, 0, -6.2e4],
+                                      init_velocity=[0, 0, 0]) \
+        .set_ignore_gravity(True)
+
+    return space_station
+
+
 def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
-                         bg_texture=None, append_satellite=True):
+                         bg_texture=None, bg_texture_rotation=False, append_satellite=True, append_space_station=False,
+                         size_scale=1):
     trans_texture = transformed_texture.replace(".jpg", "_trans.png")
     # texture = transformed_texture
     # texture = os.path.join("transformed", texture)
@@ -54,21 +72,21 @@ def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
         rotate_angle=0,
         init_position=[0, 0, 0],
         init_velocity=[0, 0, 0],
-        size_scale=1)
+        size_scale=size_scale)
 
     transformed_mars = Earth(
         texture=os.path.join("transformed", transformed_texture),
         rotate_angle=0,
         init_position=[0, 0, 0],
         init_velocity=[0, 0, 0],
-        size_scale=0.9996).set_ignore_gravity(True)
+        size_scale=size_scale * 0.9996).set_ignore_gravity(True)
 
     trans_mars = Earth(
         texture=os.path.join("transformed", trans_texture),
         rotate_angle=0,
         init_position=[0, 0, 0],
         init_velocity=[0, 0, 0],
-        size_scale=0.9990).set_ignore_gravity(True)
+        size_scale=size_scale * 0.9990).set_ignore_gravity(True)
 
     bodies = [mars, transformed_mars, trans_mars]
 
@@ -76,13 +94,17 @@ def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
     clouds = Earth(name="云层", texture="transparent_clouds.png",
                    rotate_angle=0,
                    rotation_speed=18,
-                   size_scale=1.015, parent=mars)
+                   size_scale=size_scale * 1.015, parent=mars)
 
     if append_satellite:
         satellites = create_satellites()
         bodies = bodies + satellites
     bodies.append(clouds)
 
+    if append_space_station:
+        space_station = create_space_station(mars.radius)
+        bodies.append(space_station)
+
     if camera3d:
         # TODO: 开启3D摄像机
         from simulators.ursina.entities.camera3d import Camera3d
@@ -90,19 +112,50 @@ def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
         Camera3d.init(eye_distance=1000)
 
     def on_ready():
+
         if bg_texture is not None:
-            sky = create_sphere_sky(texture=bg_texture, scale=10000, rotation_y=180)
+            sky = create_sphere_sky(texture=bg_texture, scale=10000, rotation_x=10, rotation_y=180, rotation_z=-70)
+
+            if bg_texture_rotation:
+                def sky_rotation():
+                    sky.rotation_y += 0.01
+
+                sky.update = sky_rotation
+
         # 为了较好的立体效果，可以增加太阳光线，光线指向火星（target=mars）
         create_directional_light(position=(300, 0, -300), light_num=3, target=mars)
         if append_satellite:
             for satellite in satellites:
                 satellite.planet.enabled = False
 
-    def on_timer_changed(time_data: TimeData):
-        # mars.planet.opacity = 0.01
+        if append_space_station:
+            space_station.dt_scale = 1
+
+        application.time_scale = 0.01
+        camera.clip_plane_near = 0.0005
+        camera.clip_plane_far = 1000000
+
+        if append_space_station:
+            def update():
+                from ursina import Vec3, distance
+                if camera.step_index == 3:
+                    space_station.planet.rotation_z += 0.005 * space_station.dt_scale
+                    space_station.planet.position += Vec3(0, 0, 0.00000005 * space_station.dt_scale)
+                    d = distance(space_station.planet, mars.planet) / UrsinaConfig.SCALE_FACTOR
+                    if d < mars.radius * 5:
+                        space_station.planet.scale *= pow(0.999, 72 / space_station.dt_scale)
+
+                    if d < mars.radius * 1.01:
+                        camera.step_index = 4
+
+            space_station.planet.update = update
+
+        camera.step_index = 1
+
+    def step_01(time_data: TimeData):
         opacity = round((time_data.total_hours - 1) / 10, 2)
         clouds_opacity = round(opacity - 0.5, 2)
-        show_satellites = False
+
         if opacity > 1.0:
             opacity = 1.0
         elif opacity < 0.0:
@@ -110,20 +163,35 @@ def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
         # clouds.planet.enabled = False
         if opacity >= 1.0:
             mars.planet.enabled = False  # 原火星完全消失
+            camera.step_index = 2
 
         if clouds_opacity > 1.0:
             clouds_opacity = 1.0
-            show_satellites = True
+
         elif clouds_opacity < 0.0:
             clouds_opacity = 0.0
 
         clouds.planet.alpha = clouds_opacity  # 火星云层渐渐显示
         mars.planet.alpha = 1 - opacity  # 原火星渐渐消失
-        # show_satellites = True
-        if show_satellites and append_satellite:
+
+    def step_02(time_data: TimeData):
+        if append_satellite:
             for satellite in satellites:
                 satellite.planet.enabled = True
-                # satellite.planet.look_at(mars.planet)
+
+        camera.step_index = 3
+
+    def on_timer_changed(time_data: TimeData):
+        # mars.planet.opacity = 0.01
+        if append_space_station:
+            space_station.dt_scale = time_data.dt
+
+        if camera.step_index == 1:
+            step_01(time_data)
+        elif camera.step_index == 2:
+            step_02(time_data)
+        elif camera.step_index == 4:
+            exit(0)
 
         # if time_data.total_hours > 10:
         #     trans_mars.planet.enabled = True
@@ -148,7 +216,9 @@ def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
         # 使用 ursina 查看的运行效果
         # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
         # position = 左-右+、上+下-、前+后-
-        ursina_run(bodies, SECONDS_PER_HOUR / 10,
+        ursina_run(bodies,
+                   SECONDS_PER_HOUR / 10,
+                   # SECONDS_PER_HOUR,
                    position=(0, 0, -38000),
                    show_grid=False,
                    show_camera_info=False,
@@ -168,7 +238,7 @@ def transformed_mars_ani(transformed_texture=None, texture=None, camera3d=False,
                    show_control_info=False,
                    timer_enabled=True,
                    # show_timer=True,
-                   view_closely=0.001)  # 近距离观看 view_closely=True或0.001
+                   view_closely=0.0002)  # 近距离观看 view_closely=True或0.001
 
 
 if __name__ == '__main__':