import math
import random
from solar_objects import Star, Planet


TICKET_CONFIG = {
    "window": {"width": 1200, "height": 720, "min_width": 850, "min_height": 560},
    "stars": [
        {
            "name": "Star 1",
            "planet_count": 11,
            "planets_per_orbit": 1,
            "orbit_radii": [70, 85, 100, 115, 130, 145, 160, 175, 190, 205, 220],
            "color": (255, 190, 40),
            "mass": 1.989e30,
        },
        {
            "name": "Star 2",
            "planet_count": 12,
            "planets_per_orbit": 2,
            "orbit_radii": [80, 110, 140, 170, 200, 230],
            "color": (255, 150, 55),
            "mass": 1.650e30,
        },
    ],
    "rules": {"even_orbit_direction": -1, "odd_orbit_direction": 1, "orbits_must_intersect": True},
}


SAFE_START_ANGLES = [
    3.009640, 2.737011, 5.484431, 2.979849, 1.049068, 1.478066,
    1.191136, 1.579170, 4.454420, 4.226254, 2.025695,
    0.705927, 3.488630, 2.293706, 6.173658, 0.440441, 5.055767,
]

class SolarSystem:
    def __init__(self, width=1200, height=720):
        self.width = width
        self.height = height
        self.stars = []
        self.show_orbits = True
        self.paused = False
        self.create_system()
        self.resize(width, height)

    def create_system(self):
        self.stars = []

        for star_index, star_data in enumerate(TICKET_CONFIG["stars"]):
            star = Star(
                0,
                0,
                radius=34,
                color=star_data["color"],
                name=star_data["name"],
                mass=star_data["mass"],
            )

            planet_count = star_data["planet_count"]
            planets_per_orbit = star_data["planets_per_orbit"]
            created = 0

            for orbit_number, orbit_radius in enumerate(star_data["orbit_radii"], start=1):
                direction = self.get_direction(orbit_number)

                # Ближние к звезде планеты двигаются быстрее, дальние — медленнее.
                # Скорости и начальные углы подобраны так, чтобы планеты не пересекались.
                angular_speed = (0.055 / (orbit_number ** 0.45)) * (1.0 if star_index == 0 else 0.945)

                # Начальные углы подобраны заранее: планеты не стартуют кучей и не встречаются на пересечениях орбит.
                angle_index = created if star_index == 0 else 11 + (orbit_number - 1)
                base_angle = SAFE_START_ANGLES[angle_index]

                for place in range(planets_per_orbit):
                    if created >= planet_count:
                        break

                    # Если на одной орбите две планеты, они стоят напротив друг друга.
                    angle = base_angle + place * (2 * math.pi / planets_per_orbit)
                    planet = Planet(
                        star,
                        orbit_number,
                        orbit_radius,
                        angle,
                        angular_speed,
                        direction,
                        random.uniform(3.0e24, 8.0e24),
                    )
                    star.add_planet(planet)
                    created += 1

            self.stars.append(star)

    def get_direction(self, orbit_number):
        if orbit_number % 2 == 0:
            return TICKET_CONFIG["rules"]["even_orbit_direction"]
        return TICKET_CONFIG["rules"]["odd_orbit_direction"]

    def resize(self, width, height):
        self.width = width
        self.height = height
        center_x = width // 2
        center_y = height // 2
        distance = 285
        self.stars[0].set_position(center_x - distance // 2, center_y)
        self.stars[1].set_position(center_x + distance // 2, center_y)
        for star in self.stars:
            for planet in star.planets:
                planet.x, planet.y = planet.calculate_position()

    def update(self):
        if self.paused:
            return
        for star in self.stars:
            star.update()

    def draw(self, screen, pygame):
        for star in self.stars:
            star.draw(screen, pygame, self.show_orbits)

    def toggle_pause(self):
        self.paused = not self.paused

    def toggle_orbits(self):
        self.show_orbits = not self.show_orbits

    def get_all_planets(self):
        planets = []
        for star in self.stars:
            planets.extend(star.planets)
        return planets

    def get_all_objects(self):
        objects = []
        for star in self.stars:
            objects.append(star)
            objects.extend(star.planets)
        return objects

    def get_total_mass(self):
        return sum(space_object.mass for space_object in self.get_all_objects())

    def get_planet_count(self):
        return sum(len(star.planets) for star in self.stars)