# -*- coding:utf-8 -*-
# title           :天狼星进入太阳系场景模拟
# description     :天狼星把土星带走了
# author          :Python超人
# date            :2023-02-11
# link            :https://gitcode.net/pythoncr/
# python_version  :3.8
# ==============================================================================
from bodies import Sirius, Sun, Mercury, Venus, Earth, \
    Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Moon, Asteroids
from common.celestial_data_service import init_bodies_pos_vels
from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_MONTH, AU
from sim_scenes.func import ursina_run, create_sphere_sky
from ursina import camera, application

from simulators.ursina.ursina_event import UrsinaEvent

if __name__ == '__main__':
    #  由于宇宙空间尺度非常大，如果按照实际的天体大小，则无法看到天体，因此需要对天体的尺寸进行放大
    sun = Sun(name="太阳",
              init_velocity=[0, 2, 0],
              size_scale=0.8e2)  # 太阳放大 80 倍，距离保持不变
    sirius = Sirius(name="天狼星",
                    init_position=[-10 * AU, 4 * AU, 4 * AU],
                    init_velocity=[20, -10, 0],
                    size_scale=0.8e2)  # 天狼星放大 80 倍，距离保持不变
    bodies = [
        sirius,
        sun,
        Mercury(name="水星", size_scale=4e3),  # 水星放大 4000 倍，距离保持不变
        Venus(name="金星", size_scale=4e3),  # 金星放大 4000 倍，距离保持不变
        Earth(name="地球", size_scale=4e3),  # 地球放大 4000 倍，距离保持不变
        Mars(name="火星", size_scale=4e3),  # 火星放大 4000 倍，距离保持不变
        # Asteroids(name="小行星群", size_scale=1e2,
        #           parent=sun),                     # 小行星群模拟(仅 ursina 模拟器支持)
        Jupiter(name="木星", size_scale=0.8e3),  # 木星放大 800 倍，距离保持不变
        Saturn(name="土星", size_scale=0.8e3),  # 土星放大 800 倍，距离保持不变
        Uranus(name="天王星", size_scale=0.8e3),  # 天王星放大 800 倍，距离保持不变
        Neptune(name="海王星", size_scale=1e3),  # 海王星放大 1000 倍，距离保持不变
        Pluto(name="冥王星", size_scale=10e3),  # 冥王星放大 10000 倍，距离保持不变(从太阳系的行星中排除)
    ]

    init_bodies_pos_vels(bodies)

    def on_ready():
        """
        事件绑定后，模拟器运行前会触发
        @return:
        """
        # 创建天空
        # camera.clip_plane_near = 0.1
        camera.clip_plane_far = 1000000
        create_sphere_sky(scale=200000)
        application.time_scale = 5

    # 运行前会触发 on_ready
    UrsinaEvent.on_ready_subscription(on_ready)

    # 使用 ursina 查看的运行效果
    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
    # position = 左-右+、上+下-、前+后-
    ursina_run(bodies, SECONDS_PER_MONTH,
               position=(0, 2 * AU, -11 * AU),
               cosmic_bg='',
               show_grid=False,
               # bg_music="sounds/interstellar.mp3"
               bg_music='sounds/no_glory.mp3',
               )