# -*- coding:utf-8 -*-
# title           :哈雷彗星场景模拟
# description     :哈雷彗星场景模拟
# author          :Python超人
# date            :2023-10-25
# link            :https://gitcode.net/pythoncr/
# python_version  :3.9
# ==============================================================================
import time

from ursina import camera, application

from bodies import Sun, Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune, Pluto
from common.celestial_data_service import get_init_pos_vels, init_bodies_pos_vels
from common.consts import SECONDS_PER_WEEK, AU
from objs import QuadObj, CircleObj, Obj
from sim_scenes.func import camera_look_at, two_bodies_colliding
from sim_scenes.func import ursina_run, create_sphere_sky
from simulators.ursina.ursina_event import UrsinaEvent


class HalleyCometSim:
    """
    哈雷彗星场景模拟
    """

    def __init__(self):
        pass

    def build_solar_system(self):
        # region 构建太阳系
        self.sun = Sun(size_scale=0.8e2)
        self.bodies = [
            self.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, rotate_angle=0),  # 地球放大 4000 倍，距离放大 1.3 倍
            Mars(size_scale=4e3, distance_scale=1.2),  # 火星放大 4000 倍，距离放大 1.2 倍
            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=1.5e3, distance_scale=0.36),  # 天王星放大 1500 倍，距离缩小到真实距离的 0.36
            Neptune(size_scale=1.5e3, distance_scale=0.27),  # 海王星放大 1500 倍，距离缩小到真实距离的 0.27
            Pluto(size_scale=10e3, distance_scale=0.23),  # 冥王星放大 10000 倍，距离缩小到真实距离的 0.23(从太阳系的行星中排除)
        ]
        # endregion

        init_bodies_pos_vels(self.bodies)

    def build_halley_comet(self):
        """
        创建哈雷彗星
        @return:
        """
        self.halley_comet = QuadObj(texture='two_way_foil.png',
                                    # size_scale=4e7,
                                    size_scale=1e7,
                                    init_velocity=[20, -6, 150],
                                    init_position=[0, 0.5 * AU, -10 * AU]) \
            .set_light_disable(True)

        self.bodies.append(self.halley_comet)


    def build(self):
        self.build_solar_system()
        self.build_halley_comet()

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



    def on_timer_changed(self, time_data):
        """

        @param time_data:
        @return:
        """
        # 原始方形二向箔飞行的翻转效果
        if self.halley_comet.planet.enabled:
            self.halley_comet.planet.rotation_x += 0.1
            self.halley_comet.planet.rotation_y += 1
        # 摄像机始终看向二向箔
        camera_look_at(self.halley_comet)



if __name__ == '__main__':
    """
    三体二向箔场景模拟
    """
    sim = HalleyCometSim()
    sim.build()
    # 订阅事件后，上面2个函数功能才会起作用
    # 运行中，每时每刻都会触发 on_timer_changed
    UrsinaEvent.on_timer_changed_subscription(sim.on_timer_changed)
    # 运行前会触发 on_ready
    UrsinaEvent.on_ready_subscription(sim.on_ready)
    # 使用 ursina 查看的运行效果
    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
    # position = 左-右+、上+下-、前+后-
    ursina_run(sim.bodies,
               SECONDS_PER_WEEK,
               # position=(0, 2 * AU, -11 * AU),
               position=(0, 0.5 * AU, -5 * AU),
               cosmic_bg='',
               bg_music='sounds/no_glory.mp3',
               show_camera_info=False,
               show_control_info=False,
               timer_enabled=True,
               show_grid=False)