본문 바로가기

[Pygame] 🏰 2D 타워 디펜스 게임 만들기 11강 | 총알 만들기

@도마22026. 2. 11. 19:00
728x90


10강에서는 타워가 타겟을 잡으면 곧바로 HP를 감소시키는 방식으로 공격을 구현했습니다.
이번 11강에서는 공격을 더 자연스럽게 만들기 위해 총알(Bullet) 투사체 방식으로 변경합니다.

  • Bullet 클래스 추가
  • 타워가 공격할 때 총알 생성
  • 총알이 타겟을 향해 이동
  • 충돌 시 데미지 적용 후 총알 삭제

1. Bullet이 가져야 할 정보

총알은 최소한 아래 정보를 갖습니다.

  • 현재 위치(x, y)
  • 목표 타겟(Enemy)
  • 이동 속도(speed)
  • 데미지(damage)
  • 삭제 여부(is_dead)

2. Bullet 클래스 구현

총알은 “타겟을 따라가는 유도탄” 방식으로 구현합니다.
(초보 단계에서 안정적이고, 타겟 변경 처리도 쉽습니다.)

class Bullet:
    def __init__(self, start_pos, target, speed=8, damage=10):
        self.x, self.y = start_pos
        self.target = target
        self.speed = speed
        self.damage = damage
        self.radius = 4
        self.is_dead = False

    def update(self):
        # 타겟이 이미 죽었거나 사라졌으면 총알도 제거
        if self.target.is_dead or self.target.reached_end:
            self.is_dead = True
            return

        tx, ty = self.target.x, self.target.y
        dx = tx - self.x
        dy = ty - self.y
        dist = (dx * dx + dy * dy) ** 0.5

        if dist == 0:
            dist = 0.0001

        # 타겟 방향으로 이동
        self.x += (dx / dist) * self.speed
        self.y += (dy / dist) * self.speed

        # 충돌 판정(원-원)
        hit_dist = self.radius + self.target.radius
        if dist <= hit_dist:
            self.target.hp -= self.damage
            if self.target.hp <= 0:
                self.target.is_dead = True
            self.is_dead = True

    def draw(self, screen):
        pygame.draw.circle(screen, (240, 240, 240), (int(self.x), int(self.y)), self.radius)

3. 타워 공격을 “총알 생성”으로 변경

10강에서는 target.hp -= damage를 타워에서 바로 실행했지만,
이번 강의부터는 “총알을 리스트에 추가”하는 방식으로 바꿉니다.

타워에서 총알을 만들어 bullets 리스트에 넣습니다.

def try_attack(self, bullets):
    if not self.target:
        return
    if self.target.is_dead or self.target.reached_end:
        return

    now = pygame.time.get_ticks()
    if now - self.last_fire_time >= self.fire_delay:
        bullets.append(Bullet((self.x, self.y), self.target, speed=8, damage=self.damage))
        self.last_fire_time = now

4. 메인 루프에서 bullets 업데이트/삭제

총알도 적과 마찬가지로 매 프레임 업데이트하고 그립니다.

for b in bullets:
    b.update()

bullets = [b for b in bullets if not b.is_dead]

5. 실행 결과


전체 코드

더보기
import pygame
import sys

pygame.init()

# -----------------------------
# 화면 설정
# -----------------------------
SCREEN_WIDTH = 800
SCREEN_HEIGHT = 600
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption("Tower Defense")

clock = pygame.time.Clock()
FPS = 60

# -----------------------------
# 경로 설정
# -----------------------------
path_points = [
    (0, 300),
    (100, 300),
    (100, 100),
    (300, 100),
    (300, 500),
    (650, 500),
    (650, 300),
    (450, 300),
    (450, 100),
    (700, 100),
    (700, 200),
    (800, 200)
]

PATH_WIDTH = 40
TOWER_RADIUS = 16
TOWER_COST = 50
TOWER_RANGE = 140

# -----------------------------
# 유틸: 점-선분 거리
# -----------------------------
def point_to_segment_distance(px, py, ax, ay, bx, by):
    abx = bx - ax
    aby = by - ay
    apx = px - ax
    apy = py - ay

    ab_len_sq = abx * abx + aby * aby
    if ab_len_sq == 0:
        return ((px - ax) ** 2 + (py - ay) ** 2) ** 0.5

    t = (apx * abx + apy * aby) / ab_len_sq
    t = max(0.0, min(1.0, t))

    cx = ax + abx * t
    cy = ay + aby * t

    dx = px - cx
    dy = py - cy
    return (dx * dx + dy * dy) ** 0.5

# -----------------------------
# 유틸: 경로 근처 설치 금지(타워 반지름 포함)
# -----------------------------
def is_on_path(pos, path_points):
    px, py = pos
    threshold = PATH_WIDTH / 2 + TOWER_RADIUS

    for i in range(len(path_points) - 1):
        ax, ay = path_points[i]
        bx, by = path_points[i + 1]
        if point_to_segment_distance(px, py, ax, ay, bx, by) <= threshold:
            return True
    return False

# -----------------------------
# 유틸: 타워 겹침 방지
# -----------------------------
def is_overlapping_tower(pos, towers):
    px, py = pos
    min_dist = TOWER_RADIUS * 2

    for t in towers:
        dx = px - t.x
        dy = py - t.y
        if (dx * dx + dy * dy) ** 0.5 < min_dist:
            return True
    return False

# -----------------------------
# Enemy 클래스
# -----------------------------
class Enemy:
    def __init__(self, path_points, speed=2):
        self.path_points = path_points
        self.speed = speed
        self.x, self.y = path_points[0]
        self.target_index = 1

        self.max_hp = 50
        self.hp = self.max_hp
        self.reward = 10

        self.reached_end = False
        self.is_dead = False
        self.radius = 12

    def update(self):
        if self.reached_end or self.is_dead:
            return

        if self.target_index >= len(self.path_points):
            self.reached_end = True
            return

        tx, ty = self.path_points[self.target_index]
        dx = tx - self.x
        dy = ty - self.y

        dist = (dx * dx + dy * dy) ** 0.5
        if dist == 0:
            self.target_index += 1
            return

        self.x += (dx / dist) * self.speed
        self.y += (dy / dist) * self.speed

        if dist <= self.speed:
            self.x, self.y = tx, ty
            self.target_index += 1
            if self.target_index >= len(self.path_points):
                self.reached_end = True

    def draw(self, screen):
        pygame.draw.circle(screen, (220, 80, 80), (int(self.x), int(self.y)), self.radius)

        # HP 바
        bar_width = 24
        bar_height = 4
        hp_ratio = max(0, self.hp) / self.max_hp

        pygame.draw.rect(screen, (60, 60, 60), (self.x - 12, self.y - 20, bar_width, bar_height))
        pygame.draw.rect(screen, (80, 220, 80), (self.x - 12, self.y - 20, bar_width * hp_ratio, bar_height))

# -----------------------------
# Bullet 클래스
# -----------------------------
class Bullet:
    def __init__(self, start_pos, target, speed=8, damage=10):
        self.x, self.y = start_pos
        self.target = target
        self.speed = speed
        self.damage = damage
        self.radius = 4
        self.is_dead = False

    def update(self):
        # 타겟이 사라졌으면 총알 제거
        if self.target.is_dead or self.target.reached_end:
            self.is_dead = True
            return

        tx, ty = self.target.x, self.target.y
        dx = tx - self.x
        dy = ty - self.y
        dist = (dx * dx + dy * dy) ** 0.5
        if dist == 0:
            dist = 0.0001

        # 이동
        self.x += (dx / dist) * self.speed
        self.y += (dy / dist) * self.speed

        # 충돌(원-원)
        hit_dist = self.radius + self.target.radius
        if dist <= hit_dist:
            self.target.hp -= self.damage
            if self.target.hp <= 0:
                self.target.is_dead = True
            self.is_dead = True

    def draw(self, screen):
        pygame.draw.circle(screen, (240, 240, 240), (int(self.x), int(self.y)), self.radius)

# -----------------------------
# Tower 클래스
# -----------------------------
class Tower:
    def __init__(self, pos):
        self.x, self.y = pos
        self.radius = TOWER_RADIUS
        self.range = TOWER_RANGE

        self.damage = 10
        self.fire_delay = 500
        self.last_fire_time = 0

        self.target = None

    def find_target(self, enemies):
        self.target = None
        best_dist = None

        for e in enemies:
            if e.is_dead or e.reached_end:
                continue

            dx = e.x - self.x
            dy = e.y - self.y
            dist = (dx * dx + dy * dy) ** 0.5

            if dist <= self.range:
                if best_dist is None or dist < best_dist:
                    best_dist = dist
                    self.target = e

    def try_attack(self, bullets):
        if not self.target:
            return
        if self.target.is_dead or self.target.reached_end:
            return

        now = pygame.time.get_ticks()
        if now - self.last_fire_time >= self.fire_delay:
            bullets.append(Bullet((self.x, self.y), self.target, speed=8, damage=self.damage))
            self.last_fire_time = now

    def draw(self, screen, show_range=False):
        pygame.draw.circle(screen, (80, 160, 255), (self.x, self.y), self.radius)
        pygame.draw.circle(screen, (30, 30, 30), (self.x, self.y), self.radius, 3)

        if show_range:
            pygame.draw.circle(screen, (120, 120, 120), (self.x, self.y), self.range, 1)

        if self.target and not self.target.is_dead and not self.target.reached_end:
            pygame.draw.line(
                screen, (255, 255, 255),
                (self.x, self.y),
                (int(self.target.x), int(self.target.y)),
                2
            )

# -----------------------------
# 게임 변수
# -----------------------------
enemies = []
towers = []
bullets = []  # 총알 리스트

spawn_interval = 1000
last_spawn_time = 0

base_hp = 20
gold = 100

font = pygame.font.SysFont(None, 32)
selected_tower_index = -1

# -----------------------------
# 메인 루프
# -----------------------------
while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()

        # 타워 설치(좌클릭)
        if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
            click_pos = pygame.mouse.get_pos()

            if not is_on_path(click_pos, path_points) and not is_overlapping_tower(click_pos, towers):
                if gold >= TOWER_COST:
                    towers.append(Tower(click_pos))
                    gold -= TOWER_COST
                    selected_tower_index = len(towers) - 1

    game_over = base_hp <= 0

    if not game_over:
        # 적 스폰
        now = pygame.time.get_ticks()
        if now - last_spawn_time >= spawn_interval:
            enemies.append(Enemy(path_points))
            last_spawn_time = now

        # 적 업데이트
        for enemy in enemies:
            enemy.update()

        # 타워 타겟/공격
        for tower in towers:
            tower.find_target(enemies)
            tower.try_attack(bullets)

        # 총알 업데이트
        for b in bullets:
            b.update()

        # 도착한 적 처리(베이스 HP 감소)
        arrived = [e for e in enemies if e.reached_end]
        base_hp -= len(arrived)

        # 죽은 적 처리(골드 획득)
        dead = [e for e in enemies if e.is_dead]
        gold += sum(e.reward for e in dead)

        # 적/총알 정리
        enemies = [e for e in enemies if not e.reached_end and not e.is_dead]
        bullets = [b for b in bullets if not b.is_dead]

    # -----------------------------
    # 화면 그리기
    # -----------------------------
    screen.fill((30, 30, 30))
    pygame.draw.lines(screen, (200, 180, 100), False, path_points, PATH_WIDTH)

    # 타워 출력
    for i, tower in enumerate(towers):
        tower.draw(screen, show_range=(i == selected_tower_index))

    # 적 출력
    for enemy in enemies:
        enemy.draw(screen)

    # 총알 출력
    for b in bullets:
        b.draw(screen)

    # UI
    screen.blit(font.render(f"HP: {base_hp}", True, (255, 255, 255)), (10, 10))
    screen.blit(font.render(f"Gold: {gold}", True, (255, 255, 0)), (10, 40))
    screen.blit(font.render(f"Tower Cost: {TOWER_COST}", True, (200, 200, 200)), (10, 70))

    if game_over:
        over_text = font.render("GAME OVER", True, (255, 255, 255))
        screen.blit(
            over_text,
            (SCREEN_WIDTH // 2 - over_text.get_width() // 2,
             SCREEN_HEIGHT // 2)
        )

    pygame.display.update()
    clock.tick(FPS)

728x90
도마2
@도마2 :: 도마의 코드노트

초보자를 위한 코딩 강의를 정리합니다. 파이썬부터 C#, Unity 게임 제작까지 차근차근 기록합니다. — 도마

목차