# -*- coding:utf-8 -*- # title :三体二向箔场景模拟 # description :三体二向箔场景模拟 # author :Python超人 # date :2023-10-11 # link :https://gitcode.net/pythoncr/ # python_version :3.9 # ============================================================================== import time 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 from common.consts import SECONDS_PER_WEEK, AU from objs import QuadObj, CircleObj, Obj from sim_scenes.func import camera_look_at, two_bodies_colliding from sim_scenes.func import ursina_run, create_sphere_sky from simulators.ursina.entities.entity_utils import create_rings from simulators.ursina.ursina_event import UrsinaEvent class TwoWayFoilSim: """ 三体二向箔场景模拟 """ def __init__(self): # self.three_dim_bodies = [] # self.two_dim_bodies = [] self.current_stage = self.stage_01 def build_solar_system(self): # region 构建太阳系 self.sun = Sun(size_scale=0.8e2) self.bodies = [ self.sun, # 太阳放大 80 倍 Mercury(size_scale=4e3, distance_scale=1.3), # 水星放大 4000 倍，距离放大 1.3 倍 Venus(size_scale=4e3, distance_scale=1.3), # 金星放大 4000 倍，距离放大 1.3 倍 Earth(size_scale=4e3, distance_scale=1.3, rotate_angle=0), # 地球放大 4000 倍，距离放大 1.3 倍 Mars(size_scale=4e3, distance_scale=1.2), # 火星放大 4000 倍，距离放大 1.2 倍 Jupiter(size_scale=0.68e3, distance_scale=0.72), # 木星放大 680 倍，距离缩小到真实距离的 0.72 Saturn(size_scale=0.68e3, distance_scale=0.52), # 土星放大 680 倍，距离缩小到真实距离的 0.52 Uranus(size_scale=1.5e3, distance_scale=0.36), # 天王星放大 1500 倍，距离缩小到真实距离的 0.36 Neptune(size_scale=1.5e3, distance_scale=0.27), # 海王星放大 1500 倍，距离缩小到真实距离的 0.27 Pluto(size_scale=10e3, distance_scale=0.23), # 冥王星放大 10000 倍，距离缩小到真实距离的 0.23(从太阳系的行星中排除) ] # endregion init_pos_vels = get_init_pos_vels() for body in self.bodies: pos_vels = init_pos_vels.get(type(body).__name__.lower(), None) if pos_vels is None: continue body.init_position = pos_vels['pos'] body.init_velocity = pos_vels['vel'] def build_two_way_foil(self): self.two_way_foil = QuadObj(texture='two_way_foil.png', # size_scale=4e7, size_scale=1e7, init_velocity=[0, -6, 150], init_position=[0, 0.5 * AU, -10 * AU]) \ .set_light_disable(True).set_ignore_gravity(True) 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 create_two_dim(self, body): # body_2dim = CircleObj(texture=body.texture, # size_scale=body.diameter * body.size_scale, # distance_scale=body.distance_scale, # init_position=body.position) \ # .set_light_disable(True).set_ignore_gravity(True) # # body.two_dim = body_2dim # body_2dim.three_dim = body # return body_2dim # def build_two_dim_bodies(self): # for body in self.bodies: # # two_dim_body = self.create_two_dim(body) # self.three_dim_bodies.append(body) # # self.two_dim_bodies.append(two_dim_body) # # self.bodies.append(self.two_way_foil) # self.bodies.append(self.two_way_foil_circle) # # for body in self.two_dim_bodies: # self.bodies.append(body) def build(self): self.build_solar_system() self.build_two_way_foil() # self.build_two_dim_bodies() def on_ready(self): """ 事件绑定后，模拟器运行前会触发 @return: """ # 创建天空 # camera.clip_plane_near = 0.1 camera.clip_plane_far = 1000000 create_sphere_sky(scale=200000) application.time_scale = 5 # for body in self.two_dim_bodies: # body.planet.enabled = False # model = body.planet.model # if body.three_dim.has_rings: # rings = create_rings(body.planet) # rings.scale *= 2e10 self.two_way_foil_circle.planet.rotation_x = 90 self.two_way_foil_circle.planet.enabled = False # self.two_way_foil.planet.alpha = 0.6 # for b in self.two_dim_bodies: # b.planet.rotation_x = 90 def flatten_animation(self, body): def flatten_update(planet): def warp(): planet.init_update() # time.sleep(0.2) if planet.scale_y > planet.init_scale_y / 20: planet.scale_y_v /= 1.01 # planet.enabled = False # flatten_planet.enabled = True # planet.update = planet.init_update # elif planet.scale_y < planet.init_scale_y / 1.01: # planet.set_light_off(True) planet.set_light_off(True) planet.scale_y = planet.scale_y_v return warp body.planet.rotation_speed /= 20 body.planet.scale_y_v = body.planet.scale_y body.planet.init_scale_y = body.planet.scale_y body.planet.init_update = body.planet.update body.planet.update = flatten_update(body.planet) # for i in range(10): # # body.planet.scale_y /= 2 # body.init_scale = [body.init_scale[0], body.init_scale[1] / 2, body.init_scale[2]] # time.sleep(0.2) # body.planet.enabled = False # flatten_body.planet.enabled = True # flatten_body.planet.x += 100000 def stage_01(self): """ 二向箔飞向太阳 @return: """ if two_bodies_colliding(self.two_way_foil, self.sun): # t = threading.Thread(target=self.flatten_animation, args=[self.sun, self.sun.two_dim]) # t.start() # for i in range(10): # self.sun.planet.scale_y /= 0.5 # time.sleep(1) # self.sun.planet.enabled = False # self.sun.two_dim.planet.enabled = True # self.sun.two_dim.planet.x += 100000 # two_way_foil.explode(sun) self.two_way_foil.planet.enabled = False self.two_way_foil_circle.planet.alpha = 0.9 self.two_way_foil_circle.planet.enabled = True self.current_stage = self.stage_02 def blink(self, body): return if not hasattr(body, "blink_d"): body.blink_d = 1 body.blink_v = 1 if body.blink_d > 0: body.blink_v -= 0.05 body.planet.alpha = body.blink_v if body.blink_v <= 0.7: body.blink_d = -1 elif body.blink_d < 0: body.blink_v += 0.05 body.planet.alpha = body.blink_v if body.blink_v >= 1: body.blink_d = 1 def stage_02(self): """ @return: """ # self.sun.two_dim.planet.init_scale += 0.05 self.two_way_foil_circle.planet.init_scale += 0.8 # self.blink(self.sun.two_dim) # 二维化延时的时间 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.planet.enabled = False # b.two_dim.planet.enabled = True b.two_way_time = None elif two_bodies_colliding(self.two_way_foil_circle, b): b.two_way_time = time.time() # b.planet.enabled = False # b.two_dim.planet.enabled = True # if b.two_dim.planet.enabled: # # b.two_dim.planet.init_scale += 0.05 # self.blink(b.two_dim) self.two_way_foil_circle.planet.rotation_z += 0.4 def on_timer_changed(self, time_data): self.two_way_foil.planet.rotation_x += 0.1 self.two_way_foil.planet.rotation_y += 1 # self.two_way_foil.planet.rotation_z += 10 camera_look_at(self.two_way_foil) self.current_stage() # for b in self.two_dim_bodies: # b.planet.rotation_y -= 0.5 # b.init_position = b.three_dim.position if __name__ == '__main__': """ 三体二向箔场景模拟 """ sim = TwoWayFoilSim() 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_WEEK, # position=(0, 2 * AU, -11 * AU), position=(0, 0.5 * AU, -5 * AU), cosmic_bg='', bg_music='sounds/no_glory.mp3', show_camera_info=False, show_control_info=False, timer_enabled=True, show_grid=False)