screenW, screenH = 1440, 900 screen = pygame.display.set_mode((screenW, screenH)) vis = Visualizer(screen, simul) # Run Simulation clock = pygame.time.Clock() while True: clock.tick(FPS) total_frame += 1 for event in pygame.event.get(): if event.type == pygame.QUIT: pygame.quit() sys.exit(0) screen.fill(BLACK) # Calculate 1: Infect Healthy People ## rearrange grid comm.calculate_infection() # Calculate 2: Infected People probabistically Dead or Cured comm.calculate_change_of_infected() # Calculate 3: PCR scan and quarantine comm.calculate_pcr_scan() # Calculate ?: Move comm.move_people() # visualize vis.draw_all() pygame.display.flip() # quit pygame.quit()