interstellar.py

import pygame
from const import Const
from random import randint


class Interstellar:
    def __init__(self, images, speed, explode_images=None, explode_sounds=None, x=0, y=0):
        self.original_images = images
        self.images = self.original_images
        self.speed = speed
        self.x, self.y = x, y
        self.width, self.height = 0, 0
        self.away = False
        self.hitbox = (0, 0, 0, 0)
        self.hitsize = (0, 0, 0, 0)
        self.num_of_images = len(self.images) if self.images else 0
        self.frame_time = Const.FRAME_TIME_SEC
        self.next_frame = 0
        self.frame_num = 0
        self.current_image_set = self.images
        self.allow_off_the_screen = False
        self.exploding = False
        self.explode_images = explode_images
        self.explode_sounds = explode_sounds
        if explode_images:
            self.num_of_explosion_frames = len(self.explode_images) - 1
        if explode_sounds:
            self.num_of_explode_sounds = len(explode_sounds) - 1

    def get_xy(self):
        """
        Return the position
        """
        return self.x, self.y

    def get_hitbox(self):
        """
        Return the hitbox
        """
        return self.hitbox

    def set_xy(self, x, y):
        """
        Set the position
        """
        self.x = x
        self.y = y

    def move(self):
        """
        Move the hitbox, if exists
        """
        # Move the hitbox only if the hit width is > 0, otherwise it isn't a hittable object
        if self.hitsize[2] != 0:
            self.hitbox = tuple(map(sum, zip((self.x, self.y, -self.hitsize[0], -self.hitsize[1]), self.hitsize)))

    def get_current_pic(self):
        """
        Return the current picture
        """
        return self.images[self.frame_num]

    def is_away(self):
        """
        Return true if the object is off the screen
        """
        return self.away

    def is_hit(self):
        """
        Check whether the enemy is already hit
        """
        return self.exploding

    def get_width(self):
        """
        Return object's width
        """
        return self.width

    def get_height(self):
        """
        Return object's height
        """
        return self.height

    @staticmethod
    def rescale(images_source, scale_x, scale_y):
        """
        Rescale images
        """
        images_target = []
        for image in images_source:
            images_target.append(pygame.transform.scale(image, (scale_x, scale_y)))
        return images_target

    def off_the_screen(self):
        """
        Return true if the object has vanished from the screen
        """
        if not self.allow_off_the_screen and \
                (self.x > Const.OFF_THE_SCREEN_RIGHT or self.y > Const.OFF_THE_SCREEN_BOTTOM or
                 self.x + self.width < Const.OFF_THE_SCREEN_LEFT or self.y + self.height < Const.OFF_THE_SCREEN_TOP):
            return True
        else:
            return False

    def hit(self):
        """
        Cause the enemy to become hit
        """
        self.exploding = True
        self.frame_num = 0
        self.frame_time = Const.EXPLOSION_ANIMATE_SPEED
        self.current_image_set = self.explode_images
        orig_width = self.width
        orig_height = self.height
        self.width, self.height = self.current_image_set[0].get_rect().size
        self.x = self.x - self.width // 2 + orig_width // 2
        self.y = self.y - self.height // 2 + orig_height // 2
        self.hitsize = (Const.EXPLOSION_HIT_DELTA, Const.EXPLOSION_HIT_DELTA,
                        self.width - Const.EXPLOSION_HIT_DELTA, self.height - Const.EXPLOSION_HIT_DELTA)
        if self.explode_sounds:
            explosion = self.explode_sounds[randint(0, self.num_of_explode_sounds)]
            explosion.play()