diff --git a/sim_scenes/science/comets_jupiter.py b/sim_scenes/science/comets_jupiter.py new file mode 100644 index 0000000000000000000000000000000000000000..c00ec3dcefa4efbcfa9cd8ad7284af9213e2922f --- /dev/null +++ b/sim_scenes/science/comets_jupiter.py @@ -0,0 +1,105 @@ +# -*- coding:utf-8 -*- +# title :彗木相撞模拟 +# description :天象奇观：彗木相撞模拟 +# author :Python超人 +# date :2023-04-09 +# link :https://gitcode.net/pythoncr/ +# python_version :3.8 +# ============================================================================== +from bodies import Jupiter, Body +from objs import RockSnow, Rock, create_rock +from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY +from sim_scenes.func import ursina_run, camera_look_at, two_bodies_colliding +from simulators.ursina.entities.body_timer import TimeData +from simulators.ursina.entities.entity_utils import create_directional_light +from simulators.ursina.ursina_event import UrsinaEvent + + +def create_comet(index, gravity_only_for): + """ + 随机生成石头 + @param index: 索引号 + @param gravity_only_for: 指定一个天体，石头的引力只对该天体有效 + @return: + """ + # 随机生成石头的位置和初始速度信息 + pos = [-r * random.randint(120, 200) / 100, + -r * random.randint(120, 200) / 1000, + -r * random.randint(100, 300) / 100] + # 随机速度 + vel = [0, -random.randint(90, 200) / 30, 0] + # 石头随机大小 + size_scale = random.randint(600, 1200) + # 随机创建石头 + comet = create_rock(no=index % 8 + 1, name=f'岩石{index + 1}', mass=4.4e10, + size_scale=size_scale, color=(255, 200, 0), + init_position=pos, init_velocity=vel, + gravity_only_for=[gravity_only_for], + ) + + # 给石头一个随机旋转的方向和值 + comet.rotation = [0, 0, 0] + comet.rotation[random.randint(0, 2)] = random.randint(90, 200) / 100 + return comet + + +if __name__ == '__main__': + """ + 彗木相撞模拟 + """ + import random + + # 木星在中心位置 + jupiter = Jupiter(init_position=[0, 0, 0], init_velocity=[0, 0, 0], + size_scale=1, texture="jupiter_hd.jpg") + + bodies = [jupiter] + comets = [] + r = jupiter.raduis + + for i in range(30): + # 随机生成石头 + comet = create_comet(i, gravity_only_for=jupiter) + bodies.append(comet) + comets.append(comet) + + + def on_reset(): + # 重置后恢复所有石头的状态（石头可见、引力有效） + for comet in comets: + comet.set_visible(True) + comet.ignore_mass = False + + + def on_ready(): + # 运行前触发 + # 创建太阳光 + create_directional_light(position=(200, 0, -300), target=jupiter) + # 摄像机看向木星 + camera_look_at(jupiter, rotation_z=0) + + + def on_timer_changed(time_data: TimeData): + # 运行中，每时每刻都会触发 + for comet in comets: + if comet.visibled: + # 如果是否可见，则旋转石头 + comet.planet.rotation += comet.rotation + # 循环判断每个石头与木星是否相碰撞，如果相碰撞就爆炸 + if two_bodies_colliding(comet, jupiter): + # 将石头隐藏、设置引力无效后，展示爆炸效果 + comet.explode(jupiter) + + + # 订阅事件 + UrsinaEvent.on_reset_subscription(on_reset) # 重置了会触发 on_reset + UrsinaEvent.on_ready_subscription(on_ready) # 运行前会触发 on_ready + UrsinaEvent.on_timer_changed_subscription(on_timer_changed) # 运行中，每时每刻都会触发 on_timer_changed + + # 使用 ursina 查看的运行效果 + # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 + # position = 左-右+、上+下-、前+后- + ursina_run(bodies, SECONDS_PER_HOUR / 10, + position=(0, 0, -300000), + show_timer=True, + view_closely=0.001) diff --git a/sim_scenes/science/jupiter_rocks.py b/sim_scenes/science/jupiter_rocks.py deleted file mode 100644 index 516d92670ef8dd311526501db871d87b44aaa36f..0000000000000000000000000000000000000000 --- a/sim_scenes/science/jupiter_rocks.py +++ /dev/null @@ -1,81 +0,0 @@ -# -*- coding:utf-8 -*- -# title :彗木相撞模拟 -# description :天象奇观：彗木相撞模拟 -# author :Python超人 -# date :2023-04-09 -# link :https://gitcode.net/pythoncr/ -# python_version :3.8 -# ============================================================================== -from bodies import Jupiter, Body -from objs import RockSnow, Rock, create_rock -from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY -from sim_scenes.func import ursina_run, camera_look_at, two_bodies_colliding -from simulators.ursina.entities.body_timer import TimeData -from simulators.ursina.ursina_event import UrsinaEvent - -if __name__ == '__main__': - """ - 彗木相撞模拟 - """ - import random - - # 木星在中心位置 - jupiter = Jupiter(init_position=[0, 0, 0], init_velocity=[0, 0, 0], - size_scale=1, texture="jupiter_hd.jpg") - - bodies = [jupiter] - rocks = [] - r = jupiter.raduis - - for i in range(30): - # 随机生成岩石位置和初始速度信息 - pos = [-r * random.randint(120, 200) / 100, - -r * random.randint(120, 200) / 1000, - -r * random.randint(100, 300) / 100] - # 随机速度 - vel = [0, -random.randint(90, 200) / 30, 0] - size_scale = random.randint(600, 1200) - rock = create_rock(no=i % 8 + 1, name=f'岩石{i + 1}', mass=4.4e10, - size_scale=size_scale, color=(255, 200, 0), - init_position=pos, init_velocity=vel, - gravity_only_for=[jupiter], - ) - # 岩石随机旋转量 - rock.rotation = [0, 0, 0] - rock.rotation[random.randint(0, 2)] = random.randint(90, 200) / 100 - - bodies.append(rock) - rocks.append(rock) - - - def on_reset(): - for rock in rocks: - rock.set_visible(True) - rock.ignore_mass = False - - - def on_ready(): - camera_look_at(jupiter, rotation_z=0) - - - def on_timer_changed(time_data: TimeData): - for rock in rocks: - if rock.visibled: - rock.planet.rotation += rock.rotation - # 循环判断每个抛出物与木星是否相碰撞 - if two_bodies_colliding(rock, jupiter): - # 岩石爆炸 - rock.explode(jupiter) - - - UrsinaEvent.on_reset_subscription(on_reset) - UrsinaEvent.on_ready_subscription(on_ready) - UrsinaEvent.on_timer_changed_subscription(on_timer_changed) - - # 使用 ursina 查看的运行效果 - # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 - # position = 左-右+、上+下-、前+后- - ursina_run(bodies, SECONDS_PER_HOUR / 10, - position=(0, 0, -300000), - show_timer=True, - view_closely=0.001) diff --git a/simulators/ursina/entities/entity_utils.py b/simulators/ursina/entities/entity_utils.py index 617540a2142d05e9faab5b6ce184f993440f3e36..52d37e8ffc197d508b82e434dc2c436fa2a6bdf3 100644 --- a/simulators/ursina/entities/entity_utils.py +++ b/simulators/ursina/entities/entity_utils.py @@ -9,7 +9,7 @@ # pip install -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com ursina from ursina import Ursina, window, Entity, Mesh, SmoothFollow, Texture, clamp, time, \ camera, color, mouse, Vec2, Vec3, Vec4, Text, \ - load_texture, held_keys, destroy, PointLight, distance + load_texture, held_keys, destroy, PointLight, DirectionalLight, distance from simulators.ursina.entities.body_trail import BodyTrail, BodyTrailLine from simulators.ursina.ursina_config import UrsinaConfig @@ -37,7 +37,7 @@ def create_name_text(parent): if text_color is None: text_color = get_inverse_color(b_color) else: - text_color = (text_color[0]/255, text_color[1]/255, text_color[2]/255) + text_color = (text_color[0] / 255, text_color[1] / 255, text_color[2] / 255) name_text.set_light_off() name_text.color = color.rgba(text_color[0], text_color[1], text_color[2], 1) parent.name_text = name_text @@ -159,6 +159,33 @@ def create_trail_sphere(parent, pos): # return value, direction +def create_directional_light(position, target=None, shadows=False, light_color=None): + """ + 创建平行光(DirectionalLight) + @param position: 光源位置 + @param target: 光源指向目标 + @param shadows: 是否支持阴影 + @param light_color: 光的颜色 + @return: + """ + from ursina.shaders import lit_with_shadows_shader + if shadows: + Entity.default_shader = lit_with_shadows_shader + if light_color is None: + light_color = color.white + else: + light_color = color.rgba(light_color[0] / 255, light_color[1] / 255, light_color[2] / 255, 1) + light = DirectionalLight(position=position, intensity=10, range=10, color=light_color) + if target is not None: + if hasattr(target, "planet"): + if hasattr(target.planet, "main_entity"): + light.look_at(target.planet.main_entity) + else: + light.look_at(target.planet) + else: + light.look_at(target) + + def create_trail_line(parent, pos): """ 在天体当前的位置创建一个拖尾球体 diff --git a/simulators/ursina_simulator.py b/simulators/ursina_simulator.py index 8183e5c799abafe1eeb7b03c2a0450b91cf5d663..5210d3a2f2cca074c93fc557235d2d093c856215 100644 --- a/simulators/ursina_simulator.py +++ b/simulators/ursina_simulator.py @@ -105,6 +105,7 @@ class UrsinaSimulator(Simulator): position=body.planet.position, scale=scale, fps=6, loop=False, autoplay=True) + explode_ani.set_light_off() if target is not None: if hasattr(target, "planet"):