42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
|
import pygame
|
||
|
|
|
||
|
|
|
||
|
|
class MovementHandler:
|
||
|
|
|
||
|
|
def __init__(self, sprite_type):
|
||
|
|
|
||
|
|
self.direction = pygame.math.Vector2()
|
||
|
|
|
||
|
|
def move(self, speed, hitbox, obstacle_sprites, rect):
|
||
|
|
if self.direction.magnitude() != 0:
|
||
|
|
self.direction = self.direction.normalize()
|
||
|
|
|
||
|
|
hitbox.x += self.direction.x * speed
|
||
|
|
self.collision('horizontal', hitbox, obstacle_sprites)
|
||
|
|
hitbox.y += self.direction.y * speed
|
||
|
|
self.collision('vertical', hitbox, obstacle_sprites)
|
||
|
|
rect.center = hitbox.center
|
||
|
|
|
||
|
|
def collision(self, direction, hitbox, obstacle_sprites):
|
||
|
|
if direction == 'horizontal':
|
||
|
|
for sprite in obstacle_sprites:
|
||
|
|
# The following works for static obstacles only
|
||
|
|
if sprite.hitbox.colliderect(hitbox):
|
||
|
|
# Moving Right
|
||
|
|
if self.direction.x > 0:
|
||
|
|
hitbox.right = sprite.hitbox.left
|
||
|
|
# Moving Left
|
||
|
|
if self.direction.x < 0:
|
||
|
|
hitbox.left = sprite.hitbox.right
|
||
|
|
|
||
|
|
if direction == 'vertical':
|
||
|
|
for sprite in obstacle_sprites:
|
||
|
|
# The following works for static obstacles only
|
||
|
|
if sprite.hitbox.colliderect(hitbox):
|
||
|
|
# Moving Down
|
||
|
|
if self.direction.y > 0:
|
||
|
|
hitbox.bottom = sprite.hitbox.top
|
||
|
|
# Moving Up
|
||
|
|
if self.direction.y < 0:
|
||
|
|
hitbox.top = sprite.hitbox.bottom
|