From c23e7465ddab867f56c919a0b238c3a87ea9ec64 Mon Sep 17 00:00:00 2001 From: march3 Date: Sun, 23 Apr 2023 22:03:39 +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 --- sim_lab/lagrangian_points_2.py | 227 +++++++++++++++++++++++++++ simulators/ursina/entities/planet.py | 20 ++- 2 files changed, 246 insertions(+), 1 deletion(-) create mode 100644 sim_lab/lagrangian_points_2.py diff --git a/sim_lab/lagrangian_points_2.py b/sim_lab/lagrangian_points_2.py new file mode 100644 index 0000000..cfb2913 --- /dev/null +++ b/sim_lab/lagrangian_points_2.py @@ -0,0 +1,227 @@ +""" +https://www.163.com/dy/article/G5F1016F053102ZV.html +https://www.sciencedirect.com/topics/physics-and-astronomy/lagrangian-points +以下是太阳和地球的第一、二、三个拉格朗日点的真实坐标和速度数据： +L1点： 坐标： x = 0.010205 AU， y = 0 AU， z = 0 AU 速度： vx = 0 m/s， vy = 246.593 m/s， vz = 0 m/s +L2点： 坐标： x = -0.010205 AU， y = 0 AU， z = 0 AU 速度： vx = 0 m/s， vy = -246.593 m/s， vz = 0 m/s +L3点： 坐标： x = 0.990445 AU， y = 0 AU， z = 0 AU 速度： vx = 0 m/s， vy = 11.168 m/s， vz = 0 m/s +L4点： 坐标： x = 0.500 AU， y = 0.866025 AU， z = 0 AU 速度： vx = -2446.292 m/s， vy = -1412.901 m/s， vz = 0 m/s +L5点： 坐标： x = 0.500 AU， y = -0.866025 AU， z = 0 AU 速度： vx = -2446.292 m/s， vy = 1412.901 m/s， vz = 0 m/s +https://baike.baidu.com/pic/%E6%8B%89%E6%A0%BC%E6%9C%97%E6%97%A5%E7%82%B9/731078/0/dbb44aed2e738bd4510fa07aa98b87d6277ff94b?fr=lemma&fromModule=lemma_content-image&ct=single#aid=0&pic=dbb44aed2e738bd4510fa07aa98b87d6277ff94b +""" +# +# AU = 1.496e8 +# +# +# def compute_barycenter(masses, positions): +# """ +# Compute the barycenter position of celestial objects in 3D space +# masses: a list of masses of celestial objects +# positions: a list of positions of celestial objects, each position is a tuple (x, y, z) +# """ +# m_sum = sum(masses) +# x_sum = 0 +# y_sum = 0 +# z_sum = 0 +# for i in range(len(masses)): +# x_sum += masses[i] * positions[i][0] / m_sum +# y_sum += masses[i] * positions[i][1] / m_sum +# z_sum += masses[i] * positions[i][2] / m_sum +# return (x_sum, y_sum, z_sum) +# +# +# def get_lagrangian_points(m1, m2, r): +# """ +# https://baike.baidu.com/item/%E6%8B%89%E6%A0%BC%E6%9C%97%E6%97%A5%E7%82%B9/731078 +# +# @param m1: 大质量 +# @param m2: 小质量 +# @param r: 半径 +# @return: +# """ +# a = m2 / (m1 + m2) +# l1 = (r * (1 - pow(a / 3, 1 / 3)), 0) +# l2 = (r * (1 + pow(a / 3, 1 / 3)), 0) +# l3 = (-r * (1 + (5 * a) / 12), 0) +# l4 = ((r / 2) * ((m1 - m2) / (m1 + m2)), pow(3, 1 / 2) / 2 * r) +# l5 = ((r / 2) * ((m1 - m2) / (m1 + m2)), -pow(3, 1 / 2) / 2 * r) +# +# # print(l1[0]/AU, l2[0]/AU, l3[0]/AU, l4[0]/AU, l5[0]/AU) +# return l1, l2, l3, l4, l5 +# +# +# def show_figure(points, p1_name, p2_name, unit, barycenter=None): +# import matplotlib.pyplot as plt +# +# plt.figure(figsize=(16, 12)) +# plt.plot(0, 0, "ro", markersize=20, label=p1_name) +# plt.text(-unit / 20, -unit / 10, p1_name, fontsize=30, color="r") +# plt.plot(unit, 0, "b.", markersize=4, label=p2_name) +# plt.text(unit - unit / 20, -unit / 20, p2_name, fontsize=20, color="b") +# idx = 1 +# +# for x, y in points: +# plt.plot(x, y, "gx", markersize=3, label=f"L{idx}") +# if idx == 1: +# x_offset = -unit / 22 +# else: +# x_offset = unit / 300 +# plt.text(x + x_offset, y + unit / 300, f"L{idx}", fontsize=18, color="g") +# idx += 1 +# +# if barycenter is not None: +# plt.plot(barycenter[0], barycenter[1], "gx", markersize=10, label=f"L{idx}") +# +# # plt.plot(x, y, "b.") # b：蓝色，.：点 +# # plt.plot(x, y1, "ro") # r：红色，o：圆圈 +# # plt.plot(x, y2, "kx") # k：黑色，x：x字符(小叉) +# plt.show() # 在窗口显示该图片 +# +# +# barycenter = compute_barycenter([5.97237e24, 7.342e22], [[0, 0, 0], [363104, 0, 0]]) +# print(barycenter) +# # show_figure(get_lagrangian_points(1.9891e30, 5.97237e24, AU), "Sun", "Earth", AU) +# show_figure(get_lagrangian_points(5.97237e24, 7.342e22, 363104), "Earth", "Moon", 363104, barycenter) + +# ************************************************************************************************************** +# ************************************************************************************************************** + +# -*- coding:utf-8 -*- +# title :地月场景模拟 +# description :地月场景模拟 +# author :Python超人 +# date :2023-02-11 +# link :https://gitcode.net/pythoncr/ +# python_version :3.8 +# ============================================================================== +from bodies import Sun, Earth, Moon +from objs import Satellite, Satellite2 +from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH +from sim_scenes.func import ursina_run, camera_look_at +from bodies.body import AU +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 +from simulators.ursina.ursina_mesh import create_line + + +def compute_barycenter(masses, positions): + """ + Compute the barycenter position of celestial objects in 3D space + masses: a list of masses of celestial objects + positions: a list of positions of celestial objects, each position is a tuple (x, y, z) + """ + m_sum = sum(masses) + x_sum = 0 + y_sum = 0 + z_sum = 0 + for i in range(len(masses)): + x_sum += masses[i] * positions[i][0] / m_sum + y_sum += masses[i] * positions[i][1] / m_sum + z_sum += masses[i] * positions[i][2] / m_sum + return (x_sum, y_sum, z_sum) + + +def get_lagrangian_points(m1, m2, r): + """ + https://baike.baidu.com/item/%E6%8B%89%E6%A0%BC%E6%9C%97%E6%97%A5%E7%82%B9/731078 + + @param m1: 大质量 + @param m2: 小质量 + @param r: 半径 + @return: + """ + a = m2 / (m1 + m2) + l1 = (0, 0, r * (1 - pow(a / 3, 1 / 3))) + l2 = (0, 0, r * (1 + pow(a / 3, 1 / 3))) + l3 = (0, 0, -r * (1 + (5 * a) / 12)) + l4 = (pow(3, 1 / 2) / 2 * r, 0, (r / 2) * ((m1 - m2) / (m1 + m2))) + l5 = (-pow(3, 1 / 2) / 2 * r, 0, (r / 2) * ((m1 - m2) / (m1 + m2))) + + return l1, l2, l3, l4, l5 + + +if __name__ == '__main__': + """ + 地球、月球 + """ + OFFSETTING = 0 + # TODO: 可以抵消月球带动地球的力，保持地球在原地 + # OFFSETTING = 0.01265 + bodies = [ + Earth(init_position=[0, 0, 0], texture="earth_hd.jpg", + init_velocity=[OFFSETTING, 0, 0], size_scale=0.5e1), # 地球放大 5 倍，距离保持不变 + Moon(init_position=[0, 0, 363104], # 距地距离约: 363104 至 405696 km + init_velocity=[-1.05435, 0, 0], size_scale=1e1) # 月球放大 10 倍，距离保持不变 + ] # -1.0543 < -1.05435 < -1.0545 + earth = bodies[0] + moon = bodies[1] + + points = get_lagrangian_points(earth.mass, moon.mass, 363104) + offset_points = [ + [0, 0, 0], # 调整加速度为0 + [0, 0, 0], + [0, 0, 0], + [0, 0, 0], + [0, 0, 0], + ] + velocities = [ + [-0.889, -0.1, 0], # [-0.859, 0, 0], + [-1.265, 0, 0], + [1.03, 0, 0], + [0, 0, 0], + [0, 0, 0], + ] + velocities = [ + [-0.885, 0, 0], # [-0.859, 0, 0], + [-0.879, 0, 0], + [-0.869, 0, 0], + ] + velocities=[] + for i in range(30): + v = round(-0.846-(i/10000),4) + print(v) + velocities.append([v, 0, 0]) + + satellites = [] + for i, point in enumerate(points[0:1]): + for j,velocitie in enumerate(velocities): + satellite = Satellite(name=f'卫星{j + 1}', mass=1.4e10, size_scale=1e3, color=(255, 200, 0), + init_position=[point[0] + offset_points[i][0], + point[1] + offset_points[i][1], + point[2] + offset_points[i][2]], + init_velocity=velocities[j]) + bodies.append(satellite) + satellites.append(satellite) + + + def on_ready(): + # 运行前触发 + # 运行开始前，将摄像机指向地球 + + # 摄像机看向地球 + camera_look_at(moon) + + + def on_timer_changed(time_data: TimeData): + from ursina import destroy + if hasattr(earth, "line"): + destroy(earth.line) + earth.line = create_line(from_pos=earth.planet.position, to_pos=moon.planet.main_entity.position) + for satellite in satellites: + satellite.look_at(earth) + + + # 订阅事件后，上面的函数功能才会起作用 + # 运行前会触发 on_ready + UrsinaEvent.on_ready_subscription(on_ready) + # 运行中，每时每刻都会触发 on_timer_changed + UrsinaEvent.on_timer_changed_subscription(on_timer_changed) + + # 使用 ursina 查看的运行效果 + # 常用快捷键： P：运行和暂停 O：重新开始 I：显示天体轨迹 + # position = 左-右+、上+下-、前+后- + ursina_run(bodies, SECONDS_PER_HOUR*10, + position=(-5000, 500000, -10), + show_timer=True, + show_trail=True) diff --git a/simulators/ursina/entities/planet.py b/simulators/ursina/entities/planet.py index b612d08..5c76eea 100644 --- a/simulators/ursina/entities/planet.py +++ b/simulators/ursina/entities/planet.py @@ -16,7 +16,7 @@ from simulators.ursina.ursina_event import UrsinaEvent from common.color_utils import adjust_brightness, conv_to_vec4_color, get_inverse_color from common.func import find_file from simulators.views.body_view import BodyView -from simulators.ursina.ursina_mesh import create_sphere, create_torus, create_arrow_line, create_line +from simulators.ursina.ursina_mesh import create_sphere, create_torus, create_arrow_line, create_line, create_label import math @@ -97,6 +97,7 @@ class Planet(Entity): collider=collider, position=pos, rotation=rotation, + ignore_paused=True, double_sided=True ) if hasattr(self.body, "rotate_angle"): @@ -294,3 +295,20 @@ class Planet(Entity): self.body_view.appeared = False # 最后删除自己 destroy(self) + + def input(self, key): + if self.hovered: + if key == 'left mouse down': + # print(key, self) + self.show_name() + + def show_name(self): + if hasattr(self, "label_name"): + destroy(self.label_name) + delattr(self, "label_name") + else: + self.label_name = create_label(parent=self, + label=self.body.name, + pos=Vec3(-0.5, -0.5, -0.5), + color=color.red) + self.label_name.set_light_off() \ No newline at end of file -- GitLab