# TurnGame.py

import pygame
from Constant import scrrr_width, scrrr_height, GAMEOVER
import random
from Map import Map
from Sunshine import Sunshine
from PeaShooter import PeaShooter
from PeaBullet import PeaBullet
from Zombie import Zombie


# 主程序
class MainGame:

    score = 0

    # 存储所有地图坐标点
    map_points_list = []
    # 存储所有的地图块
    map_list = []
    # 存储所有植物的列表
    plants_list = []
    plants_pool = [0, 0, 0, 0, 0, 0]
    plants_pool_list = []
    # 存储所有豌豆子弹的列表
    peabullet_list = []
    peabullet_count = 0
    # 新增存储所有僵尸的列表
    zombie_list = []
    window = pygame.display.set_mode([scrrr_width, scrrr_height])
    # 0为植物回合，1为僵尸回合，2为玩家回合
    game_turn = 2

    something_in_mouse_store = False
    mouse_pressed = False
    mouse_get: PeaBullet = None
    mouse_store = []

    timecount = 50

    shoule_creat_zombie = False

    turn_count = 0

    # 开始游戏
    def start_game(self):
        # 初始化窗口
        # pygame.display.init()
        # 初始化坐标和地图
        self.init_plant_points()
        pygame.font.init()
        self.init_map()
        initmap = self.map_list[1][1]
        self.creat_plant(PeaShooter(initmap.position[0], initmap.position[1], 1), 1)
        self.creat_plant(PeaShooter(initmap.position[1], initmap.position[1], 1), 1)
        for i in range(0, 6):
            # self.creat_plant(PeaShooter((i - 1) * 80 + 4 * 80, 0, 1), 1)
            peashooter = PeaShooter((i - 1) * 80 + 5 * 80, 0, i + 1)
            MainGame.plants_pool_list.append(peashooter)
        # 只要游戏没结束，就一直循环
        while not GAMEOVER:
            self.load_ui()
            # 需要反复加载地图
            self.load_map()
            # 调用加载植物的方法
            self.load_plants()
            # 调用加载所有子弹的方法
            self.load_peabullets()
            # 调用事件处理的方法
            # 调用展示僵尸的方法
            self.load_zombies()
            #  计数器增长，每数到100，调用初始化僵尸的方法
            self.turn()
            self.update_pool()
            self.debugg()
            #  pygame自己的休眠
            pygame.time.wait(10)
            #  实时更新
            pygame.display.update()

    # 显示哪里种植了植物和棋盘上有多少植物
    def debugg(self):
        self.timecount += 1
        if self.timecount > 500:
            for i in range(len(self.map_list)):
                print()
                for j in range(len(self.map_list[i])):
                    print(self.map_list[i][j].can_grow, end="\t")
            print()
            print(len(self.peabullet_list))

            self.timecount = 0

    # 绘制物体和提示文本
    def draw_object(self, object):
        self.window.blit(object.image, object.rect)
        if not isinstance(object, Sunshine):
            self.window.blit(
                self.draw_text(f"{object.damage}", 26, (255, 0, 0)), object.rect
            )
        else:
            self.window.blit(
                self.draw_text(f"{object.money}", 26, (255, 0, 0)), object.rect
            )

    # 更新植物池列表
    def update_pool(self):
        for i in range(0, 6):
            self.plants_pool_list[i].count = self.plants_pool[i]

            self.window.blit(
                self.draw_text(f"{self.plants_pool_list[i].count}", 26, (255, 0, 0)),
                self.plants_pool_list[i].rect.move(50, 45),
            )

    # 加载僵尸或者移除僵尸
    def load_zombies(self):
        for zombie in MainGame.zombie_list:
            if zombie.live:
                self.draw_object(zombie)
                # self.window.blit(zombie.image, zombie.rect)
            else:
                MainGame.zombie_list.remove(zombie)

    def turn(self):
        # 主死循环每一次更新一帧
        # 0为植物回合，1为僵尸回合，3为操控回合
        if self.game_turn == 0:
            self.plants_event()
            # self.load_peabullets()
            if self.peabullet_count <= 0:
                self.zombie_turn_start()

        elif self.game_turn == 1:
            if self.shoule_creat_zombie == True:
                self.init_zombies()
                self.shoule_creat_zombie = False
            self.zombies_event()
            # self.zombie_turn_count += 1
            if self.is_zombie_end():
                self.play_turn_start()

        elif self.game_turn == 2:
            self.move_plant()
            pass

    # 初始化玩家回合
    def play_turn_start(self):
        self.game_turn = 2

    # 僵尸回合结束
    def is_zombie_end(self):
        for zombie in MainGame.zombie_list:
            if zombie.stop == False:
                return False

        return True

    # 植物回合开始
    def plant_turn_start(self):
        self.turn_count += 1
        self.game_turn = 0
        for plant in MainGame.plants_list:
            plant.should_fire = True
            plant.should_count_fire = True
            plant.shoot_number_count = plant.shoot_number

    # 僵尸回合开始
    def zombie_turn_start(self):
        self.shoule_creat_zombie = True
        self.game_turn = 1
        for zombie in MainGame.zombie_list:
            zombie.stop = False
            zombie.distance_count = zombie.distance

    # 加载ui
    def load_ui(self):
        self.window.fill((255, 255, 255))
        self.window.blit(
            self.draw_text(
                f"当前回合数{self.turn_count}",
                26,
                (255, 0, 0),
            ),
            (5, 40),
        )

        text1 = self.draw_text("按右键拖动豌豆射手", 26, (255, 0, 0))
        self.window.blit(text1, (5, 5))

    # 文本绘制
    def draw_text(self, content, size, color):

        font = pygame.font.SysFont("kaiti", size)
        text = font.render(content, True, color)
        return text

    # 植物创建逻辑
    def creat_plant(self, plant, grade):
        a = plant.rect[1] // 80 - 1
        b = plant.rect[0] // 80
        map = MainGame.map_list[a][b]
        map.can_grow = False

        peashooter = PeaShooter(map.position[0], map.position[1], grade)
        MainGame.plants_list.append(peashooter)

    # 移动植物逻辑
    def move_plant(self):
        # try:
        eventList = pygame.event.get()
        for event in eventList:
            if event.type == pygame.QUIT:
                self.gameOver()
            elif event.type == pygame.MOUSEBUTTONDOWN:
                x = event.pos[0] // 80
                y = (event.pos[1] // 80) - 1
                if y == -1:
                    map = MainGame.map_list[6][x - 4]
                else:
                    map = MainGame.map_list[y][x]
                print(event.pos)
                print(y, x)
                print(map.position)

                self.mouse_pressed = True
                if y == -1:
                    number = (map.position[0] // 80) - 3
                    print(f"number={number}")
                    # 此植物位置存mouse_store，植物存mouse_get

                    self.mouse_get = PeaShooter(
                        map.position[0],
                        map.position[1],
                        MainGame.plants_pool_list[number - 1].grade,
                    )
                    self.mouse_store = [
                        [map.position[1], map.position[0]],
                        MainGame.plants_pool_list[number - 1].grade,
                    ]

                    MainGame.plants_pool_list[number - 1].count -= 1
                    MainGame.plants_pool[number - 1] -= 1

                else:

                    # 有东西
                    can_grow = map.can_grow
                    if not can_grow:

                        print(f"{map.position} 1 not can grow")

                        # 此植物位置存mouse_store，植物存mouse_get
                        for plant in MainGame.plants_list:

                            if (
                                plant.rect.x == map.position[0]
                                and plant.rect.y == map.position[1]
                            ):
                                self.mouse_get = plant
                                self.mouse_store = [
                                    [map.position[1], map.position[0]],
                                    plant.grade,
                                ]
                                print("mouse_store get", self.mouse_store)
                                # 删除原植物
                                MainGame.plants_list.remove(plant)
                                MainGame.map_list[y][x].can_grow = True

                    else:
                        print(f"{map.position} 1 can grow")

            elif event.type == pygame.MOUSEBUTTONUP:
                x = event.pos[0] // 80
                y = (event.pos[1] // 80) - 1
                if y == -1:
                    map = MainGame.map_list[6][x - 4]
                else:
                    map = MainGame.map_list[y][x]
                print(map.position)

                self.mouse_pressed = False
                # 能放，清空mouse_store，此位置创建植物
                can_grow = map.can_grow
                if can_grow:
                    print(f"{map.position} 2 can grow")
                    # 清空mouse_store
                    if self.mouse_get is not None:
                        self.creat_plant(
                            self.mouse_get,
                            self.mouse_get.grade,
                        )
                        # self.turn_count += 1
                        self.plant_turn_start()
                    else:
                        print("1 Empty mouse_get")
                    self.mouse_get = None

                else:
                    print(f"{map.position} 2 not can grow")

                if not can_grow:
                    # 不能放，且等级同，清空mouse_store，mouse_get，本植物等级加一
                    for plant in MainGame.plants_list:
                        if (
                            plant.rect.x == map.position[0]
                            and plant.rect.y == map.position[1]
                        ):
                            if self.mouse_store[1] == plant.grade:
                                plant.grade += 1
                                MainGame.plants_list.remove(plant)
                                self.creat_plant(plant, plant.grade)
                                self.mouse_store = None
                                self.mouse_get = None
                                self.plant_turn_start()

                    # 不能放，且等级不同，在原位置恢复，清空mouse_store，mouse_get
                    print(f"{map.position} 3 not can grow")
                    if self.mouse_get is not None:
                        self.creat_plant(
                            PeaShooter(
                                self.mouse_store[0][1],
                                self.mouse_store[0][0],
                                self.mouse_store[1],
                            ),
                            self.mouse_store[1],
                        )
                        print("mouse_store out", self.mouse_store)

                        self.mouse_store = None
                        self.mouse_get = None

                    else:
                        print("2 Empty mouse_store")
                else:
                    print(f"{map.position} 3 can grow")

            if self.mouse_pressed:
                if self.mouse_get is not None:
                    self.mouse_get.rect = pygame.mouse.get_pos()
                    self.draw_object(self.mouse_get)
                    # self.window.blit(self.mouse_get.image, self.mouse_get.rect)
                else:
                    print("3 Empty mouse_get")
        # except IndexError:
        #     print("Empty Index!")

    #  初始化坐标点
    def init_plant_points(self):
        for y in range(1, 7):
            points = []
            for x in range(10):
                point = (x, y)
                points.append(point)
            MainGame.map_points_list.append(points)
            print("MainGame.map_points_list", MainGame.map_points_list)

        points = []
        for x in range(4, 10):
            point = (x, 0)
            points.append(point)
        MainGame.map_points_list.append(points)
        print("MainGame.map_points_list", MainGame.map_points_list)

    #  初始化地图
    def init_map(self):
        for points in MainGame.map_points_list:
            temp_map_list = list()
            for point in points:
                # map = None
                if (point[0] + point[1]) % 2 == 0:
                    map = Map(point[0] * 80, point[1] * 80, 0)
                else:
                    map = Map(point[0] * 80, point[1] * 80, 1)
                # 将地图块加入到窗口中
                temp_map_list.append(map)
                print("temp_map_list", temp_map_list)
            MainGame.map_list.append(temp_map_list)
        print("MainGame.map_list", MainGame.map_list)

    # 新增初始化僵尸的方法
    def init_zombies(self):
        # 不同的
        first_numbers = random.sample(range(1, 7), random.randint(1, 5))

        # 生成0或1的随机数,列表推导式
        second_numbers = [random.choice([0, 1]) for _ in range(len(first_numbers))]

        # 组合成元组
        random_tuples = list(zip(first_numbers, second_numbers))

        for i in random_tuples:
            if i[1] == 1:
                sunshine = Sunshine(800, i[0] * 80)
                MainGame.zombie_list.append(sunshine)
            else:
                zombie = Zombie(800, i[0] * 80)
                MainGame.zombie_list.append(zombie)

    #  将地图加载到窗口中
    def load_map(self):
        for temp_map_list in MainGame.map_list:
            for map in temp_map_list:
                # map.load_map()

                self.window.blit(map.image, map.position)

    # 更新加载植物或移除植物
    def load_plants(self):
        for plant in MainGame.plants_list + MainGame.plants_pool_list:

            if plant.live:
                self.draw_object(plant)
                # self.window.blit(plant.image, plant.rect)

            else:
                MainGame.plants_list.remove(plant)

    # 植物处理
    def plants_event(self):
        for plant in MainGame.plants_list:
            self.draw_object(plant)
            # self.window.blit(plant.image, plant.rect)

            #  如果活着
            if plant.should_fire:
                if plant.should_count_fire == True:
                    self.peabullet_count += plant.shoot_number
                    print("plus", self.peabullet_count)
                    plant.should_count_fire = False
                self.plant_shoot(plant)

    # 植物射击逻辑
    def plant_shoot(self, plant):
        plant.shot_count += 1
        #  计数器到25发射一次
        if plant.shot_count >= 25 and plant.shoot_number_count >= 1:
            #  基于当前豌豆射手的位置，创建子弹
            print(plant.shoot_number_count)
            plant.shoot_number_count -= 1
            peabullet = PeaBullet(plant, plant.bullet_damage)
            #  将子弹存储到子弹列表中
            MainGame.peabullet_list.append(peabullet)
            plant.shot_count = 0

    #  加载所有子弹的方法
    def load_peabullets(self):
        for bullet in MainGame.peabullet_list:
            if bullet.live:
                self.draw_object(bullet)
                # self.window.blit(bullet.image, bullet.rect)
                bullet.move_bullet()
                #  调用子弹是否打中僵尸的方法

                for enemy in MainGame.zombie_list:
                    if pygame.sprite.collide_rect(bullet, enemy):
                        # 打中僵尸之后，修改子弹的状态，
                        bullet.live = False
                        # 僵尸掉血
                        enemy.damage -= bullet.damage
                        if enemy.damage <= 0:
                            enemy.live = False
                            if isinstance(enemy, Sunshine):
                                self.plants_pool[enemy.money - 1] += 1
            else:
                MainGame.peabullet_list.remove(bullet)
                self.peabullet_count -= 1

    # 将所有僵尸加载到地图中
    def zombies_event(self):
        for enemy in MainGame.zombie_list:
            if enemy.live:
                self.zombie_move(enemy)
            else:
                MainGame.zombie_list.remove(enemy)

    # 僵尸移动逻辑
    def zombie_move(self, zombie):
        self.draw_object(zombie)
        # self.window.blit(zombie.image, zombie.rect)

        if zombie.live and not zombie.stop:
            zombie.distance_count -= zombie.speed
            zombie.rect.x -= zombie.speed
            if zombie.rect.x < -80:
                #  调用游戏结束方法
                MainGame().gameOver()
            if zombie.distance_count <= 0:
                zombie.stop = True

    #  程序结束方法
    def gameOver(self):
        self.window.blit(self.draw_text("游戏结束", 50, (255, 0, 0)), (300, 200))
        print("游戏结束")
        pygame.time.wait(400)
        global GAMEOVER
        GAMEOVER = True