/
snapshot_parser.py
182 lines (152 loc) · 7.99 KB
/
snapshot_parser.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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
"""Imports"""
import exceptions
def exception_handler(snapshot):
# Checks to see if snapshot is valid. See exceptions.py for details
if snapshot.get("allplayers") is None:
raise exceptions.MatchNotStarted
if snapshot.get("map").get("phase") == 'warmup':
raise exceptions.WarmUp
if len(snapshot["allplayers"].keys()) == 0:
raise exceptions.EmptyServer
return snapshot
def snapshot_formatter(ssoriginal):
# Converts Player names to Generic names for easy dict access
ss1 = ssoriginal
playernames = list(ss1["allplayers"].keys())
ss1_ap = ss1["allplayers"]
for i in range(len(playernames)):
ss1_ap[str("player" + str(i+1))] = ss1_ap.pop(str(playernames[i]))
return ss1
def snapshot_arrayfier(snapshot_formatted):
"""Creating list of attributes for prediction
# # Order of attributes -
# # 'map', 'bomb_planted','ct_score', 't_score', 'time_left',
# # 'ct_players_alive','t_players_alive', 'ct_health', 't_health',
# # 'ct_armor', 't_armor','ct_pistols_special', 't_pistols_special',
# # 't_pistols_standard', 'ct_pistols_standard', 'ct_primaries_force',
# # 't_primaries_force', 'ct_primaries_fullbuy', 't_primaries_fullbuy',
# # 'ct_grenades', 't_grenades', 'ct_helmets', 't_helmets', 'ct_defuse_kits'
# # Encoded Lables:Original Values
# # round_winner = {'CT': 0, 'T': 1}
# # map = {'de_cache': 0, 'de_dust2': 1, 'de_inferno': 2, 'de_mirage': 3,
# 'de_nuke': 4, 'de_overpass': 5, 'de_train': 6, 'de_vertigo': 7}
# # bomb_planted = {False: 0, True: 1}
"""
snap = snapshot_formatted
predictors = [
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
]
# adding var0 - map
map_codes = {
'de_cache': 0, 'de_dust2': 1, 'de_inferno': 2, 'de_mirage': 3,
'de_nuke': 4, 'de_overpass': 5, 'de_train': 6, 'de_vertigo': 7, 'de_ancient': 4
}
map_string = str(snap["map"]["name"])
if map_codes.get(map_string) is None:
predictors[0] = 1 # Default Map set to de_dust2 (most balanced map)
else:
predictors[0] = map_codes.get(map_string)
# adding var1 - bomb_planted
if "bomb" in snap["round"].keys():
predictors[1] = 1
# adding var2,3 - scores
predictors[2] = snap["map"]["team_ct"]["score"]
predictors[3] = snap["map"]["team_t"]["score"]
# adding var4 - time_left
predictors[4] = int(float(snap["phase_countdowns"]["phase_ends_in"]))
# adding var 5,6,7,8 - players alive and team health
counter_t = 0
counter_ct = 0
health_t = 0
health_ct = 0
for i in range(len(snap["allplayers"].keys())):
if snap["allplayers"][str("player" + str(i + 1))]["state"]["health"] > 0:
if snap["allplayers"][str("player" + str(i + 1))]["team"] == "T":
counter_t += 1
health_t += snap["allplayers"][str("player" + str(i + 1))]["state"]["health"]
if snap["allplayers"][str("player" + str(i + 1))]["team"] == "CT":
counter_ct += 1
health_ct += snap["allplayers"][str("player" + str(i + 1))]["state"]["health"]
predictors[5] = counter_ct
predictors[6] = counter_t
predictors[7] = health_ct
predictors[8] = health_t
# adding var 9,10 - armor
armor_t = 0
armor_ct = 0
for i in range(len(snap["allplayers"].keys())):
if snap["allplayers"][str("player" + str(i + 1))]["state"]["armor"] > 0:
if snap["allplayers"][str("player" + str(i + 1))]["team"] == "T":
armor_t += snap["allplayers"][str("player" + str(i + 1))]["state"]["armor"]
if snap["allplayers"][str("player" + str(i + 1))]["team"] == "CT":
armor_ct += snap["allplayers"][str("player" + str(i + 1))]["state"]["armor"]
predictors[9] = armor_ct
predictors[10] = armor_t
# adding vars 11 to 20 - weapons and grenades
ct_pistols_special = 0; t_pistols_special = 0
pistols_special = ["weapon_cz75auto", "weapon_elite", 'weapon_r8revolver', 'weapon_deagle', 'weapon_fiveseven',
'weapon_p250', 'weapon_tec9']
t_pistols_standard = 0; ct_pistols_standard = 0
pistols_standard = ['weapon_usps', 'weapon_glock', 'weapon_hkp2000']
ct_primaries_force = 0; t_primaries_force = 0
primaries_force = ['weapon_bizon', 'weapon_famas', 'weapon_galilar', 'weapon_mac10', 'weapon_mag7', 'weapon_mp5sd',
'weapon_mp7', 'weapon_mp9', 'weapon_negev', 'weapon_nova', 'weapon_p90', 'weapon_sawedoff',
'weapon_ssg08', 'weapon_ump45', 'weapon_xm1014', ]
ct_primaries_fullbuy = 0; t_primaries_fullbuy = 0
primaries_fullbuy = ['weapon_ak47', 'weapon_aug', 'weapon_awp', 'weapon_g3sg1', 'weapon_m249', 'weapon_m4a1s', "weapon_m4a1",
"weapon_m4a1_silencer", 'weapon_m4a4', 'weapon_scar20', 'weapon_sg556']
ct_grenades = 0; t_grenades = 0
grenades = ["weapon_hegrenade", "weapon_frag_grenade", "weapon_flashbang", "weapon_smokegrenade", "weapon_decoy",
"weapon_molotov", "weapon_incgrenade", ]
ignore = ["weapon_knife", "weapon_knife_t", "weapon_c4"]
for i in range(len(snap["allplayers"].keys())):
player = str("player" + str(i + 1))
if snap["allplayers"][player]["team"] == "T":
for iterator in range(len(list(snap["allplayers"][player]["weapons"].keys()))):
weapon = str("weapon_" + str(iterator))
if snap["allplayers"][player]["weapons"][weapon]["name"] in ignore:
continue
elif snap["allplayers"][player]["weapons"][weapon]["name"] in pistols_special:
t_pistols_special += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in pistols_standard:
t_pistols_standard += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in primaries_force:
t_primaries_force += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in primaries_fullbuy:
t_primaries_fullbuy += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in grenades:
t_grenades += 1
elif snap["allplayers"][player]["team"] == "CT":
for iterator in range(len(list(snap["allplayers"][player]["weapons"].keys()))):
weapon = str("weapon_" + str(iterator))
if snap["allplayers"][player]["weapons"][weapon]["name"] in ignore:
continue
elif snap["allplayers"][player]["weapons"][weapon]["name"] in pistols_special:
ct_pistols_special += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in pistols_standard:
ct_pistols_standard += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in primaries_force:
ct_primaries_force += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in primaries_fullbuy:
ct_primaries_fullbuy += 1
elif snap["allplayers"][player]["weapons"][weapon]["name"] in grenades:
ct_grenades += 1
weapons_count = [ct_pistols_special, t_pistols_special, t_pistols_standard, ct_pistols_standard,
ct_primaries_force, t_primaries_force, ct_primaries_fullbuy, t_primaries_fullbuy, ct_grenades, t_grenades]
predictors[11:21] = weapons_count[0:10]
# adding vars 21,22 - helmets
helmets_t = 0
helmets_ct = 0
for i in range(len(snap["allplayers"].keys())):
player = str("player" + str(i + 1))
if snap["allplayers"][player]["state"]["helmet"] is True:
if snap["allplayers"][player]["team"] == "T":
helmets_t += 1
if snap["allplayers"][player]["team"] == "CT":
helmets_ct += 1
predictors[21] = helmets_ct
predictors[22] = helmets_t
# adding var23 - defuse kits
# # data not transmitted, check it out later
return predictors