Python超人-宇宙模拟器

sim_scenes/interest/the_eye_of_god.py

+# -*- coding:utf-8 -*-
+# title           :上帝之眼
+# description     :上帝之眼
+# author          :Python超人
+# date            :2023-03-26
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.8
+# ==============================================================================
+from bodies import Sun, Earth, Moon, FixedStar, Body
+from common.consts import SECONDS_PER_HOUR, SECONDS_PER_HALF_DAY, SECONDS_PER_DAY, SECONDS_PER_WEEK
+from sim_scenes.func import mayavi_run, ursina_run
+from bodies.body import AU
+import random
+
+
+def show_eye_of_god():
+    """
+    上帝之眼
+    """
+    D = 600
+    mass = 0.9e25
+    # camera_pos = 左-右+、上+下-、前+后-
+    camera_pos = (-100 * D, 0, -5000 * D)
+
+    def get_position(pos, scale):
+        # [ 远+近-  , 左+右-  , 上+下-]
+        return pos[0] + (scale - 1.0) * 300 * (random.randint(90, 110)) * D, pos[1], pos[2]
+        # return pos[0], pos[1], pos[2]
+
+    bodies: list = gen_eye_bodies(
+        {"D": D, "Body": Body, "mass": mass, "get_position": get_position, "camera_pos": camera_pos})
+    face = FixedStar(name="face", texture="fixed_star.png", mass=mass * 3000, color=(0xff, 0xf8, 0xd4),
+                     init_position=[2000 * D, 240 * D, -90 * D],  # [ 远+近-  , 左+右-  , 上+下-]
+                     ignore_mass=True)
+    face.light_on = False
+    bodies.append(face)
+
+    # 使用 ursina 查看的运行效果
+    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
+    # position = 左-右+、上+下-、前+后-
+    ursina_run(bodies, SECONDS_PER_WEEK * 2, position=camera_pos)
+
+
+def gen_eye_bodies(params):
+    """
+    根据像素图片以及参数，自动生成星球，注意图片像素不能太多，否则会导致电脑运行太慢
+    @param params:
+    @return:
+    """
+    from PIL import Image
+
+    img = Image.open("./images/eye.png").convert('RGBA')
+    width, height = img.size
+    interval_factor = 20  # 星球间距因子
+    body_template = 'Body(name="%s", mass=mass, color=(%d, %d, %d), size_scale=%.4f, ' \
+                    'init_position=get_position([0, %g * D, %g * D], %.4f), init_velocity=[0, 0, 0], ignore_mass=True)'
+    bodies_str = "["
+
+    # 以图片像素为坐标，对角线的距离
+    distance_hw = pow(pow(width, 2) + pow(height, 2), 1 / 2)
+
+    for h in range(0, height):
+        # row = []
+        for w in range(0, width):
+            # 以图片像素为坐标，每个像素点到中心的距离
+            distance_to_center = pow(pow(w - width / 2, 2) + pow(h - height / 2, 2), 1 / 2)
+            # 让 body 从中心开始，离摄像机越远， body 的缩放值越大（scale 就越大，）
+            scale = (distance_to_center / (distance_hw * 10) + 1)  # 中心最近 1.0 ~ 1.05
+            # TODO: 队列反向排列（中心最远 1.05 ~ 1.0）
+            # scale = 1.05 - scale + 1.0
+            # print(scale)
+            # 获取像素的颜色
+            pixel = img.getpixel((w, h))
+            # 对于纯白色的颜色，就忽略，不生成星球（这样像素中纯白色越多，对电脑的压力就越少）
+            if pixel[0] >= 255 and pixel[1] >= 255 and pixel[1] >= 255:
+                continue
+            body_str = body_template % (f"星球{h}:{w}", pixel[0], pixel[1], pixel[2], scale,
+                                        w * interval_factor, -h * interval_factor, scale)
+            bodies_str += body_str + ",\n"
+
+    bodies_str += "]"
+    return eval(bodies_str, params)
+
+
+if __name__ == '__main__':
+    show_eye_of_god()

sim_scenes/interest/utils/body_utils.py

+# -*- coding:utf-8 -*-
+# title           :Body工具类
+# description     :Body工具类
+# author          :Python超人
+# date            :2023-03-26
+# link            :https://gitcode.net/pythoncr/
+# python_version  :3.8
+# ==============================================================================
+
+
+def gen_eye_bodies(pixel_image, params, body_template=None):
+    """
+    根据像素图片以及参数，自动生成星球，注意图片像素不能太多，否则会导致电脑运行太慢
+    @param pixel_image:
+    @param params:
+    @return:
+    """
+    from PIL import Image
+
+    img = Image.open(pixel_image).convert('RGBA')
+    width, height = img.size
+    interval_factor = 20  # 星球间距因子
+    if body_template is None:
+        body_template = 'Body(name="%s", mass=mass, color=(%d, %d, %d), size_scale=%.4f, ' \
+                        'init_position=get_position([0, %g * D, %g * D], %.4f), init_velocity=[0, 0, 0], ignore_mass=True)'
+    bodies_str = "["
+
+    # 以图片像素为坐标，对角线的距离
+    distance_hw = pow(pow(width, 2) + pow(height, 2), 1 / 2)
+
+    for h in range(0, height):
+        # row = []
+        for w in range(0, width):
+            # 以图片像素为坐标，每个像素点到中心的距离
+            distance_to_center = pow(pow(w - width / 2, 2) + pow(h - height / 2, 2), 1 / 2)
+            # 让 body 从中心开始，离摄像机越远， body 的缩放值越大（scale 就越大，）
+            scale = (distance_to_center / (distance_hw * 10) + 1)  # 中心最近 1.0 ~ 1.05
+            # TODO: 队列反向排列（中心最远 1.05 ~ 1.0）
+            # scale = 1.05 - scale + 1.0
+            # print(scale)
+            # 获取像素的颜色
+            pixel = img.getpixel((w, h))
+            # 对于纯白色的颜色，就忽略，不生成星球（这样像素中纯白色越多，对电脑的压力就越少）
+            if pixel[0] >= 255 and pixel[1] >= 255 and pixel[1] >= 255:
+                continue
+            body_str = body_template % (f"星球{h}:{w}", pixel[0], pixel[1], pixel[2], scale,
+                                        w * interval_factor, -h * interval_factor, scale)
+            bodies_str += body_str + ",\n"
+
+    bodies_str += "]"
+    return eval(bodies_str, params)
+
+
+if __name__ == '__main__':
+    import random
+    from bodies import Body
+
+    D = 600
+    mass = 0.9e25
+    # camera_pos = 左-右+、上+下-、前+后-
+    camera_pos = (-100 * D, 0, -5000 * D)
+
+
+    def get_position(pos, scale):
+        # [ 远+近-  , 左+右-  , 上+下-]
+        return pos[0] + (scale - 1.0) * 300 * (random.randint(90, 110)) * D, pos[1], pos[2]
+        # return pos[0], pos[1], pos[2]
+
+
+    bodies: list = gen_eye_bodies(pixel_image="../images/eye.png",
+                                  params={"D": D,
+                                          "Body": Body,
+                                          "mass": mass,
+                                          "get_position": get_position,
+                                          "camera_pos": camera_pos})
+
+    print(bodies)

simulators/views/ursina_view.py

@@ -95,8 +95,14 @@ class Planet(Entity):
         self.init_scale = scale
         if hasattr(body_view, "texture"):
             texture = load_texture(body_view.texture)
+            # color.white
+            self.plant_color = color.white
         else:
             texture = None
+            b_color = self.body_view.color
+            if len(b_color) == 3:
+                b_color = (b_color[0], b_color[1], b_color[2], 1.0)
+            self.plant_color = color.rgba(*b_color)
 
         if hasattr(self.body_view.body, "torus_stars"):
             # 创建一个星环小天体群（主要模拟小行星群，非一个天体）
@@ -108,9 +114,7 @@ class Planet(Entity):
             rotation = (0, 0, 0)
 
         UrsinaEvent.on_reset_subscription(self.on_reset)
-        # color.white
-        self.plant_color = color.white
-        # self.plant_color = color.rgba(*self.body_view.color)
+
         super().__init__(
             # model="sphere",
             model=model,