Add bouncing spot based on transform helper

This commit is contained in:
Patrick Moessler 2023-02-17 02:07:06 +01:00
parent fec4f0334a
commit 5a28453d2e

View file

@ -1,12 +1,12 @@
from typing import Any
import pygame as pg
from effects.effect import Effect, Colors
from effects.effect import Effect, Colors, transform_bounce
import random
import math
from typing import Union, Generator
class BouncingSpot(Effect):
class BouncingSpot2(Effect):
def __init__(
self,
bounds: pg.Rect,
@ -24,8 +24,6 @@ class BouncingSpot(Effect):
self.velocity = random.randint(self.min_velocity, self.max_velocity)
self.ticks = random.randint(0, 360)
self.x_factor = random.uniform(x_factor[0], x_factor[1])
self.y_factor = random.uniform(y_factor[0], y_factor[1])
self.color = color
size = (math.sin(self.ticks) / 2 + 0.5) * (
self.max_size - self.min_size
@ -44,6 +42,10 @@ class BouncingSpot(Effect):
*groups
)
self.bounds = bounds
self.bouncer = transform_bounce(
bounds=bounds, velocity=velocity, x_factor=x_factor, y_factor=y_factor
)
next(self.bouncer)
self.update()
def update(self, *args: Any, **kwargs: Any) -> None:
@ -53,14 +55,8 @@ class BouncingSpot(Effect):
new_scale = new_size - self.rect.width
self.rect.inflate_ip(new_scale, new_scale)
self.rect.centerx = (
0.4 * math.cos(self.x_factor * self.ticks) * self.bounds.width
+ self.bounds.centerx
)
self.rect.centery = (
0.4 * math.sin(self.y_factor * self.ticks) * self.bounds.height
+ self.bounds.centery
)
self.rect.center = self.bouncer.send(self.rect.size)
self.image.fill(Colors.Black)
pg.draw.ellipse(