BioGym/config_parser.py at main · AnmolS99/BioGym · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import configparser
import ast

from sns_renderer import SNS_Renderer
from bio_environment import BioEnvironment
from bio_world import BioGymWorld


class ConfigParser:

    def __init__(self, config_filename) -> None:
        self.config = configparser.ConfigParser(inline_comment_prefixes=";")
        self.config.read(config_filename)

    def create_renderer(self, num_species, grid_size, action_unit_size):
        render_mode = self.config["Renderer"]["render_mode"]
        sim_height = int(self.config["Renderer"]["sim_height"])
        pix_padding = int(self.config["Renderer"]["render_pix_padding"])
        display_population = self.config["Renderer"].getboolean(
            "display_population")

        return SNS_Renderer(render_mode, sim_height, pix_padding, num_species,
                            grid_size, action_unit_size, display_population)

    def create_bio_environment(self, num_species, grid_size, action_unit_size,
                               action_multiplier,
                               reduced_actions) -> BioEnvironment:
        diagonal_neighbours = self.config["BioEnvironment"].getboolean(
            "diagonal_neighbours")
        migration_rate = ast.literal_eval(
            self.config["BioEnvironment"]["migration_rate"])
        species_ranges = ast.literal_eval(
            self.config["BioEnvironment"]["species_ranges"])

        r = self.config["BioEnvironment"].getfloat("r")
        k = self.config["BioEnvironment"].getfloat("k")
        a = self.config["BioEnvironment"].getfloat("a")
        b = self.config["BioEnvironment"].getfloat("b")
        e = self.config["BioEnvironment"].getfloat("e")
        d = self.config["BioEnvironment"].getfloat("d")
        a_2 = self.config["BioEnvironment"].getfloat("a_2")
        b_2 = self.config["BioEnvironment"].getfloat("b_2")
        e_2 = self.config["BioEnvironment"].getfloat("e_2")
        d_2 = self.config["BioEnvironment"].getfloat("d_2")
        s = self.config["BioEnvironment"].getfloat("s")
        gamma = self.config["BioEnvironment"].getfloat("gamma")

        return BioEnvironment(num_species, grid_size, action_unit_size,
                              action_multiplier, diagonal_neighbours,
                              reduced_actions, migration_rate, species_ranges,
                              r, k, a, b, e, d, a_2, b_2, e_2, d_2, s, gamma)

    def create_bio_gym_world(self):
        num_species = int(self.config["BioGymWorld"]["num_species"])
        grid_size = int(self.config["BioGymWorld"]["grid_size"])
        action_unit_size = int(self.config["BioGymWorld"]["action_unit_size"])
        action_multiplier = int(
            self.config["BioGymWorld"]["action_multiplier"])
        max_steps = int(self.config["BioGymWorld"]["max_steps"])
        reduced_actions = self.config["BioGymWorld"].getboolean(
            "reduced_actions")

        bio_env = self.create_bio_environment(num_species, grid_size,
                                              action_unit_size,
                                              action_multiplier,
                                              reduced_actions)
        renderer = self.create_renderer(num_species, grid_size,
                                        action_unit_size)

        return bio_env, renderer, max_steps, reduced_actions