pneuma-pygame/game/effects/particle_effects.py

91 lines
3.7 KiB
Python

import os
import pygame
from utils.resource_loader import import_folder
from random import choice
class AnimationPlayer:
def __init__(self):
script_dir = os.path.dirname(os.path.abspath(__file__))
asset_path = os.path.join(
script_dir, '../..', 'assets')
self.frames = {
# magic
'flame': import_folder(f'{asset_path}/graphics/particles/flame/frames'),
'aura': import_folder(f'{asset_path}/graphics/particles/aura'),
'heal': import_folder(f'{asset_path}/graphics/particles/heal/frames'),
# attacks
'claw': import_folder(f'{asset_path}/graphics/particles/claw'),
'slash': import_folder(f'{asset_path}/graphics/particles/slash'),
'sparkle': import_folder(f'{asset_path}/graphics/particles/sparkle'),
'leaf_attack': import_folder(f'{asset_path}/graphics/particles/leaf_attack'),
'thunder': import_folder(f'{asset_path}/graphics/particles/thunder'),
# monster deaths
'squid': import_folder(f'{asset_path}/graphics/particles/smoke_orange'),
'raccoon': import_folder(f'{asset_path}/graphics/particles/raccoon'),
'spirit': import_folder(f'{asset_path}/graphics/particles/nova'),
'bamboo': import_folder(f'{asset_path}/graphics/particles/bamboo'),
# leafs
'leaf': (
import_folder(f'{asset_path}/graphics/particles/leaf1'),
import_folder(f'{asset_path}/graphics/particles/leaf2'),
import_folder(f'{asset_path}/graphics/particles/leaf3'),
import_folder(f'{asset_path}/graphics/particles/leaf4'),
import_folder(f'{asset_path}/graphics/particles/leaf5'),
import_folder(f'{asset_path}/graphics/particles/leaf6'),
self.reflect_images(import_folder(
f'{asset_path}/graphics/particles/leaf1')),
self.reflect_images(import_folder(
f'{asset_path}/graphics/particles/leaf2')),
self.reflect_images(import_folder(
f'{asset_path}/graphics/particles/leaf3')),
self.reflect_images(import_folder(
f'{asset_path}/graphics/particles/leaf4')),
self.reflect_images(import_folder(
f'{asset_path}/graphics/particles/leaf5')),
self.reflect_images(import_folder(
f'{asset_path}/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()