ursina_simulator.py

# -*- coding:utf-8 -*-
# title           :ursina天体运行模拟器
# description     :ursina天体运行模拟器
# author          :Python超人
# date            :2023-02-11
# link            :https://gitcode.net/pythoncr/
# python_version  :3.8
# ==============================================================================
# pip install -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com ursina
from ursina import Ursina, window, Entity, Grid, Mesh, camera, Text, application, color, mouse, Vec2, Vec3, \
    load_texture, held_keys, distance
from ursina.prefabs.first_person_controller import FirstPersonController

from simulators.ursina.ursina_event import UrsinaEvent
from simulators.ursina.ursina_ui import UrsinaUI
from simulators.views.ursina_view import UrsinaView, UrsinaPlayer
from simulators.ursina.ursina_config import UrsinaConfig
from simulators.simulator import Simulator
from common.system import System
import time
import datetime
import math
import os
from ursina import EditorCamera, PointLight, SpotLight, AmbientLight, DirectionalLight
from scenes.func import ursina_run


class WorldGrid(Entity):  # Entity # 定义构造方法
    def __init__(self):
        super().__init__()
        s = 100
        grid = Entity(model=Grid(s, s), scale=s * 20, color=color.rgba(255, 255, 255, 20), rotation_x=90,
                      position=(0, -80, 0))
        # 坐标轴
        # vertsx = ((0, 0, 0), (10, 0, 0))
        # Entity(model=Mesh(vertices=vertsx, mode='line', thickness=3), color=color.cyan).set_light_off()
        # vertsyz = [(0, 0, 0), (0, 10, 0), (0, 0, 0), (0, 0, 10)]
        # Entity(model=Mesh(vertices=vertsyz, mode='line', thickness=3), color=color.yellow).set_light_off()
        grid.set_light_off()


class UrsinaSimulator(Simulator):
    """
    Ursina官网： https://www.ursinaengine.org/
    """
    def __init__(self, bodies_sys: System):
        # self.app = Ursina()
        import os
        os.environ['CUDA_VISIBLE_DEVICES'] = '1'  # 选择第二个GPU
        self.app = Ursina(window_title='GPU模拟',
                     window_kwargs={'vsync': True, 'fullscreen': False, 'borderless': False, 'show_ursina_splash': True,
                                    'high_resolution': True})

        self.ursina_views = []
        window.color = color.black

        super().__init__(bodies_sys, UrsinaView)

        # ps = ["sun", "mercury", "venus", "earth", "mars", "jupiter", "saturn", "uranus", "neptune"]
        # cp = [200, 15, 35, 42, 20, 160, 145, 90, 80]
        # x, y, z = 0, 0, 0
        for i, view in enumerate(self.body_views):
            # pos = tuple(body.position)
            # ursina_view = UrsinaView(body)
            view.update()
            self.ursina_views.append(view)
            # planets.append(newPlanet)
            # x += cp[i] * 10
        self.adj_run_params()
        UrsinaEvent.on_searching_bodies_subscription(type(self).__name__, self.on_searching_bodies)

    def adj_run_params(self):
        max_distance = 0
        for b1 in self.body_views:
            for b2 in self.body_views:
                if b1 is b2:
                    continue
            d = distance(b1.planet, b2.planet)
            if d > max_distance:
                max_distance = d

        # UrsinaConfig.control_camera_speed = round(max_distance * 10, 2)
        time_scale = round(pow(max_distance, 1 / 4), 2)
        if time_scale < 0.01:
            time_scale = 0.01

        application.time_scale = time_scale
        # UrsinaConfig.auto_scale_factor = 1.0e-9

    def on_searching_bodies(self, **kwargs):
        views = []
        for view in self.body_views:
            # if view.appeared:
            views.append(view)
        return views

    def check_elapsed_time(self):
        """检查时间间隔是否已过"""
        now = datetime.datetime.now()
        elapsed_time = now - self.last_time
        value = elapsed_time >= self.interval
        if value:
            self.last_time = now
        return value

    def check_and_evolve(self):
        if self.check_elapsed_time():
            run_speed_factor = UrsinaConfig.run_speed_factor
            if UrsinaConfig.seconds_per == 0:
                evolve_dt = self.evolve_dt * run_speed_factor
            else:
                evolve_dt = UrsinaConfig.seconds_per * run_speed_factor

            evolve_dt = evolve_dt * self.interval_fator
            super().evolve(evolve_dt)

    def cosmic_background(self, texture='../textures/cosmic2.jpg'):
        """
        加入宇宙背景
        :param texture:
        :return:
        """
        # Add skybox
        from ursina import Sky
        Sky(texture=texture).scale = 10000

        # texture = load_texture(texture)
        # sky_dome = Entity(model='sky_dome', texture=texture, scale=10000,
        #                   color=color.white,
        #                   position=(0, 0, 0),
        #                   rotation=(0, 0, 0))

    def __add_glow(self, entity, intensity=2, light_color=color.white, attenuation=3):
        """
        未用，保留代码
        :param entity:
        :param intensity:
        :param light_color:
        :param attenuation:
        :return:
        """
        lights = []
        import math
        for i in range(5):
            glow_entity = Entity(parent=entity, model='sphere', color=color.rgba(1.0, 0.6, 0.2, 1),
                                 scale=math.pow(1.03, i), alpha=0.2)
            lights.append(glow_entity)
        # 创建一个新的 Entity 对象，作为光晕的容器
        # glow_entity = Entity(parent=entity, model='sphere', scale=entity.scale * 1.2)
        # 创建 PointLight 对象，并设置它的属性
        for i in range(2):
            light = PointLight(parent=lights[0], intensity=intensity, color=light_color, attenuation=attenuation)
            lights.append(light)

        # 把 Entity 对象放到星星的后面，使得光晕看起来像是从星星发出来的
        glow_entity.world_position = entity.world_position
        glow_entity.world_parent = entity.parent
        glow_entity.y += entity.scale_y * 0.1
        glow_entity.depth_test = False
        return lights

    def create_fixed_star_lights(self, entity):
        """
        创建恒星的发光的效果、并作为灯光源
        :param entity:
        :return:
        """

        # 如果是恒星（如：太阳），自身会发光，则需要关闭灯光
        entity.set_light_off()

        lights = []
        # 创建多个新的 Entity 对象，作为光晕的容器
        for i in range(10):
            glow_entity = Entity(parent=entity, model='sphere', color=color.rgba(1.0, 0.6, 0.2, 1),
                                 scale=math.pow(1.03, i), alpha=0.1)

            lights.append(glow_entity)
        for i in range(2):
            # 创建 PointLight 对象，作为恒星的灯光源
            light = PointLight(parent=entity, intensity=10, range=10, color=color.white)
            lights.append(light)

        # light = DirectionalLight(shadows=True, direction=Vec3(0, 0, 1), color=color.white)
        # light.look_at(Vec3(0, 0, -1))
        # light = SpotLight(parent=entity,shadows=True, direction=Vec3(1,1,1), color=color.white)

        return lights

    def switch_to_english_input_method(self):
        pass
        # os.system("powershell -Command \"Set-WinUserLanguageList -LanguageList 'en-US', 'zh-CN' -Force\"")

        # if os.name == 'nt':  # Windows 系统
        #     os.system('powershell -Command "Set-WinUserLanguageList en-US -Force"')
        # elif os.name == 'posix':  # macOS 或 Linux 系统
        #     os.system(
        #         'defaults write ~/Library/Preferences/com.apple.HIToolbox.plist AppleSelectedInputSources -array-add \'{ "InputSourceKind" = "Keyboard Layout"; "KeyboardLayout ID" = 252; "KeyboardLayout Name" = "ABC"; }\'')
        #     os.system('killall cfprefsd')  # 重启 CoreFoundation 进程使更改生效

    def run(self, dt, **kwargs):

        window.title = '宇宙模拟器'

        # 设置 camera 的裁剪面和位置
        # camera.clip_plane_near = 0.00000009
        # camera.fov = 120
        # camera.clip_plane_far = 1000
        # camera.position = (0, 10, -20)
        # camera.rotation_x = -30

        # 设定时间间隔为0.01秒
        self.interval_fator = 0.01
        self.evolve_dt = dt
        self.interval = datetime.timedelta(seconds=self.interval_fator)

        self.last_time = datetime.datetime.now() - datetime.timedelta(seconds=2)
        if "light" in kwargs:
            if kwargs["light"]:
                for v in self.ursina_views:
                    if v.body.is_fixed_star:
                        # self.lights = self.create_fixed_star_lights(v.planet)
                        pass

        if "show_grid" in kwargs:
            if kwargs["show_grid"]:
                WorldGrid()

        if "cosmic_bg" in kwargs:
            cosmic_bg = kwargs["cosmic_bg"]
            if cosmic_bg is None:
                # cosmic_bg = '../textures/cosmic1.png'
                # cosmic_bg = '../textures/cosmic2.jpg'
                cosmic_bg = '../textures/cosmic3.jpg'
            import os
            if cosmic_bg is not None and os.path.exists(cosmic_bg):
                self.cosmic_background(cosmic_bg)

        ui = UrsinaUI()
        EditorCamera(ignore_paused=True)
        # 防止打开中文输入法
        # self.switch_to_english_input_method()
        self.app.run()


if __name__ == '__main__':
    from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Moon
    from bodies.body import AU
    from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_HALF_DAY

    """
    3个太阳、1个地球
    """
    bodies = [
        Sun(name='太阳1', mass=1.5e30, init_position=[849597870.700, 0, 0], init_velocity=[0, 7.0, 0],
            size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
        Sun(name='太阳2', mass=2e30, init_position=[0, 0, 0], init_velocity=[0, -8.0, 0],
            size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
        Sun(name='太阳3', mass=2.5e30, init_position=[0, -849597870.700, 0], init_velocity=[18.0, 0, 0],
            size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
        Earth(name='地球', init_position=[0, -349597870.700, 0], init_velocity=[15.50, 0, 0],
              size_scale=4e3, texture="earth.png", distance_scale=1),  # 地球放大 4000 倍，距离保持不变
    ]
    # bodies = [
    #
    #     Sun(name='太阳2', mass=1.5e30, init_position=[0, 0, 0], init_velocity=[0, -8.0, 0],
    #         size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
    #     Sun(name='太阳2', mass=1.5e30, init_position=[849597870.700, 0, 0], init_velocity=[0, -8.0, 0],
    #         size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
    #     Sun(name='太阳2', mass=1.5e30, init_position=[0, -849597870.700, 0], init_velocity=[0, -8.0, 0],
    #         size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
    #     Earth(name='地球', mass=5.97237e24, init_position=[0, -349597870.700, 0], init_velocity=[15.50, 0, 0],
    #           size_scale=4e3, texture="earth.png", distance_scale=1),  # 地球放大 4000 倍，距离保持不变
    # ]
    bodies = [
        Sun(size_scale=0.8e2),  # 太阳放大 80 倍
        Mercury(size_scale=4e3, distance_scale=1.3),  # 水星放大 4000 倍，距离放大 1.3 倍
        Venus(size_scale=4e3, distance_scale=1.3),  # 金星放大 4000 倍，距离放大 1.3 倍
        Earth(init_position=[1.12 * AU, 0, 0],
              init_velocity=[0, 29.79, 0], size_scale=4e3, distance_scale=1.3),  # 地球放大 4000 倍，距离放大 1.3 倍
        Moon(init_position=[363104 + 1.12 * AU, 0, 0],
             init_velocity=[-9, 29.79 + 1.023, 0], size_scale=4e3, distance_scale=1.3),
        Mars(size_scale=4e3, distance_scale=1.3),  # 火星放大 4000 倍，距离放大 1.3 倍
        Jupiter(size_scale=0.68e3, distance_scale=0.65),  # 木星放大 680 倍，距离缩小到真实距离的 0.65
        Saturn(size_scale=0.68e3, distance_scale=0.52),  # 土星放大 680 倍，距离缩小到真实距离的 0.52
        Uranus(size_scale=0.8e3, distance_scale=0.36),  # 天王星放大 800 倍，距离缩小到真实距离的 0.36
        Neptune(size_scale=1e3, distance_scale=0.27),  # 海王星放大 1000 倍，距离缩小到真实距离的 0.27
        Pluto(size_scale=10e3, distance_scale=0.23),  # 冥王星放大 10000 倍，距离缩小到真实距离的 0.23(从太阳系的行星中排除)
    ]

    ursina_run(bodies, SECONDS_PER_DAY, position=(AU * 2, AU * 2, AU * 3), show_grid=True)