person_model.py

from mesa import Model
from mesa.time import RandomActivation
from mesa.space import MultiGrid
from person_agent import PersonAgent
from utils import width, height, update_tx
from random import randrange


# Gera coordenadas para serem utilizadas como posição inical do agente no grid
def genPos():
    x = randrange(width)
    y = randrange(height)
    return x, y


# Model Pessoa
# Responsável pela comunicação com a interface
# Recebe parâmetros da interface 
# Criação dos agentes (Agente Pessoa) 
class PersonModel(Model):
  def __init__(self, persons, initial_infected, tx_death, tx_transmission):
        self.numAgents = persons
        self.schedule = RandomActivation(self)
        self.grid = MultiGrid(width,height, True)
        self.running = True
        self.filledPositions = []
        update_tx(tx_death,tx_transmission)

        # Cria os agentes de acordo com o número especificado na interface
        for i in range(self.numAgents):
            x, y = genPos()
            while (x,y) in self.filledPositions:
                x, y = genPos()
            self.filledPositions.append((x,y))
            agent = PersonAgent(i, self)
            self.schedule.add(agent)
            self.grid.place_agent(agent, (x , y))
            
            # Inicializa a quantidade informada pelo usuário de agentes infectados com o vírus 
            if i < initial_infected:
                agent.isContaminated = True
                agent.isTransmitter = True

  def step(self):
        self.schedule.step()