83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
import pygame
|
|
import math
|
|
from game.config import TOWER_DATA, COLOR_BLACK, COLOR_WHITE
|
|
from game.utils import distance
|
|
from game.projectile import Projectile
|
|
|
|
|
|
class Tower:
|
|
def __init__(self, tower_type, grid_col, grid_row, px, py):
|
|
data = TOWER_DATA[tower_type]
|
|
self.type = tower_type
|
|
self.col = grid_col
|
|
self.row = grid_row
|
|
self.x = px
|
|
self.y = py
|
|
self.damage = data["damage"]
|
|
self.range = data["range"]
|
|
self.fire_rate = data["fire_rate"]
|
|
self.price = data["price"]
|
|
self.color = data["color"]
|
|
self.name = data["name"]
|
|
self.splash = data.get("splash", 0)
|
|
self.slow_factor = data.get("slow_factor", 0)
|
|
self.slow_duration = data.get("slow_duration", 0)
|
|
self.cooldown = 0
|
|
self.target = None
|
|
|
|
def find_target(self, enemies):
|
|
self.target = None
|
|
best = None
|
|
best_progress = -1
|
|
for e in enemies:
|
|
if not e.alive:
|
|
continue
|
|
d = distance(self.x, self.y, e.x, e.y)
|
|
if d <= self.range:
|
|
progress = e.waypoint_index + (1 - distance(e.x, e.y, *self._wp(e.waypoint_index)) / 100)
|
|
if progress > best_progress:
|
|
best_progress = progress
|
|
best = e
|
|
self.target = best
|
|
|
|
def _wp(self, idx):
|
|
from game.config import PATH_WAYPOINTS
|
|
if idx < len(PATH_WAYPOINTS):
|
|
return PATH_WAYPOINTS[idx]
|
|
return PATH_WAYPOINTS[-1]
|
|
|
|
def update(self, dt, enemies, projectiles):
|
|
if self.cooldown > 0:
|
|
self.cooldown -= dt
|
|
|
|
self.find_target(enemies)
|
|
|
|
if self.target and self.cooldown <= 0:
|
|
self.cooldown = self.fire_rate
|
|
proj_type = {"arrow": "arrow", "cannon": "cannon", "slow": "slow"}[self.type]
|
|
projectiles.append(Projectile(
|
|
self.x, self.y, self.target, self.damage,
|
|
proj_type=proj_type, splash=self.splash,
|
|
slow_factor=self.slow_factor, slow_duration=self.slow_duration,
|
|
))
|
|
|
|
def draw(self, surface):
|
|
ix, iy = int(self.x), int(self.y)
|
|
if self.type == "arrow":
|
|
pygame.draw.rect(surface, self.color, (ix - 12, iy - 12, 24, 24))
|
|
pygame.draw.rect(surface, COLOR_BLACK, (ix - 12, iy - 12, 24, 24), 2)
|
|
elif self.type == "cannon":
|
|
pygame.draw.circle(surface, self.color, (ix, iy), 14)
|
|
pygame.draw.circle(surface, COLOR_BLACK, (ix, iy), 14, 2)
|
|
pygame.draw.circle(surface, (180, 100, 30), (ix, iy), 8)
|
|
elif self.type == "slow":
|
|
points = [(ix, iy - 14), (ix + 14, iy), (ix, iy + 14), (ix - 14, iy)]
|
|
pygame.draw.polygon(surface, self.color, points)
|
|
pygame.draw.polygon(surface, COLOR_BLACK, points, 2)
|
|
|
|
if self.target and self.target.alive:
|
|
pygame.draw.line(surface, (*self.color, ), (ix, iy), (int(self.target.x), int(self.target.y)), 1)
|
|
|
|
def draw_range(self, surface):
|
|
pygame.draw.circle(surface, COLOR_WHITE, (int(self.x), int(self.y)), self.range, 1)
|