HTequi/blob-simulation/interface.py

import pygame
import random
import time
import datetime

from pygame.locals import *
from board import Board
from blob.main import Blob_Manager
from player import Player


class Interface:

    def __init__(self, board, player, blob, scale, save_dir):
        """
        :type board: Board
        :type player: Player
        :type blob: Blob_Manager
        :type scale: float
        :type save_dir: str
        """
        pygame.init()
        # pygame.key.set_repeat(5, 50)

        self.board = board
        self.player = player
        self.blob = blob
        self.scale = scale

        self.save_dir = save_dir

        self.debug_mode = False
        self.play = False
        self.do_step = False
        self.show_ants = True

        width = self.board.width * scale
        height = self.board.height * scale
        self.window = pygame.display.set_mode((width, height))

        discovered_food = pygame.image.load("cross.png").convert()
        discovered_food.set_colorkey((255, 255, 255))
        self.discovered_food = pygame.transform.scale(discovered_food, (scale, scale))

    def draw(self):

        width = self.board.width * self.scale
        height = self.board.height * self.scale

        game_surface = pygame.Surface((self.board.width, self.board.height))
        board_array = pygame.PixelArray(game_surface)
        for x in range(self.board.width):
            for y in range(self.board.height):
                board_array[x,y] = (0,0,0)
                if self.board.is_touched(x, y):
                    board_array[x, y] = (50, 50, 0)

                val = self.board.get_blob(x, y)
                val = max(0, min(val, 255))
                if val != 0:
                    board_array[x, y] = (255, 255 - val, 0)

                if self.board.has_food(x, y):
                    board_array[x, y] = (0, 150, 0)

                if self.show_ants:
                    for scouter in self.blob.scouters:
                        board_array[scouter.x, scouter.y] = (255, 255, 255)

        del board_array

        game_window = pygame.transform.scale(game_surface, (width, height))

        pygame.draw.line(game_window, (125, 125, 125), (0, height / 2), (width, height / 2))

        self.window.blit(game_window, (0, 0))
        for food in self.blob.knowledge['food']:
            self.window.blit(self.discovered_food, (food[0] * self.scale, food[1] * self.scale))
        pygame.display.flip()

    def event_listener(self, event):

        # ADMIN ACTIONS
        if event.type == KEYDOWN and event.key == 100:  # D Letter
            self.debug_mode = not self.debug_mode
            print("Debug Mode: " + str(self.debug_mode))

        elif event.type == KEYDOWN and event.key == K_UP:
            Board.DECREASE_BLOB += 0.1
            print("Pheromone evaporation : " + str(Board.DECREASE_BLOB))

        elif event.type == KEYDOWN and event.key == K_DOWN:
            Board.DECREASE_BLOB -= 0.1
            print("Pheromone evaporation : " + str(Board.DECREASE_BLOB))

        elif event.type == KEYDOWN and event.key == K_SPACE:
            self.show_ants = not self.show_ants

        elif event.type == KEYDOWN and event.key == 115:  # S Letter
            ts = datetime.datetime.fromtimestamp(time.time()).strftime('%Y-%m-%d_%H.%M.%S')

            print("Data saved at " + ts)
            f = open(self.save_dir + ts + ".board", 'w')
            f.write(self.board.save())
            f.close()

            f = open(self.save_dir + ts + ".blob", 'w')
            f.write(self.blob.save())
            f.close()

            f = open(self.save_dir + ts + ".player", 'w')
            f.write(self.player.save())
            f.close()

            pygame.image.save(self.window, self.save_dir + ts + ".jpg")

        # DEBUG ACTIONS
        elif self.debug_mode and event.type == MOUSEBUTTONDOWN and event.button == 1:  # Right Click
                x = int(pygame.mouse.get_pos()[0] / self.scale)
                y = int(pygame.mouse.get_pos()[1] / self.scale)
                self.board.update_blob(x, y, 10)

        # PLAYER ACTIONS
        elif event.type == MOUSEBUTTONDOWN and event.button == 1:  # Right Click
            x = int(pygame.mouse.get_pos()[0]/self.scale)
            y = int(pygame.mouse.get_pos()[1]/self.scale)
            self.player.set_food(x, y)

        elif event.type == KEYDOWN and event.key == 110:  # N Letter
            self.player.clean_board()

        elif event.type == KEYDOWN and event.key == 114:  # R Letter
            self.player.set_random_food(10, not self.player.clean_top)

        elif event.type == KEYDOWN and event.key == 104:  # H Letter
            size_percent = self.player.check_blob_size()
            print("Current blob size percent: " + str(size_percent))

        # BLOB ACTIONS
        elif event.type == KEYDOWN and event.key == 112:  # P Letter
            self.play = not self.play

        elif event.type == KEYDOWN and event.key == K_RETURN:
            self.do_step = True

        elif event.type == KEYDOWN and event.key == 107:  # K Letter
            nbr = random.randrange(len(self.blob.scouters))
            self.blob.knowledge['max_scouters'] -= 1
            del self.blob.scouters[nbr]
            print("Scouter killed ! - Total : " + str(len(self.blob.scouters)))

        elif event.type == KEYDOWN and event.key == 113:  # A letter
            self.blob.knowledge['max_scouters'] += 1
            self.blob.add_scouter()
            print("New scouter ! - Total : " + str(len(self.blob.scouters)))

        elif event.type == KEYDOWN:
            print("Unrecognised key code : " + str(event.key))