import math
import turtle
import random

window = turtle.Screen()
window.setup(1200 + 3, 800 + 3)
#window.bgcolor('blue')
window.bgpic("images/background.png")
window.screensize(1200, 800)
# window.tracer(n=2)

BASE_X, BASE_Y = 0, -300

our_missiles = []
enemy_missiles = []

def calc_distance(x1, y1, x2, y2):
    dx = x2 - x1
    dy = y2 - y1
    length = (dx ** 2 + dy ** 2) ** 0.5
    return length


def calc_heading(x1, y1, x2, y2):
    dx = x2 - x1
    dy = y2 - y1
    length = (dx ** 2 + dy ** 2) ** 0.5
    cos_alpha = dx / length
    alpha = math.acos(cos_alpha)
    alpha = math.degrees(alpha)
    if dy < 0:
        alpha = -alpha
    return alpha


def fire_missile(x, y):
    missile = turtle.Turtle(visible=False)
    missile.speed(0)
    missile.color('white')
    missile.penup()
    missile.setpos(x=BASE_X, y=BASE_Y)
    missile.pendown()
    heading = calc_heading(x1=BASE_X, y1=BASE_Y, x2=x, y2=y)
    missile.setheading(heading)
    missile.showturtle()
    info = {'missile': missile, 'target': [x, y],
            'state': 'launched', 'radius': 0}
    our_missiles.append(info)


def launch_enemy_rocket(x,y):
    missile = turtle.Turtle(visible=False)
    missile.speed(0)
    missile.color('red')
    missile.penup()
    missile.setpos(x=x, y=y)
    missile.pendown()
    heading = calc_heading(x1=x, y1=y, x2=BASE_X, y2=BASE_Y)
    missile.setheading(heading)
    missile.showturtle()
    info = {'missile': missile, 'target': [BASE_X, BASE_Y],
            'state': 'launched', 'radius': 0}
    enemy_missiles.append(info)


for rocket in range(6):
    launch_enemy_rocket(random.randint(-500, 500), 300)

window.onclick(fire_missile)

while True:
    window.update()

    for info in our_missiles:
        state = info['state']
        missile = info['missile']
        if state == 'launched':
            missile.forward(4)
            target = info['target']
            if missile.distance(x=target[0], y=target[1]) < 20:
                info['state'] = 'explode'
                missile.shape('circle')
        elif state == 'explode':
            missile.shape('circle')
            info['radius'] += 1
            if info['radius'] > 5:
                missile.clear()
                missile.hideturtle()
                info['state'] = 'dead'
            else:
                missile.shapesize(info['radius'])

    dead_missiles = [info for info in our_missiles if info['state'] == 'dead']
    for dead in dead_missiles:
        our_missiles.remove(dead)

    for info in enemy_missiles:
        state = info['state']
        missile = info['missile']
        if state == 'launched':
            missile.forward(4)
            target = info['target']
            cur_pos = missile.pos()
            for our_missile in our_missiles:
                our_mis = our_missile['missile']
                our_pos = our_mis.pos()
                if calc_distance(cur_pos[0], cur_pos[1], our_pos[0], our_pos[1]) < 20:
                    our_missile['state'] = 'explode'
                    info['state'] = 'dead'
                    missile.clear()
                    missile.hideturtle()
            if missile.distance(x=BASE_X, y=BASE_Y) < 20:
                info['state'] = 'explode'
                missile.shape('circle')
        elif state == 'explode':
            info['radius'] += 1
            if info['radius'] > 5:
                missile.clear()
                missile.hideturtle()
                info['state'] = 'dead'
            else:
                missile.shapesize(info['radius'])

    dead_missiles = [info for info in enemy_missiles if info['state'] == 'dead']
    for dead in dead_missiles:
        enemy_missiles.remove(dead)