Python超人-宇宙模拟器(代码备份，代码的清理未删除)

common/celestial_data_service.py

@@ -268,7 +268,6 @@ def show_bodies_posvels(dt):
 
 
 def gen_real_pos_vel_bodies(body_names, dt):
-
     import pytz
     # 安装 astropy 包
     # pip install -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com astropy
@@ -281,12 +280,13 @@ def gen_real_pos_vel_bodies(body_names, dt):
 
     for body_name in str(body_names).split(","):
         earth_pos_vel = get_body_barycentric_posvel(body_name, dt)
-        pos,vel = earth_pos_vel[0],earth_pos_vel[1]
+        pos, vel = earth_pos_vel[0], earth_pos_vel[1]
         gen_body_str += f"     init_pos_vels['{body_name.lower()}']={{}}\n"
         gen_body_str += f"     init_pos_vels['{body_name.lower()}']['pos']=[%.4f,%.4f,%.4f]\n" % (pos.x.to(u.km).value,
                                                                                                   pos.y.to(u.km).value,
                                                                                                   pos.z.to(u.km).value)
-        gen_body_str += f"     init_pos_vels['{body_name.lower()}']['vel']=[%.4f,%.4f,%.4f]\n" % (vel.x.to(u.km / u.second).value,
+        gen_body_str += f"     init_pos_vels['{body_name.lower()}']['vel']=[%.4f,%.4f,%.4f]\n" % (
+        vel.x.to(u.km / u.second).value,
         vel.y.to(u.km / u.second).value,
         vel.z.to(u.km / u.second).value)
 
@@ -304,36 +304,45 @@ def gen_real_pos_vel_bodies(body_names, dt):
 def get_init_pos_vels():
     init_pos_vels = {}
     init_pos_vels['sun'] = {}
-    init_pos_vels['sun']['pos'] = [120062.8111, -462734.9565, -198328.7512]
-    init_pos_vels['sun']['vel'] = [0.0093, 0.0037, 0.0014]
+    init_pos_vels['sun']['pos'] = [-4638653.0000, 0.0000, 1314332.6250]
+    init_pos_vels['sun']['vel'] = [-0.0165, 0.0000, -0.0097]
     init_pos_vels['mercury'] = {}
-    init_pos_vels['mercury']['pos'] = [-27459444.3141, 33659309.8266, 20887619.1358]
-    init_pos_vels['mercury']['vel'] = [-49.8268, -24.7415, -8.0569]
+    init_pos_vels['mercury']['pos'] = [-54412616.0000, 0.0000, 31280100.0000]
+    init_pos_vels['mercury']['vel'] = [-25.2667, 0.0000, -40.0561]
     init_pos_vels['venus'] = {}
-    init_pos_vels['venus']['pos'] = [21823958.6720, -97202911.8077, -45109059.0044]
-    init_pos_vels['venus']['vel'] = [34.0912, 7.0463, 1.0154]
+    init_pos_vels['venus']['pos'] = [-32341672.0000, 0.0000, -99861352.0000]
+    init_pos_vels['venus']['vel'] = [34.3770, 0.0000, -10.6676]
     init_pos_vels['earth'] = {}
-    init_pos_vels['earth']['pos'] = [-25983996.5057, 132369339.0025, 57373518.2429]
-    init_pos_vels['earth']['vel'] = [-29.7869, -4.9606, -2.1489]
+    init_pos_vels['earth']['pos'] = [140427392.0000, 0.0000, 45724572.0000]
+    init_pos_vels['earth']['vel'] = [-8.1280, 0.0000, 28.2277]
     init_pos_vels['mars'] = {}
-    init_pos_vels['mars']['pos'] = [-230623216.7918, -71477306.3132, -26561224.6080]
-    init_pos_vels['mars']['vel'] = [8.4564, -18.9614, -8.9250]
+    init_pos_vels['mars']['pos'] = [218751648.0000, 0.0000, -45263688.0000]
+    init_pos_vels['mars']['vel'] = [5.2205, 0.0000, 23.4601]
+    init_pos_vels['asteroids'] = {}
+    init_pos_vels['asteroids']['pos'] = [0.0000, 0.0000, 0.0000]
+    init_pos_vels['asteroids']['vel'] = [0.0000, 0.0000, 0.0000]
     init_pos_vels['jupiter'] = {}
-    init_pos_vels['jupiter']['pos'] = [-357712686.7572, 637570826.6050, 281973598.8065]
-    init_pos_vels['jupiter']['vel'] = [-11.7897, -5.0252, -1.8671]
+    init_pos_vels['jupiter']['pos'] = [769445760.0000, 0.0000, 2017638.5000]
+    init_pos_vels['jupiter']['vel'] = [0.0389, 0.0000, 13.0879]
     init_pos_vels['saturn'] = {}
-    init_pos_vels['saturn']['pos'] = [713384516.7012, -1202372203.9542, -527442445.4970]
-    init_pos_vels['saturn']['vel'] = [7.9649, 4.3679, 1.4610]
+    init_pos_vels['saturn']['pos'] = [-1330536704.0000, 0.0000, -454165280.0000]
+    init_pos_vels['saturn']['vel'] = [3.2189, 0.0000, -9.1974]
     init_pos_vels['uranus'] = {}
-    init_pos_vels['uranus']['pos'] = [-2666283945.3730, 544495773.6564, 276202868.3064]
-    init_pos_vels['uranus']['vel'] = [-1.5476, -6.3725, -2.7691]
+    init_pos_vels['uranus']['pos'] = [-1757163648.0000, 0.0000, 2272503808.0000]
+    init_pos_vels['uranus']['vel'] = [-5.3918, 0.0000, -4.1634]
     init_pos_vels['neptune'] = {}
-    init_pos_vels['neptune']['pos'] = [2602697275.5219, 3374419276.7397, 1316379531.2695]
-    init_pos_vels['neptune']['vel'] = [-4.4365, 2.9291, 1.3095]
+    init_pos_vels['neptune']['pos'] = [-1480893824.0000, 0.0000, 4351772160.0000]
+    init_pos_vels['neptune']['vel'] = [-5.1466, 0.0000, -1.7189]
+    init_pos_vels['pluto'] = {}
+    init_pos_vels['pluto']['pos'] = [-1293909504.0000, 0.0000, 5770579968.0000]
+    init_pos_vels['pluto']['vel'] = [-4.5847, 0.0000, -1.0475]
+
     return init_pos_vels
 
+
 if __name__ == '__main__':
     from astropy.time import Time  # 时间
+
     #    bodies = [
     #         sun,  # 太阳放大 80 倍
     #         Mercury(size_scale=4e3, distance_scale=1.3),  # 水星放大 4000 倍，距离放大 1.3 倍

sim_scenes/tri_bodies/two_way_foil.py

@@ -6,23 +6,17 @@
 # link            :https://gitcode.net/pythoncr/
 # python_version  :3.9
 # ==============================================================================
-import threading
 import time
-import datetime
 
-from bodies import Sun, Earth, Moon
-from objs import CoreValagaClas, SciFiBomber, WaterDrop, QuadObj, CircleObj
-from common.consts import AU, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH
-from sim_scenes.func import ursina_run, camera_look_at, create_main_entity, two_bodies_colliding
-from simulators.ursina.entities.body_timer import TimeData
-from simulators.ursina.entities.entity_utils import create_directional_light, create_rings
-from simulators.ursina.ursina_config import UrsinaConfig
-from simulators.ursina.ursina_event import UrsinaEvent
 from ursina import camera, application
 
-from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Asteroids
-from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_MONTH, SECONDS_PER_YEAR, AU
-from sim_scenes.func import mayavi_run, ursina_run, create_sphere_sky
+from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto
+from common.celestial_data_service import get_init_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.entities.entity_utils import create_rings
 from simulators.ursina.ursina_event import UrsinaEvent
 
 
@@ -33,48 +27,10 @@ class TwoWayFoilSim:
 
     def __init__(self):
 
-        self.three_dim_bodies = []
-        self.two_dim_bodies = []
+        # self.three_dim_bodies = []
+        # self.two_dim_bodies = []
         self.current_stage = self.stage_01
 
-    def get_init_pos_vels(self):
-        init_pos_vels = {}
-        init_pos_vels['sun'] = {}
-        init_pos_vels['sun']['pos'] = [-4638653.0000, 0.0000, 1314332.6250]
-        init_pos_vels['sun']['vel'] = [-0.0165, 0.0000, -0.0097]
-        init_pos_vels['mercury'] = {}
-        init_pos_vels['mercury']['pos'] = [-54412616.0000, 0.0000, 31280100.0000]
-        init_pos_vels['mercury']['vel'] = [-25.2667, 0.0000, -40.0561]
-        init_pos_vels['venus'] = {}
-        init_pos_vels['venus']['pos'] = [-32341672.0000, 0.0000, -99861352.0000]
-        init_pos_vels['venus']['vel'] = [34.3770, 0.0000, -10.6676]
-        init_pos_vels['earth'] = {}
-        init_pos_vels['earth']['pos'] = [140427392.0000, 0.0000, 45724572.0000]
-        init_pos_vels['earth']['vel'] = [-8.1280, 0.0000, 28.2277]
-        init_pos_vels['mars'] = {}
-        init_pos_vels['mars']['pos'] = [218751648.0000, 0.0000, -45263688.0000]
-        init_pos_vels['mars']['vel'] = [5.2205, 0.0000, 23.4601]
-        init_pos_vels['asteroids'] = {}
-        init_pos_vels['asteroids']['pos'] = [0.0000, 0.0000, 0.0000]
-        init_pos_vels['asteroids']['vel'] = [0.0000, 0.0000, 0.0000]
-        init_pos_vels['jupiter'] = {}
-        init_pos_vels['jupiter']['pos'] = [769445760.0000, 0.0000, 2017638.5000]
-        init_pos_vels['jupiter']['vel'] = [0.0389, 0.0000, 13.0879]
-        init_pos_vels['saturn'] = {}
-        init_pos_vels['saturn']['pos'] = [-1330536704.0000, 0.0000, -454165280.0000]
-        init_pos_vels['saturn']['vel'] = [3.2189, 0.0000, -9.1974]
-        init_pos_vels['uranus'] = {}
-        init_pos_vels['uranus']['pos'] = [-1757163648.0000, 0.0000, 2272503808.0000]
-        init_pos_vels['uranus']['vel'] = [-5.3918, 0.0000, -4.1634]
-        init_pos_vels['neptune'] = {}
-        init_pos_vels['neptune']['pos'] = [-1480893824.0000, 0.0000, 4351772160.0000]
-        init_pos_vels['neptune']['vel'] = [-5.1466, 0.0000, -1.7189]
-        init_pos_vels['pluto'] = {}
-        init_pos_vels['pluto']['pos'] = [-1293909504.0000, 0.0000, 5770579968.0000]
-        init_pos_vels['pluto']['vel'] = [-4.5847, 0.0000, -1.0475]
-
-        return init_pos_vels
-
     def build_solar_system(self):
         # region 构建太阳系
         self.sun = Sun(size_scale=0.8e2)
@@ -84,7 +40,6 @@ class TwoWayFoilSim:
             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 倍
-            # Asteroids(size_scale=1e2, parent=self.sun),  # 小行星模拟(仅 ursina 模拟器支持)
             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
@@ -92,7 +47,8 @@ class TwoWayFoilSim:
             Pluto(size_scale=10e3, distance_scale=0.23),  # 冥王星放大 10000 倍，距离缩小到真实距离的 0.23(从太阳系的行星中排除)
         ]
         # endregion
-        init_pos_vels = self.get_init_pos_vels()
+
+        init_pos_vels = get_init_pos_vels()
 
         for body in self.bodies:
             pos_vels = init_pos_vels.get(type(body).__name__.lower(), None)
@@ -114,33 +70,36 @@ class TwoWayFoilSim:
                                              ) \
             .set_light_disable(True).set_ignore_gravity(True)
 
-    def create_two_dim(self, body):
-        body_2dim = CircleObj(texture=body.texture,
-                              size_scale=body.diameter * body.size_scale,
-                              distance_scale=body.distance_scale,
-                              init_position=body.position) \
-            .set_light_disable(True).set_ignore_gravity(True)
-
-        body.two_dim = body_2dim
-        body_2dim.three_dim = body
-        return body_2dim
-
-    def build_two_dim_bodies(self):
-        for body in self.bodies:
-            two_dim_body = self.create_two_dim(body)
-            self.three_dim_bodies.append(body)
-            self.two_dim_bodies.append(two_dim_body)
-
         self.bodies.append(self.two_way_foil)
         self.bodies.append(self.two_way_foil_circle)
 
-        for body in self.two_dim_bodies:
-            self.bodies.append(body)
+    # def create_two_dim(self, body):
+    #     body_2dim = CircleObj(texture=body.texture,
+    #                           size_scale=body.diameter * body.size_scale,
+    #                           distance_scale=body.distance_scale,
+    #                           init_position=body.position) \
+    #         .set_light_disable(True).set_ignore_gravity(True)
+    #
+    #     body.two_dim = body_2dim
+    #     body_2dim.three_dim = body
+    #     return body_2dim
+
+    # def build_two_dim_bodies(self):
+    #     for body in self.bodies:
+    #         # two_dim_body = self.create_two_dim(body)
+    #         self.three_dim_bodies.append(body)
+    #         # self.two_dim_bodies.append(two_dim_body)
+    #
+    #     self.bodies.append(self.two_way_foil)
+    #     self.bodies.append(self.two_way_foil_circle)
+    #
+    #     for body in self.two_dim_bodies:
+    #         self.bodies.append(body)
 
     def build(self):
         self.build_solar_system()
         self.build_two_way_foil()
-        self.build_two_dim_bodies()
+        # self.build_two_dim_bodies()
 
     def on_ready(self):
         """
@@ -152,22 +111,22 @@ class TwoWayFoilSim:
         camera.clip_plane_far = 1000000
         create_sphere_sky(scale=200000)
         application.time_scale = 5
-        for body in self.two_dim_bodies:
-            body.planet.enabled = False
-            model = body.planet.model
-            if body.three_dim.has_rings:
-                rings = create_rings(body.planet)
-                rings.scale *= 2e10
+        # for body in self.two_dim_bodies:
+        #     body.planet.enabled = False
+        #     model = body.planet.model
+        #     if body.three_dim.has_rings:
+        #         rings = create_rings(body.planet)
+        #         rings.scale *= 2e10
 
         self.two_way_foil_circle.planet.rotation_x = 90
         self.two_way_foil_circle.planet.enabled = False
         # self.two_way_foil.planet.alpha = 0.6
 
-        for b in self.two_dim_bodies:
-            b.planet.rotation_x = 90
+        # for b in self.two_dim_bodies:
+        #     b.planet.rotation_x = 90
 
-    def flatten_animation(self, body, flatten_body):
-        def flatten_update(planet, flatten_planet):
+    def flatten_animation(self, body):
+        def flatten_update(planet):
             def warp():
                 planet.init_update()
                 # time.sleep(0.2)
@@ -187,7 +146,7 @@ class TwoWayFoilSim:
         body.planet.scale_y_v = body.planet.scale_y
         body.planet.init_scale_y = body.planet.scale_y
         body.planet.init_update = body.planet.update
-        body.planet.update = flatten_update(body.planet, flatten_body.planet)
+        body.planet.update = flatten_update(body.planet)
 
         # for i in range(10):
         #     # body.planet.scale_y /= 2
@@ -241,17 +200,19 @@ class TwoWayFoilSim:
         """
         # self.sun.two_dim.planet.init_scale += 0.05
         self.two_way_foil_circle.planet.init_scale += 0.8
-        self.blink(self.sun.two_dim)
+        # self.blink(self.sun.two_dim)
         # 二维化延时的时间
         two_way_delay_times = [0.5,  # 太阳
                                0.5, 0.8, 1.0,  # 水星 金星 地球
                                1.2, 2.0, 3.0,  # 火星 木星 土星
                                4.0, 4.5, 5.5]  # 天王星 海王星 冥王星
-        for idx, b in enumerate(self.three_dim_bodies):
+        for idx, b in enumerate(self.bodies):
+            if isinstance(b, Obj):
+                continue
             if hasattr(b, "two_way_time"):
                 if b.two_way_time is not None:
                     if time.time() - b.two_way_time > two_way_delay_times[idx]:
-                        self.flatten_animation(b, b.two_dim)
+                        self.flatten_animation(b)
                         # b.planet.enabled = False
                         # b.two_dim.planet.enabled = True
                         b.two_way_time = None
@@ -259,9 +220,9 @@ class TwoWayFoilSim:
                 b.two_way_time = time.time()
                 # b.planet.enabled = False
                 # b.two_dim.planet.enabled = True
-            if b.two_dim.planet.enabled:
-                # b.two_dim.planet.init_scale += 0.05
-                self.blink(b.two_dim)
+            # if b.two_dim.planet.enabled:
+            #     # b.two_dim.planet.init_scale += 0.05
+            #     self.blink(b.two_dim)
         self.two_way_foil_circle.planet.rotation_z += 0.4
 
     def on_timer_changed(self, time_data):
@@ -270,9 +231,9 @@ class TwoWayFoilSim:
         # self.two_way_foil.planet.rotation_z += 10
         camera_look_at(self.two_way_foil)
         self.current_stage()
-        for b in self.two_dim_bodies:
-            b.planet.rotation_y -= 0.5
-            b.init_position = b.three_dim.position
+        # for b in self.two_dim_bodies:
+        #     b.planet.rotation_y -= 0.5
+        #     b.init_position = b.three_dim.position
 
 
 if __name__ == '__main__':