# -*- coding:utf-8 -*- # title :三体二向箔场景模拟 # description :三体二向箔场景模拟 # author :Python超人 # date :2024-01-20 # link :https://gitcode.net/pythoncr/ # python_version :3.9 # ============================================================================== from ursina import camera, application, lerp, Vec3 from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto from common.celestial_data_service import init_bodies_reality_pos_vels, conv_to_astropy_time, \ set_solar_system_celestial_position, init_bodies_pos_vels from common.consts import SECONDS_PER_YEAR, AU from common.func import calculate_distance from bodies import Earth from objs import QuadObj, CircleObj, Obj from sim_scenes.func import create_text_panel, camera_look_at, get_run_speed_factor, \ camera_move_update, camera_move_to_target_update, camera_move_control, two_bodies_colliding from sim_scenes.func import ursina_run, create_sphere_sky from sim_scenes.solar_system.halley_comet_lib import HalleyCometSimBase, HalleyCometParams, \ create_halley_comet, create_orbit_line from sim_scenes.universe_sim_scenes import UniverseSimScenes from simulators.func import ext_fun_for_method from simulators.ursina.entities.body_timer import TimeData from simulators.ursina.entities.entity_utils import get_value_direction_vectors from simulators.ursina.ui.control_ui import ControlUI from simulators.ursina.ursina_config import UrsinaConfig from simulators.ursina.ursina_event import UrsinaEvent from simulators.ursina.ursina_mesh import create_label import time class TwoWayFoilSim(UniverseSimScenes): """ 三体二向箔场景模拟 """ def build_solar_system(self): # region 构建太阳系 show_trail = True self.sun = Sun(size_scale=0.6e2, show_trail=show_trail, # color=(255, 250, 245), # texture="", texture="sun_light.jpg" ) self.mercury = Mercury(size_scale=2e3, distance_scale=1.5, show_trail=show_trail) self.venus = Venus(size_scale=2e3, distance_scale=1.5, show_trail=show_trail) self.earth = Earth(size_scale=2e3, distance_scale=1.5, rotate_angle=0, show_trail=show_trail) self.mars = Mars(size_scale=2e3, distance_scale=1.4, show_trail=show_trail) self.jupiter = Jupiter(size_scale=0.3e3, distance_scale=0.72, show_trail=show_trail) self.saturn = Saturn(size_scale=0.3e3, distance_scale=0.52, show_trail=show_trail) self.uranus = Uranus(size_scale=0.8e3, distance_scale=0.33, show_trail=show_trail) self.neptune = Neptune(size_scale=0.8e3, distance_scale=0.25, show_trail=show_trail) self.pluto = Pluto(size_scale=1e4, distance_scale=0.23, 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.two_way_foil_dt_factor = 1 self.sun.two_way_foil_dt = 0.003 * self.two_way_foil_dt_factor self.saturn.two_way_foil_dt = 0.005 * self.two_way_foil_dt_factor self.uranus.two_way_foil_dt = 0.002 * self.two_way_foil_dt_factor init_bodies_pos_vels(self.bodies) def build(self): """ 构建太阳系系统以及哈雷彗星 @return: """ self.build_solar_system() self.build_two_way_foil() def init_settings(self): """ 初始化设置 @return: """ from ursina import color # 创建天空 create_sphere_sky(scale=50000, rotation_x=200, rotation_y=None, rotation_z=45) # UrsinaConfig.trail_type = "curve_line" # UrsinaConfig.trail_length = 300 UrsinaConfig.trail_type = "line" # UrsinaConfig.trail_length = 152 # 尾巴数量刚刚好 UrsinaConfig.trail_length = 130 UrsinaConfig.trail_thickness_factor = 3 # UrsinaConfig.trail_length = 180 UrsinaConfig.trail_factor = 3 # camera.clip_plane_near = 0.1 camera.clip_plane_far = 51000 # camera.fov = 60 self.current_stage = self.stage_01 def build_two_way_foil(self): """ 创建二向箔（一个原始的方形二向箔，一个不断扩展的圆形二向箔） @return: """ # 原始的方形二向箔 self.two_way_foil = QuadObj(texture='two_way_foil.png', # size_scale=4e7, size_scale=1e6, init_velocity=[0, -50, 60], init_position=[0, 5 * AU, -6 * AU]) \ .set_light_disable(True) # .set_ignore_gravity(False) # 不断扩展的圆形二向箔 self.two_way_foil_circle = CircleObj(texture="two_way_foil_circle.png", size_scale=self.two_way_foil.size_scale * 2, ) \ .set_light_disable(True).set_ignore_gravity(True) self.bodies.append(self.two_way_foil) self.bodies.append(self.two_way_foil_circle) def s_f(self, value=1): if value == 0: return 0 return get_run_speed_factor() * value def camera_move(self, time_data): """ 摄像机移动控制 @param dt: @return: """ # 摄像机移动控制数据 camera_move_infos = [ # 条件：年份 # 移动的信息： # 按坐标系方向移动 x:右+左-, y:升+降-, z:前+(接近太阳)后-(远离太阳) # 以摄像机视角移动 f:前 b:后 l:左 r:右 u:上 d:下 (0, {"f": 5.2}), # (1983, {"to": {"ct_id": 1, "t": 10}}), (10, {}), (12, {"f": 0.2}), (15, {"f": 0.3}), (18, {"f": 0.4}), # (30, {"f": 0.5}), # (40, {"f": 0.6}), (160, {"b": 2}), (1988, {"y": -3, "z": -12}), (1989, {"z": -8, "f": -5}), (1993, {"z": -8, "f": -3}), (1995, {"z": -8}), (2000, {"z": -8, "y": -0.2}), (2013, {}), (2048, {"f": 3}), (2062, {"y": -3}), (2063, {"y": -10, "z": 2}), (2181, {}), # (2082, {"exit": True}) ] def cond_cb(ps): return ps["next_cond"] > time_data.total_days >= ps["cond"] camera_move_control(camera_move_infos, cond_cb=cond_cb, value_conv=self.s_f, smooth=1) def flatten_animation(self, body): """ 天体二维化的动画 @param body: 天体 @return: """ def flatten_update(planet): def warp(): # 原始的 update 方法中有计算天体的运行 planet.init_update() # 对Y轴进行压平动画，如果压平大小不足 1/50，则继续压缩，直到压缩到 1/50 就不压缩了 # （如果觉得 1/50 的厚度压的不够，还可以继续压缩，基本上就够了） # if planet.scale_y_v > planet.init_scale_y / 50: if planet.scale_y_v > 0.5: planet.scale_y_v /= 1.01 # else: # self.gen_pixcel_image(planet) planet.scale_x_v += 0.05 # 灯光关闭，不然压到2纬就会是黑色 planet.set_light_off(True) planet.scale_y = planet.scale_y_v planet.scale_x = planet.scale_x_v planet.scale_z = planet.scale_x_v return warp body.is_2d = True # 压平时，转速将为以前的 1/200 body.planet.rotation_speed /= 200 # 记录原始的厚度大小 body.planet.scale_y_v = body.planet.scale_y body.planet.init_scale_y = body.planet.scale_y body.planet.scale_x_v = body.planet.scale_x body.planet.init_scale_x = body.planet.scale_x # 原始的 update 方法中有计算天体的运行，需要保留 body.planet.init_update = body.planet.update # 替换 update body.planet.update = flatten_update(body.planet) def get_target(self): for body in self.bodies: if hasattr(body, "is_2d"): continue return body return None def stage_01(self): """ 二向箔飞向太阳 @return: """ # 如果二向箔和太阳碰撞 if two_bodies_colliding(self.two_way_foil, self.sun): # 隐藏原始二向箔，保持在原地，不在飞行 # self.two_way_foil.planet.enabled = False self.two_way_foil.init_velocity = [0, 0, 0] def two_way_foil_update(): target = self.get_target() if target is self.sun: # 关闭太阳的光晕 self.sun.hide_children() # 让实体A朝向实体B # self.two_way_foil.planet.look_at(target) if target is not None: # 让实体A向着实体B移动 dt = target.two_way_foil_dt if hasattr(target, "two_way_foil_dt") \ else 0.003 * self.two_way_foil_dt_factor # current_time = time.time() # if current_time - self.two_way_foil.last_time >= 0.1: # dt = dt * 2 # self.two_way_foil.last_time = current_time # target.two_way_foil_dt = dt self.two_way_foil.planet.position = \ lerp(self.two_way_foil.planet.position, target.planet.position, dt) else: self.two_way_foil.planet.enabled = False self.two_way_foil.planet.update = two_way_foil_update # 圆形二向箔显示并设置透明度为0.8 self.two_way_foil_circle.planet.alpha = 0.8 self.two_way_foil_circle.planet.enabled = True # 当前阶段为 stage_02：二向箔压平天体的阶段 self.current_stage = self.stage_02 def stage_02(self): """ 二向箔压平天体（二维化）的阶段 @return: """ self.two_way_foil.planet.enabled = False # self.sun.two_dim.planet.init_scale += 0.05 # 圆形二向箔不断扩展变大 self.two_way_foil_circle.planet.init_scale += 0.8 # 调整天体二维化的时间，一般需要延时，保证扩展和二维化同步的真实效果 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.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_way_time = None elif two_bodies_colliding(self.two_way_foil_circle, b): # 二向箔和天体碰撞，但暂时先不二维化，记下时间，延时二维化 b.two_way_time = time.time() # 圆形二向箔不断旋转的效果 self.two_way_foil_circle.planet.rotation_z += 0.4 def on_ready(self): """ 事件绑定后，模拟器运行前会触发 @return: """ # 初始化设置 self.init_settings() # self.set_window_size((int(1920 * r), int(1080 * r)), False) # 最大分辨率的高度-1，保证不全屏 # self.set_window_size((1920, 1079), False) application.time_scale = 0.00001 # 显示网格以及坐标线 # self.show_grid_axises() self.two_way_foil.planet.alpha = 0.95 # 圆形二向箔初始化（一开始不显示） self.two_way_foil_circle.planet.rotation_x = 90 self.two_way_foil_circle.planet.enabled = False self.two_way_foil.scale_factor = 0.6 def change_two_way_foil(): # 长方形膜状物，长八点五厘米，宽五点二厘米，比一张信用卡略大一些，极薄，看不出任何厚度。 # 封装状态下晶莹剔透、无色透明。待机模式下由于封装力场逐渐蒸发，会发出白光，使其表面呈纯白色，看上去就是一张纸条 self.two_way_foil.planet.scale_x = 8.5 * self.two_way_foil.scale_factor self.two_way_foil.planet.scale_y = 5.2 * self.two_way_foil.scale_factor # self.two_way_foil.scale_factor -= 1e-4 ext_fun_for_method(self.two_way_foil.planet, after_run_fun=change_two_way_foil) def on_timer_changed(self, time_data): """ @param time_data: @return: """ # 原始方形二向箔飞行的翻转效果 if self.two_way_foil.planet.enabled: self.two_way_foil.planet.rotation_x += 0.01 self.two_way_foil.planet.rotation_y += 0.5 # dt = time_data.get_datetime(self.start_time) # # 摄像机看向哈雷彗星 camera_look_at(self.two_way_foil, rotation_z=0) # camera.look_at_2d(self.halley_comet.planet.position, axis='y') time_total_hours = time_data.total_hours # if not hasattr(self, "last_total_hours"): # self.last_total_hours = time_total_hours # self.update_halley_comet_info(dt) # print("time_total_hours", time_total_hours) # # 摄像机移动每10小时更新一次 # self.interval_run(fun=self.camera_move, fun_args=[dt], # total_times=time_total_hours, # interval=20) self.camera_move(time_data) self.current_stage() if __name__ == '__main__': """ """ params = TwoWayFoilSim() sim = TwoWayFoilSim(params) 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 / 48 * get_run_speed_factor(), # position=(0, 2 * AU, -11 * AU), # position=(0, 0.5 * AU, -5 * AU), # position=(2 * AU, -5 * AU, -20 * AU), position=(2 * AU, 2 * AU, -10 * AU), cosmic_bg='', # show_trail=True, # bg_music='sounds/no_glory.mp3', show_camera_info=False, # video_recoder=True, show_control_info=False, timer_enabled=True, show_timer=True, show_grid=False )