Python超人-宇宙模拟器

sim_scenes/solar_system/earth_moon6.py

+# -*- 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 common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH
+from sim_scenes.func import mayavi_run, ursina_run
+from bodies.body import AU
+
+if __name__ == '__main__':
+    """
+    地球、6个月球
+    """
+    # 地球的Y方向初始速度
+    EARTH_INIT_VELOCITY = 0
+    # 地球在中心位置
+    e = Earth(init_position=[0, 0, 0], texture="earth_hd.jpg", init_velocity=[0, 0, 0])
+    bodies = [
+        e,
+        Moon(init_position=[e.diameter, 0, 0], init_velocity=[0, 0, 0]),   # 月球在地球前面
+        Moon(init_position=[-e.diameter, 0, 0], init_velocity=[0, 0, 0]),  # 月球在地球后面
+        Moon(init_position=[0, e.diameter, 0], init_velocity=[0, 0, 0]),   # 月球在地球左面
+        Moon(init_position=[0, -e.diameter, 0], init_velocity=[0, 0, 0]),  # 月球在地球右面
+        Moon(init_position=[0, 0, e.diameter], init_velocity=[0, 0, 0]),   # 月球在地球上面
+        Moon(init_position=[0, 0, -e.diameter], init_velocity=[0, 0, 0]),  # 月球在地球下面
+    ]
+
+    # 使用 ursina 查看的运行效果
+    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
+    # position = 左-右+、上+下-、前+后-
+    ursina_run(bodies, 60, position=(0, 0, 0), show_trail=True, view_closely=0.001)

simulators/ursina/entities/body_trail.py

@@ -9,7 +9,6 @@
 # pip install -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com ursina
 from ursina import Entity, Mesh, Text, color, destroy, Vec3
 
-from simulators.ursina.ursina_config import UrsinaConfig
 from simulators.ursina.ursina_mesh import create_arrow_line
 
 
@@ -23,12 +22,6 @@ class BodyTrail(Entity):
         )
 
     def input(self, key):
-        # if self.hovered:
-        #     if key == 'left mouse down':
-        #         print(f"{self} was clicked!")
-        #         self.text.enabled = True
-        #     elif key == 'left mouse up':
-        #         self.text.enabled = False
         if self.hovered:
             if key == 'left mouse down':
                 # print(key, self)
@@ -41,95 +34,11 @@ class BodyTrail(Entity):
             for c in self.children:
                 destroy(c)
             return
-        # entity_infos = {"velocity": [vel_info, vel_direction, vel_position],
-        #                       "acceleration": [acc_info, acc_direction, acc_position]}
+
         vel_info, vel_direction, vel_position = self.entity_infos["velocity"]
         acc_info, acc_direction, acc_position = self.entity_infos["acceleration"]
-        #
-        # verts_acc = [(0, 0, 0), tuple(acc_direction)]
-        # verts_vel = [(0, 0, 0), tuple(vel_direction)]
-
-        # acc_arrow = Arrow(parent=self,from_pos=Vec3((0, 0, 0)), to_pos=Vec3(tuple(acc_direction)),color=color.yellow, alpha=0.5)
-        # acc_arrow.set_light_off()
 
         v_arrow, v_line, v_text = create_arrow_line((0, 0, 0), tuple(vel_direction), parent=self,
                                                     label=vel_info, color=color.red, alpha=0.8, arrow_scale=0.5)
         a_arrow, a_line, a_text = create_arrow_line((0, 0, 0), tuple(acc_direction), parent=self,
                                                     label=acc_info, color=color.yellow, alpha=0.8, arrow_scale=0.5)
-
-        # acc_line = Entity(parent=self, model=Mesh(vertices=verts_acc, mode='line', thickness=3),
-        #                   color=color.yellow, alpha=0.5)
-        # acc_line.set_light_off()
-        #
-        # # vel_arrow = Arrow(parent=self,from_pos=Vec3((0, 0, 0)), to_pos=Vec3(tuple(vel_direction)),color=color.red, alpha=0.5)
-        # # vel_arrow.set_light_off()
-        #
-        # vel_line = Entity(parent=self, model=Mesh(vertices=verts_vel, mode='line', thickness=3),
-        #                   color=color.red, alpha=0.5)
-        # vel_line.set_light_off()
-        #
-        # vel_text = Text(vel_info, scale=50, billboard=True, parent=self,
-        #                 font=UrsinaConfig.CN_FONT, background=False, color=color.red,
-        #                 position=vel_position, alpha=0.5)
-        # vel_text.set_light_off()
-        #
-        # acc_text = Text(acc_info, scale=50, billboard=True, parent=self,
-        #                 font=UrsinaConfig.CN_FONT, background=False, color=color.yellow,
-        #                 position=acc_position, alpha=0.5)
-        # acc_text.set_light_off()
-
-
-class Arrow(Entity):
-    def __init__(self, parent, from_pos=(0, 0, 0), to_pos=(1, 0, 0), **kwargs):
-        from_pos = to_pos / 2
-        super().__init__(parent=parent, model='arrow', position=from_pos, **kwargs)
-        # self.x = -pos[1]
-        # self.y = pos[2]
-        # self.z = pos[0]
-        to_pos = 1000 * to_pos
-        # to_pos = -to_pos[2],-to_pos[1],to_pos[0]
-        # to_pos = -to_pos[2],-to_pos[0],to_pos[1]
-        # to_pos = to_pos[0],to_pos[2],to_pos[1]
-        # to_pos = -to_pos[0],to_pos[1],to_pos[2]
-        # to_pos = to_pos[1], -to_pos[0], -to_pos[2]
-        # to_pos = to_pos[1], -to_pos[0], -to_pos[2]
-        self.rotation = (0, 0, 0)
-        self.look_at(to_pos)
-        print(self.rotation)
-        # self.model = Mesh(vertices=[
-        #     from_pos,
-        #     from_pos + (to_pos - from_pos) * 0.9,
-        #     from_pos + (to_pos - from_pos) * 0.9 + Vec3(0, 0.1, 0),
-        #     to_pos,
-        #     from_pos + (to_pos - from_pos) * 0.9 - Vec3(0, 0.1, 0),
-        #     from_pos + (to_pos - from_pos) * 0.9
-        # ], mode='triangle', thickness=thickness)
-        #
-        # self.color = color
-        # self.look_at(to_pos)
-        # self.scale_z = (to_pos - from_pos).length()
-
-# class Arrow(Entity):
-#     def __init__(self,parent, from_pos, to_pos, **kwargs):
-#         super().__init__(parent=parent, model='arrow', **kwargs)
-#         self.position = from_pos
-#         self.look_at(to_pos)
-
-
-# import numpy as np
-# def draw_arrow(parent, from_pos, to_pos, color, alpha):
-#     # 计算方向向量和长度
-#     p1 = np.array(from_pos)
-#     p2 = np.array(to_pos)
-#     direction = p2 - p1
-#     length = np.linalg.norm(direction)
-#
-#     # 创建箭头实体
-#     arrow = Entity(parent=parent, model="arrow", scale=(1, 1, length),color=color,alpha=alpha)
-#
-#     # 设置箭头位置和方向
-#     arrow_position = p1 + direction/2
-#     arrow.position = arrow_position
-#     arrow.look_at(p2)
-#
-#     return arrow

simulators/ursina/entities/entity_utils.py

@@ -255,9 +255,6 @@ def create_trail_info(body, trail):
     acc_direction = acceleration[1]
     acc_direction = np.array(acc_direction) * 2
 
-    verts_acc = [(0, 0, 0), tuple(acc_direction)]
-    verts_vel = [(0, 0, 0), tuple(vel_direction)]
-
     vel_position = vel_direction
     vel_position = (vel_position[0], vel_position[1], vel_position[2])
 
@@ -267,57 +264,3 @@ def create_trail_info(body, trail):
     trail.entity_infos = {"velocity": [vel_info, vel_direction, vel_position],
                           "acceleration": [acc_info, acc_direction, acc_position]}
 
-
-def create_trail_text_xxx(body, trail):
-    velocity = merge_vectors(body.velocity)
-    acceleration = merge_vectors(body.acceleration)
-    vel_info = "%.2fkm/s" % (velocity[0])
-    acc_m = acceleration[0] * 1000
-
-    if acc_m >= 0.01:
-        acc_info = "%.2fm/s²" % (acc_m)
-    else:
-        acc_info = "%.2fmm/s²" % (acc_m * 1000)
-
-    vel_direction = velocity[1]
-    vel_direction = np.array(vel_direction) * 5
-
-    acc_direction = acceleration[1]
-    acc_direction = np.array(acc_direction) * 2
-    # acc_direction = np.array(acc_direction)*UrsinaConfig.SCALE_FACTOR
-    # vertsyz = [tuple(body.position),tuple(acc_direction)] #  [(0, 0, 0), (0, 10, 0), (0, 0, 0), (0, 0, 10)]
-    verts_acc = [(0, 0, 0), tuple(acc_direction)]
-    acc_line = Entity(parent=trail, model=Mesh(vertices=verts_acc, mode='line', thickness=3),
-                      color=color.yellow, alpha=0.5)
-    acc_line.set_light_off()
-
-    verts_vel = [(0, 0, 0), tuple(vel_direction)]
-    vel_line = Entity(parent=trail, model=Mesh(vertices=verts_vel, mode='line', thickness=3),
-                      color=color.red, alpha=0.5)
-    vel_line.set_light_off()
-
-    vel_position = vel_direction
-    vel_position = (vel_position[0], vel_position[1], vel_position[2])
-
-    vel_text = Text(vel_info, scale=50, billboard=True, parent=trail,
-                    font=UrsinaConfig.CN_FONT, background=False, color=color.red,
-                    position=vel_position, alpha=0.5)
-    vel_text.set_light_off()
-    acc_position = acc_direction
-    acc_position = (acc_position[0], acc_position[1], acc_position[2])
-
-    acc_text = Text(acc_info, scale=50, billboard=True, parent=trail,
-                    font=UrsinaConfig.CN_FONT, background=False, color=color.yellow,
-                    position=acc_position, alpha=0.5)
-    acc_text.set_light_off()
-    # self.name_text.background.color = color.rgba(b_color[0], b_color[1], b_color[2], 0.3)
-    # # self.name_text.scale = self.scale
-    # inverse_color = get_inverse_color(b_color)
-    # self.name_text.color = color.rgba(inverse_color[0], inverse_color[1], inverse_color[2], 1)
-    # acc_line.enabled = False
-    # vel_line.enabled = False
-    # vel_text.enabled = False
-    # acc_text.enabled = False
-
-    trail.entity_infos = [acc_line, vel_line, vel_text, acc_text]
-    return acc_line, vel_line, vel_text, acc_text

simulators/ursina/entities/planet.py

@@ -62,7 +62,7 @@ class Planet(Entity):
                 subdivisions = self.body.resolution
 
             model = create_sphere(0.5, subdivisions)
-            rotation = (0,  self.body.rotate_angle,0)
+            rotation = (0, 0, 0)
 
         UrsinaEvent.on_reset_subscription(self.on_reset)
 

simulators/ursina/ursina_mesh.py

@@ -307,7 +307,7 @@ if __name__ == '__main__':
     # arrow = Entity(model=create_arrow(), color=color.yellow)
     # arrow.set_light_off()
 
-    arrow, line = create_arrow_line((0, 0, 0), (10, 0, 0))
+    arrow, line, text = create_arrow_line((0, 0, 0), (10, 0, 0))
 
     EditorCamera()
     app.run()

simulators/ursina_simulator.py

@@ -37,7 +37,11 @@ class WorldGrid(Entity):
     """
 
     def draw_axises(self):
-        # 坐标轴
+        """
+        画坐标轴
+        @return:
+        """
+
         arrow_x, line_x, text_x = create_arrow_line((0, 0, 0), (10, 0, 0), label="X", color=color.red)
         arrow_y, line_y, text_y = create_arrow_line((0, 0, 0), (0, 10, 0), label="Y", color=color.green)
         arrow_z, line_z, text_z = create_arrow_line((0, 0, 0), (0, 0, 10), label="Z", color=color.yellow)