From fc8c8bdab99dba27b2cf38b9f1023067aa9a2bdb Mon Sep 17 00:00:00 2001 From: march3 Date: Wed, 11 Oct 2023 14:58:44 +0800 Subject: [PATCH] =?UTF-8?q?Python=E8=B6=85=E4=BA=BA-=E5=AE=87=E5=AE=99?= =?UTF-8?q?=E6=A8=A1=E6=8B=9F=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- objs/__init__.py | 1 + objs/obj.py | 10 ++- objs/quad_obj.py | 48 ++++++++++++++ sim_scenes/tri_bodies/two_way_foil.py | 92 +++++++++++++++++++++++++++ 4 files changed, 149 insertions(+), 2 deletions(-) create mode 100644 objs/quad_obj.py create mode 100644 sim_scenes/tri_bodies/two_way_foil.py diff --git a/objs/__init__.py b/objs/__init__.py index 0addc77..a69ce88 100644 --- a/objs/__init__.py +++ b/objs/__init__.py @@ -1,5 +1,6 @@ from objs.obj import Obj from objs.diamond import Diamond +from objs.quad_obj import QuadObj from objs.football import Football from objs.satellite import Satellite, Satellite2 from objs.space_ship import SpaceShip diff --git a/objs/obj.py b/objs/obj.py index b3ea346..6614e03 100644 --- a/objs/obj.py +++ b/objs/obj.py @@ -51,7 +51,10 @@ class Obj(metaclass=ABCMeta): self.__his_acc = [] self.__his_reserved_num = 200 self.gravity_only_for = gravity_only_for - self.model = self.find_model(model) + if isinstance(model, str): + self.model = self.find_model(model) + else: + self.model = model self.rotation = rotation if name is None: @@ -350,7 +353,9 @@ class Obj(metaclass=ABCMeta): def __repr__(self): return '<%s(%s):%s(%s)> m=%.3e(kg), d=%.3e(kg/m³), p=[%.3e,%.3e,%.3e](km), v=%s(km/s), s=%.3e' % \ - (self.name, self.__class__.__name__, os.path.basename(self.model), os.path.basename(self.texture), + (self.name, self.__class__.__name__, + os.path.basename(self.model) if isinstance(self.model, str) else self.model, + os.path.basename(self.texture) if isinstance(self.texture, str) else self.texture, self.mass, self.density, self.position[0], self.position[1], self.position[2], self.velocity, self.size_scale) @@ -485,6 +490,7 @@ class Obj(metaclass=ABCMeta): self.size_scale = size_scale # self.init_position = np.array([0, -3000000, 9000000]) self.init_position = np.array([0, 0, 0]) + def on_ready(): camera_look_at(self) if callable(on_ready_fun): diff --git a/objs/quad_obj.py b/objs/quad_obj.py new file mode 100644 index 0000000..9c43ee2 --- /dev/null +++ b/objs/quad_obj.py @@ -0,0 +1,48 @@ +# -*- coding:utf-8 -*- +# title :四边形 +# description :四边形 +# author :Python超人 +# date :2023-10-11 +# link :https://gitcode.net/pythoncr/ +# python_version :3.9 +# ============================================================================== +from objs.obj import Obj + + +class QuadObj(Obj): + """ + 四边形 + 密度：3.51g/cm³ + """ + + def __init__(self, name="四边形", mass=5.97237e24, + init_position=[0, 0, 0], + init_velocity=[0, 0, 0], + texture=None, size_scale=1.0, distance_scale=1.0, + ignore_mass=False, density=3.51e3, color=(7, 0, 162), + trail_color=None, show_name=False, + parent=None, gravity_only_for=[]): + from ursina.models.procedural.quad import Quad + params = { + "name": name, + "mass": mass, + "init_position": init_position, + "init_velocity": init_velocity, + "density": density, + "color": color, + "texture": texture, + "size_scale": size_scale, + "distance_scale": distance_scale, + "ignore_mass": ignore_mass, + "trail_color": trail_color, + "show_name": show_name, + "parent": parent, + "gravity_only_for": gravity_only_for, + "model": Quad(0), + } + super().__init__(**params) + + +if __name__ == '__main__': + quad = QuadObj() + print(quad) diff --git a/sim_scenes/tri_bodies/two_way_foil.py b/sim_scenes/tri_bodies/two_way_foil.py new file mode 100644 index 0000000..c23d5f3 --- /dev/null +++ b/sim_scenes/tri_bodies/two_way_foil.py @@ -0,0 +1,92 @@ +# -*- 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 +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 +from simulators.ursina.ursina_config import UrsinaConfig +from simulators.ursina.ursina_event import UrsinaEvent +from ursina import camera + +if __name__ == '__main__': + """ + 三体二向箔场景模拟 + """ +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__': + # 八大行星：木星(♃)、土星(♄)、天王星(♅)、海王星(♆)、地球(⊕)、金星(♀)、火星(♂)、水星(☿) + # 排列顺序 + # 1、体积：(以地球为1)木星 ：土星 ：天王星 ：海王星 ：地球 ：金星 ：火星 ：水星 = 1330：745：65：60：1：0.86：0.15：0.056 + # 2、质量：(以地球为1)木星 ：土星 ：天王星 ：海王星 ：地球 ：金星 ：火星 ：水星 = 318：95：14.53：17.15：1：0.8：0.11：0.0553 + # 3、离太阳从近到远的顺序：水星、金星、地球、火星、木星、土星、天王星、海王星 + # ===================================================================== + # 以下展示的效果非太阳系真实的距离和大小 + # 1、由于宇宙空间尺度非常大，如果按照实际的天体大小，则无法看到天体，因此需要对天体的尺寸进行放大 + # 2、为了达到最佳的显示效果，对每个行星天体的距离进行了缩放 + + # 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=2e8, + init_velocity=[-10, -10, 0], + init_position=[0, 0, 39.55 * AU]) \ + .set_light_disable(True).set_ignore_gravity(True) + + bodies.append(two_way_foil) + + + def on_ready(): + """ + 事件绑定后，模拟器运行前会触发 + @return: + """ + # 创建天空 + create_sphere_sky(scale=8000) + + + def on_timer_changed(time_data): + two_way_foil.planet.rotation_x += 1 + two_way_foil.planet.rotation_y += 2 + 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, position=(0, 2 * AU, -11 * AU), + cosmic_bg='', + timer_enabled=True, + show_grid=False) -- GitLab