# -*- coding:utf-8 -*-
# title           :三体场景模拟01
# description     :三体场景模拟（3个太阳、1个地球）
# author          :Python超人
# date            :2023-02-11
# link            :https://gitcode.net/pythoncr/
# python_version  :3.8
# ==============================================================================
from bodies import Sun, Earth
from common.consts import SECONDS_PER_WEEK, SECONDS_PER_YEAR, SECONDS_PER_MONTH, AU
from sim_scenes.func import mayavi_run, ursina_run, create_sphere_sky
from simulators.ursina.entities.body_timer import TimeData
from simulators.ursina.ursina_config import UrsinaConfig
from simulators.ursina.ursina_event import UrsinaEvent
from ursina import application, window, camera, EditorCamera
import numpy as np

from simulators.ursina_simulator import UrsinaSimulator

if __name__ == '__main__':
    """
    3个太阳、1个地球（效果1）
    可以修改影响效果的参数为： 
    1、三个方向的初始位置 init_position[x, y, z]
    2、三个方向的初始速度 init_velocity[x, y, z]
    3、天体质量 mass    
    """
    bodies = [
        Sun(mass=1.55e30, init_position=[8.5e8, 0, 1000000], init_velocity=[0.0, 7.1, 0],
            size_scale=5e1, texture="sun21.jpg", color=(111, 140, 255)),  # 蓝色太阳
        Sun(mass=2.1e30, init_position=[0, 0, 0], init_velocity=[0, -8.0, 0],
            size_scale=5e1, texture="sun21.jpg", color=(250, 195, 47)),  # 红色太阳
        Sun(mass=2.5e30, init_position=[0, -8.5e8, -2000000], init_velocity=[12.0, 0, 0],
            size_scale=5e1, texture="sun21.jpg", color=(198, 29, 3)),  # 黄色太阳
        Earth(init_position=[0, -3.5e8, 3000000], init_velocity=[13.360, 0, 0], rotate_angle=0,
              size_scale=4e3, distance_scale=1),  # 地球放大 4000 倍，距离保持不变
    ]


    # 使用 mayavi 查看的运行效果
    # mayavi_run(bodies, SECONDS_PER_WEEK, view_azimuth=0)
    def on_ready():
        from ursina import Text, color
        window.borderless = False
        sky = create_sphere_sky(scale=80000, texture="bg_pan_deep_blue.jpg")
        camera.clip_plane_near = 1
        camera.clip_plane_far = sky.scale_x * 1.5
        font = "fonts/DroidSansFallback.ttf"
        from common.func import find_file
        font = find_file(f"{font}", UrsinaConfig.CN_FONT)

        text = Text(text="万有引力模拟三体", color=color.white,scale=1.5, position=(-0.85, 0.48),
                    font=font)


    def on_timer_changed(time_data: TimeData):
        total_days = time_data.total_days
        ec = UrsinaSimulator.EditorCamera
        print(total_days)
        if 100 < total_days < 3000:
            camera.position += camera.right * 0.1
            camera.position += camera.down * 0.01
        elif 3200 < total_days < 3600:
            ec.target_z -= 7.5
        elif 6000 < total_days < 7500:
            camera.position += camera.right * 1
            camera.position += camera.down * 0.5
        elif 10800 < total_days < 12000:
            camera.position += camera.right * 1
        # elif total_days > 10765:
        elif 12800 < total_days < 16000:
            ec.target_z -= 5
            camera.position += camera.up * 0.3
        elif total_days > 17000:
            UrsinaEvent.on_reset()
            # camera.position = UrsinaConfig.SCALE_FACTOR * np.array([10 * AU, -2 * AU, -20 * AU]),
        # elif total_days > 3500:
        #     camera.position += camera.back * 0.1


    def on_reset():
        reset_camera()


    def reset_camera():
        # 参考 EditorCamera 中的 elif key == 'scroll up':
        ec = UrsinaSimulator.EditorCamera
        if not hasattr(ec, "original_target_z"):
            ec.original_target_z = ec.target_z
            camera.original_position = camera.position
        else:
            ec.target_z = ec.original_target_z
            camera.position = camera.original_position

    UrsinaConfig.trail_type = "line"
    # UrsinaConfig.trail_length = 91
    # UrsinaConfig.trail_type = "curve_line"
    UrsinaConfig.trail_length = 50
    # UrsinaConfig.trail_length = 1000
    UrsinaConfig.trail_thickness_factor = 2
    # 运行前会触发 on_ready
    UrsinaEvent.on_ready_subscription(on_ready)
    # 运行中，每时每刻都会触发 on_timer_changed
    UrsinaEvent.on_timer_changed_subscription(on_timer_changed)

    UrsinaEvent.on_reset_subscription(on_reset)

    # 使用 ursina 查看的运行效果
    # 常用快捷键： P：运行和暂停  O：重新开始  I：显示天体轨迹
    # position = 左-右+、上+下-、前+后-
    ursina_run(bodies, SECONDS_PER_YEAR / 3,
               cosmic_bg='',
               show_grid=False,
               # position=(20 * AU, 0, -40 * AU),
               position=(10 * AU, -2 * AU, -20 * AU),
               show_camera_info=False,
               show_control_info=False,
               show_trail=True,
               timer_enabled=True)