ursina_simulator.py 7.4 KB
Newer Older
三月三net's avatar
三月三net 已提交
1 2 3 4 5 6 7 8 9
# -*- 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
三月三net's avatar
三月三net 已提交
10
from ursina import Ursina, window, Entity, Grid, Mesh, camera, color, mouse, Vec2, Vec3, load_texture, held_keys
M
march3 已提交
11
from ursina.prefabs.first_person_controller import FirstPersonController
M
march3 已提交
12 13

from simulators.views.ursina_view import UrsinaView, UrsinaPlayer
三月三net's avatar
三月三net 已提交
14 15 16

from simulators.simulator import Simulator
from common.system import System
M
march3 已提交
17 18
import time
import datetime
三月三net's avatar
三月三net 已提交
19
from ursina import EditorCamera, PointLight, SpotLight, AmbientLight
三月三net's avatar
三月三net 已提交
20
from scenes.func import ursina_run
三月三net's avatar
三月三net 已提交
21

M
march3 已提交
22

三月三net's avatar
三月三net 已提交
23 24 25 26 27 28 29 30 31 32 33 34 35
class WorldGrid(Entity):  # Entity # 定义构造方法
    def __init__(self):
        super().__init__()
        s = 100
        grid = Entity(model=Grid(s, s), scale=s * 20, color=color.color(0, 0, .1, 1), 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()


三月三net's avatar
三月三net 已提交
36 37 38
class UrsinaSimulator(Simulator):
    def __init__(self, bodies_sys: System):
        self.app = Ursina()
M
march3 已提交
39
        self.ursina_views = []
三月三net's avatar
三月三net 已提交
40 41 42 43 44 45 46
        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
M
march3 已提交
47
        for i, view in enumerate(self.body_views):
三月三net's avatar
三月三net 已提交
48
            # pos = tuple(body.position)
M
march3 已提交
49 50
            # ursina_view = UrsinaView(body)
            view.update()
M
march3 已提交
51
            self.ursina_views.append(view)
三月三net's avatar
三月三net 已提交
52 53 54
            # planets.append(newPlanet)
            # x += cp[i] * 10

M
march3 已提交
55 56 57 58
    def check_elapsed_time(self):
        """检查时间间隔是否已过"""
        now = datetime.datetime.now()
        elapsed_time = now - self.last_time
M
march3 已提交
59 60 61 62
        value = elapsed_time >= self.interval
        if value:
            self.last_time = now
        return value
M
march3 已提交
63 64 65 66 67

    def check_and_evolve(self):
        if self.check_elapsed_time():
            super().evolve(self.evolve_dt)

三月三net's avatar
三月三net 已提交
68
    def cosmic_background(self, texture='../textures/cosmic1.jpg'):
三月三net's avatar
三月三net 已提交
69 70 71 72 73
        """
        加入宇宙背景
        :param texture:
        :return:
        """
三月三net's avatar
三月三net 已提交
74 75 76 77 78 79 80 81
        # Add skybox
        from ursina import Sky
        Sky(texture=texture)
        # 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))
三月三net's avatar
三月三net 已提交
82

三月三net's avatar
三月三net 已提交
83
    def run(self, dt, **kwargs):
三月三net's avatar
三月三net 已提交
84
        from ursina import EditorCamera, PointLight, SpotLight, AmbientLight, DirectionalLight
M
march3 已提交
85 86
        self.evolve_dt = dt
        # 设定时间间隔为1秒
M
march3 已提交
87 88
        self.interval = datetime.timedelta(seconds=0.01)
        self.last_time = datetime.datetime.now() - datetime.timedelta(seconds=2)
三月三net's avatar
三月三net 已提交
89 90
        if "light" in kwargs:
            if kwargs["light"]:
三月三net's avatar
三月三net 已提交
91 92
                for v in self.ursina_views:
                    if v.body.is_fixed_star:
三月三net's avatar
三月三net 已提交
93
                        v.light = PointLight(parent=v)
三月三net's avatar
三月三net 已提交
94 95 96 97 98
                # PointLight(parent=camera, color=color.white, position=(0, 0, 0))
                # AmbientLight(color=color.rgba(100, 100, 100, 0.1))
                # DirectionalLight
                # SpotLight

三月三net's avatar
三月三net 已提交
99 100 101 102
        if "show_grid" in kwargs:
            if kwargs["show_grid"]:
                WorldGrid()

三月三net's avatar
三月三net 已提交
103 104
        if "cosmic_bg" in kwargs:
            cosmic_bg = kwargs["cosmic_bg"]
三月三net's avatar
三月三net 已提交
105 106
            if cosmic_bg is None:
                cosmic_bg = '../textures/cosmic1.png'
三月三net's avatar
三月三net 已提交
107 108 109 110
            import os
            if cosmic_bg is not None and os.path.exists(cosmic_bg):
                self.cosmic_background(cosmic_bg)

三月三net's avatar
三月三net 已提交
111
        EditorCamera()
三月三net's avatar
三月三net 已提交
112 113
        self.app.run()

M
march3 已提交
114

三月三net's avatar
三月三net 已提交
115
if __name__ == '__main__':
M
march3 已提交
116
    from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Moon
三月三net's avatar
三月三net 已提交
117
    from bodies.body import AU
三月三net's avatar
三月三net 已提交
118
    from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_HALF_DAY
三月三net's avatar
三月三net 已提交
119 120 121 122 123

    """
    3个太阳、1个地球
    """
    bodies = [
M
march3 已提交
124
        Sun(name='太阳1', mass=1.5e30, init_position=[849597870.700, 0, 0], init_velocity=[0, 7.0, 0],
三月三net's avatar
三月三net 已提交
125
            size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
M
march3 已提交
126
        Sun(name='太阳2', mass=2e30, init_position=[0, 0, 0], init_velocity=[0, -8.0, 0],
三月三net's avatar
三月三net 已提交
127
            size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
M
march3 已提交
128
        Sun(name='太阳3', mass=2.5e30, init_position=[0, -849597870.700, 0], init_velocity=[18.0, 0, 0],
三月三net's avatar
三月三net 已提交
129
            size_scale=5e1, texture="sun.png"),  # 太阳放大 100 倍
M
march3 已提交
130
        Earth(name='地球', init_position=[0, -349597870.700, 0], init_velocity=[15.50, 0, 0],
三月三net's avatar
三月三net 已提交
131 132
              size_scale=4e3, texture="earth.png", distance_scale=1),  # 地球放大 4000 倍,距离保持不变
    ]
M
march3 已提交
133 134 135 136 137 138 139 140 141 142 143
    # 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 倍,距离保持不变
    # ]
M
march3 已提交
144
    bodies = [
三月三net's avatar
三月三net 已提交
145 146 147
        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 倍
三月三net's avatar
三月三net 已提交
148 149 150
        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],
三月三net's avatar
三月三net 已提交
151
             init_velocity=[-9, 29.79 + 1.023, 0], size_scale=4e3, distance_scale=1.3),
三月三net's avatar
三月三net 已提交
152
        Mars(size_scale=4e3, distance_scale=1.3),  # 火星放大 4000 倍,距离放大 1.3 倍
M
march3 已提交
153
        Jupiter(size_scale=0.68e3, distance_scale=0.65),  # 木星放大 680 倍,距离缩小到真实距离的 0.65
三月三net's avatar
三月三net 已提交
154 155 156 157
        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(从太阳系的行星中排除)
M
march3 已提交
158
    ]
M
march3 已提交
159

三月三net's avatar
三月三net 已提交
160
    ursina_run(bodies, SECONDS_PER_DAY, position=(AU * 2, AU * 2, AU * 3), show_grid=True)