-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathenvs.py
82 lines (67 loc) · 3.26 KB
/
envs.py
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
71
72
73
74
75
76
77
78
79
80
81
82
from vizdoom import *
from arguments import get_args
import time
def add_game_vars(game):
game.add_available_game_variable(GameVariable.AMMO0)
game.add_available_game_variable(GameVariable.AMMO1)
game.add_available_game_variable(GameVariable.AMMO2)
game.add_available_game_variable(GameVariable.AMMO3)
game.add_available_game_variable(GameVariable.AMMO4)
game.add_available_game_variable(GameVariable.AMMO5)
game.add_available_game_variable(GameVariable.AMMO6)
game.add_available_game_variable(GameVariable.AMMO7)
game.add_available_game_variable(GameVariable.AMMO8)
game.add_available_game_variable(GameVariable.AMMO9)
game.add_available_game_variable(GameVariable.WEAPON0)
game.add_available_game_variable(GameVariable.WEAPON1)
game.add_available_game_variable(GameVariable.WEAPON2)
game.add_available_game_variable(GameVariable.WEAPON3)
game.add_available_game_variable(GameVariable.WEAPON4)
game.add_available_game_variable(GameVariable.WEAPON5)
game.add_available_game_variable(GameVariable.WEAPON6)
game.add_available_game_variable(GameVariable.WEAPON7)
game.add_available_game_variable(GameVariable.WEAPON8)
game.add_available_game_variable(GameVariable.WEAPON9)
game.add_available_game_variable(GameVariable.POSITION_X)
game.add_available_game_variable(GameVariable.POSITION_Y)
game.add_available_game_variable(GameVariable.ON_GROUND)
game.add_available_game_variable(GameVariable.KILLCOUNT)
game.add_available_game_variable(GameVariable.DEATHCOUNT)
game.add_available_game_variable(GameVariable.ARMOR)
game.add_available_game_variable(GameVariable.FRAGCOUNT)
game.add_available_game_variable(GameVariable.HEALTH)
def make_env(worker_id, config_file_path=None, visual=False, cig=False, bots=7):
print("Initializing doom environment", worker_id, "...")
game = DoomGame()
game.load_config(config_file_path)
add_game_vars(game)
if cig:
game.set_doom_map("map02")
game.add_game_args("-host 1 -deathmatch +timelimit 10.0 "
"+sv_forcerespawn 1 +sv_noautoaim 1 +sv_respawnprotect 1 +sv_spawnfarthest 1 +sv_nocrouch 1 "
"+viz_respawn_delay 10 +viz_nocheat 1")
game.add_game_args("+name AI +colorset 0")
game.set_mode(Mode.PLAYER)
game.set_window_visible(visual)
game.init()
if cig:
game.send_game_command("removebots")
for i in range(bots):
game.send_game_command("addbot")
#print(f"{len(server_state.players_in_game)} players in the game")
return game
def make_cig_env(worker_id, visual=False):
game = DoomGame()
# Use CIG example config or your own.
game.load_config("../../scenarios/cig.cfg")
game.set_doom_map("map02") # Full deathmatch.
# Join existing game.
game.add_game_args("-join 127.0.0.1") # Connect to a host for a multiplayer game.
# Name your agent and select color
# colors: 0 - green, 1 - gray, 2 - brown, 3 - red, 4 - light gray, 5 - light brown, 6 - light red, 7 - light blue
game.add_game_args(f"+name AI-{worker_id} +colorset {worker_id}")
# During the competition, async mode will be forced for all agents.
game.set_mode(Mode.ASYNC_PLAYER)
game.set_window_visible(visual)
game.init()
return game