utils.py

import pygame
import engine
import globals
import gamecomponents

pygame.font.init()
font = pygame.font.Font(pygame.font.get_default_font(), 24)

# function from:
# https://nerdparadise.com/programming/pygameblitopacity
def blit_alpha(target, source, location, opacity):
    x = location[0]
    y = location[1]
    temp = pygame.Surface((source.get_width(), source.get_height())).convert()
    temp.blit(target, (-x, -y))
    temp.blit(source, (0, 0))
    temp.set_alpha(opacity)        
    target.blit(temp, location)

def drawText(screen, t, x, y, fg, alpha):
    text = font.render(t, True, fg)
    text_rectangle = text.get_rect()
    text_rectangle.topleft = (x,y)

    blit_alpha(screen, text, (x,y), alpha)

heart_image = pygame.image.load('images/heart.png')

def setHealth(entity):
    if entity.battle:
        entity.battle.lives = 3

def setInvisible(entity):
    if entity.imageGroups:
        entity.imageGroups.alpha = 50

def endInvisible(entity):
    if entity.imageGroups:
        entity.imageGroups.alpha = 255

powerups = ['health', 'invisible']

powerupImages = {
    'health' : [pygame.image.load('images/powerup_health.png')],
    'invisible' : [pygame.image.load('images/powerup_invisible.png')]
}

powerupSound = {
    'health' : 'coin',
    'invisible' : 'coin'
}

powerupApply = {
    'health' : setHealth,
    'invisible' : setInvisible
}

powerupEnd = {
    'health' : None,
    'invisible' : endInvisible
}

powerupEffectTimer = {
    'health' : 0,
    'invisible' : 1000
}

def makePowerup(type, x, y):
    entity = engine.Entity()
    entity.position = engine.Position(x,y,40,40)
    entityAnimation = engine.ImageGroup(powerupImages[type])
    entity.imageGroups.add('idle', entityAnimation)
    entity.effect = gamecomponents.Effect(
        powerupApply[type], 
        powerupEffectTimer[type],
        powerupSound[type],
        powerupEnd[type]
    )
    return entity    

coin0 = pygame.image.load('images/coin_0.png')
coin1 = pygame.image.load('images/coin_1.png')
coin2 = pygame.image.load('images/coin_2.png')
coin3 = pygame.image.load('images/coin_3.png')
coin4 = pygame.image.load('images/coin_4.png')
coin5 = pygame.image.load('images/coin_5.png')

def makeCoin(x,y):
    entity = engine.Entity()
    entity.position = engine.Position(x,y,23,23)
    entityAnimation = engine.ImageGroup([coin1, coin2, coin3, coin4, coin5], delay=12)
    entity.imageGroups.add('idle', entityAnimation)
    entity.type = 'collectable'
    return entity

enemy0 = pygame.image.load('images/spike_monster.png')

def makeEnemy(x,y):
    entity = engine.Entity()
    entity.position = engine.Position(x,y,50,26)
    entityAnimation = engine.ImageGroup([enemy0])
    entity.imageGroups.add('idle', entityAnimation)
    entity.type = 'dangerous'
    return entity

playing = pygame.image.load('images/playing.png')
not_playing = pygame.image.load('images/not_playing.png')

idle0 = pygame.image.load('images/vita_00.png')
idle1 = pygame.image.load('images/vita_01.png')
idle2 = pygame.image.load('images/vita_02.png')
idle3 = pygame.image.load('images/vita_03.png')

walking0 = pygame.image.load('images/vita_04.png')
walking1 = pygame.image.load('images/vita_05.png')
walking2 = pygame.image.load('images/vita_06.png')
walking3 = pygame.image.load('images/vita_07.png')
walking4 = pygame.image.load('images/vita_08.png')
walking5 = pygame.image.load('images/vita_09.png')

jumping = pygame.image.load('images/vita_11.png')

def orderPlayers():
    newPlayerOrder = []
    for player in [globals.player1, globals.player2, globals.player3, globals.player4]:
        if player in globals.players:
            newPlayerOrder.append(player)
    globals.players = newPlayerOrder

def setPlayerCameras():

    screenWidth, screenHeight = pygame.display.get_surface().get_size()

    # 1 player game
    if len(globals.players) == 1:
        cameraWidth = screenWidth - (2*10)
        cameraHeight = screenHeight - (2*10)
        p = globals.players[0]
        p.camera = engine.Camera(10,10,cameraWidth, cameraHeight)
        p.camera.setWorldPos(p.position.initial.x, p.position.initial.y)
        p.camera.trackEntity(p)
    
    # 2 player game
    if len(globals.players) == 2:

        cameraWidth = (screenWidth - (3*10)) / 2
        cameraHeight = screenHeight - (2*10)

        p1 = globals.players[0]
        p1.camera = engine.Camera(10,10,cameraWidth, cameraHeight)
        p1.camera.setWorldPos(p1.position.initial.x, p1.position.initial.y)
        p1.camera.trackEntity(p1)

        p2 = globals.players[1]
        p2.camera = engine.Camera((2*10)+cameraWidth,10,cameraWidth, cameraHeight)
        p2.camera.setWorldPos(p2.position.initial.x, p2.position.initial.y)
        p2.camera.trackEntity(p2)

    # 3 or 4 player game
    if len(globals.players) >= 3:
        cameraWidth = (screenWidth - (3*10)) / 2
        cameraHeight = (screenHeight - (3*10)) / 2
        p1 = globals.players[0]
        p1.camera = engine.Camera(10,10,cameraWidth, cameraHeight)
        p1.camera.setWorldPos(p1.position.initial.x, p1.position.initial.y)
        p1.camera.trackEntity(p1)

        p2 = globals.players[1]
        p2.camera = engine.Camera((2*10)+cameraWidth,10,cameraWidth, cameraHeight)
        p2.camera.setWorldPos(p2.position.initial.x, p2.position.initial.y)
        p2.camera.trackEntity(p2)

        p3 = globals.players[2]
        p3.camera = engine.Camera(10,(2*10)+cameraHeight,cameraWidth, cameraHeight)
        p3.camera.setWorldPos(p3.position.initial.x, p3.position.initial.y)
        p3.camera.trackEntity(p3)

        if len(globals.players) == 4:
            p4 = globals.players[3]
            p4.camera = engine.Camera((2*10)+cameraWidth,(2*10)+cameraHeight,cameraWidth, cameraHeight)
            p4.camera.setWorldPos(p4.position.initial.x, p4.position.initial.y)
            p4.camera.trackEntity(p4)

def resetPlayer(entity):
    entity.score.score = 0
    entity.battle.lives = 3
    entity.position.rect.x = entity.position.initial.x
    entity.position.rect.y = entity.position.initial.y
    entity.speed = 0
    entity.acceleration = entity.initialAcceleration
    entity.camera.setWorldPos(entity.position.initial.x, entity.position.initial.y)
    entity.direction = 'right'
    entity.imageGroups.alpha = 255
    entity.effect = None
    entity.state = 'idle'
    if entity.camera is not None:
        entity.camera.zoomLevel = 1

def makePlayer(x,y):
    entity = engine.Entity()
    entity.position = engine.Position(x,y,45,51)
    entityIdleAnimation = engine.ImageGroup([idle0, idle1, idle2, idle3])
    entityWalkingAnimation = engine.ImageGroup([walking0, walking1, walking2, walking3, walking4, walking5], delay=6)
    entityJumpingAnimation = engine.ImageGroup([jumping])
    entity.imageGroups.add('idle', entityIdleAnimation)
    entity.imageGroups.add('walking', entityWalkingAnimation)
    entity.imageGroups.add('jumping', entityJumpingAnimation)
    entity.score = gamecomponents.Score()
    entity.battle = gamecomponents.Battle()
    entity.intention = engine.Intention()
    entity.acceleration = 0.3
    entity.initialAcceleration = entity.acceleration
    entity.type = 'player'
    entity.reset = resetPlayer
    entity.rigidBody = engine.RigidBody(10,1,25,50)
    return entity