# coding=utf8
"""
游戏主程序
"""
# 系统模块
import random
# 第三方模块
import pygame
# 自定义模块
from plane_sprites import * # 导入模块中所有成员
 
 
class PlaneGame(object):
 """
 游戏主程序
 """
 
 def __init__(self):
 """
 游戏初始化
 """
 # 1.创建游戏窗口
 self.screen = pygame.display.set_mode(SCREEN_RECT.size)
 # 2.创建游戏时钟
 self.clock = pygame.time.Clock()
 # 3.创建精灵和精灵组
 self.__create_sprites()
 # 4.设置定时器事件创建敌机
 pygame.time.set_timer(CREATE_ENEMY_EVENT, 100)
 # 5.发射子弹的定时器
 pygame.time.set_timer(HERO_FIRE_EVENT, 500)
 
 def __create_sprites(self):
 """
 创建游戏精灵
 :return:
 """
 # 1.创建背景精灵和精灵组
 bg1 = Backgroud()
 bg2 = Backgroud(True)
 self.back_group = pygame.sprite.Group(bg1, bg2)
 
 # 2.创建英雄精灵
 self.hero = Hero()
 self.hero_group = pygame.sprite.Group(self.hero)
 
 # 3.创建敌机精灵组
 self.enemy_group = pygame.sprite.Group()
 
 def start_game(self):
 """
 开始游戏循环
 :return:
 """
 print("开始游戏")
 while True:
  # 1.设置游戏屏幕的刷新频率
  self.clock.tick(FRAME_PRE_SEC)
  # 2.游戏事件监听
  self.__event_handler()
  # 3.精灵碰撞检测
  self.check_collide()
  # 4.更新和绘制精灵组
  self.update_sprites()
  # 5.更新显示图像
  pygame.display.update()
 
 def __event_handler(self):
 """
 事件监听处理
 :return:
 """
 for event in pygame.event.get():
  # 判断是否退出游戏
  if event.type == pygame.QUIT:
  PlaneGame.__game_over()
  elif event.type == CREATE_ENEMY_EVENT:
  print("敌机出场")
  # 1.创建敌机精灵
  enemy = Enemy()
  # 2.将敌机精灵添加到精灵组
  self.enemy_group.add(enemy)
  elif event.type == pygame.KEYDOWN:
  keys_pressed = pygame.key.get_pressed()
  if keys_pressed[pygame.K_RIGHT]:
   self.hero.speed = 1
  elif keys_pressed[pygame.K_LEFT]:
   self.hero.speed = -1
  else:
   self.hero.speed = 0
  elif event.type == pygame.KEYUP:
  # 1.按键抬起不在移动
  self.hero.speed = 0
 
  elif event.type == HERO_FIRE_EVENT:
  # 英雄发射子弹
  self.hero.fire()
  else:
  print(event.type)
 
 def check_collide(self):
 """
 精灵碰撞检测
 :return:
 """
 # 1.子弹摧毁敌机
 pygame.sprite.groupcollide(self.hero.bullets, self.enemy_group, True, True)
 
 # 2.敌机撞毁英雄
 """
 enemies = pygame.sprite.spritecollide(self.hero, self.enemy_group, True)
 if len(enemies) > 0:
  self.hero.kill()
  PlaneGame.__game_over()
 """
 
 def update_sprites(self):
 """
 更新精灵的位置
 :return:
 """
 # 更新位置
 self.back_group.update()
 self.hero_group.update()
 self.enemy_group.update()
 self.hero.bullets.update()
 # 绘制图形到屏幕
 self.back_group.draw(self.screen)
 self.hero_group.draw(self.screen)
 self.enemy_group.draw(self.screen)
 self.hero.bullets.draw(self.screen)
 
 @staticmethod
 def __game_over():
 """
 游戏结束的处理
 :return:
 """
 pygame.quit()
 exit()
 
 
if __name__ == "__main__": # 模块内置属性__name__默认为"__main__"，主要为了方便测试
 # 1.创建游戏对象
 game = PlaneGame()
 # 2.开始游戏
 game.start_game()

# coding=utf8
"""
游戏精灵
"""
# 系统模块
import random
# 第三方模块
import pygame
# 自定义模块
from plane_sprites import * # 导入模块中所有成员
 
# 游戏常量 ，python中没有意义上的常量，一般字母全部大写就是认为是常量
SCREEN_RECT = pygame.Rect(0, 0, 480, 700) # 游戏屏幕大小设定
FRAME_PRE_SEC = 60 # 设置游戏屏幕的刷新频率
# 游戏图片的路径
BACKGROUND_IMAGE_NAME = "./images/background.png"
HERO_IMAGE_NAME = "./images/hero.png"
ENEMY_IMAGE_NAME = "./images/enemy.png"
BULLET_IMAGE_NAME = "./images/bullet.png"
 
# 创建定时器
CREATE_ENEMY_EVENT = pygame.USEREVENT
HERO_FIRE_EVENT = pygame.USEREVENT + 1
 
 
class GameSprites(pygame.sprite.Sprite):
 """
 飞机大战游戏精灵基类
 """
 
 def __init__(self, image_name, speed=1):
 # 调用父类的初始化方法
 super().__init__()
 # 加载图片到内存
 self.image = pygame.image.load(image_name)
 # 获取图像的初始位置
 self.rect = self.image.get_rect()
 self.speed = speed
 
 def update(self, *args):
 # 垂直方向向下移动
 self.rect.y += self.speed
 
 
class Backgroud(GameSprites):
 """
 游戏背景精灵
 """
 
 def __init__(self, is_alt=False):
 # 1.调用父类创建精灵
 super().__init__(BACKGROUND_IMAGE_NAME)
 # 2.判断是否在屏幕上方
 if is_alt:
  self.rect.y = -self.rect.height
 
 def update(self):
 # 1.调用父类的实现
 super().update()
 # 2.判断背景是否飞出屏幕,如果是将背景设置到屏幕上方
 if self.rect.y >= SCREEN_RECT.height:
  self.rect.y = -self.rect.height
 
 
class Hero(GameSprites):
 """
 游戏英雄精灵
 """
 
 def __init__(self):
 # 1.调用父类设置图像和速度
 super().__init__(HERO_IMAGE_NAME, 0)
 
 # 2.设置英雄的初始位置
 self.rect.centerx = SCREEN_RECT.centerx
 self.rect.bottom = SCREEN_RECT.bottom - 120
 
 # 3.创建子弹精灵组
 self.bullets = pygame.sprite.Group()
 
 def update(self):
 # 1.英雄在水平方向移动
 self.rect.x += self.speed
 
 # 2.控制英雄不飞出屏幕
 if self.rect.x < 0:
  self.rect.x = 0
 elif self.rect.right > SCREEN_RECT.right:
  self.rect.right = SCREEN_RECT.right
 
 def fire(self):
 for i in (0, 1, 2):
  # 1.创建子弹精灵
  bullet = Bullet()
  # 2.设置精灵位置
  bullet.rect.bottom = self.rect.y - 20 * i
  bullet.rect.centerx = self.rect.centerx
  # 3.添加到精灵组
  self.bullets.add(bullet)
 
 
class Enemy(GameSprites):
 """
 敌机精灵
 """
 
 def __init__(self):
 super().__init__(ENEMY_IMAGE_NAME)
 # 1.指定敌机的初始速度
 self.speed = random.randint(1, 3)
 # 2.指定敌机的随机位置
 self.rect.bottom = 0
 max_x = SCREEN_RECT.width - self.rect.width
 self.rect.x = random.randint(0, max_x)
 
 def update(self):
 # 1. 调用父类方法，保持垂直飞行
 super().update()
 # 2.判断是否飞出屏幕，如果是则从精灵组中删除敌机
 if self.rect.y >= SCREEN_RECT.height:
  # 自动销毁,调用__del()__内置方法
  self.kill()
 
 def __del__(self):
 print("敌机消失 %s" % self.rect)
 
 
class Bullet(GameSprites):
 """
 子弹精灵
 """
 
 def __init__(self):
 super().__init__(BULLET_IMAGE_NAME, -2)
 self.bullets = pygame.sprite.Group()
 
 def update(self):
 super().update()
 if self.rect.bottom < 0:
  self.kill()