Python超人-宇宙模拟器

sim_scenes/solar_system/halley_comet_sim.py

@@ -151,8 +151,9 @@ class HalleyCometSim(HalleyCometSimBase):
                              scale=scale, alpha=1.0, background=background
                              )
         label.set_light_off()
+        self.last_year = year
 
-    def set_comet_trail_alpha(self, distance_sun):
+    def update_comet_trail_alpha(self, distance_sun):
         """
         根据彗哈雷星和太阳的距离，设置彗星尾巴的透明度来模仿接近太阳有慧尾，离开太阳到一定距离就渐渐消失
         @param distance_sun: 彗哈雷星和太阳的距离
@@ -204,6 +205,12 @@ class HalleyCometSim(HalleyCometSimBase):
         return get_run_speed_factor() * value
 
     def camera_move(self, dt):
+        """
+        摄像机移动控制
+        @param dt:
+        @return:
+        """
+        # 摄像机移动控制数据
         camera_move_infos = [
             # 条件：年份
             # 移动的信息：
@@ -229,65 +236,95 @@ class HalleyCometSim(HalleyCometSimBase):
         @param time_data:
         @return:
         """
-        dt = time_data.get_datetime(str(self.start_time))
-        year = dt.strftime("%Y")
-
+        dt = time_data.get_datetime(self.start_time)
+        # 摄像机看向哈雷彗星
         camera_look_at(self.halley_comet, rotation_z=0)
 
-        self.camera_move(dt)
+        time_total_hours = time_data.total_hours
 
-        if hasattr(self, "halley_comet"):
-            if self.halley_comet.planet.enabled:
-                self.halley_comet.planet.look_at(self.sun.planet)
-                d_sun = calculate_distance(self.halley_comet.position, self.sun.position)
-                d_earth = calculate_distance(self.halley_comet.position, self.earth.position)
-                trail_keys = self.halley_comet.planet.trails.keys()
-                if len(trail_keys) > 0:
-                    last_trail = list(trail_keys)[-1]
-                    if hasattr(last_trail, "entity_infos"):
-                        last_trail.entity_infos["distance_from_sun"] = d_sun
-                        last_trail.entity_infos["distance_from_earth"] = d_earth
-                        last_trail.entity_infos["time"] = dt.strftime("%Y-%m-%d")
+        if not hasattr(self, "last_total_hours"):
+            self.last_total_hours = time_total_hours
+            self.update_halley_comet_info(dt)
 
-                    pos = self.halley_comet.planet.position
+        # 每隔一段时间运行一次更新（不要太频繁更新，会导致摄像机抖动）
+        if time_total_hours - self.last_total_hours > 100:
+            self.camera_move(dt)
+            self.update_halley_comet_info(dt)
+            self.update_ui(time_data, dt)
+            # 记录最后更新的总耗时（小时）
+            self.last_total_hours = time_total_hours
 
-                    import copy
+    def check_create_year_label(self, last_trail, dt):
+        """
+        检测并创建年标签
+        @param last_trail:
+        @param dt:
+        @return:
+        """
+        if last_trail is None:
+            return
 
+        import copy
+        year = dt.strftime("%Y")
+        pos = self.halley_comet.planet.position
+        # 里程碑标签
         if self.show_milestone_lable(last_trail, dt):
             pass
         elif not hasattr(self, "last_year"):
-                        self.create_year_label(last_trail, year, pos)
-                    elif self.last_year != year:
-                        if not hasattr(self, "last_label_pos"):
+            # 第一次运行，则创建“年标签”
             self.create_year_label(last_trail, year, pos)
             self.last_label_pos = copy.deepcopy(self.halley_comet.position)
-                        else:
-                            # 防止标签非常紧密
+        elif self.last_year != year:
+            if hasattr(self, "last_label_pos"):
+                # 防止“年标签”显示非常紧密
                 d = calculate_distance(self.halley_comet.position, self.last_label_pos)
                 if d > 2 * AU:
                     self.create_year_label(last_trail, year, pos)
                     self.last_label_pos = copy.deepcopy(self.halley_comet.position)
 
-                self.last_year = year
+    def update_halley_comet_info(self, dt):
+        """
+        更新哈雷彗星的信息
+        @param dt:
+        @return:
+        """
+        # 哈雷彗星面向太阳，这样彗尾就远离太阳的方向
+        self.halley_comet.planet.look_at(self.sun.planet)
 
-                # 哈雷彗星离太阳最近的点称为 "perihelion of Halley's Comet"（近日点：comet_peri），
-                if not hasattr(self, "comet_peri"):
-                    self.comet_peri = d_sun
-                    self.comet_peri_dt = dt.strftime("%Y-%m-%d")
-                elif d_sun < self.comet_peri:
-                    self.comet_peri = d_sun
-                    self.comet_peri_dt = dt.strftime("%Y-%m-%d")
+        # 计算哈雷彗星与太阳、地球的距离
+        d_sun = calculate_distance(self.halley_comet.position, self.sun.position)
+        d_earth = calculate_distance(self.halley_comet.position, self.earth.position)
+        # 哈雷彗星所有轨迹线（注意不是彗尾）
+        trail_keys = self.halley_comet.planet.trails.keys()
+        # 哈雷彗星有轨迹线
+        if len(trail_keys) > 0:
+            # 找的最后的轨迹线
+            last_trail = list(trail_keys)[-1]
+            if hasattr(last_trail, "entity_infos"):
+                # 如果是轨迹球，则轨迹球会记录当前位置与太阳、地球的距离，并记录了当前的日期
+                last_trail.entity_infos["distance_from_sun"] = d_sun
+                last_trail.entity_infos["distance_from_earth"] = d_earth
+                last_trail.entity_infos["time"] = dt.strftime("%Y-%m-%d")
 
-                # 哈雷彗星离太阳最远的点称为 "aphelion of Halley's Comet"（远日点）
-                if not hasattr(self, "comet_aphel"):
-                    self.comet_aphel = d_sun
-                    self.comet_aphel_dt = dt.strftime("%Y-%m-%d")
-                elif d_sun > self.comet_aphel:
-                    self.comet_aphel = d_sun
-                    self.comet_aphel_dt = dt.strftime("%Y-%m-%d")
+            # 检测并再轨迹线上创建年标签
+            self.check_create_year_label(last_trail, dt)
 
-                self.set_comet_trail_alpha(d_sun)
+        # 更新彗星尾巴的透明度
+        self.update_comet_trail_alpha(d_sun)
 
+        # 计算和更新近日点数据
+        self.update_comet_peri(d_sun, dt)
+        # 计算和更新远日点数据
+        self.update_comet_aphel(d_sun, dt)
+        # 更新文字信息面板
+        self.update_text_panel(d_sun)
+
+    def update_text_panel(self, d_sun):
+        """
+        更新文字信息面板
+        @param d_sun:
+        @return:
+        """
         panel_text = "哈雷彗星:\n\n距离太阳：%.3f\tAU" % (d_sun / AU)
         panel_text += "\n\n离日最远：%.3f\tAU" % (self.comet_aphel / AU)
         panel_text += "\n\n离日最近：%.3f\tAU" % (self.comet_peri / AU)
@@ -296,22 +333,50 @@ class HalleyCometSim(HalleyCometSimBase):
 
         self.text_panel.text = panel_text
 
-        time_total_hours = time_data.total_hours
+    def update_comet_aphel(self, d_sun, dt):
+        """
+        计算和更新远日点数据
+        @param d_sun:
+        @param dt:
+        @return:
+        """
+        # 哈雷彗星离太阳最远的点称为 "aphelion of Halley's Comet"（远日点）
+        if not hasattr(self, "comet_aphel"):
+            self.comet_aphel = d_sun
+            self.comet_aphel_dt = dt.strftime("%Y-%m-%d")
+        elif d_sun > self.comet_aphel:
+            self.comet_aphel = d_sun
+            self.comet_aphel_dt = dt.strftime("%Y-%m-%d")
 
-        if not hasattr(self, "last_total_hours"):
-            self.last_total_hours = time_total_hours
+    def update_comet_peri(self, d_sun, dt):
+        """
+        计算和更新近日点数据
+        @param d_sun:
+        @param dt:
+        @return:
+        """
+        # 哈雷彗星离太阳最近的点称为 "perihelion of Halley's Comet"（近日点：comet_peri），
+        if not hasattr(self, "comet_peri"):
+            self.comet_peri = d_sun
+            self.comet_peri_dt = dt.strftime("%Y-%m-%d")
+        elif d_sun < self.comet_peri:
+            self.comet_peri = d_sun
+            self.comet_peri_dt = dt.strftime("%Y-%m-%d")
 
-        # 每个一段时间运行一次更新（不要太频繁更新，会导致摄像机抖动）
-        if time_total_hours - self.last_total_hours > 100:
+    def update_ui(self, time_data, dt):
+        """
+        更新UI，包含天体的位置、时钟、轨道的位置
+        @param time_data:
+        @param dt:
+        @return:
+        """
         # 更新天体的位置
         self.set_bodies_position(time_data)
         # 更新时钟
         self.show_clock(dt)
-            # 更新轨道的位置
-            for i, orbit_line in enumerate(self.orbit_lines):
-                orbit_line.position = self.sun.planet.position
-
-            self.last_total_hours = time_total_hours
+        # 更新轨道的位置（TODO:短时间可以忽略）
+        # for i, orbit_line in enumerate(self.orbit_lines):
+        #     orbit_line.position = self.sun.planet.position
 
 
 if __name__ == '__main__':

sim_scenes/universe_sim_scenes.py

@@ -29,3 +29,5 @@ class UniverseSimScenes:
         # self.set_window_size((int(1920 * r), int(1080 * r)))
         # self.set_window_size((2376, 1080))
         # self.set_window_size((520, 540), fullscreen=False)
+
+

simulators/ursina/entities/body_timer.py

@@ -59,15 +59,21 @@ class TimeData:
             else:
                 self.time_text = f'{self.hours:02d}:{self.minutes:02d}:{self.seconds:02d}'
 
-
     def get_datetime(self, init_datetime):
+        """
+        根据初始化的时间，获取当前北京时间
+        @param init_datetime:
+        @return:
+        """
         import datetime
+        if not isinstance(init_datetime, str):
+            init_datetime = str(init_datetime)
         # UTC_format = "%Y-%m-%dT%H:%M:%S.%fZ"
         if len(init_datetime) == 19:
             init_datetime = init_datetime + ".000"
         UTC = datetime.datetime.strptime(init_datetime + "Z", "%Y-%m-%d %H:%M:%S.%fZ")
         # BJS_format = "%Y-%m-%d %H:%M:%S"
-        BJS = UTC + datetime.timedelta(hours=8+self.total_hours)
+        BJS = UTC + datetime.timedelta(hours=8 + self.total_hours)
         # BJS = BJS.strftime(BJS_format)
         # dt = datetime(init_datetime)
         return BJS
@@ -89,6 +95,7 @@ class AppTimeUtil:
     """
     应用计数器工具类
     """
+
     def __init__(self):
         self.arrival_time = -1
         self.current_time = 0