bullet.py

import math

import pygame as pg
from pygame.sprite import Sprite


class Bullet(Sprite):
    """A class to manage bullets fired from the ship"""

    def __init__(self, setting, screen, ship, traj, damage, charge=0):
        """Create a bullet object at the ships current position"""
        super(Bullet, self).__init__()
        self.screen = screen

        # decide the trajectory of bullets
        self.traj = traj
        # To specify the bullet is ultimate
        self.isUltimate = False

        # load the bullet image and set its rect attribute
        self.image = pg.image.load('gfx/pBullet.png')
        if (self.traj == 1):
            self.image = pg.transform.rotate(self.image, 315)
        elif (self.traj == 2):
            self.image = pg.transform.rotate(self.image, 45)
        self.rect = self.image.get_rect()
        if (charge != 0):
            bulletSize = (self.rect.width * (charge + 1), self.rect.height * (charge + 1))
            self.image = pg.transform.scale(self.image, bulletSize)
            self.rect = self.image.get_rect()

            # Create a bullet rect at (0,0)
        ##self.rect = pg.Rect(0, 0, setting.bulletWidth, setting.bulletHeight)
        self.rect.centerx = ship.rect.centerx
        self.rect.top = ship.rect.top

        # store the bullets position as a decimal value
        self.x = float(self.rect.centerx)
        self.y = float(self.rect.y)
        self.setting = setting

        # damage of basic bullet (default : 1)
        self.damage = damage

    def update(self):
        """Move the bullet -y up the screen"""
        # update the decimal position of the bullet
        bulletSpeed = self.setting.bulletSpeed
        if (self.traj == 0):
            self.y -= 1.5 * bulletSpeed
        elif (self.traj == 1):
            self.x += 0.5 * bulletSpeed
            self.y -= 0.5 * 2.0 * bulletSpeed
        elif (self.traj == 2):
            self.x -= 0.5 * bulletSpeed
            self.y -= 0.5 * 2.0 * bulletSpeed
        else:
            self.x -= 0.8 * math.sin(0.05 * self.y)
            self.y -= 0.8 * bulletSpeed

            # Update the rect position899
        self.rect.centerx = self.x
        self.rect.y = self.y

    def drawBullet(self):
        """Draw the bullet to the screen"""
        # pg.draw.rect(self.screen, self.color, self.rect)
        self.screen.blit(self.image, self.rect)


class SpecialBullet(Sprite):
    """A class to manage special bullet which can be fired only by the ultimate"""

    def __init__(self, setting, screen, pos, damage):
        """Create the bullet object at the some position"""
        super(SpecialBullet, self).__init__()
        self.screen = screen

        # To specify the bullet is ultimate
        self.isUltimate = True

        # load the bullet image and set its rect attribute
        self.image = pg.image.load('gfx/sBullet.png')
        self.rect = self.image.get_rect()

        # Create a bullet rect at (0,0)
        ##self.rect = pg.Rect(0, 0, setting.bulletWidth, setting.bulletHeight)
        self.rect.centerx = pos[0]
        self.rect.top = pos[1]

        # store the bullets position as a decimal value
        self.y = float(self.rect.y)
        self.setting = setting

        # damage of SpecialBullet
        self.damage = damage * 10

    def update(self):
        """Move the bullet -y up the screen"""
        # update the decimal position of the bullet
        bulletSpeed = self.setting.bulletSpeed
        self.y -= bulletSpeed
        self.rect.y = self.y

    def drawBullet(self):
        """Draw the bullet to the screen"""
        # pg.draw.rect(self.screen, self.color, self.rect)
        self.screen.blit(self.image, self.rect)