Projekt_Sztuczna_Inteligencja/json_generator.py

import json
import random

import project_constants as const

# import tile class
from tile import Tile

# import mine models
from mines.mine_models.standard_mine import StandardMine
from mines.mine_models.time_mine import TimeMine
from mines.mine_models.chained_mine import ChainedMine


class JsonGenerator:
    grid = dict()

    # constructor that can be used to set agent's initial state
    def __init__(self, agents_initial_position=(0, 0), agents_initial_direction=const.Direction.UP.value):
        # saving agent's initial state (position & direction)
        self.agents_initial_position = format_position_to_tuple(agents_initial_position)
        self.agents_initial_direction = agents_initial_direction

        # saving data to the grid dictionary
        self.grid["agents_initial_state"] = {
            "position": format_position_to_str(agents_initial_position),
            "direction": agents_initial_direction
        }

    # sets agent's initial state
    def set_agents_initial_state(self, position=(0, 0), direction=const.Direction.UP.value):
        # setting fields in the instance
        self.agents_initial_position = format_position_to_tuple(position)
        self.agents_initial_direction = direction

        # setting new agent's initial state
        self.grid["agents_initial_state"] = {
            "position": format_position_to_str(position),
            "direction": direction
        }

    # overwrites grid field with a new grid with randomized colors and mines
    def generate_randomized_grid(self, dimensions, mine_appearance_chance=0.15, predecessor_chance_decrease=0.25):
        # clearing grid field
        self.clear_grid()

        # formatting dimensions to tuple
        dimensions = format_position_to_tuple(dimensions)

        # getting grid dimensions
        num_of_rows, num_of_columns = dimensions

        tile_pool = []

        for i in range(num_of_rows):
            for j in range(num_of_columns):

                # picking random values for tiles
                random_tile_color = random.choice(const.STRUCT_TILE_TERRAINS)

                # adding added tile's indexes to a pool
                tile_pool.append((i, j))

                # creating random tile
                self.add_tile((i, j), random_tile_color)

        # deleting agent's starting tile from the pool
        deleted_row, deleted_column = self.agents_initial_position
        tile_pool.remove((int(deleted_row), int(deleted_column)))

        for i in range(num_of_rows):
            for j in range(num_of_columns):

                # checking if a mine will appear
                if random.random() < mine_appearance_chance and len(tile_pool) > 0 and tile_pool.__contains__((i, j)):
                    # removing current tile from the pool
                    tile_pool.remove((i, j))

                    # choosing random mine parameters
                    random_mine_type = random.choice(const.STRUCT_MINE_TYPES)
                    random_attribute_values = []

                    for attr_type in const.STRUCT_MINE_ATTRIBUTE_TYPES[random_mine_type]:
                        random_attribute_values.append(_get_random_attribute_values(attr_type, dimensions))

                    # adding the mine
                    self.set_a_mine((i, j), random_mine_type, random_attribute_values)

                    # if is ChainedMine create predecessors
                    if random_mine_type == "chained":
                        predecessor_appearance_chance = 1.0
                        current_tile = format_position_to_str((i, j))

                        # create chained predecessors
                        while random.random() < predecessor_appearance_chance and len(tile_pool) > 0:
                            predecessor_appearance_chance -= predecessor_chance_decrease

                            predecessor_position = random.choice(tile_pool)
                            predecessor = format_position_to_str(predecessor_position)
                            tile_pool.remove(predecessor_position)

                            self.set_a_mine(predecessor_position, "chained", [])

                            self.grid[current_tile]["mine"]["predecessor"] = predecessor
                            self.grid[predecessor]["mine"]["predecessor"] = None

                            current_tile = predecessor

    # adds a new tile or edits an existing one in the grid field
    def add_tile(self, position, color):
        # creating new tile without a mine
        self.grid[format_position_to_str(position)] = {
            "color": color,
            "mine": None
        }

    # adds a new tile with a mine or edits an existing one in the grid field
    def add_tile_with_a_mine(self, position, color, mine_type, attribute_values):
        # setting mine data using attribute_values
        mine_values = const.STRUCT_MINE_ATTRIBUTES[mine_type]

        for key in mine_values.keys():
            if key not in const.HARDCODED_VALUES and len(attribute_values) > 0:
                mine_values[key] = attribute_values.pop(0)

        # creating a new tile
        self.grid[format_position_to_str(position)] = {
            "color": color
        }

        # updating the tile with a mine field
        self.grid[format_position_to_str(position)]["mine"] = {}

        for key in mine_values.keys():
            self.grid[format_position_to_str(position)]["mine"][key] = mine_values[key]

    # deletes a mine with a given position from the grid field
    def delete_a_tile(self, position):
        # deleting a tile with given key
        self.grid.pop(format_position_to_str(position))

    # adds a mine to a tile stored in the grid field
    def set_a_mine(self, position, mine_type, attribute_values):
        # setting mine data using attribute_values
        mine_values = const.STRUCT_MINE_ATTRIBUTES[mine_type]

        for key in mine_values.keys():
            if key not in const.HARDCODED_VALUES and len(attribute_values) > 0:
                mine_values[key] = attribute_values.pop(0)

        # adding a mine to the edited tile
        self.grid[format_position_to_str(position)]["mine"] = {}

        for key in mine_values.keys():
            self.grid[format_position_to_str(position)]["mine"][key] = mine_values[key]

    # deletes a mine from a tile stored in the grid field
    def delete_a_mine(self, position):
        # removing mine from the edited tile
        self.grid[format_position_to_str(position)]["mine"] = None

    # returns chosen tiles data
    def get_tile(self, position):
        return self.grid[format_position_to_str(position)]

    # returns chosen mines data
    def get_mine(self, position):
        return self.grid[format_position_to_str(position)]["mine"]

    # returns the grid field
    def get_grid(self):
        return self.grid

    # clears the grid field
    def clear_grid(self):
        # clearing grid dict
        self.grid.clear()

        # re-setting the agent's initial state
        self.grid["agents_initial_state"] = {
            "position": format_position_to_str(self.agents_initial_position),
            "direction": self.agents_initial_direction
        }

    # loads a grid from a file and overwrites the grid field
    def load_from_a_file(self, file_path):
        # opening a file for reading
        with open(file_path, 'r') as input_file:
            # overwriting the grid field with the grid stored in a file
            self.grid = json.load(input_file)

    # saves the current grid field to a file
    def save_to_a_file(self, file_path, access_mode):
        # opening a file with a given access mode (w - write / a - append)
        with open(file_path, access_mode) as output_file:
            # saving the grid to the file
            json.dump(self.grid, output_file, indent=2, sort_keys=True)

    # edits a grid in a file. doesn't delete data, only overwrites and adds new entries
    def edit_a_file(self, file_path):
        # opening a file for reading
        with open(file_path, "r") as input_file:
            # loading data that was stored in the file previously
            previous_data = json.load(input_file)

            # creating and updating a new grid using it's own grid field
            new_grid = previous_data
            new_grid.update(self.grid)

            # opening the file for writing
            with open(file_path, "w") as output_file:
                # saving the newly created grid
                json.dump(new_grid, output_file, indent=2, sort_keys=True)


# STATIC "PUBLIC" FUNCTIONS

# creates a Mine (Standard or Time or Chained) instance from mine dict data
# doesn't link chained mines, since it has no actual access to a grid
def create_a_mine(mine_dict, position):
    # initializing a mine with no value
    mine = None

    # if mine doesn't exist - returning None
    if mine_dict is None:
        return mine

    # formatting position to right format - in case it's not
    position = format_position_to_tuple(position)

    # if mine's type is "standard" - creating standard mine
    if mine_dict["mine_type"] == "standard":
        mine = StandardMine(position)

    # if mine's type is "time" - creating time mine
    elif mine_dict["mine_type"] == "time":
        timer_value = mine_dict["timer"]
        mine = TimeMine(position, timer_value)

    # if mine's type is "chained" - creating chained mine (no successors assigned yet)
    elif mine_dict["mine_type"] == "chained":
        mine = ChainedMine(position)

    return mine


# creates a Tile instance with a mine (if a mine exists) from tile dict data
def create_a_tile(tile_dict, position):
    # formatting position to right format - in case it's not
    position = format_position_to_tuple(position)

    # getting tile's parameters
    color = tile_dict["color"]
    mine = create_a_mine(tile_dict["mine"], position)

    # creating and returning a tile with the parameters set above
    return Tile(position, color, mine)


# returns a list of tuples containing chained mine's position and it's predecessors position
# data is in format [(chained_mine_position, its_predecessors_position), ...]
def get_chained_mine_and_its_predecessor_pairs(minefield_dictionary):
    predecessors = list()

    # iterate for each key in the grid field
    for key in minefield_dictionary.keys():
        # if a chained mine with predecessor exists - adding it's and it's predecessors positions as a tuple to a list
        if key != "agents_initial_state" and minefield_dictionary[key]["mine"] is not None\
                and minefield_dictionary[key]["mine"]["mine_type"] == "chained"\
                and minefield_dictionary[key]["mine"]["predecessor"] is not None:

            # getting the chained mines and it's predecessors positions
            this_mines_position = tuple(int(i) for i in key.split(','))
            its_predecessors_position = \
                tuple(int(i) for i in minefield_dictionary[key]["mine"]["predecessor"].split(','))

            # adding the positions to the list as a tuple
            predecessors.append((this_mines_position, its_predecessors_position))

    return predecessors


# changes position from str or tuple format to str format
def format_position_to_str(position):
    # if mine's position is in "row,column" format - that means position parameter is already in good format
    if isinstance(position, str):
        pass

    # if mine's position is in (row:int, column:int) - creating string from tuple
    elif isinstance(position, tuple) and list(map(type, position)) == [int, int]:
        row, column = position
        return str(row) + ',' + str(column)

    # if mine's position is not in any of 2 formats above - returning None - unsupported format
    else:
        return None

    return position


# changes position from str or tuple format to tuple format
def format_position_to_tuple(position):
    # if mine's position is in "row,column" format - getting parameters from string
    if isinstance(position, str):
        position = tuple(int(i) for i in position.split(','))

    # if mine's position is in (row:int, column:int) - that means position parameter is already in good format
    elif isinstance(position, tuple) and list(map(type, position)) == [int, int]:
        pass

    # if mine's position is not in any of 2 formats above - returning None - unsupported format
    else:
        return None

    return position


# AUXILIARY "PRIVATE" FUNCTIONS

# auxiliary function that returns random attribute values based on their type
def _get_random_attribute_values(attr_type, grid_dimensions):
    num_of_rows, num_of_columns = grid_dimensions

    if attr_type == int:
        # temporary solution
        return random.randint(num_of_rows + num_of_columns, (2 * num_of_rows + num_of_columns))

    else:
        return None