import math
import random
from time import sleep
import pygame
from pygame.locals import *

W_W = 1000
W_H = 800

BLACK = (0, 0, 0)
WHITE = (200, 200, 200)
YELLOW = (250, 250, 20)
BLUE = (20, 20, 250)

pygame.init()
screen = pygame.display.set_mode((W_W, W_H))
pygame.display.set_caption('주방에서 살아남기')
pygame.display.set_icon(pygame.image.load('cheese.png'))
fps_clock = pygame.time.Clock()
FPS = 60
score = 0

default_font = pygame.font.Font('NanumGothic.ttf', 28)
background_img = pygame.image.load('background.png')
trapped_sound = pygame.mixer.Sound('trapped.wav')
yelling_sound = pygame.mixer.Sound('yelling.wav')
cheese_sound = pygame.mixer.Sound('cheese.wav')
eating_sound = pygame.mixer.Sound('eating.wav')
y_b_s = pygame.mixer.Sound('you_better_stop.wav')
pygame.mixer.music.load('nabibobettau.wav')

class Mouse(pygame.sprite.Sprite):
    def __init__(self):
        super(Mouse, self).__init__()
        self.image = pygame.image.load('mouse.png')
        self.rect = self.image.get_rect()
        self.centerx = self.rect.centerx
        self.centery = self.rect.centery

    def set_position(self, x, y):
        self.rect.x = x - self.centerx
        self.rect.y = y - self.centery

    def collide(self, sprite):
       for sprite in sprite:
            if pygame.sprite.collide_rect(self, sprite):
                return sprite


class Trap(pygame.sprite.Sprite):
    def __init__(self, xpos, ypos, hspeed, vspeed):
        super(Trap, self).__init__()
        traps = ('trap01.png', 'trap02.png', 'trap03.png', 'trap04.png', 'trap05.png',\
                 'trap06.png', 'trap07.png', 'trap08.png', 'trap09.png', 'trap10.png',)

        self.image = pygame.image.load(random.choice(traps))
        self.rect = self.image.get_rect()
        self.rect.x = xpos
        self.rect.y = ypos
        self.hspeed = hspeed
        self.vspeed = vspeed

        self.set_direction()

    def set_direction(self):
        if self.hspeed > 0:
            self.image = pygame.transform.rotate(self.image, 270)
        elif self.hspeed < 0:
            self.image = pygame.transform.rotate(self.image, 90)
        elif self.vspeed > 0:
            self.image = pygame.transform.rotate(self.image, 180)

    def update(self):
        self.rect.x += self.hspeed
        self.rect.y += self.vspeed
        if self.collide():
            self.kill()

    def collide(self):
        if self.rect.x < 0 - self.rect.height or self.rect.x > W_W:
            return True
        elif self.rect.y < 0 - self.rect.height or self.rect.y > W_H:
            return True


def random_trap(speed):
    random_direction = random.randint(1, 4)
    if random_direction == 1:
        return Cheese(random.randint(0, W_W), 0, 0, speed)
    elif random_direction == 2:
        return Cheese(W_W, random.randint(0, W_H), -speed, 0)
    elif random_direction == 3:
        return Cheese(random.randint(0, W_W), W_H, 0, -speed)
    elif random_direction == 4:
        return Cheese(0, random.randint(0, W_H), speed, 0)


class Cheese(pygame.pygame.sprite.Sprite):
    def __init__(self, x, y):
        super(Cheese, self).__init__()
        self.image = pygame.image.load('cheese.png')
        self.rect = self.image.get_rect()
        self.rect.x = x - self.rect.centerx
        self.rect.y = y - self.rect.centery


def d_r_b(background_img):
    background_rect = background_img.get_rect()
    for i in range(int(math.ceil(W_W / background_rect.width))):
        for j in range(int(math.ceil(W_H / background_rect.height))):
            screen.blit(background_img, Rect(i * background_rect.width,
                                             j * background_rect.width,
                                             background_rect.width,
                                             background_rect.height))


def draw_text(text, font, surface, x, y, main_color):
    text_obj = font.render(text, True, main_color)
    text_rect = text_obj.get_rect()
    text_rect.centerx = x
    text_rect.centery = y
    surface.blit(text_obj, text_rect)


def game_loop():
    global score

    pygame.mixer.music.play(-1)
    pygame.mouse.set_visible(False)

    mouse = Mouse()
    mouse.set_position(*pygame.mouse.get_pos())
    trap = pygame.sprite.Group()
    cheeses = pygame.sprite.Group()

    min_trap_speed = 1
    max_trap_speed = 1
    occur_of_traps = 1
    occur_prob = 15
    cheese_count = 1
    paused = False

    while True:
        pygame.display.update()
        fps_clock.tick(FPS)

        if paused:
            for event in pygame.event.get():
                if event.type == pygame.KEYDOWN:
                    if event.key == pygame.K_p:
                        paused = not paused
                        pygame.mouse.set_visible(False)
                if event.type == QUIT:
                    return 'quit'

        else:
           d_r_b(background_img)

            occur_of_traps = 1 + int(score / 500)
            min_of_traps = 1 + int(score / 400)
            max_of_traps = 1 + int(score/ 300)

            if random.randint(1, occur_prob) == 1:
                for i in range(occur_of_traps):
                    trap.add(random_trap(random.randint(min_trap_speed, max_trap_speed)))
                    score += 1

                if random.randint(1, occur_prob * 10)== 1:
                    cheese = Cheese(random.randint(30, W_W - 30),
                                    random.randint(30, W_H - 30))
                    cheeses.add(cheese)

            draw_text('점수: {}'.format(score), default_font, screen, 80, 20, YELLOW)
            draw_text('워프: {}'.format(cheese_count), default_font, screen, 700, 20, BLUE)
            trap.update()
            cheeses.update()
            trap.draw(screen)
            cheeses.draw(screen)

            cheese = mouse.collide(cheeses)
            if mouse.collide(trap):
                trapped_sound.play()
                yelling_sound.play()
                trap.empty()
                return 'game_screen'

            elif cheese:
                cheese_sound.play()
                cheese_count += 1
                cheese.kill()

            screen.blit(mouse.image, mouse.rect)

            for event in pygame.event.get():
                if event.type == pygame.MOUSEMOTION:

                    mouse_pos = pygame.mouse.get_pos()
                    if mouse_pos[0] <= 10:
                        pygame.mouse.set_position(W_W - 10, mouse_pos[1])
                    elif mouse_pos[0] >= W_W - 10:
                        pygame.mouse.set_position(0 + 10, mouse_pos[1])
                    elif mouse_pos[1] <= 10:
                        pygame.mouse.set_position(mouse_pos[0], W_H - 10)
                    elif mouse_pos[1] >= W_H - 10:
                        pygame.mouse.set_position(mouse_pos[0], 0 + 10)
                    mouse.set_position(*mouse_pos)
                if event.type == pygame.MOUSEBUTTONDOWN:
                    if cheese_count > 0:
                        cheese_count -= 1
                        y_b_s.play()
                        yelling_sound.play()
                        eating_sound.play()
                        sleep(1)
                        trap.empty()
                if event.type == pygame.KEYDOWN:
                    if event.key == pygame.K_p:
                        paused = not paused
                        if paused:
                            transp_surf = pygame.Surface((W_W, W_H))
                            transp_surf.set_alpha(150)
                            screen.blit(transp_surf, transp_surf.get_rect())
                            pygame.mouse.set_visible(True)
                            draw_text('일시정지',
                                      pygame.font.Font('NanumGothic.ttf', 60),
                                      screen, W_W/ 2, W_H / 2, YELLOW)
                if event.type == QUIT:
                    return 'quit'

    return 'game_screen'


def game_screen():
    global score
    pygame.mouse.set_visible(True)

    start_image = pygame.image.load('game_screen.png')
    screen.blit(start_image, [0, 0])

    draw_text('주방에서 살아남기',
             pygame.font.Font('NanumGothic.ttf', 70), screen,
            W_W / 2, W_H / 3.4, WHITE)
    draw_text('점수: {}'.format(score),
              default_font, screen,
              W_W / 2, W_H / 2.4, YELLOW)
    draw_text("마우스 버튼이나 's'키를 누르면 게임이 시작됩니다.",
              default_font, screen,
              W_W / 2, W_H / 2.0, WHITE)
    draw_text("게임을 종료하려면 'Q'키를 누르세요.",
              default_font, screen,
              W_W / 2, W_H / 1.8, WHITE)

    pygame.display.update()

    for event in pygame.event.get():
        if event.type == pygame.KEYDOWN:
            if event.type == pygame.K_q:
                return 'quit'
            elif event.key == pygame.K_s:
                return 'play'
        if event.type == pygame.MOUSEBUTTONDOWN:
            return 'play'
        if event.type == QUIT:
            return 'quit'

    return 'game_screen'


def main_loop():
    action = 'game_screen'
    while action != 'quit':
        if action == 'game_screen':
            action = game_screen()
        elif action == 'play':
            action = game_loop()

    pygame.quit()


main_loop()
#이 코드는 마우스로 쥐를 움직여 쥐덫을 피하는 게임인데, 쥐덫이 쥐라는 객체를 따라다니게 했으면 좋겠는데 도저히
#모르겠어요 도와주세요! ㅠㅠ