pneuma-pygame/effects/particle_effects.py

161 lines
6 KiB
Python
Raw Normal View History

2024-05-16 23:15:22 +00:00
import os
import pygame
from utils.resource_loader import import_folder
from random import choice
class AnimationPlayer:
def __init__(self):
self.frames = {
# Spells
'flame': import_folder(os.path.join('graphics',
'particles',
'flame',
'frames')),
'aura': import_folder(os.path.join('graphics',
'particles',
'aura')),
'heal': import_folder(os.path.join('graphics',
'particles',
'heal',
'frames')),
# Attacks
'claw': import_folder(os.path.join('graphics',
'particles',
'claw')),
'slash': import_folder(os.path.join('graphics',
'particles',
'slash')),
'sparkle': import_folder(os.path.join('graphics',
'particles',
'sparkle')),
'leaf_attack': import_folder(os.path.join('graphics',
'particles',
'leaf_attack')),
'thunder': import_folder(os.path.join('graphics',
'particles',
'thunder')),
# Monster Deaths
'squid': import_folder(os.path.join('graphics',
'particles',
'smoke_orange')),
'raccoon': import_folder(os.path.join('graphics',
'particles',
'raccoon')),
'spirit': import_folder(os.path.join('graphics',
'particles',
'nova')),
'bamboo': import_folder(os.path.join('graphics',
'particles',
'bamboo')),
# Leafs
'leaf': (
import_folder(os.path.join('graphics',
'particles',
'leaf1')),
import_folder(os.path.join('graphics',
'particles',
'leaf2')),
import_folder(os.path.join('graphics',
'particles',
'leaf3')),
import_folder(os.path.join('graphics',
'particles',
'leaf4')),
import_folder(os.path.join('graphics',
'particles',
'leaf5')),
import_folder(os.path.join('graphics',
'particles',
'leaf6')),
self.reflect_images(
import_folder(os.path.join('graphics',
'particles',
'leaf1'))),
self.reflect_images(
import_folder(os.path.join('graphics',
'particles',
'leaf2'))),
self.reflect_images(
import_folder(
os.path.join('graphics',
'particles',
'leaf3'))),
self.reflect_images(
import_folder(
os.path.join('graphics',
'particles',
'leaf4'))),
self.reflect_images(
import_folder(
os.path.join('graphics',
'particles',
'leaf5'))),
self.reflect_images(
import_folder(
os.path.join('graphics',
'particles',
'leaf6')))
)
}
def reflect_images(self, frames):
new_frames = []
for frame in frames:
flipped_frame = pygame.transform.flip(frame, True, False)
new_frames.append(flipped_frame)
return new_frames
def create_grass_particles(self, position, groups):
animation_frames = choice(self.frames['leaf'])
ParticleEffect(position, animation_frames, groups)
def generate_particles(self, animation_type, position, groups):
animation_frames = self.frames[animation_type]
ParticleEffect(position, animation_frames, groups)
class ParticleEffect(pygame.sprite.Sprite):
def __init__(self, position, animation_frames, groups):
super().__init__(groups)
self.frame_index = 0
self.animation_speed = 0.15
self.frames = animation_frames
self.image = self.frames[self.frame_index]
self.rect = self.image.get_rect(center=position)
self.sprite_type = 'magic'
def animate(self):
self.frame_index += self.animation_speed
if self.frame_index >= len(self.frames):
self.kill()
else:
self.image = self.frames[int(self.frame_index)]
def update(self):
self.animate()