import pygame
import math
import time
from datetime import datetime

# to debug, we'll print to log
########## having issues with this, works on one machine but not another
# import sys
# sys.stdout = open('C:/breakout/output.txt', 'w')
# print('test')


# Define some colors
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
OFFWHITE = (237, 240, 245)
LIGHTBLUE = (73, 122, 204)
GREEN = (0, 255, 0)
RED = (255, 0, 0)

paddle_y_pos = 400
size = [700, 500]
points_dic = {1: WHITE, 2: GREEN, 3: RED}
target_dic = {}
reset = False
lives = 1
score = 0

def reset_game(screen):
    font = pygame.font.Font('freesansbold.ttf', 32)
    text1 = font.render('GAME OVER', True, WHITE, BLACK)
    text2 = font.render('A TO RESTART', True, WHITE, BLACK)
    text3 = font.render('X TO EXIT', True, WHITE, BLACK)
    text1_rect = text1.get_rect()
    text2_rect = text2.get_rect()
    text3_rect = text3.get_rect()
    screen.fill(BLACK)
    text1_rect.center = size[0]/2, size[1]/2 - 100
    text2_rect.center = size[0]/2, size[1]/2
    text3_rect.center = size[0]/2, size[1]/2 + 100
    screen.blit(text1, text1_rect)
    screen.blit(text2, text2_rect)
    screen.blit(text3, text3_rect)
    pygame.display.flip()
    return True

def print_score():
    font = pygame.font.Font('freesansbold.ttf', 10)
    score_text = font.render('{}'.format(score), True, WHITE, BLACK)
    score_text_rect = score_text.get_rect()
    score_text_rect.center = size[0]/2, 495
    screen.blit(score_text, score_text_rect)

def get_target_area(target_dic):
    # x0 leftmost, x1 rightmost
    # y0 topmost, y1 bottommost
    target_area = {}
    for key in target_dic:
        x0y0 = target_dic[key].x_pos, target_dic[key].y_pos
        x0y1 = target_dic[key].x_pos, target_dic[key].y_pos + target_dic[key].height
        x1y0 = target_dic[key].x_pos + target_dic[key].width, target_dic[key].y_pos
        x1y1 = target_dic[key].x_pos + target_dic[key].width, target_dic[key].y_pos + target_dic[key].height
        target_area[key] = (x0y0, x0y1, x1y0, x1y1)
    return target_area


def initialize_screen(level=1):
    if level == 1:
        for i in range(6):
            target_dic[i] = Target(70 * i, 50, 2, i)


def populate_screen(level=1):
    for key in target_dic:
        target_dic[key].draw(screen)
    pygame.draw.rect(screen, WHITE, [0, size[1] - 20, size[0], 10])
    for i in range(0, ball.lives - 1):
        pygame.draw.circle(screen, WHITE, (5 + 10*i, size[1] - 5), 5)


class Target:
    def __init__(self, x_pos=0, y_pos=0, points=1, key=0, width=70, height=10):
        self.points = points
        self.x_pos = x_pos
        self.y_pos = y_pos
        self.width = width
        self.height = height
        self.color = points_dic[self.points]
        self.key = key

    def draw(self, screen):
        pygame.draw.rect(screen, LIGHTBLUE, [self.x_pos, self.y_pos, self.width, self.height])
        pygame.draw.rect(screen, self.color, [self.x_pos + 2, self.y_pos + 2, self.width - 4, self.height - 4])

    def collision(self):
        # print('collision! on key {}'.format(self.key))
        # print('upper x:{}, upper y:{}, lower x:{}, lower y:{}'.format(self.x_pos, self.y_pos, self.x_pos +
        #                                                               self.width, self.y_pos + self.height))
        self.points -= 1
        if self.points == 0:
            target_dic.pop(self.key)
            return
        self.color = points_dic[self.points]


class Paddle:
    def __init__(self, x_pos=300, y_pos=paddle_y_pos, width=100, height=10, color=WHITE):
        self.x_pos = x_pos
        self.y_pos = y_pos
        self.width = width
        self.height = height
        self.color = color
        self.speed = 0

    def draw(self, screen):
        pygame.draw.rect(screen, self.color, [self.x_pos, self.y_pos, self.width, self.height])

    def move(self):
        self.x_pos = self.x_pos + self.speed
        if self.x_pos <= 0:
            self.x_pos = 0
        if self.x_pos >= 700 - self.width:
            self.x_pos = 700 - self.width

class Ball:
    def __init__(self, x_pos=200, y_pos=200, width=10, height=10, color=WHITE, speed=5, angle=.75*math.pi, lives=3):
        self.x_pos = x_pos
        self.y_pos = y_pos
        self.width = width
        self.height = height
        self.color = color
        self.speed = [speed*math.cos(angle), speed*math.sin(angle)]
        self.angle = angle
        self.lives = lives

    def check_lose(self):
        return self.y_pos + self.height >= size[1] - 20

    def reset(self):
        self.x_pos = 200
        self.y_pos = 200
        self.angle = .75*math.pi

    def draw(self, screen):
        pygame.draw.rect(screen, self.color, [self.x_pos, self.y_pos, self.width, self.height])

    def move(self):
        self.x_pos = self.x_pos + self.speed[0]*math.cos(self.angle)
        self.y_pos = self.y_pos + self.speed[1]*math.sin(self.angle)
        if self.x_pos <= 0:
            self.x_pos = 0
        if self.x_pos >= 700 - self.width:
            self.x_pos = 700 - self.width
        if self.x_pos >= 700 - self.width or self.x_pos <= 0:
            self.speed[0] = self.speed[0] * -1
        if self.y_pos <= 0:
            self.y_pos = 0
        if self.y_pos >= 500 - self.width:
            self.y_pos = 500 - self.width
        if self.y_pos >= 500 - self.width or self.y_pos <= 0:
            self.speed[1] = self.speed[1] * -1


def check_paddle(paddle, ball):
    if ball.y_pos + ball.height >= paddle.y_pos:
    # if (paddle.x_pos - ball.width <= ball.x_pos <= paddle.x_pos + paddle.width) and ball.y_pos + ball.height >= paddle.y_pos:
        ball.speed[1] *= -1


def check_target(target_area, ball):
    global score
    v, h = 1, 1
    collision = False
    # print('starting list: {}'.format('.'.join([str(x) for x in list(target_dic.keys())])))
    for key in list(target_dic.keys()):
        collision_0 = False
        if target_dic[key].x_pos - ball.width <= ball.x_pos <= target_dic[key].x_pos + ball.width + target_dic[
            key].width and (
                target_dic[key].y_pos - ball.height <= ball.y_pos <= target_dic[key].y_pos or
                target_dic[key].y_pos + target_dic[key].height - abs(ball.speed[1]) <= ball.y_pos <=
                target_dic[key].y_pos + target_dic[key].height):
            print('collision detected - vertical')
            print('ball location: x_pos = {}, y_pos = {}'.format(ball.x_pos, ball.y_pos))
            print('target location: x_pos = {}, y_pos = {}, width = {}, height = {}'.format(target_dic[key].x_pos,
                                                                                            target_dic[key].y_pos,
                                                                                            target_dic[key].width,
                                                                                            target_dic[key].height))
            if target_dic[key].y_pos - ball.height <= ball.y_pos <= target_dic[key].y_pos:
                print('top')
            if target_dic[key].y_pos + target_dic[key].height - abs(ball.speed[1]) <= ball.y_pos <= target_dic[key].y_pos + target_dic[key].height:
                print('bottom')
            # print('test used to detect collision:')
            # print('(target_dic[key].x_pos - ball.width <= ball.x_pos <= target_dic[key].x_pos + paddle.width) and ball.y_pos <= target_dic[key].y_pos + target_dic[key].height')
            # print('subbed values:')
            # print('{} - {} <= {} <= {} + {}) and {} <= {} + {}'.format(target_dic[key].x_pos, ball.width, ball.x_pos, target_dic[key].x_pos, paddle.width, ball.y_pos, target_dic[key].y_pos, target_dic[key].height))

            collision = True
            collision_0 = True
            score += 1
            v = -1
        if target_dic[key].y_pos - ball.height <= ball.y_pos <= target_dic[key].y_pos + target_dic[key].height + ball.height \
            and (target_dic[key].x_pos - ball.width <= ball.x_pos <= target_dic[key].x_pos
                 or target_dic[key].x_pos + target_dic[key].width - abs(ball.speed[0]) <= ball.x_pos <= target_dic[key].x_pos + target_dic[key].width):
            print('collision detected - horizontal')
            print('ball location: x_pos = {}, y_pos = {}'.format(ball.x_pos, ball.y_pos))
            print('target location: x_pos = {}, y_pos = {}, width = {}, height = {}'.format(target_dic[key].x_pos,
                                                                                            target_dic[key].y_pos,
                                                                                            target_dic[key].width,
                                                                                            target_dic[key].height))
            collision = True
            collision_0 = True
            score += 1
            h = -1
            # if target_dic[key].x_pos - ball.width <= ball.x_pos <= target_dic[key].x_pos:
            #     print('1')
            # if target_dic[key].x_pos + target_dic[key].width - ball.width <= ball.x_pos <= target_dic[key].x_pos + target_dic[key].width:
            #     print('2')
        if collision_0:
            print('removing target {}'.format(key))
            target_dic[key].collision()
    print('ending list: {}'.format('.'.join([str(x) for x in list(target_dic.keys())])))
    print('exiting out of target loop')
    return collision, v, h


# Setup

pygame.init()

# Set the width and height of the screen [width,height]

screen = pygame.display.set_mode(size)

pygame.display.set_caption("My Game")

# Loop until the user clicks the close button.
done = False

# Used to manage how fast the screen updates
clock = pygame.time.Clock()

# Hide the mouse cursor
pygame.mouse.set_visible(0)

paddle = Paddle()
ball = Ball()
initialize_screen(1)

# find target with max y_pos
max_y_pos = 0
for key in target_dic:
    if target_dic[key].y_pos > max_y_pos:
        max_y_pos = target_dic[key].y_pos + target_dic[key].height + max(20, target_dic[key].height)


i = 0
# -------- Main Program Loop -----------
while not done:
    i += 1
    # --- Event Processing
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            done = True
            # User pressed down on a key

        elif event.type == pygame.KEYDOWN:
            # Figure out if it was an arrow key. If so
            # adjust speed.
            if event.key == pygame.K_LEFT:
                paddle.speed = -6
            elif event.key == pygame.K_RIGHT:
                paddle.speed = 6
            if reset and event.key == pygame.K_a:
                ball.lives = lives
                reset = False
                lose = False
                score = 0
                ball.reset()
            if reset and event.key == pygame.K_x:
                done = True

        # User let up on a key
        elif event.type == pygame.KEYUP:
            # If it is an arrow key, reset vector back to zero
            if event.key == pygame.K_LEFT or event.key == pygame.K_RIGHT:
                paddle.speed = 0

    # --- Game Logic
    lose = ball.check_lose()
    if lose:
        time.sleep(1)
        ball.lives -= 1
        if ball.lives > 0:
            ball.reset()
    else:
        target_area = get_target_area(target_dic)

        # Move the object according to the speed vector.
        paddle.move()
        # ball.move()
        result = [False]
        if ball.y_pos > paddle.y_pos - 20:
            check_paddle(paddle, ball)
        if ball.y_pos < max_y_pos:
            result = check_target(target_area, ball)
            print('ball location: x_pos = {}, y_pos = {}'.format(ball.x_pos, ball.y_pos))
        if result[0]:
            print('flipped')
            print(result)
            ball.speed[1] *= result[1]
            ball.speed[0] *= result[2]
        ball.move()


    # --- Drawing Code

    # First, clear the screen to WHITE. Don't put other drawing commands
    # above this, or they will be erased with this command.
    screen.fill(BLACK)
    populate_screen(1)
    paddle.draw(screen)
    ball.draw(screen)
    print_score()
    if lose:
        reset = reset_game(screen)
    # Go ahead and update the screen with what we've drawn.
    pygame.display.flip()

    # for debugging, we want to print

    # if i % 2 == 0:
    # dateTimeObj = datetime.now()
    # timestampStr = dateTimeObj.strftime("%m_%d_%Y_%H_%M_%S_%f")
    # print('{}, x: {}, y: {}'.format(timestampStr, ball.x_pos, ball.y_pos))
    # pygame.image.save(screen, "C:/breakout/screenshot" + timestampStr + ".jpg")

    # Limit frames per second
    clock.tick(60)

# Close the window and quit.
pygame.quit()