Projekt_Si/classes/agent.py

import pygame
from collections import deque
import prefs
class Agent:
    def __init__(self, x, y, cells, baseScore=0):
        self.sprite = pygame.image.load("sprites/BartenderNew64.png").convert_alpha()
        self.sprite = pygame.transform.scale(self.sprite, (prefs.CELL_SIZE, prefs.CELL_SIZE))
        self.current_cell = cells[x][y]
        self.moved=False
        self.last_move_time = pygame.time.get_ticks()
        self.last_interact_time = pygame.time.get_ticks()
        self.last_update_time = pygame.time.get_ticks()
        self.cells = cells
        self.score = baseScore
        self.multiplier = 1
        self.direction = 0

        self.textures = [
            pygame.image.load("sprites/BartenderNew64.png").convert_alpha(),
            pygame.image.load("sprites/AgentLewo.png").convert_alpha(), 
            pygame.image.load("sprites/AgentTyl.png").convert_alpha(),
            pygame.image.load("sprites/AgentPrawo.png").convert_alpha()
        ]

    def move_up(self):
        if pygame.time.get_ticks()-self.last_move_time > 125 and self.current_cell.Y > 0 and not self.cells[self.current_cell.X][self.current_cell.Y-1].blocking_movement:
            self.current_cell = self.cells[self.current_cell.X][self.current_cell.Y-1]
            self.moved=True
            self.last_move_time=pygame.time.get_ticks()

    def move_down(self):
        if pygame.time.get_ticks()-self.last_move_time > 125 and self.current_cell.Y < prefs.GRID_SIZE-1 and not self.cells[self.current_cell.X][self.current_cell.Y+1].blocking_movement:
            self.current_cell = self.cells[self.current_cell.X][self.current_cell.Y+1]
            self.moved=True
            self.last_move_time=pygame.time.get_ticks()

    def move_left(self):
        if pygame.time.get_ticks()-self.last_move_time > 125 and self.current_cell.X > 0 and not self.cells[self.current_cell.X-1][self.current_cell.Y].blocking_movement:
            self.current_cell = self.cells[self.current_cell.X-1][self.current_cell.Y]
            self.moved=True
            self.last_move_time=pygame.time.get_ticks()

    def move_right(self):
        if pygame.time.get_ticks()-self.last_move_time > 125 and self.current_cell.X < prefs.GRID_SIZE-1 and not self.cells[self.current_cell.X+1][self.current_cell.Y].blocking_movement:
            self.current_cell = self.cells[self.current_cell.X+1][self.current_cell.Y]
            self.moved=True
            self.last_move_time=pygame.time.get_ticks()

    def update(self, surface):
        surface.blit(self.sprite, (self.current_cell.X * prefs.CELL_SIZE, 
                                   self.current_cell.Y * prefs.CELL_SIZE))
        
        current_update_time = pygame.time.get_ticks()
        # różnca czasu między wyoływaniami, używa do uniezależnienia od ilości wywołań funkcji
        delta_time = ((current_update_time - self.last_update_time)/1000)
        self.increase_multiplier(-(1 / 16) * delta_time)

                
                
        self.last_update_time = current_update_time

    def increase_score(self, amount):
        self.score += amount * round(self.multiplier,2)
        print("Agent score changed from {} to {} (multiplied by {}!)".format(self.score - amount, self.score, round(self.multiplier,2)))

    def increase_multiplier(self, amount):
        self.multiplier += amount
        if self.multiplier > 2: 
            self.multiplier = 2
            print("Agent score changed from {} to {}".format(self.multiplier , self.multiplier + amount  if self.multiplier + amount <= 2 else 2))
            return
        if self.multiplier < 1: 
            self.multiplier = 1

    def moveto(self,x,y):
        if not self.cells[x][y].blocking_movement:
            self.current_cell = self.cells[x][y]
            self.moved=True
            self.last_move_time=pygame.time.get_ticks()
            print("Agent moved to x,y: ",x,y)
        else:
            print("Agent cannot move to this direction")


    def bfs(self, start, target, cells):
        queue = deque([(start,[])])
        visited = set()

        while queue:
            current, path = queue.popleft()
            if current==target:
                return path + [current]
            
            if current in visited:
                continue

            visited.add(current)

            for neighbor in self.get_neighbors(current, cells):
                queue.append((neighbor, path + [current]))

        return None

    def get_neighbors(self, cell, cells):
        neighbors = []
        x, y = cell.X, cell.Y
        if x > 0 and not cells[x-1][y].blocking_movement:
            neighbors.append(cells[x-1][y])
        if x < prefs.GRID_SIZE - 1 and not cells[x+1][y].blocking_movement:
            neighbors.append(cells[x+1][y])
        if y > 0 and not cells[x][y-1].blocking_movement:
            neighbors.append(cells[x][y-1])
        if y < prefs.GRID_SIZE - 1 and not cells[x][y+1].blocking_movement:
            neighbors.append(cells[x][y+1])

        return neighbors

    #oddaje tablice punktow jako sciezke agenta
    def convert_to_coordinates(self, shortest_path):
        coordinates = [(cell.X, cell.Y) for cell in shortest_path]
        return coordinates

    #Wyjmuje pierwsze koordynaty do ruszenia agenta a potem usuwa go z listy
    def pop_first_coordinates(self, coordinates):
        if coordinates:
            x, y = coordinates.pop(0)
            return x, y
        else:
            print("Lista współrzędnych jest pusta.")
            return None, None

        
    #Funkcja pomocnicza dla watku bo chcemy zeby agent poruszal sie ale zeby to normalnie wygladalo
    def sciezkaAgenta(self, agent, path):
        x,y = self.pop_first_coordinates(path)
        if x is not None and y is not None:
            agent.moveto(x,y)


    def rotate_left(self):
        if pygame.time.get_ticks()-self.last_move_time > 125:
            self.direction +=1
            if self.direction==4:
                self.direction=0
            self.sprite = self.textures[self.direction]
            self.sprite = pygame.transform.scale(self.sprite, (prefs.CELL_SIZE, prefs.CELL_SIZE))
            self.last_move_time=pygame.time.get_ticks()
            print(self.direction)

    def rotate_right(self):
        if pygame.time.get_ticks()-self.last_move_time > 125:
            self.direction-=1
            if self.direction==-1:
                self.direction=3
            self.sprite = self.textures[self.direction]
            self.sprite = pygame.transform.scale(self.sprite, (prefs.CELL_SIZE, prefs.CELL_SIZE))
            self.last_move_time=pygame.time.get_ticks()
            print(self.direction)