import pygame
 
 
SCREEN_RECT = pygame.Rect(0, 0, 512, 768)   # 用一個(gè)常量來存儲(chǔ)屏幕的位置和大小,常量用全大寫表示
SCREEN_CENTER = [SCREEN_RECT.width/2, SCREEN_RECT.height/2]
 
CREATE_ENEMY_EVENT = pygame.USEREVENT       # 敵機(jī)事件定時(shí)器常量值
OWN_FIRE_EVENT = pygame.USEREVENT + 1       # 發(fā)射子彈定時(shí)器常量值
BONUS_ENEMY_EVENT = pygame.USEREVENT + 2    # 加分?jǐn)硻C(jī)的定時(shí)器常量值
STAR_ENEMY_EVENT = pygame.USEREVENT + 3     # 星星敵機(jī)的定時(shí)器常量值
BOSS_ENEMY_EVENT = pygame.USEREVENT + 4     # BOSS敵機(jī)的定時(shí)器常量值
BULLET_BOSS_EVENT = pygame.USEREVENT + 5    # BOSS敵機(jī)的子彈定時(shí)器常量值
 
ENEMY_TIMER = 600                           # 敵機(jī)出現(xiàn)的時(shí)間頻率
BULLET_TIMER = 400                          # 子彈出現(xiàn)的時(shí)間頻率
BONUS_TIMER = 5000                          # 加分?jǐn)硻C(jī)出現(xiàn)的時(shí)間頻率
START_TIMER = 20000                         # 星星敵機(jī)出現(xiàn)的時(shí)間頻率
BOSS_TIMER = 120000                         # BOSS敵機(jī)出現(xiàn)的時(shí)間頻率
BULLET_BOSS_TIMER = 1500                    # BOSS敵機(jī)子彈出現(xiàn)的時(shí)間頻率
 
LIFE_NUMS = 5                               # 生命值
MOVE_SPEED = 5                              # 已方飛機(jī)移動(dòng)速度
STAR_ENEMY_LIFE = 6                         # 打掉星星敵機(jī)需要的子彈
BOSS_ENEMY_LIFE = 100                       # 打掉BOSS敵機(jī)需要的子彈
 
BACKGROUND = r'.\images\background.png'
GAME_OVER = r'.\images\gameover.png'
OWN_IMAGE = r'.\images\own.png'
OWN_DOWN_IMAGE = r'.\images\own_died.png'
ENEMY_RED_IMAGE = r'.\images\enemy_red.png'
ENEMY_BLUE_IMAGE = r'.\images\enemy_blue.png'
ENEMY_DOWN_IMAGE = r'.\images\enemy_down.png'
BULLET_IMAGE = r'.\images\bullet.png'
BULLET_BOSS_IMAGE = r'.\images\bullet_boss.png'
LIFE_IMAGE = r'.\images\life.png'
GOLD_COIN_IMAGE = r'.\images\bonus.png'
STAR_IMAGE = r'.\images\star.png'
READY_IMAGE = r'.\images\ready.png'
BOSS_IMAGE = r'.\images\boss.png'
LOGO_IMAGE = r'.\images\logo.ico'
 
SYS_FONT = r'.\fonts\BOLDER.ttf'
GAME_NAME = '小飛機(jī)'
 
FPS = 60            # 屏幕刷新頻率

import threading
from sprites import *
from initialize import *
 
 
# noinspection PyPep8Naming
# 主游戲類
class PlaneGame(object):
    def __init__(self):
        self.score = 0
        self.bullet_grade = 1   # 初始的子彈等級
        self.boss_flag = 0      # 判斷BOSS是否出現(xiàn)
        self.ready_time = pygame.time.get_ticks()   # 用來控制顯示ready畫面的時(shí)鐘
        self.star_enemy_life = STAR_ENEMY_LIFE      # 星星敵機(jī)需要幾顆子彈
        self.boss_enemy_life = BOSS_ENEMY_LIFE      # BOSS敵機(jī)需要幾顆子彈
        self.bullet_type = {1: 'One', 2: 'Two', 3: 'Three', 4: 'Four'}
        self.screen = pygame.display.set_mode(SCREEN_RECT.size)
        logo = pygame.image.load(LOGO_IMAGE)        # 設(shè)置游戲窗口圖標(biāo)
        pygame.display.set_icon(logo)
        pygame.display.set_caption(GAME_NAME)       # 設(shè)置游戲窗口名稱
 
        self.clock = pygame.time.Clock()  # 創(chuàng)建游戲時(shí)鐘
        self.__createSprite()  # 調(diào)用私有方法, 創(chuàng)建精靈和組
 
        pygame.time.set_timer(CREATE_ENEMY_EVENT, ENEMY_TIMER)  # 設(shè)置定時(shí)器事件
        pygame.time.set_timer(OWN_FIRE_EVENT, BULLET_TIMER)
        pygame.time.set_timer(BONUS_ENEMY_EVENT, BONUS_TIMER)
        pygame.time.set_timer(STAR_ENEMY_EVENT, START_TIMER)
        pygame.time.set_timer(BOSS_ENEMY_EVENT, BOSS_TIMER)
        pygame.time.set_timer(BULLET_BOSS_EVENT, BULLET_BOSS_TIMER)
        self.life_num = LIFE_NUMS
 
    def __createSprite(self):
        """ 私有方法，創(chuàng)建精靈和組 """
        bg1 = BackGround(BACKGROUND)                    # 創(chuàng)建背景和組
        bg2 = BackGround(BACKGROUND)
        bg2.rect.y = -bg2.rect.height                   # ???
        self.back_group = pygame.sprite.Group(bg1, bg2)
 
        self.ready_go_group = pygame.sprite.Group()     # 創(chuàng)建ready組
        self.ready_go_group.add(ReadyGo())
        self.enemy_red_group = pygame.sprite.Group()    # 創(chuàng)建紅色敵機(jī)組
        self.enemy_down_group = pygame.sprite.Group()   # 創(chuàng)建被擊中敵機(jī)組
        self.bonus_enemy_group = pygame.sprite.Group()  # 創(chuàng)建加分?jǐn)硻C(jī)組
        self.gold_coin_group = pygame.sprite.Group()    # 創(chuàng)建金幣組
        self.star_enemy_group = pygame.sprite.Group()   # 創(chuàng)建星星敵機(jī)組
 
        self.boss_enemy = EnemyBoss()                   # ???
        self.boss_enemy_group = pygame.sprite.Group()   # 創(chuàng)建BOSS敵機(jī)組
        self.bullet_boss_group = pygame.sprite.Group()  # 創(chuàng)建BOSS子彈組
 
        self.star_coin_group = pygame.sprite.Group()    # 創(chuàng)建星星組
        self.own_died_group = pygame.sprite.Group()     # 創(chuàng)建自己死亡飛機(jī)的組
        self.__resetOwn()                               # 創(chuàng)建自己的飛機(jī)和組
 
    def start(self):
        """ 游戲開始 """
        while True:
            self.clock.tick(FPS)        # 刷新幀率
            self.__eventHandler()       # 事件監(jiān)聽
            self.__checkCollide()       # 碰撞檢測
            self.__updateSprites()      # 更新/繪制畫面
 
            pygame.display.update()     # 更新屏幕顯示
            self.__gameOver()           # 游戲結(jié)束
 
    def __eventHandler(self):
        """ 事件監(jiān)聽(監(jiān)聽定時(shí)器常量) """
        for event in pygame.event.get():
            if event.type == pygame.QUIT:  # 判斷是否退出游戲
                PlaneGame.quit()
            elif event.type == CREATE_ENEMY_EVENT:  # 當(dāng)事件值等于CREATE_ENEMY_EVENT時(shí)，將敵機(jī)精靈添加到組
                enemy = EnemyRed()
                self.enemy_red_group.add(enemy)
            elif event.type == BONUS_ENEMY_EVENT:  # 當(dāng)事件值等于BONUS_ENEMY_EVENT時(shí)，將加分?jǐn)硻C(jī)加入到組
                bonus_enemy = EnemyRed()
                self.bonus_enemy_group.add(bonus_enemy)
            elif event.type == STAR_ENEMY_EVENT:  # 當(dāng)事件值等于STAR_ENEMY_EVENT時(shí)，將星星敵機(jī)加入到組
                self.star_enemy_life = STAR_ENEMY_LIFE  # 賦予新的星星敵機(jī)生命值
                star_enemy = EnemyBlue()
                self.star_enemy_group.add(star_enemy)
 
            elif event.type == BOSS_ENEMY_EVENT:  # 當(dāng)事件值等于BOSS_ENEMY_EVENT時(shí)，將BOSS敵機(jī)加入到組
                self.boss_enemy_life = BOSS_ENEMY_LIFE  # 賦予新的BOSS敵機(jī)生命值
                self.boss_enemy = EnemyBoss()            # ????
                self.boss_enemy_group.add(self.boss_enemy)
                self.boss_flag = 1          # 修改BOSS出現(xiàn)的flag
 
            elif event.type == OWN_FIRE_EVENT:  # 當(dāng)事件值等于OWN_FIRE_EVENT時(shí)，按子彈等級開火
                for n in self.bullet_type:
                    grade = 4 if self.bullet_grade > 4 else self.bullet_grade
                    if grade == n:
                        exec('self.own.fire{}()'.format(self.bullet_type[n]))
 
            if self.boss_flag == 1:     # 如果BOSS出現(xiàn)，則發(fā)射子彈
                if event.type == BULLET_BOSS_EVENT:
                    self.boss_enemy.fire()
 
        keys_pressed = pygame.key.get_pressed()  # 使用鍵盤提供的方法獲取按鍵, 得到一個(gè)按鍵元組
        if keys_pressed[pygame.K_RIGHT]:  # 按右方向鍵時(shí)向右移動(dòng)，水平速度為4
            self.own.speed_y = 0  # 同時(shí)將垂直方向的速度變?yōu)?
            self.own.speed_x = MOVE_SPEED
        elif keys_pressed[pygame.K_LEFT]:  # 按左方向鍵時(shí)向右移動(dòng)，水平速度為4
            self.own.speed_y = 0
            self.own.speed_x = -MOVE_SPEED
        elif keys_pressed[pygame.K_UP]:  # 按上方向鍵時(shí)向右移動(dòng)，垂直速度為-4
            self.own.speed_x = 0  # 同時(shí)將水平方向的速度變?yōu)?
            self.own.speed_y = -MOVE_SPEED
        elif keys_pressed[pygame.K_DOWN]:  # 按下方向鍵時(shí)向右移動(dòng)，垂直速度為4
            self.own.speed_x = 0
            self.own.speed_y = MOVE_SPEED
        elif keys_pressed[pygame.K_SPACE]:  # 按空格鍵時(shí)暫停游戲
            self.__gamePause(True)
        else:  # 按其他鍵或不按時(shí)不移動(dòng)
            self.own.speed_x = 0
            self.own.speed_y = 0
 
    def __checkCollide(self):
        """ 碰撞檢測 """
        # 當(dāng)子彈和敵機(jī)碰撞，敵機(jī)爆炸，得分加1，groupcollide()是精靈基類中的方法，返回一個(gè)字典，子彈是鍵，敵機(jī)是值
        killers = pygame.sprite.groupcollide(self.own.bullets, self.enemy_red_group, True, True)  # 子彈摧毀敵機(jī)
        if len(killers) > 0:  # 如果返回值長度大于0時(shí)，得分加1
            self.score += 1
            enemy_down = SpriteDown(ENEMY_DOWN_IMAGE, list(killers.keys())[0].rect, 0)  # 取出碰撞字典中的鍵用來定位顯示爆炸
            self.enemy_down_group.add(enemy_down)
 
        # 當(dāng)敵機(jī)和已機(jī)碰撞，飛機(jī)數(shù)量減1，1秒后重新顯示一架飛機(jī)
        died_1 = pygame.sprite.groupcollide(self.enemy_red_group, self.own_group, True, True)       # 敵機(jī)撞毀自己飛機(jī)
        died_2 = pygame.sprite.groupcollide(self.bonus_enemy_group, self.own_group, True, True)     # 加分?jǐn)硻C(jī)撞毀自己飛機(jī)
        died_3 = pygame.sprite.groupcollide(self.star_enemy_group, self.own_group, True, True)      # 星星敵機(jī)撞毀自己飛機(jī)
        died_4 = pygame.sprite.groupcollide(self.boss_enemy.bullets, self.own_group, True, True)    # BOSS敵機(jī)子彈擊毀已機(jī)
        died_5 = pygame.sprite.groupcollide(self.boss_enemy_group, self.own_group, True, True)      # BOSS敵機(jī)撞毀已機(jī)
        died_owns = {**died_1, **died_2, **died_3, **died_4, **died_5}  # 合并字典
        if len(died_owns) > 0:  # 列表有內(nèi)容代表發(fā)生碰撞了
            self.life_num -= 1  # 飛機(jī)數(shù)量減1
            own_died = SpriteDown(OWN_DOWN_IMAGE, list(died_owns.keys())[0].rect, 0)
            self.own_died_group.add(own_died)
            self.s = threading.Timer(1, self.__resetOwn)  # 設(shè)置一個(gè)多線程定時(shí)調(diào)用函數(shù)，1秒后開始執(zhí)行
            self.s.start()
 
        # 當(dāng)子彈和加分?jǐn)硻C(jī)碰撞，出現(xiàn)金幣繼續(xù)下落
        bonus_enemies = pygame.sprite.groupcollide(self.own.bullets, self.bonus_enemy_group, True, True)
        if len(bonus_enemies) > 0:
            self.score += 1
            gold_coin = SpriteDown(GOLD_COIN_IMAGE, list(bonus_enemies.keys())[0].rect, 1)
            self.gold_coin_group.add(gold_coin)
 
        # 當(dāng)已機(jī)和硬幣碰撞，得分+5，金幣消失
        gold = pygame.sprite.groupcollide(self.gold_coin_group, self.own_group, True, False)
        if len(gold) > 0:
            self.score += 5
 
        # 當(dāng)子彈和星星敵機(jī)碰撞，子彈消失，超過n顆子彈后出現(xiàn)星星繼續(xù)下落
        if self.star_enemy_life <= 0:
            star_enemy = pygame.sprite.groupcollide(self.own.bullets, self.star_enemy_group, True, True)
            if len(star_enemy) > 0:
                self.score += 5
                star_coin = SpriteDown(STAR_IMAGE, list(star_enemy.keys())[0].rect, 1)
                self.star_coin_group.add(star_coin)
        else:
            star_enemies = pygame.sprite.groupcollide(self.own.bullets, self.star_enemy_group, True, False)
            if len(star_enemies) > 0:
                self.star_enemy_life -= 1
 
        # 當(dāng)子彈和BOSS敵機(jī)碰撞，子彈消失，超過n顆子彈后BOSS敵機(jī)爆炸
        if self.boss_enemy_life <= 0:
            boss_enemy = pygame.sprite.groupcollide(self.own.bullets, self.boss_enemy_group, True, True)
            if len(boss_enemy) > 0:
                self.score += 300
                enemy_down = SpriteDown(ENEMY_DOWN_IMAGE, list(boss_enemy.keys())[0].rect, 1)
                self.enemy_down_group.add(enemy_down)
                self.boss_flag = 0
        else:
            boss_enemies = pygame.sprite.groupcollide(self.own.bullets, self.boss_enemy_group, True, False)
            if len(boss_enemies) > 0:
                self.boss_enemy_life -= 1
 
        # 當(dāng)已機(jī)和星星碰撞，得分+10，子彈等級+1
        star = pygame.sprite.groupcollide(self.star_coin_group, self.own_group, True, False)
        if len(star) > 0:
            self.score += 10
            self.bullet_grade += 1
 
    def __resetOwn(self):
        """ 自己飛機(jī)重生 """
        self.own = Own()  # 創(chuàng)建自己的飛機(jī)精靈和精靈組
        self.own_group = pygame.sprite.Group(self.own)
        self.bullet_grade = 1  # 初始的子彈等級
 
    def __infoDisplay(self, text, pos, angle):
        """
        -- 顯示游戲信息
        :param text: 文字內(nèi)容
        :param pos: 位置[x, y]
        :param angle: 用來定位的角
        :return:
        """
        score_font = pygame.font.Font(SYS_FONT, 24)
        score_text = score_font.render(text, True, (255, 255, 255))
        text_rect = score_text.get_rect()
        exec('text_rect.{}={}'.format(angle, pos))
        self.screen.blit(score_text, text_rect)
 
    def __updateSprites(self):
        """ 更新/繪制精靈組 """
        self.back_group.update()
        self.back_group.draw(self.screen)
 
        self.__infoDisplay('score ' + str(self.score), [20, 10], 'topleft')
        self.__infoDisplay('level 01', [492, 10], 'topright')
        self.__infoDisplay('life ' + str(self.life_num), [230, 10], 'topleft')
 
        if pygame.time.get_ticks() < self.ready_time + 3000:    # 顯示ready畫面3秒
            self.ready_go_group.update()
            self.ready_go_group.draw(self.screen)
 
        if pygame.time.get_ticks() > self.ready_time + 4000:    # ready畫面1秒后開始顯示游戲元素
            self.enemy_red_group.update()
            self.enemy_red_group.draw(self.screen)    # 顯示敵機(jī)
            self.bonus_enemy_group.update()
            self.bonus_enemy_group.draw(self.screen)  # 顯示加分?jǐn)硻C(jī)
            self.star_enemy_group.update()
            self.star_enemy_group.draw(self.screen)  # 顯示星星敵機(jī)
            self.boss_enemy_group.update()
            self.boss_enemy_group.draw(self.screen)  # 顯示BOSS敵機(jī)
            self.enemy_down_group.update()
            self.enemy_down_group.draw(self.screen)  # 顯示敵機(jī)爆炸
            self.own.bullets.update()
            self.own.bullets.draw(self.screen)      # 顯示已方子彈
 
            self.boss_enemy.bullets.update()
            self.boss_enemy.bullets.draw(self.screen)  # 顯示BOSS子彈
 
            self.own_died_group.update()
            self.own_died_group.draw(self.screen)  # 顯示已方飛機(jī)爆炸
            self.gold_coin_group.update()
            self.gold_coin_group.draw(self.screen)  # 顯示金幣
            self.star_coin_group.update()
            self.star_coin_group.draw(self.screen)  # 顯示星星
            self.own_group.update()
            self.own_group.draw(self.screen)  # 顯示已方飛機(jī)
 
    def __gamePause(self, pause=False):
        """ 暫停游戲， 按任意鍵退出"""
        self.__infoDisplay('Any key to continue...', SCREEN_CENTER, 'center')
        pygame.display.update()  # 更新屏幕顯示
        while pause:
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    PlaneGame.quit()
                elif event.type == pygame.KEYDOWN:
                    pause = False
 
    def __gameOver(self):
        """ 游戲結(jié)束，按ESC重新開始 """
        if self.life_num == 0:  # 如果飛機(jī)數(shù)量為0，退出游戲
            game_over = pygame.image.load(GAME_OVER)  # 顯示GAMEOVER圖片
            self.screen.blit(game_over, (50, 150))
            self.__infoDisplay('press ESC play again...', SCREEN_CENTER, 'center')
            pygame.display.update()  # 更新屏幕顯示
            flag = True
            while flag:
                for event in pygame.event.get():
                    if event.type == pygame.QUIT:
                        PlaneGame.quit()
                    elif event.type == pygame.KEYDOWN:
                        if event.key == pygame.K_ESCAPE:  # 按SPACE重新開始游戲
                            flag = False
                            PlaneGame().start()
 
    @staticmethod
    def quit():
        """ 靜態(tài)函數(shù)，退出游戲 """
        pygame.quit()
        exit()