# -*- coding:utf-8 -*-
# title           :三体二向箔场景模拟
# description     :三体二向箔场景模拟
# author          :Python超人
# date            :2023-10-11
# link            :https://gitcode.net/pythoncr/
# python_version  :3.9
# ==============================================================================


from bodies import Sun, Earth, Moon
from objs import CoreValagaClas, SciFiBomber, WaterDrop, QuadObj
from common.consts import AU, SECONDS_PER_DAY, SECONDS_PER_WEEK, SECONDS_PER_MONTH
from sim_scenes.func import ursina_run, camera_look_at, create_main_entity, two_bodies_colliding
from simulators.ursina.entities.body_timer import TimeData
from simulators.ursina.entities.entity_utils import create_directional_light
from simulators.ursina.ursina_config import UrsinaConfig
from simulators.ursina.ursina_event import UrsinaEvent
from ursina import camera

if __name__ == '__main__':
    """
    三体二向箔场景模拟
    """
from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto, Asteroids
from common.consts import SECONDS_PER_WEEK, SECONDS_PER_DAY, SECONDS_PER_MONTH, SECONDS_PER_YEAR, AU
from sim_scenes.func import mayavi_run, ursina_run, create_sphere_sky
from simulators.ursina.ursina_event import UrsinaEvent

if __name__ == '__main__':
    # 八大行星：木星(♃)、土星(♄)、天王星(♅)、海王星(♆)、地球(⊕)、金星(♀)、火星(♂)、水星(☿)
    # 排列顺序
    # 1、体积：(以地球为1)木星 ：土星 ：天王星 ：海王星 ：地球 ：金星 ：火星 ：水星 = 1330：745：65：60：1：0.86：0.15：0.056
    # 2、质量：(以地球为1)木星 ：土星 ：天王星 ：海王星 ：地球 ：金星 ：火星 ：水星 = 318：95：14.53：17.15：1：0.8：0.11：0.0553
    # 3、离太阳从近到远的顺序：水星、金星、地球、火星、木星、土星、天王星、海王星
    #  =====================================================================
    #  以下展示的效果非太阳系真实的距离和大小
    #  1、由于宇宙空间尺度非常大，如果按照实际的天体大小，则无法看到天体，因此需要对天体的尺寸进行放大
    #  2、为了达到最佳的显示效果，对每个行星天体的距离进行了缩放

    # region 构建太阳系
    sun = Sun(size_scale=0.8e2)
    bodies = [
        sun,  # 太阳放大 80 倍
        Mercury(size_scale=4e3, distance_scale=1.3),  # 水星放大 4000 倍，距离放大 1.3 倍
        Venus(size_scale=4e3, distance_scale=1.3),  # 金星放大 4000 倍，距离放大 1.3 倍
        Earth(size_scale=4e3, distance_scale=1.3),  # 地球放大 4000 倍，距离放大 1.3 倍
        Mars(size_scale=4e3, distance_scale=1.2),  # 火星放大 4000 倍，距离放大 1.2 倍
        Asteroids(size_scale=1e2, parent=sun),  # 小行星模拟(仅 ursina 模拟器支持)
        Jupiter(size_scale=0.68e3, distance_scale=0.72),  # 木星放大 680 倍，距离缩小到真实距离的 0.72
        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(从太阳系的行星中排除)
    ]
    # endregion

    two_way_foil = QuadObj(texture='caustic.jpg', size_scale=2e8,
                           init_velocity=[-10, -10, 0],
                           init_position=[0, 0, 39.55 * AU]) \
        .set_light_disable(True).set_ignore_gravity(True)

    bodies.append(two_way_foil)


    def on_ready():
        """
        事件绑定后，模拟器运行前会触发
        @return:
        """
        # 创建天空
        create_sphere_sky(scale=8000)


    def on_timer_changed(time_data):
        two_way_foil.planet.rotation_x += 1
        two_way_foil.planet.rotation_y += 2
        pass


    # 订阅事件后，上面2个函数功能才会起作用
    # 运行中，每时每刻都会触发 on_timer_changed
    UrsinaEvent.on_timer_changed_subscription(on_timer_changed)
    # 运行前会触发 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='',
               timer_enabled=True,
               show_grid=False)