def make_world(self):
map_size = FLAGS.map_size
world = World(width=map_size, height=map_size)
# list of all agents
agents = []
self.atype_to_idx = {
"predator": [],
"prey": []
}
# make agents
# add predators
n_predator = FLAGS.n_predator
for i in range(n_predator):
agents.append(Predator())
self.atype_to_idx['predator'].append(i)
# add preys
n_prey = FLAGS.n_prey
for i in range(n_prey):
agents.append(Prey())
self.atype_to_idx['prey'].append(n_predator + i)
# used by BaseScenario
# assign id to agents
world.agents = agents
for i, agent in enumerate(world.agents):
agent.id = i + 1
agent.silent = True
# make initial conditions
self.reset_world(world)
return world
def make_world(self):
world = World(width=map_size, height=map_size)
agents = []
self.atype_to_idx = {
"predator": [],
"prey": [],
"prey2": []
}
# add predators
for i in xrange(n_predator):
agents.append(Predator())
self.atype_to_idx["predator"].append(i)
# add preys
for i in xrange(n_prey1):
agents.append(Prey())
self.atype_to_idx["prey"].append(n_predator + i)
for i in xrange(n_prey2):
agents.append(Prey2())
self.atype_to_idx["prey2"].append(n_predator + n_prey1 + i)
world.agents = agents
for i, agent in enumerate(world.agents):
agent.id = i + 1
agent.silent = True
# make initial conditions
self.reset_world(world)
return world
def make_world(self):
world = World(width=map_size, height=map_size)
agents = []
self.atype_to_idx = {"predator": [], "prey": []}
# add predators
for i in xrange(n_predator):
agents.append(
Predator(power_threshold=power_threshold, power_dec=power_dec))
self.atype_to_idx["predator"].append(i)
# add preys
for i in xrange(n_prey):
agents.append(Prey())
self.atype_to_idx["prey"].append(n_predator + i)
world.agents = agents
for i, agent in enumerate(world.agents):
agent.id = i + 1
# make initial conditions
self.reset_world(world)
return world