unit.py

import pygame as pg
from pygame.sprite import Sprite
from settings import *


# =============================================== Owner Classes ================================================== #
class Unit(Sprite):
    """This class handles knights."""
    def __init__(self, game):
        self.groups = game.g.all_sprites, game.g.collision_sprites, game.g.units
        pg.sprite.Sprite.__init__(self, self.groups)
        self.game = game
        # Image information
        self.image_deselected = pg.image.load(UNIT_DESELECTED)
        self.image_selected = pg.image.load(UNIT_DESELECTED)
        self.image = self.image_deselected
        # Location information.
        self.size = self.image.get_rect().size
        self.rect = self.image.get_rect()
        self.rect.x = pg.mouse.get_pos()[0] - (self.size[0] / 2) - self.game.camera.x
        self.rect.y = pg.mouse.get_pos()[1] - (self.size[1] / 2) - self.game.camera.y
        # Movement slope information.
        self.move_x = 0
        self.move_y = 0
        self.move_slope = 0
        self.move_quadrant = 0
        self.move_counter = 0
        self.speed = KNIGHT_SPEED
        self.health = KNIGHT_HP
        # Boolean logic.
        self.is_selected = False
        self.is_moving = False
        self.is_hotkeyed = False

    def update(self):
        """This fires every single frame."""
        if (self.game.mouse.is_commanding and self.is_selected) or self.is_moving:
            self.set_move_slope()
        self.collision_detection()

    def set_move_location(self):
        """Defines new movement location for selected units."""
        self.move_counter = 0
        self.is_moving = True
        self.move_x = pg.mouse.get_pos()[0] - self.game.camera.x
        self.move_y = pg.mouse.get_pos()[1] - self.game.camera.y

    def set_move_slope(self):
        """This calculates the movement to a given location."""
        self.game.slope_movement_logic(self)

    def collision_detection(self):
        """If collision with another sprite occurs, this bumps us away."""
        for other in pg.sprite.spritecollide(self, self.game.g.collision_sprites, False):
            self.game.collision_bump(self, other)

    def set_selected(self):
        """Adds unit into the selected units group."""
        pg.sprite.Group.add(self.game.g.selected_units, self)
        self.is_selected = True
        self.image = self.image_selected

    def set_deselected(self):
        """Removes unit from the selected units group."""
        pg.sprite.Group.remove(self.game.g.selected_units, self)
        self.is_selected = False
        self.image = self.image_deselected

    def delete(self):
        """Deletes self."""
        pg.sprite.spritecollide(self, self.game.g.units, True)


class Knight(Unit):
    def __init__(self, game):
        super().__init__(game)
        # Image information.
        self.image_deselected = pg.image.load(KNIGHT_DESELECTED)
        self.image_selected = pg.image.load(KNIGHT_SELECTED)
        self.image = pg.image.load(KNIGHT_DESELECTED)
        self.health = KNIGHT_HP


# =============================================== Enemy Classes ================================================== #
class EnemyUnit(Sprite):
    """This class handles enemy units."""
    def __init__(self, game):
        self.groups = game.g.all_sprites, game.g.collision_sprites, game.g.enemy_units
        pg.sprite.Sprite.__init__(self, self.groups)
        self.game = game
        # Image information
        self.image = pg.image.load(UNIT_DESELECTED)
        self.size = self.image.get_rect().size
        self.rect = self.image.get_rect()
        self.rect.x = pg.mouse.get_pos()[0] - (self.size[0] / 2) - self.game.camera.x
        self.rect.y = pg.mouse.get_pos()[1] - (self.size[1] / 2) - self.game.camera.y
        # Movement slope information.
        self.move_x = 0
        self.move_y = 0
        self.move_slope = 0
        self.move_quadrant = 0
        self.move_counter = 0

        # Health information
        self.health = KNIGHT_HP

    def child_init(self):
        pass

    def update(self):
        """I need an AI class one day."""
        self.collision_detection()

    def collision_detection(self):
        """If collision with another sprite occurs, this bumps us away."""
        for other in pg.sprite.spritecollide(self, self.game.g.collision_sprites, False):
            self.game.collision_bump(self, other)

    def delete(self):
        """Deletes self."""
        pg.sprite.spritecollide(self, self.game.g.enemy_units, True)


class EnemyKnight(EnemyUnit):
    def __init__(self, game):
        super().__init__(game)
        # Image information.
        self.image = pg.image.load(ENEMY_KNIGHT)