106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
import pygame
|
|
import sys
|
|
|
|
from sprite.BoundingBox import BoundingBox
|
|
from sprite.PositionScale import PositionScale
|
|
from sprite.Spritesheet import Spritesheet
|
|
|
|
sys.path.append('./sprite')
|
|
|
|
|
|
class Sprite:
|
|
def __init__(self, spritesheet: Spritesheet):
|
|
self.spritesheet = spritesheet
|
|
|
|
self.animation_state = list(self.spritesheet.animations.keys())[0]
|
|
self.animation_frame = 0
|
|
self.animation_delay = 0
|
|
|
|
self.position_scale = PositionScale()
|
|
|
|
self.visible = True
|
|
self.animated = True
|
|
|
|
self.image = None
|
|
|
|
self.is_collider = True
|
|
|
|
def tick(self, dt: float):
|
|
animation = self.spritesheet.animations[self.animation_state]
|
|
|
|
if self.animated:
|
|
self.animation_delay += dt
|
|
|
|
while self.animation_delay >= animation['delays'][self.animation_frame % len(animation['delays'])]:
|
|
self.animation_delay -= animation['delays'][self.animation_frame % len(animation['delays'])]
|
|
self.animation_frame = (self.animation_frame + 1) % len(animation['images'])
|
|
|
|
self.image = animation['images'][self.animation_frame % len(animation['images'])]
|
|
|
|
def set_animation_state(self, state: str):
|
|
if state in self.spritesheet.animations:
|
|
self.animation_state = state
|
|
self.animation_delay = 0
|
|
self.tick(0)
|
|
|
|
def set_animation_frame(self, frame: int):
|
|
self.animation_frame = frame
|
|
self.tick(0)
|
|
|
|
def draw(self, screen: pygame.Surface, screen_transform: PositionScale):
|
|
if not self.visible:
|
|
return
|
|
|
|
if self.image is not None:
|
|
target_scale = (screen_transform.scale[0] * self.position_scale.scale[0],
|
|
screen_transform.scale[1] * self.position_scale.scale[1])
|
|
|
|
target_position = ((self.position_scale.position[0] + screen_transform.position[0]) * target_scale[0],
|
|
(self.position_scale.position[1] + screen_transform.position[1]) * target_scale[1])
|
|
|
|
target_size = (int(target_scale[0] * self.image.get_width()),
|
|
int(target_scale[1] * self.image.get_height()))
|
|
|
|
target_image = self.get_scaled_image(self.image, target_size)
|
|
screen.blit(target_image, target_position)
|
|
|
|
def get_bounding_box(self) -> BoundingBox:
|
|
return BoundingBox(
|
|
self.position_scale.position[0],
|
|
self.position_scale.position[1],
|
|
self.image.get_width() * self.position_scale.scale[0],
|
|
self.image.get_height() * self.position_scale.scale[1]
|
|
)
|
|
|
|
def get_scaled_image(self, image, resize):
|
|
return pygame.transform.scale(image, resize)
|
|
|
|
def dump(self, file):
|
|
# re-attach all the images to a single surface and save it to a file
|
|
width = 0
|
|
height = 0
|
|
|
|
for animation in self.spritesheet.animations.values():
|
|
max_height = 0
|
|
total_width = 0
|
|
for image in animation['images']:
|
|
total_width += image.get_width()
|
|
max_height = max(max_height, image.get_height())
|
|
width = max(width, total_width)
|
|
height += max_height
|
|
|
|
sheet = pygame.Surface((width, height))
|
|
|
|
x = 0
|
|
y = 0
|
|
for animation in self.spritesheet.animations.values():
|
|
max_height = 0
|
|
for image in animation['images']:
|
|
sheet.blit(image, (x, y))
|
|
x += image.get_width()
|
|
max_height = max(max_height, image.get_height())
|
|
y += max_height
|
|
x = 0
|
|
|
|
pygame.image.save(sheet, file)
|