arctic-masher/enemies.py

import pygame
import uuid
import random

from globals import *
from utilities import *
from player import *
from blocks import *

def findClosestPlayer(x, y):
    closestPlayerPosition = {"difference": (1,1), "distance": 9999}
    for id, player in players.items():
        difference = (player.x - x, player.y - y)
        distance = max(abs(difference[0]), abs(difference[1]))
        if distance > closestPlayerPosition["distance"]:
            continue
        closestPlayerPosition = {"difference": difference, "distance": distance, "player": player}
    return closestPlayerPosition

def randomizeMovement(weightsx, weightsy, diff, wrongChance):
    weightsx = [-diff[0], abs(diff[1])+0.1, diff[0]]
    if weightsx[0] < 0:
        weightsx[0] = weightsx[2]*wrongChance
    elif weightsx[2] < 0:
        weightsx[2] = weightsx[0]*wrongChance
    else:
        weightsx[0] = max(weightsx[1]/(15.1-weightsx[1]*0.5), 1)
        weightsx[2] = weightsx[0]
    weightsy = [-diff[1], abs(diff[0])+0.1, diff[1]]
    if weightsy[0] < 0:
        weightsy[0] = weightsy[2]*wrongChance
    elif weightsy[2] < 0:
        weightsy[2] = weightsy[0]*wrongChance
    else:
        weightsy[0] = max(weightsy[1]/(15.1-weightsy[1]*0.5), 1)
        weightsy[2] = weightsy[0]
    return (weightsx, weightsy)

class Enemy:
    def __init__(self, id, x, y):
        self.id = id
        self.ENEMY = True
        self.x = x
        self.y = y
        self.renderPos = (x,y)
        # self.AITime = random.randint(350, 600)
        self.AITime = random.randint(10, 50)
        self.deathColor = white
        self.sprite = pygame.Surface((32,32), pygame.SRCALPHA)
        pygame.draw.circle(self.sprite, white, (16,16), 10)
        
    def die(self):
        del enemies[self.id]

    def pushed(self, x, y, caller, pusher):
        if hasattr(pusher, "PLAYER") and hasattr(caller, "BLOCK"):
            if checkOOBounds((self.x + x, self.y + y)) or hasattr(gameMap[self.x + x][self.y + y], "BLOCK"):
                pusher.kills += 1
                caller.color = self.deathColor
                self.die()
                return
        raise Exception("Not crushing enemy!")
    
    def move(self, x, y):
        try:
            if checkOOBounds((self.x + x, self.y + y)):
                raise Exception("Cannot move off edge!")
            if gameMap[self.x + x][self.y + y]:
                if hasattr(gameMap[self.x + x][self.y + y], "PLAYER"):
                    gameMap[self.x + x][self.y + y].die()
                else:
                    raise Exception("Path is blocked.")
            gameMap[self.x][self.y] = False
            self.x += x
            self.y += y
            gameMap[self.x][self.y] = self
        except Exception as error:
            raise error
    
    def runAI(self):
        self.AITime -= 1
        if self.AITime < 1:
            try:
                diff = findClosestPlayer(self.x, self.y)["difference"]
                weightsx = [1,1,1]
                weightsy = [1,1,1]
                randomWeights = randomizeMovement(weightsx, weightsy, diff, 0.15)
                weightsx = randomWeights[0]
                weightsy = randomWeights[1]
                movementx = random.choices([-1, 0, 1], weightsx)[0]
                movementy = random.choices([-1, 0, 1], weightsy)[0]
                self.move(movementx, movementy)
                self.AITime = random.randint(150, 200)
            except Exception as error:
                pass

class Smart(Enemy):
    def __init__(self, id, x, y):
        Enemy.__init__(self, id, x, y)
        self.SMART = True
        self.usesPathMap = True
        self.deathColor = red
        self.trackedPlayer = False
        self.trackingCounter = 0
        self.sprite = pygame.Surface((32,32), pygame.SRCALPHA)
        pygame.draw.circle(self.sprite, red, (16,16), 12)
    
    def runAI(self):
        self.AITime -= 1
        if self.AITime < 1:
            try:
                if self.trackingCounter < 1:
                    self.AITime = random.randint(150, 250)
                    self.trackingCounter = 9
                    self.trackedPlayer = False
                elif self.trackedPlayer and self.trackedPlayer["player"].alive:
                    pathMap = self.trackedPlayer["player"].pathMap
                    weightsx = [1,1,1]
                    weightsy = [1,1,1]
                    if pathMap[self.x][self.y]:
                        fastDir = pathMap[self.x][self.y][2]
                        weightsx[1 + fastDir[0]] = 30
                        weightsy[1 + fastDir[1]] = 30
                    else:
                        randomWeights = randomizeMovement(weightsx, weightsy, self.trackedPlayer["difference"], 0.32)
                        weightsx = randomWeights[0]
                        weightsy = randomWeights[1]
                    movementx = random.choices([-1, 0, 1], weightsx)[0]
                    movementy = random.choices([-1, 0, 1], weightsy)[0]
                    self.move(movementx, movementy)
                    self.trackingCounter -= 1
                    self.AITime = random.randint(10, 15)
                else:
                    closestPlayer = findClosestPlayer(self.x, self.y)
                    self.trackedPlayer = closestPlayer
                    self.AITime = random.randint(10, 15)
            except Exception as error:
                pass

class Mage(Enemy):
    def __init__(self, id, x, y):
        Enemy.__init__(self, id, x, y)
        self.deathColor = purple
        self.sprite = pygame.Surface((32,32), pygame.SRCALPHA)
        pygame.draw.circle(self.sprite, purple, (16,16), 6)
    
    def runAI(self):
        self.AITime -= 1
        if self.AITime < 1:
            try:
                diff = findClosestPlayer(self.x, self.y)["difference"]
                weightsx = [1,1,1]
                weightsy = [1,1,1]
                randomWeights = randomizeMovement(weightsx, weightsy, diff, 0.15)
                weightsx = randomWeights[0]
                weightsy = randomWeights[1]
                movementx = random.choices([-1, 0, 1], weightsx)[0] * random.randint(1,2)
                movementy = random.choices([-1, 0, 1], weightsy)[0] * random.randint(1,2)
                self.move(movementx, movementy)
                self.AITime = random.randint(150, 200)
            except Exception as error:
                pass