# -*- coding:utf-8 -*- # title :三体二向箔场景模拟 # description :三体二向箔场景模拟 # author :Python超人 # date :2023-10-11 # link :https://gitcode.net/pythoncr/ # python_version :3.9 # ============================================================================== 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 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 if __name__ == '__main__': """ 三体二向箔场景模拟 """ # region 构建太阳系 sun = Sun(size_scale=0.8e2) bodies = [ 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=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 two_way_foil = QuadObj(texture='caustic.jpg', size_scale=4e7, init_velocity=[0, 0, 100], init_position=[0, 0, -10*AU]) \ .set_light_disable(True).set_ignore_gravity(True) three_dim_bodies = [] two_dim_bodies = [] def create_two_dim(body): body_2 = 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_2 body_2.three_dim = body return body_2 for body in bodies: two_dim_body = create_two_dim(body) three_dim_bodies.append(body) two_dim_bodies.append(two_dim_body) bodies.append(two_way_foil) for body in two_dim_bodies: bodies.append(body) def on_ready(): """ 事件绑定后，模拟器运行前会触发 @return: """ # 创建天空 # camera.clip_plane_near = 0.1 camera.clip_plane_far = 1000000 create_sphere_sky(scale=200000) for body in 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 current_stage = None def stage_01(): """ 二向箔飞向太阳 @return: """ global current_stage if two_bodies_colliding(two_way_foil, sun): sun.planet.enabled = False sun.two_dim.planet.enabled = True # two_way_foil.explode(sun) two_way_foil.planet.enabled = False current_stage = stage_02 def blink(body): 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(): """ @return: """ global current_stage sun.two_dim.planet.init_scale += 1.1 blink(sun.two_dim) for b in three_dim_bodies: if two_bodies_colliding(sun.two_dim, b): b.planet.enabled = False b.two_dim.planet.enabled = True if b.two_dim.planet.enabled: b.two_dim.planet.init_scale += 1.1 blink(b.two_dim) current_stage = stage_01 def on_timer_changed(time_data): two_way_foil.planet.rotation_x += 1 two_way_foil.planet.rotation_y += 2 current_stage() # if sun.two_dim.planet.enabled: # sun.two_dim.planet.init_scale += 1.1 # for b in three_dim_bodies: # if two_bodies_colliding(sun.two_dim, b): # b.planet.enabled = False # b.two_dim.planet.enabled = True # if b.two_dim.planet.enabled: # b.two_dim.planet.init_scale += 1.1 # else: # if two_bodies_colliding(two_way_foil, sun): # sun.planet.enabled = False # sun.two_dim.planet.enabled = True # for b in three_dim_bodies: # b.planet.enabled = False for b in two_dim_bodies: b.planet.rotation_x = 90 b.init_position = b.three_dim.position print(b.planet.position, b.three_dim.planet.position) pass # 订阅事件后，上面2个函数功能才会起作用 # 运行中，每时每刻都会触发 on_timer_changed UrsinaEvent.on_timer_changed_subscription(on_timer_changed) # 运行前会触发 on_ready UrsinaEvent.on_ready_subscription(on_ready) # 使用 ursina 查看的运行效果 # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- ursina_run(bodies, SECONDS_PER_MONTH / 2, position=(0, 2 * AU, -11 * AU), cosmic_bg='', timer_enabled=True, show_grid=False)