create_arrow的代码备份

bodies/body.py

@@ -19,7 +19,7 @@ class Body(metaclass=ABCMeta):
     天体基类
     """
 
-    def __init__(self, name, mass, init_position, init_velocity,
+    def __init__(self, name, mass, init_position, init_velocity, rotate_angle=0,
                  density=5e3, color=(125 / 255, 125 / 255, 125 / 255),
                  texture=None, size_scale=1.0, distance_scale=1.0,
                  rotation_speed=None, parent=None, ignore_mass=False,
@@ -60,6 +60,8 @@ class Body(metaclass=ABCMeta):
         self.init_position = np.array(init_position, dtype='float32')
         self.init_velocity = np.array(init_velocity, dtype='float32')
 
+        self.rotate_angle = rotate_angle
+
         # self.__position = copy.deepcopy(self.init_position)
         # self.__velocity = copy.deepcopy(self.init_velocity)
 

bodies/earth.py

@@ -23,7 +23,7 @@ class Earth(Body):
     　平均密度: 5507.85 kg/m³
     """
 
-    def __init__(self, name="地球", mass=5.97237e24,
+    def __init__(self, name="地球", mass=5.97237e24,rotate_angle=0,
                  init_position=[1.12 * AU, 0, 0],
                  init_velocity=[0, 29.79, 0],
                  texture="earth1.jpg", size_scale=1.0, distance_scale=1.0,
@@ -33,6 +33,7 @@ class Earth(Body):
             "mass": mass,
             "init_position": init_position,
             "init_velocity": init_velocity,
+            "rotate_angle": rotate_angle,
             "density": 5507.85,
             "color": (7, 0, 162),
             "texture": texture,

simulators/ursina/entities/body_trail.py

@@ -10,6 +10,7 @@
 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
 
 
 class BodyTrail(Entity):
@@ -44,51 +45,55 @@ class BodyTrail(Entity):
         #                       "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)]
+        #
+        # 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()
 
-        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()
+        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
+        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 = 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.rotation = (0, 0, 0)
         self.look_at(to_pos)
         print(self.rotation)
         # self.model = Mesh(vertices=[
@@ -104,7 +109,6 @@ class Arrow(Entity):
         # 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)
@@ -112,7 +116,6 @@ class Arrow(Entity):
 #         self.look_at(to_pos)
 
 
-
 # import numpy as np
 # def draw_arrow(parent, from_pos, to_pos, color, alpha):
 #     # 计算方向向量和长度

simulators/ursina/ursina_mesh.py

@@ -7,11 +7,13 @@
 # python_version  :3.8
 # ==============================================================================
 # pip install -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com ursina
-from ursina import Ursina, window, Entity, Mesh, EditorCamera, color, mouse, Vec2, Vec3, load_texture, Texture
+from ursina import Ursina, window, Entity, Mesh, EditorCamera, color, mouse, Vec2, Vec3, load_texture, Texture, Text
 from math import pi, sin, cos
 import numpy as np
 import math
 
+from simulators.ursina.ursina_config import UrsinaConfig
+
 
 def create_sphere(radius, subdivisions):
     """
@@ -61,6 +63,105 @@ def create_sphere(radius, subdivisions):
     return Mesh(vertices=verts, triangles=tris, normals=normals, uvs=uvs, mode='triangle')
 
 
+def create_arrow(height=0.5, width=0.1):
+    # 创建金字塔的顶点
+    r = width / 2
+    verts = [
+        (0, height, 0),  # 顶点
+        (-r, 0, -r),  # 左后底点
+        (r, 0, -r),  # 右后底点
+        (r, 0, r),  # 右前底点
+        (-r, 0, r),  # 左前底点
+    ]
+    # 修改指向
+    verts = [(z, x, y) for x, y, z in verts]
+
+    # 定义金字塔的面
+    faces = [
+        (0, 1, 2),  # 底面1
+        (0, 2, 3),  # 底面2
+        (0, 3, 4),  # 底面3
+        (0, 4, 1),  # 底面4
+        (4, 2, 1),  # 侧面1
+        (4, 3, 2),  # 侧面1
+    ]
+
+    # 创建一个金字塔的Mesh对象
+    arrow_mesh = Mesh(vertices=verts, triangles=faces, mode='triangle')
+    return arrow_mesh
+
+
+def create_arrow_line(from_pos, to_pos, parent=None, label=None,
+                      set_light_off=True, alpha=1.0,
+                      color=color.white, thickness=2,
+                      arrow_scale=1, text_scale=50):
+    """
+    创建箭头和箭头线段
+    @param from_pos: 箭头线段开始位置
+    @param to_pos: 箭头线段结束位置
+    @param parent:
+    @param label: 箭头显示的文字
+    @param set_light_off: 是否设置为灯光关闭状态
+    @param alpha: 透明度
+    @param color: 箭头线颜色
+    @param thickness: 线段粗细
+    @param arrow_scale: 箭头缩放
+    @return:
+    """
+    height = 0.5 * thickness
+    width = 0.1 * thickness
+    arrow_mesh = create_arrow(height, width)
+    verts = (Vec3(from_pos), Vec3(to_pos))
+    line = Entity(parent=parent, model=Mesh(vertices=verts, mode='line', thickness=thickness),
+                  color=color, alpha=alpha)
+    arrow = Entity(parent=line, model=arrow_mesh, position=verts[1],
+                   scale=thickness * arrow_scale, color=color, alpha=alpha)
+    arrow.look_at(Vec3(to_pos) * 100)
+
+    if set_light_off:
+        line.set_light_off()
+        arrow.set_light_off()
+
+    if label is not None:
+        text = Text(label, parent=line, scale=text_scale, billboard=True, color=color,
+                    position=Vec3(to_pos) + Vec3(1, 1, 1), alpha=alpha,
+                    font=UrsinaConfig.CN_FONT, background=False)
+        if set_light_off:
+            text.set_light_off()
+    else:
+        text = None
+
+    return arrow, line, text
+
+
+def create_pyramid():
+    # 创建金字塔的顶点
+    verts = [
+        (0, 2, 0),  # 顶点
+        (-1, 0, -1),  # 左后底点
+        (1, 0, -1),  # 右后底点
+        (1, 0, 1),  # 右前底点
+        (-1, 0, 1),  # 左前底点
+    ]
+
+    # 定义金字塔的面
+    faces = [
+        (0, 1, 2),  # 底面1
+        (0, 2, 3),  # 底面2
+        (0, 3, 4),  # 底面3
+        (0, 4, 1),  # 底面4
+        (4, 2, 1),  # 侧面1
+        (4, 3, 2),  # 侧面1
+    ]
+
+    # 创建一个金字塔的Mesh对象
+    # verts = [(0, 1, 0), (1, 0, 1), (1, 0, -1), (-1, 0, -1), (-1, 0, 1)]
+    # faces = [(0, 1, 2), (0, 2, 3), (0, 3, 4), (0, 4, 1), (1, 3, 2)]
+    pyramid_mesh = Mesh(vertices=verts, triangles=faces, mode='triangle')
+
+    return pyramid_mesh
+
+
 def create_body_torus(inner_radius, outer_radius, subdivisions):
     vertices = []
     uvs = []
@@ -195,9 +296,18 @@ if __name__ == '__main__':
     # torus = create_torus(1.5, 3, 64)
     # entity = Entity(model=torus, texture=textureRings, rotation=(85, 0, 0), double_sided=True)
 
-    body_torus = create_torus(9, 10, 64)
-    entities = Entity(model=body_torus, texture=textureAsteroids, rotation=(85, 0, 0), double_sided=True)
-    entities.set_light_off()
+    # body_torus = create_torus(9, 10, 64)
+    # entities = Entity(model=body_torus, texture=textureAsteroids, rotation=(85, 0, 0), double_sided=True)
+    # entities.set_light_off()
+
+    # 创建金字塔的实体对象
+    # pyramid = Entity(model=create_pyramid(), texture='brick', color=color.yellow)
+    # pyramid.set_light_off()
+
+    # arrow = Entity(model=create_arrow(), color=color.yellow)
+    # arrow.set_light_off()
+
+    arrow, line = create_arrow_line((0, 0, 0), (10, 0, 0))
 
     EditorCamera()
     app.run()

simulators/ursina_simulator.py

@@ -15,6 +15,7 @@ from simulators.ursina.ursina_event import UrsinaEvent
 # from simulators.ursina.ursina_ui import UrsinaUI
 from simulators.ursina.ui.control_ui import ControlUI
 from simulators.ursina.ui.control_handler import ControlHandler
+from simulators.ursina.ursina_mesh import create_arrow_line
 
 from simulators.views.ursina_view import UrsinaView
 from simulators.ursina.entities.ursina_player import UrsinaPlayer
@@ -35,6 +36,12 @@ class WorldGrid(Entity):
     创建一个宇宙网格对象
     """
 
+    def draw_axises(self):
+        # 坐标轴
+        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)
+
     def __init__(self):
         super().__init__()
         s = 100
@@ -42,15 +49,8 @@ class WorldGrid(Entity):
                       position=(0, -80, 0))
         grid.set_light_off()
 
-        # 坐标轴
-        # vertsx = ((0, 0, 0), (10, 0, 0))
-        # Entity(model=Mesh(vertices=vertsx, mode='line', thickness=3), color=color.red).set_light_off()
-        #
-        # vertsy = [(0, 0, 0), (0, 10, 0)]
-        # Entity(model=Mesh(vertices=vertsy, mode='line', thickness=3), color=color.green).set_light_off()
-        #
-        # vertsz = [(0, 0, 0), (0, 0, 10)]
-        # Entity(model=Mesh(vertices=vertsz, mode='line', thickness=3), color=color.blue).set_light_off()
+        # self.draw_axises()
+
 
 class UrsinaSimulator(Simulator):
     """