From 34a84a1478e82351c0bdd1ae0649e36e66fc91ac Mon Sep 17 00:00:00 2001
From: march3 <gang.wei@qq.com>
Date: Thu, 6 Jul 2023 20:03:43 +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/bodies_gravity_sim.py | 60 +++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 sim_lab/bodies_gravity_sim.py

diff --git a/sim_lab/bodies_gravity_sim.py b/sim_lab/bodies_gravity_sim.py
new file mode 100644
index 0000000..0c4d679
--- /dev/null
+++ b/sim_lab/bodies_gravity_sim.py
@@ -0,0 +1,60 @@
+# -*- coding:utf-8 -*-
+# title           :验证万有引力（为什么旋转的球体会导致被吸引天体轨道偏转）
+# description     :太验证万有引力（为什么旋转的球体会导致被吸引天体轨道偏转）
+# author          :Python超人
+# date            :2023-07-06
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.8
+# ==============================================================================
+from bodies import Sun, Moon
+from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, AU
+from sim_scenes.func import mayavi_run, ursina_run
+import numpy as np
+
+
+def ignore_gravity_with(body):
+    if isinstance(body, Sun):
+        return True
+
+    return False
+
+def create_circle_bodies(radius):
+    bodies = []
+    for i in range(0, 360, 30):
+        # 当前天数的角度（弧度）
+        angle = i * np.pi / 180
+        # 计算月亮的坐标（这里没有用到万有引力）
+        px = radius * np.cos(angle)
+        pz = radius * np.sin(angle)
+        body = Sun(size_scale=1e-3,
+                     init_velocity=[0, 0, 0],
+                     init_position=[px, 0, pz],
+                     # ignore_mass=True,
+                     rotation_speed=0)
+        body.ignore_gravity_with = ignore_gravity_with
+        bodies.append(body)
+    return bodies
+
+
+if __name__ == '__main__':
+    """
+    太阳、地球
+    """
+    bodies = create_circle_bodies(100000)
+    bodies.append(Moon(size_scale=100,
+                       init_position=[0, 0, 500000000],
+                       init_velocity=[0, 0, 0],
+                       # ignore_mass=True,
+                       ))
+    # bodies = [
+    #     Sun(size_scale=5e1),  # 太阳放大 50 倍
+    #     Earth(size_scale=2e3, distance_scale=1),  # 地球放大 2000 倍，距离保持不变
+    # ]
+
+    # 使用 mayavi 查看的运行效果
+    # mayavi_run(bodies, SECONDS_PER_WEEK, view_azimuth=-45)
+
+    # 使用 ursina 查看的运行效果
+    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
+    # position = 左-右+、上+下-、前+后-
+    ursina_run(bodies, SECONDS_PER_DAY, position=(0, 0, 0), show_trail=True)
-- 
GitLab