diff --git a/sim_scenes/solar_system/halley_comet_sim_02.py b/sim_scenes/solar_system/halley_comet_sim_02.py index 9fb8fdb3a34a407a7fe9c95bd16f8c814c2d20c6..735b123648970ed948fc01aba239f4a060da2c9d 100644 --- a/sim_scenes/solar_system/halley_comet_sim_02.py +++ b/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__':