# -*- coding:utf-8 -*- # title :三体二向箔场景模拟 # description :三体二向箔场景模拟 # author :Python超人 # date :2023-10-11 # link :https://gitcode.net/pythoncr/ # python_version :3.9 # ============================================================================== import time import datetime from bodies import Sun, Earth, Moon from objs import CoreValagaClas, SciFiBomber, WaterDrop, QuadObj, CircleObj from common.consts import AU, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH from sim_scenes.func import ursina_run, camera_look_at, create_main_entity, two_bodies_colliding from simulators.ursina.entities.body_timer import TimeData from simulators.ursina.entities.entity_utils import create_directional_light, create_rings from simulators.ursina.ursina_config import UrsinaConfig from simulators.ursina.ursina_event import UrsinaEvent from ursina import camera, application from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Asteroids from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_MONTH, SECONDS_PER_YEAR, AU from sim_scenes.func import mayavi_run, ursina_run, create_sphere_sky 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 get_init_pos_vels(self): init_pos_vels = {} init_pos_vels['sun'] = {} init_pos_vels['sun']['pos'] = [-4638653.0000, 0.0000, 1314332.6250] init_pos_vels['sun']['vel'] = [-0.0165, 0.0000, -0.0097] init_pos_vels['mercury'] = {} init_pos_vels['mercury']['pos'] = [-54412616.0000, 0.0000, 31280100.0000] init_pos_vels['mercury']['vel'] = [-25.2667, 0.0000, -40.0561] init_pos_vels['venus'] = {} init_pos_vels['venus']['pos'] = [-32341672.0000, 0.0000, -99861352.0000] init_pos_vels['venus']['vel'] = [34.3770, 0.0000, -10.6676] init_pos_vels['earth'] = {} init_pos_vels['earth']['pos'] = [140427392.0000, 0.0000, 45724572.0000] init_pos_vels['earth']['vel'] = [-8.1280, 0.0000, 28.2277] init_pos_vels['mars'] = {} init_pos_vels['mars']['pos'] = [218751648.0000, 0.0000, -45263688.0000] init_pos_vels['mars']['vel'] = [5.2205, 0.0000, 23.4601] init_pos_vels['asteroids'] = {} init_pos_vels['asteroids']['pos'] = [0.0000, 0.0000, 0.0000] init_pos_vels['asteroids']['vel'] = [0.0000, 0.0000, 0.0000] init_pos_vels['jupiter'] = {} init_pos_vels['jupiter']['pos'] = [769445760.0000, 0.0000, 2017638.5000] init_pos_vels['jupiter']['vel'] = [0.0389, 0.0000, 13.0879] init_pos_vels['saturn'] = {} init_pos_vels['saturn']['pos'] = [-1330536704.0000, 0.0000, -454165280.0000] init_pos_vels['saturn']['vel'] = [3.2189, 0.0000, -9.1974] init_pos_vels['uranus'] = {} init_pos_vels['uranus']['pos'] = [-1757163648.0000, 0.0000, 2272503808.0000] init_pos_vels['uranus']['vel'] = [-5.3918, 0.0000, -4.1634] init_pos_vels['neptune'] = {} init_pos_vels['neptune']['pos'] = [-1480893824.0000, 0.0000, 4351772160.0000] init_pos_vels['neptune']['vel'] = [-5.1466, 0.0000, -1.7189] init_pos_vels['pluto'] = {} init_pos_vels['pluto']['pos'] = [-1293909504.0000, 0.0000, 5770579968.0000] init_pos_vels['pluto']['vel'] = [-4.5847, 0.0000, -1.0475] return init_pos_vels 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), # 地球放大 4000 倍，距离放大 1.3 倍 Mars(size_scale=4e3, distance_scale=1.2), # 火星放大 4000 倍，距离放大 1.2 倍 # Asteroids(size_scale=1e2, parent=self.sun), # 小行星模拟(仅 ursina 模拟器支持) 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=0.8e3, distance_scale=0.36), # 天王星放大 800 倍，距离缩小到真实距离的 0.36 Neptune(size_scale=1e3, distance_scale=0.27), # 海王星放大 1000 倍，距离缩小到真实距离的 0.27 Pluto(size_scale=10e3, distance_scale=0.23), # 冥王星放大 10000 倍，距离缩小到真实距离的 0.23(从太阳系的行星中排除) ] # endregion init_pos_vels = self.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) 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 stage_01(self): """ 二向箔飞向太阳 @return: """ if two_bodies_colliding(self.two_way_foil, self.sun): 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, 5.0, 6.5] # 天王星 海王星 冥王星 for idx, b in enumerate(self.three_dim_bodies): 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]: 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 += 0.8 # 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)