# -*- coding:utf-8 -*-
# title           :三体场景模拟02
# 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_DAY, SECONDS_PER_WEEK, SECONDS_PER_YEAR, AU
from sim_scenes.func import mayavi_run, mpl_run, ursina_run

if __name__ == '__main__':
    """
    注释： 3个太阳（ChatGPT生成的效果）
    可以修改影响效果的参数为： 
    1、三个方向的初始位置 init_position[x, y, z]
    2、三个方向的初始速度 init_velocity[x, y, z]
    3、天体质量 mass    
    """
    # 代码案例如下：
    SIZE_SCALE = 5e1  # 生成的太阳放大 50 倍
    RUN_DIAMETER = 1.392e6  # 真实太阳的直径
    # 恒星A：质量 = 1.0, 速度 = (2.0, 3.0, 4.0), 坐标 = (5.0, 6.0, 7.0)
    # 恒星B：质量 = 1.5, 速度 = (-1.0, -2.0, -3.0), 坐标 = (8.0, 9.0, 10.0)
    # 恒星C：质量 = 2.0, 速度 = (1.5, 2.5, 3.5), 坐标 = (11.0, 12.0, 13.0)
    bodies = [
        Sun(name="太阳1", mass=1e30, init_position=[500 * RUN_DIAMETER, 600 * RUN_DIAMETER, 700 * RUN_DIAMETER],
            init_velocity=[2, 3, 4],
            size_scale=SIZE_SCALE, texture="sun1.jpg"),
        Sun(name="太阳2", mass=1.5e30, init_position=[800 * RUN_DIAMETER, 900 * RUN_DIAMETER, 1000 * RUN_DIAMETER],
            init_velocity=[-1, -2, -3],
            size_scale=SIZE_SCALE, texture="sun2.jpg"),
        Sun(name="太阳3", mass=2e30, init_position=[1100 * RUN_DIAMETER, 1200 * RUN_DIAMETER, 1300 * RUN_DIAMETER],
            init_velocity=[1.5, 2.5, 3.5],
            size_scale=SIZE_SCALE, texture="sun2.jpg"),
    ]

    # 按照以上代码案例格式生成代码，要求 init_position 、init_velocity、mass 生成后，要保证在万有引力的作用下，能正常运行，不会碰撞
    """
太阳1：
    初始位置：(100000, 200000, 300000) km
    初始速度：(-1.0, -2.0, 3.0) km/s
    质量：2.5 x 10^30 kg

太阳2：
    初始位置：(-150000, 250000, 350000) km
    初始速度：(2.0, -3.0, -1.0) km/s
    质量：2.0 x 10^30 kg

太阳3：
    初始位置：(200000, -300000, 400000) km
    初始速度：(-3.0, 1.0, -2.0) km/s
    质量：2.8 x 10^30 kg
    """
    # 使用 mayavi 查看的运行效果
    # mayavi_run(bodies, SECONDS_PER_WEEK, view_azimuth=0)

    # 使用 matplotlib 查看运行效果
    # mpl_run(bodies, SECONDS_PER_WEEK)

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