diff --git a/sim_scenes/featured/two_way_foil.py b/sim_scenes/featured/two_way_foil_sim.py similarity index 63% rename from sim_scenes/featured/two_way_foil.py rename to sim_scenes/featured/two_way_foil_sim.py index f228cb437e94d53fe1dda8fe4be915aaa4346ee2..58aa468d3cc88f459e7414356682af57123ec54f 100644 --- a/sim_scenes/featured/two_way_foil.py +++ b/sim_scenes/featured/two_way_foil_sim.py @@ -2,24 +2,33 @@ # title :三体二向箔场景模拟 # description :三体二向箔场景模拟 # author :Python超人 -# date :2023-10-27 +# date :2024-01-20 # link :https://gitcode.net/pythoncr/ # python_version :3.9 # ============================================================================== -import time - -from ursina import camera, application, lerp +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 get_init_pos_vels, init_bodies_pos_vels -from common.consts import SECONDS_PER_WEEK, AU -from common.image_utils import resize_image +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 camera_look_at, two_bodies_colliding, camera_move_control, get_run_speed_factor +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.ursina.ursina_event import UrsinaEvent 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): @@ -27,33 +36,36 @@ class TwoWayFoilSim(UniverseSimScenes): 三体二向箔场景模拟 """ - def __init__(self): - self.current_stage = self.stage_01 - def build_solar_system(self): # region 构建太阳系 - self.sun, self.mercury, self.venus, self.earth, self.mars, self.jupiter, \ - self.saturn, self.uranus, self.neptune, self.pluto = \ - Sun(size_scale=0.8e2, texture="sun_light.jpg"), \ - Mercury(size_scale=2e3, distance_scale=1.5), \ - Venus(size_scale=2e3, distance_scale=1.5), \ - Earth(size_scale=2e3, distance_scale=1.5, rotate_angle=0), \ - Mars(size_scale=2e3, distance_scale=1.4), \ - Jupiter(size_scale=0.3e3, distance_scale=0.72), \ - Saturn(size_scale=0.3e3, distance_scale=0.52), \ - Uranus(size_scale=0.8e3, distance_scale=0.36), \ - Neptune(size_scale=0.8e3, distance_scale=0.27), \ - Pluto(size_scale=10e3, distance_scale=0.23) - + 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.sun, # 太阳 + self.mercury, # 水星 + self.venus, # 金星 + self.earth, # 地球 + self.mars, # 火星 + self.jupiter, # 木星 + self.saturn, # 土星 + self.uranus, # 天王星 + self.neptune, # 海王星 + self.pluto, # 冥王星 ] - # endregion - # self.sun.glows = (2, 1.005, 0.01) - # # glows = (glow_num:10, glow_scale:1.03 glow_alpha:0.1~1) - # self.sun.glows = (0, 1.005, 0.1, 160) - self.sun.glows = (2, 1.005, 0.01) self.two_way_foil_dt_factor = 1 self.sun.two_way_foil_dt = 0.003 * self.two_way_foil_dt_factor @@ -62,6 +74,37 @@ class TwoWayFoilSim(UniverseSimScenes): 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): """ 创建二向箔（一个原始的方形二向箔，一个不断扩展的圆形二向箔） @@ -70,10 +113,10 @@ class TwoWayFoilSim(UniverseSimScenes): # 原始的方形二向箔 self.two_way_foil = QuadObj(texture='two_way_foil.png', # size_scale=4e7, - size_scale=1e7, + size_scale=1e6, init_velocity=[0, -50, 60], init_position=[0, 5 * AU, -6 * AU]) \ - .set_light_disable(True).set_ignore_gravity(True) + .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, @@ -83,70 +126,51 @@ class TwoWayFoilSim(UniverseSimScenes): self.bodies.append(self.two_way_foil) self.bodies.append(self.two_way_foil_circle) - def build(self): - self.build_solar_system() - self.build_two_way_foil() + def s_f(self, value=1): + if value == 0: + return 0 + return get_run_speed_factor() * value - def on_ready(self): + def camera_move(self, time_data): """ - 事件绑定后，模拟器运行前会触发 + 摄像机移动控制 + @param dt: @return: """ - # 创建天空 - # camera.clip_plane_near = 0.1 - camera.clip_plane_far = 1000000 - create_sphere_sky(scale=200000) - application.time_scale = 5 - # 圆形二向箔初始化（一开始不显示） - self.two_way_foil_circle.planet.rotation_x = 90 - self.two_way_foil_circle.planet.enabled = False - - def change_two_way_foil(): - # 长方形膜状物，长八点五厘米，宽五点二厘米，比一张信用卡略大一些，极薄，看不出任何厚度。 - # 封装状态下晶莹剔透、无色透明。待机模式下由于封装力场逐渐蒸发，会发出白光，使其表面呈纯白色，看上去就是一张纸条 - self.two_way_foil.planet.scale_x = 8.5 - self.two_way_foil.planet.scale_y = 5.2 - - glow_circle = self.sun.planet.glow_circle - fixed_star_ratio = glow_circle.scale_x / self.sun.planet.scale_x - - def glow_circle_update(): - glow_circle.enabled = self.sun.planet.enabled - glow_circle.position = self.sun.planet.position - glow_circle.scale_x = fixed_star_ratio * self.sun.planet.scale_y - if glow_circle.scale_x < 5: - glow_circle.update = lambda :None - glow_circle.enabled = False - glow_circle.scale_z = fixed_star_ratio * self.sun.planet.scale_x - # glow_circle.scale_z = fixed_star_ratio * fixed_star.scale_z - glow_circle.look_at(camera) - - glow_circle.update = glow_circle_update + # 摄像机移动控制数据 + 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}) + ] - ext_fun_for_method(self.two_way_foil.planet, after_run_fun=change_two_way_foil) + def cond_cb(ps): + return ps["next_cond"] > time_data.total_days >= ps["cond"] - def gen_pixcel_image(self, planet): - """ - 将纹理图改为像素图片 - @param planet: - @return: - """ - import os - if planet.texture is None: - return - in_img = str(planet.texture.path) - if "_pixcel" in in_img: - return - out_img = os.path.split(in_img) - out_file = out_img[1].split('.') - out_file = f'{out_file[0]}_pixcel.{out_file[1]}' - out_img = os.path.abspath(os.path.join(out_img[0], "temp", out_file)) - if not os.path.exists(out_img): - resize_image(in_img, out_img, 50, 25) - from ursina import Texture - planet.texture = Texture(out_img) - - return out_img + camera_move_control(camera_move_infos, + cond_cb=cond_cb, + value_conv=self.s_f, smooth=1) def flatten_animation(self, body): """ @@ -273,6 +297,35 @@ class TwoWayFoilSim(UniverseSimScenes): # 圆形二向箔不断旋转的效果 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): """ @@ -281,75 +334,38 @@ class TwoWayFoilSim(UniverseSimScenes): """ # 原始方形二向箔飞行的翻转效果 if self.two_way_foil.planet.enabled: - self.two_way_foil.planet.rotation_x += 0.1 - self.two_way_foil.planet.rotation_y += 1 - # # 摄像机始终看向二向箔 - # camera_look_at(self.two_way_foil) - - target = self.get_target() - if target is not None: - if target is self.sun: - # 摄像机始终看向二向箔 - camera_look_at(self.two_way_foil) - else: - # 摄像机始终看向二向箔 - camera_look_at(self.two_way_foil) - camera.position += camera.forward*10 + 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) - # camera.position = \ - # lerp(camera.position, target.planet.position, 0.001) - self.current_stage() - 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:下 - (1, {"l": 2}), - (10, {"f": 50}), - # (40, {"f": 80}), - # (1983, {"to": {"ct_id": 1, "t": 10}}), - # (2, {"x": 2, "y": -2, "z": -15}), - # (3, {"y": -6, "z": -12}), - # (4, {"y": -3, "z": -12}), - # (5, {"z": -8, "f": -5}), - # (6, {"z": -8, "f": -3}), - # (7, {"z": -8}), - # (8, {"z": -8, "y": -0.2}), - # (9, {}), - # (100, {"f": 10}), - # (200, {"f": 10}), - # (500, {"f": 10}), - # (2062, {"y": -3}), - # (2063, {"y": -10, "z": 2}), - (2181, {}), - # (2082, {"exit": True}) - ] - - camera_move_control(camera_move_infos, - cond_cb=lambda ps: ps["next_cond"] > time_data.total_days >= ps["cond"], - value_conv=self.s_f, smooth=None) - if __name__ == '__main__': """ - 三体二向箔场景模拟 + """ - sim = TwoWayFoilSim() + + params = TwoWayFoilSim() + + sim = TwoWayFoilSim(params) sim.build() + # 订阅事件后，上面2个函数功能才会起作用 # 运行中，每时每刻都会触发 on_timer_changed UrsinaEvent.on_timer_changed_subscription(sim.on_timer_changed) @@ -359,13 +375,18 @@ if __name__ == '__main__': # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- ursina_run(sim.bodies, - SECONDS_PER_WEEK, - position=(0, 2 * AU, -11 * AU), - # position=(0, 5 * AU, -5 * AU), + 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='', - bg_music='sounds/no_glory.mp3', + # 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) + show_grid=False + )