124 lines
4.0 KiB
Python
124 lines
4.0 KiB
Python
import pygame
|
|
from settings import *
|
|
|
|
|
|
class Enemy:
|
|
def __init__(self, x, y):
|
|
self.rect = pygame.Rect(x, y, 28, 48)
|
|
self.velocity = pygame.Vector2(0, 0)
|
|
self.pos = pygame.Vector2(x, y)
|
|
self.health = ENEMY_HEALTH
|
|
self.max_health = ENEMY_HEALTH
|
|
self.knockback_timer = 0
|
|
self.invulnerable_timer = 0
|
|
|
|
# Attack hitbox
|
|
self.attack_rect = pygame.Rect(x, y, 60, 60)
|
|
|
|
def update(self, dt, world, player):
|
|
dt = min(dt, 0.05)
|
|
|
|
# Apply gravity (FIX for flying)
|
|
self.velocity.y += ENEMY_GRAVITY * dt
|
|
if self.velocity.y > MAX_FALL_SPEED:
|
|
self.velocity.y = MAX_FALL_SPEED
|
|
|
|
# Basic AI: follow player
|
|
if self.knockback_timer <= 0:
|
|
if player.rect.x < self.rect.x:
|
|
self.velocity.x = -ENEMY_SPEED
|
|
else:
|
|
self.velocity.x = ENEMY_SPEED
|
|
else:
|
|
self.knockback_timer -= dt
|
|
|
|
# Update position
|
|
self.pos.x += self.velocity.x * dt
|
|
self.rect.x = round(self.pos.x)
|
|
self.handle_horizontal_collisions(world)
|
|
|
|
self.pos.y += self.velocity.y * dt
|
|
self.rect.y = round(self.pos.y)
|
|
self.handle_vertical_collisions(world)
|
|
|
|
# Update attack rect position
|
|
self.attack_rect.centerx = self.rect.centerx
|
|
self.attack_rect.centery = self.rect.centery
|
|
|
|
# Decay invulnerability
|
|
if self.invulnerable_timer > 0:
|
|
self.invulnerable_timer -= dt
|
|
|
|
def handle_horizontal_collisions(self, world):
|
|
"""Handle horizontal collision"""
|
|
for tile in world.get_nearby_tiles(self.rect):
|
|
if tile["solid"] and self.rect.colliderect(tile["rect"]):
|
|
if self.velocity.x > 0:
|
|
self.rect.right = tile["rect"].left
|
|
elif self.velocity.x < 0:
|
|
self.rect.left = tile["rect"].right
|
|
|
|
self.pos.x = self.rect.x
|
|
self.velocity.x = 0
|
|
|
|
def handle_vertical_collisions(self, world):
|
|
"""Handle vertical collision"""
|
|
for tile in world.get_nearby_tiles(self.rect):
|
|
if tile["solid"] and self.rect.colliderect(tile["rect"]):
|
|
if self.velocity.y > 0:
|
|
self.rect.bottom = tile["rect"].top
|
|
elif self.velocity.y < 0:
|
|
self.rect.top = tile["rect"].bottom
|
|
|
|
self.pos.y = self.rect.y
|
|
self.velocity.y = 0
|
|
|
|
def take_damage(self, damage, direction):
|
|
"""Take damage and apply knockback"""
|
|
if self.invulnerable_timer > 0:
|
|
return
|
|
|
|
self.health -= damage
|
|
self.invulnerable_timer = 0.3
|
|
self.knockback_timer = 0.15
|
|
|
|
# Apply knockback (affected by resistance)
|
|
knockback_force = 400 * ENEMY_KNOCKBACK_RESISTANCE
|
|
self.velocity.x = direction * knockback_force
|
|
self.velocity.y = -300 # Slight upward knockback
|
|
|
|
def draw(self, screen, camera):
|
|
draw_rect = camera.apply(self.rect)
|
|
|
|
# Flash when invulnerable
|
|
if self.invulnerable_timer > 0:
|
|
color = (100, 100, 255) # Light blue
|
|
else:
|
|
color = (0, 0, 255) # Blue
|
|
|
|
pygame.draw.rect(screen, color, draw_rect)
|
|
|
|
if SHOW_COLLIDERS:
|
|
pygame.draw.rect(screen, (0, 255, 0), draw_rect, 2)
|
|
# Draw attack hitbox
|
|
attack_draw_rect = camera.apply(self.attack_rect)
|
|
pygame.draw.rect(screen, (255, 0, 0), attack_draw_rect, 1)
|
|
|
|
# Draw health bar
|
|
health_bar_width = 28
|
|
health_bar_height = 4
|
|
health_percent = self.health / self.max_health
|
|
|
|
health_bar_rect = pygame.Rect(
|
|
draw_rect.x,
|
|
draw_rect.y - 8,
|
|
health_bar_width * health_percent,
|
|
health_bar_height
|
|
)
|
|
pygame.draw.rect(screen, (0, 255, 0), health_bar_rect)
|
|
pygame.draw.rect(screen, (255, 0, 0), pygame.Rect(
|
|
draw_rect.x,
|
|
draw_rect.y - 8,
|
|
health_bar_width,
|
|
health_bar_height
|
|
), 1) |