Python超人-宇宙模拟器(哈雷彗星模拟2代码清理备份)

sim_scenes/solar_system/halley_comet_sim_02.py

@@ -8,10 +8,13 @@
 # ==============================================================================
 import time
 import math
-from ursina import camera, application
 
+import pytz
+from ursina import camera, application
+from astropy.time import Time
 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_reality_pos_vels, get_reality_orbit_points
+from common.celestial_data_service import get_init_pos_vels, init_bodies_reality_pos_vels, get_reality_orbit_points, \
+    conv_to_astropy_time
 from common.color_utils import trail_color_brightest
 from common.consts import SECONDS_PER_YEAR, AU
 from common.func import calculate_distance
@@ -29,8 +32,14 @@ class HalleyCometSim:
     哈雷彗星场景模拟
     """
 
-    def __init__(self):
-        pass
+    def __init__(self, dt=None):
+        if dt is None:
+            self.dt = Time.now()
+        elif isinstance(dt, str):
+            self.dt = conv_to_astropy_time(dt)
+        else:
+            self.dt = dt
+        # print("北京时间：", dt.to_datetime(timezone=pytz.timezone('Asia/Shanghai')))
 
     def build_solar_system(self):
         # region 构建太阳系
@@ -73,7 +82,8 @@ class HalleyCometSim:
         # ]
         # endregion
         trail_color_brightest(self.bodies)
-        init_bodies_reality_pos_vels(self.bodies)
+
+        init_bodies_reality_pos_vels(self.bodies, self.dt)
 
     def build_halley_comet(self):
         """
@@ -135,7 +145,8 @@ class HalleyCometSim:
 
     def create_orbit_line(self, center_body, body):
         orbital_days = int(math.ceil(body.orbital_days * 1.02))
-        points = get_reality_orbit_points(type(body).__name__.lower(), days=orbital_days, segments=100)
+        points = get_reality_orbit_points(type(body).__name__.lower(), start_time=self.dt, days=orbital_days,
+                                          segments=100)
         # print(points)
         orbit_line = create_orbit_by_points(center_body.position, points, line_color=body.trail_color)
         return orbit_line
@@ -158,7 +169,7 @@ class HalleyCometSim:
         # WorldGrid().draw_axises(10)
 
         application.time_scale = 5
-        # self.orbit_lines = []
+        self.orbit_lines = []
         for body in self.bodies[1:]:
             if isinstance(body, HalleComet):
                 continue
@@ -212,7 +223,7 @@ body.position - sun.position
             # # 海王星 2.6574795246124268 93.01887512207031 -90.14009857177734
 
             # orbit_line.body = body
-            # self.orbit_lines.append(orbit_line)
+            self.orbit_lines.append(orbit_line)
 
         self.text_panel = create_text_panel()
 
@@ -232,13 +243,8 @@ body.position - sun.position
             d = calculate_distance(self.halley_comet.position, self.sun.position)
             self.text_panel.text = "哈雷彗星距离太阳：%.3f AU" % (d / AU)
 
-        # for i, orbit_line in enumerate(self.orbit_lines):
-        #     if i < 4:
-        #         adj_scale = False
-        #     else:
-        #         adj_scale = True
-        #     # 由于天体运行不是标准的圆形，则需要动态调整轨道的大小，保证轨道线始终在天体的中心位置
-        #     orbit_line.auto_adjust(adj_scale=adj_scale)
+        for i, orbit_line in enumerate(self.orbit_lines):
+            orbit_line.position = self.sun.planet.position
 
 
 if __name__ == '__main__':