# -*- coding:utf-8 -*- # title :哈雷彗星场景模拟 # description :哈雷彗星场景模拟 # author :Python超人 # date :2023-10-25 # link :https://gitcode.net/pythoncr/ # python_version :3.9 # ============================================================================== import time import math from ursina import camera, application 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.color_utils import trail_color_brightest from common.consts import SECONDS_PER_YEAR, AU from common.func import calculate_distance from objs import HalleComet, Obj from sim_scenes.func import camera_look_at, two_bodies_colliding, create_text_panel from sim_scenes.func import ursina_run, create_sphere_sky from simulators.ursina.entities.world_grid import WorldGrid from simulators.ursina.ursina_config import UrsinaConfig from simulators.ursina.ursina_event import UrsinaEvent from simulators.ursina.ursina_mesh import create_orbit_line, create_orbit_by_points class HalleyCometSim: """ 哈雷彗星场景模拟 """ def __init__(self): pass def build_solar_system(self): # region 构建太阳系 show_trail = False self.sun = Sun(size_scale=0.8e2, show_trail=show_trail) self.mercury = Mercury(size_scale=5e3, show_trail=show_trail) self.venus = Venus(size_scale=5e3, show_trail=show_trail) self.earth = Earth(size_scale=5e3, rotate_angle=0, show_trail=show_trail) self.mars = Mars(size_scale=5e3, show_trail=show_trail) self.jupiter = Jupiter(size_scale=2.5e3, show_trail=show_trail) self.saturn = Saturn(size_scale=2.5e3, show_trail=show_trail) self.uranus = Uranus(size_scale=6e3, show_trail=show_trail) self.neptune = Neptune(size_scale=6e3, show_trail=show_trail) self.pluto = Pluto(size_scale=1e5, show_trail=show_trail) self.bodies = [ self.sun, # 太阳 self.mercury, # 水星 self.venus, # 金星 self.earth, # 地球 self.mars, # 火星 self.jupiter, # 木星 self.saturn, # 土星 self.uranus, # 天王星 self.neptune, # 海王星 # self.pluto, # 冥王星 ] self.mercury.orbital_days = 87.9691 self.venus.orbital_days = 224.701 self.earth.orbital_days = 365.24219 self.mars.orbital_days = 686.971 self.jupiter.orbital_days = 11.862 * 365.24219 self.saturn.orbital_days = 29.4571 * 365.24219 self.uranus.orbital_days = 84.0205 * 365.24219 self.neptune.orbital_days = 164.8 * 365.24219 # self.bodies = [ # self.sun, # 太阳 # self.mars, # 火星 # self.neptune, # 海王星 # ] # endregion trail_color_brightest(self.bodies) init_bodies_reality_pos_vels(self.bodies) def build_halley_comet(self): """ 创建哈雷彗星 @return: """ # 哈雷彗星的平均运行速度约为每小时 70,000 英里或每小时 126,000 公里 。(35公里/秒) # 每76.1年环绕太阳一周的周期彗星 # 3.335, 0, 10.7 73 # 3.33, 0, 10.7 73 # [3.335, 0, 10.699] 71 # [3.33, 0, 10.655] 68 # [3.33, 0, 10.66] 69 self.halley_comet = HalleComet( # size_scale=4e7, size_scale=1e8, init_velocity=[3.34, 0, 10.7], # [3.33, 0, 10.6] < ? <[3.34, 0, 10.7] init_position=[0, 0.5 * AU, -10 * AU]) \ .set_light_disable(True) self.bodies.append(self.halley_comet) def build(self): self.build_solar_system() self.build_halley_comet() def calculate_angles(self, point1, point2): dx = point2.x - point1.x dy = point2.y - point1.y dz = point2.z - point1.z roll = math.degrees(math.atan2(dy, dz)) pitch = math.degrees(math.atan2(dx, math.sqrt(dy ** 2 + dz ** 2))) yaw = math.degrees(math.atan2(math.sin(roll), math.cos(roll))) return roll, pitch, yaw # # def calculate_angles(self, point1, point2): # import numpy as np # # 计算向量AB # AB = point1 - point2 # # # 计算向量AB与x轴、y轴和z轴之间的夹角 # angle_x = np.arctan2(AB.y, AB.x) * 180 / np.pi # angle_y = np.arctan2(AB.z, np.sqrt(AB.x ** 2 + AB.y ** 2)) * 180 / np.pi # angle_z = np.arctan2(np.sqrt(AB.x ** 2 + AB.y ** 2), AB.z) * 180 / np.pi # # return angle_x, angle_y, angle_z # def calculate_angles(self, point1, point2): # # # 计算向量AB # AB = point2 - point1 # # 计算向量AB与x轴、y轴和z轴之间的夹角 # angle_x = -math.degrees(-math.atan2(AB.z, AB.y)) # angle_y = -math.degrees(-math.atan2(AB.z, AB.x)) # angle_z = math.degrees(math.atan2(AB.y, AB.x)) # return angle_x, angle_y, angle_z 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) # print(points) orbit_line = create_orbit_by_points(center_body.position, points, line_color=body.trail_color) return orbit_line def on_ready(self): """ 事件绑定后，模拟器运行前会触发 @return: """ # 创建天空 from ursina import scene UrsinaConfig.trail_type = "line" UrsinaConfig.trail_length = 91 # UrsinaConfig.trail_length = 1000 UrsinaConfig.trail_thickness_factor = 3 # camera.clip_plane_near = 0.1 camera.clip_plane_far = 1000000 create_sphere_sky(scale=200000) # WorldGrid().draw_axises(10) application.time_scale = 5 # self.orbit_lines = [] for body in self.bodies[1:]: if isinstance(body, HalleComet): continue print("create_orbit_line", body) """ # 在 ursina 中，我以 sun.position 为中心， body.position 的所在位置(x,y,z) 画一个圆环（orbit_line）。 到 sun 中心点(x,y,z) 为半径，返回一个圆形轨道 orbit_line orbit_line = create_orbit_line(sun, body) # 帮我解决下面的问题，怎么修改下面的值，让 orbit_line 轨道线的倾斜和 body 匹配 orbit_line.rotation_x = ? orbit_line.rotation_y = ? orbit_line.rotation_z = ? body.position - sun.position """ orbit_line = self.create_orbit_line(self.sun, body) # # orbit_line.enabled = False # angle_x, angle_y, angle_z = self.calculate_angles(self.sun.planet.position, body.planet.position) # # # 获取body相对于self.sun的位置向量 # # relative_position = body.planet.position - self.sun.planet.position # # # # rotation_x = -math.degrees( # # math.atan2(relative_position.y, math.sqrt(relative_position.x ** 2 + relative_position.z ** 2))) # # rotation_y = math.degrees(math.atan2(relative_position.x, relative_position.z)) # # # 计算旋转角度 # # orbit_line.rotation_x = rotation_x + 90 # # orbit_line.rotation_z = 0 # # orbit_line.rotation_y = rotation_y + 120 # # # # # angle = math.atan2(relative_position.y, relative_position.x) # # # # orbit_line.rotation_x = angle_x - 110 # angle_x+90 # angle_x # angle_x # # orbit_line.rotation_y = angle_y # angle_y+90 # angle_y # - 50 # # orbit_line.rotation_z = angle_z - 90 # angle_z# angle_z # angle_z # # # print(body.name,angle_x,angle_y,angle_z) # # print(body.name, orbit_line.rotation_x, orbit_line.rotation_y, orbit_line.rotation_z) # # orbit_line.look_at(body.planet) # # print(body.name, orbit_line.rotation_x, orbit_line.rotation_y, orbit_line.rotation_z) # # 火星 -90.0 -90.0 0.0 # # 火星 -0.0 -0.0 0.0 # # 火星 113.2222958819701 137.6691000401162 -158.65250912389882 # # 火星 16.115840911865234 -132.33090209960938 104.18995666503906 # # # 海王星 -90.0 -90.0 0.0 # # 海王星 -0.0 -0.0 0.0 # # 海王星 131.3906379192235 3.0188762268681826 -2.6611704409164667 # # 海王星 2.6574795246124268 93.01887512207031 -90.14009857177734 # orbit_line.body = body # self.orbit_lines.append(orbit_line) self.text_panel = create_text_panel() def on_timer_changed(self, time_data): """ @param time_data: @return: """ if hasattr(self, "halley_comet"): # 哈雷彗星飞行的翻转效果 if self.halley_comet.planet.enabled: self.halley_comet.planet.rotation_x += 0.1 self.halley_comet.planet.rotation_y += 1 # 摄像机始终看向哈雷彗星 # camera_look_at(self.halley_comet) 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) if __name__ == '__main__': """ 哈雷彗星场景模拟 """ sim = HalleyCometSim() sim.build() # 订阅事件后，上面2个函数功能才会起作用 # 运行中，每时每刻都会触发 on_timer_changed UrsinaEvent.on_timer_changed_subscription(sim.on_timer_changed) # 运行前会触发 on_ready UrsinaEvent.on_ready_subscription(sim.on_ready) # 使用 ursina 查看的运行效果 # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- ursina_run(sim.bodies, SECONDS_PER_YEAR, # position=(0, 2 * AU, -11 * AU), position=(0, 0.5 * AU, -5 * AU), cosmic_bg='', show_trail=True, # bg_music='sounds/no_glory.mp3', show_camera_info=False, show_control_info=False, show_timer=True, show_grid=False )