From 464d2c885cb3c0bdd0427007d3ca5ef2ee29c64e Mon Sep 17 00:00:00 2001 From: march3 Date: Sat, 24 Jun 2023 18:10:06 +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 --- common/func.py | 9 +++++++-- sim_scenes/fiction/earth_orbit_stopped.py | 10 ++++++---- 2 files changed, 13 insertions(+), 6 deletions(-) diff --git a/common/func.py b/common/func.py index 5309e01..88ada0d 100644 --- a/common/func.py +++ b/common/func.py @@ -109,12 +109,17 @@ def calculate_distance(pos1, pos2=[0, 0, 0]): return d -def get_acceleration_info(acc_value): +def get_acceleration_info(acceleration): """ - @param acc_value: 加速度的值（km/s²） + @param acceleration: 加速度的值（km/s²） @return: """ + if isinstance(acceleration, list) or isinstance(acceleration, np.ndarray): + from simulators.ursina.entities.entity_utils import get_value_direction_vectors + acc_value, direction = get_value_direction_vectors(acceleration) + else: + acc_value = acceleration acc_m = acc_value * 1000 # 加速度的值（m/s²） diff --git a/sim_scenes/fiction/earth_orbit_stopped.py b/sim_scenes/fiction/earth_orbit_stopped.py index 0a8b0af..cde1d41 100644 --- a/sim_scenes/fiction/earth_orbit_stopped.py +++ b/sim_scenes/fiction/earth_orbit_stopped.py @@ -13,6 +13,7 @@ from sim_scenes.func import mayavi_run, ursina_run, camera_look_at, two_bodies_c create_text_panel from bodies.body import AU from simulators.ursina.entities.body_timer import BodyTimer, 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 @@ -89,7 +90,8 @@ class EarthOrbitStoppedSim: earth_pos = self.earth.planet.world_position camera.world_position = Vec3(earth_pos[0], earth_pos[1] + 0.01, earth_pos[2] - 0.1) - acceleration_info = get_acceleration_info(self.earth.acceleration[2]) + acceleration_info = get_acceleration_info(self.earth.acceleration) + velocity, _ = get_value_direction_vectors(self.earth.velocity) if two_bodies_colliding(self.sun, self.earth): self.arrived_sun = True @@ -98,7 +100,7 @@ class EarthOrbitStoppedSim: self.text_panel.text = self.arrived_info. \ replace("${distance}", "0 km"). \ replace("${acceleration}", "%s" % acceleration_info). \ - replace("${speed}", "%s km/s" % round(self.earth.velocity[2], 2)) \ + replace("${speed}", "%s km/s" % round(velocity, 2)) \ + "\n\n" + msg ControlUI.current_ui.show_message(msg, close_time=-1) application.pause() @@ -146,7 +148,7 @@ class EarthOrbitStoppedSim: self.text_panel.text = self.arrived_info. \ replace("${distance}", "%s km" % distance_str). \ replace("${acceleration}", "%s" % acceleration_info). \ - replace("${speed}", "%s km/s" % round(self.earth.velocity[2], 2)) + replace("${speed}", "%s km/s" % round(velocity, 2)) if __name__ == '__main__': @@ -183,6 +185,6 @@ if __name__ == '__main__': # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 # position = 左-右+、上+下-、前+后- # position=(0, 0, 0) 的位置是站在地球视角，可以观看月相变化的过程 - ursina_run(sim.bodies, SECONDS_PER_DAY, position=(0, 0.0001 * AU, -0.02 * AU), + ursina_run(sim.bodies, SECONDS_PER_DAY, position=(0, 0.0001 * AU, -0.2 * AU), show_timer=True, show_grid=True) -- GitLab