Pygame基础9-射击

import pygame
import sys

class Player(pygame.sprite.Sprite):
    def __init__(self):
        super().__init__()
        self.image = pygame.Surface((50, 50))
        self.image.fill((255, 255, 255))
        self.rect = self.image.get_rect(center = (screen_size[0]//2, screen_size[1]//2))
    def update(self):
        self.rect.center = pygame.mouse.get_pos()
    def shoot(self):
        return Bullet(self.rect.centerx, self.rect.centery)
        

class Bullet(pygame.sprite.Sprite):
    def __init__(self,pos_x, pos_y):
        super().__init__()
        self.image = pygame.Surface((50, 10))
        self.image.fill((255, 0, 0))
        self.rect = self.image.get_rect(center = (pos_x, pos_y))
        self.speed = 5
    def update(self):
        self.rect.x += self.speed
        # 如果子弹超出屏幕，就删除。否则子弹会（在屏幕外）一直存在。
        if self.rect.right > screen_size[0] + 20:
            self.kill()

# 初始化
pygame.init()
screen_size = (800, 600)
screen = pygame.display.set_mode(screen_size)
pygame.display.set_caption("Pygame Demo")
clock = pygame.time.Clock()

pygame.mouse.set_visible(False)

player = Player()
player_group = pygame.sprite.Group()
player_group.add(player)

bullet_group = pygame.sprite.Group()

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        if event.type == pygame.MOUSEBUTTONDOWN:
            bullet = player.shoot()
            bullet_group.add(bullet)

    # update
    # draw
    screen.fill((0, 0, 0))   
    # 先画子弹，再画玩家，否则玩家会被子弹挡住。
    bullet_group.update()
    bullet_group.draw(screen)
    
    player_group.update()
    player_group.draw(screen)
    
    pygame.display.flip()
    clock.tick(60)