Display.py

import pygame
from Sim import sim
import sys
import json

class display:
    def __init__(self):
        pygame.init()
        self.WIDTH, self.HEIGHT = 600, 700
        self.DISPLAY_FACTOR = 25
        self.screen = pygame.display.set_mode(
            (self.WIDTH, self.HEIGHT)
        )
        pygame.display.set_caption("Ecosystem Simulation")
        self.clock = pygame.time.Clock()
        self.FPS = 2

        self.WHITE = (255, 255, 255)
        self.BLUE = (0, 0, 255)
        self.RED = (255, 0, 0)
        self.BLACK = (0, 0, 0)

        self.font = pygame.font.Font('freesansbold.ttf', 40)
        self.font_display_y = 650

        with open("data.json") as f:
            self.data = json.load(f)

    def draw_animals(self, info):
        for animal in info:
            x = animal["x"] * self.DISPLAY_FACTOR
            y = animal["y"] * self.DISPLAY_FACTOR
            pygame.draw.rect(self.screen, self.RED, (x, y, self.DISPLAY_FACTOR, self.DISPLAY_FACTOR))
            pygame.draw.rect(self.screen, self.BLACK, (x, y, self.DISPLAY_FACTOR, self.DISPLAY_FACTOR), 1)

    def draw_food(self, foods):
        for food in foods:
            x = food["x"] * self.DISPLAY_FACTOR
            y = food["y"] * self.DISPLAY_FACTOR
            pygame.draw.rect(self.screen, self.BLUE, (x, y, self.DISPLAY_FACTOR, self.DISPLAY_FACTOR))

    def write_cycle(self, cycle):
        msg = f"Cycle: {cycle}"
        rendered_item = self.font.render(msg, False, self.BLACK)
        self.screen.blit(rendered_item, (40, self.font_display_y))

    def write_pop(self, pop):
        msg = f"Population: {pop}"
        rendered_item = self.font.render(msg, False, self.BLACK)
        self.screen.blit(rendered_item, (300, self.font_display_y))

    def run(self):
        loop = 0
        while loop < len(self.data):
            data = self.data[loop]
            self.clock.tick(self.FPS)

            #Check and handle events
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    sys.exit()
                if event.type == pygame.KEYDOWN:
                    if event.key == pygame.q:
                        sys.exit()

            #Fill the screen before drawing new items
            self.screen.fill(self.WHITE)

            ###Draw Items###
            #Create a border for the simulation area
            pygame.draw.rect(self.screen, self.BLACK, (0, 0, 600, 600), 1)

            self.draw_food(data["food_data"])
            self.draw_animals(data["animals"]["info"])
            self.write_cycle(data["cycle"])
            self.write_pop(data["pop"])
            ###

            pygame.display.flip()
            loop += 1

if __name__ == '__main__':
    d = display()
    d.run()