Colisiones de grupos¶

import sys
import random
import pygame as pg

WIDTH, HEIGHT = 720, 400
FPS = 60

pg.init()
display = pg.display.set_mode((WIDTH, HEIGHT))
background = pg.image.load("grass.png").convert_alpha()
clock = pg.time.Clock()


class Rock(pg.sprite.Sprite):
    def __init__(self, x=0, y=0, mushroom_group=None):
        super().__init__()
        self.image = pg.image.load("rock.png")
        self.rect = self.image.get_rect()
        self.rect.topleft = (x, y)

        self.mushroom_group = mushroom_group


class Mushroom(pg.sprite.Sprite):
    def __init__(self, x=0, y=0, speed=0, auto=False):
        super().__init__()
        self.image = pg.image.load("mushroom.png")
        self.rect = self.image.get_rect()
        self.rect.topleft = (x, y)

        self.speed = speed

        if auto:
            self.rect.x = random.randint(0, WIDTH - self.image.get_size()[0])
            self.rect.y = random.randint(0, HEIGHT - self.image.get_size()[1])
            self.speed = random.randint(50, 250)

        # internal management position
        self.position = pg.Vector2(self.rect.x, self.rect.y)

    def update(self, dt):
        # Movimiento dependiente del tiempo
        # Manejado en un Vector flotante y redondeado en el rect
        self.position.x += self.speed * dt
        if self.position.x > WIDTH:
            self.position.x = 0.0
        self.rect.x = int(self.position.x)


mushroom_group = pg.sprite.Group()
rock_group = pg.sprite.Group()
rock_group.add(Rock(550, 50, mushroom_group=mushroom_group))
rock_group.add(Rock(500, 250, mushroom_group=mushroom_group))


while 1:
    for event in pg.event.get():
        if event.type == pg.QUIT:
            sys.exit()
        if event.type == pg.KEYDOWN:
            if event.key == pg.K_x:
                mushroom_group.add(Mushroom(auto=True))
            if event.key == pg.K_z:
                mushroom_group.empty()
    dt = clock.tick(FPS) / 1000
    mushroom_group.update(dt)
    rock_group.update(dt)

    # Comprobar cada roca que colisiona contra las setas, borramos las setas
    pg.sprite.groupcollide(rock_group, mushroom_group, 0, 1)
    # Descomposición de las colisiones
    # colliders = pg.sprite.groupcollide(rock_group, mushroom_group, 0, 0)
    # for rock, mushrooms in colliders.items():
    #     # print(rock, mushrooms)
    #     for mushroom in mushrooms:
    #         mushroom.kill()
    display.blit(background, (0, 0))
    mushroom_group.draw(display)
    rock_group.draw(display)
    pg.display.update()