def calc_area_distance(params_q, distances_q):
while True:
try:
params = params_q.get(timeout=5)
except queue.Empty:
print('Params queue empty')
break
distance = area_distance(params['map'], params['a1'], params['a2'])
distances_q.put((params['map'], params['a1'], params['a2'], distance))
return
def frame_to_graph(frame,
metric,
map_name,
full=False,
places=False,
coordinates=False):
"""Transforms a frame to a graph
Args:
frame (dictionary) : A frame from the dict in csgo.parser.DemoParser
metric (string) : A string indicating "graph" for graph distance or one of 'euclidean', 'manhattan', 'canberra', 'cosine'
map_name (string) : A string indicating the map
full (boolean) : True for full graphs (force 10 nodes) or False for graph of only alive players
places (boolean) : True to include a 110 long OHE vector of player location, False otherwise
coordinates (boolean) : True to include player X/Y/Z, False otherwise
Returns:
A (np.array) : Adjacency matrix using graph distances
X (np.array) : Matrix with information on each node
"""
if map_name not in [
"de_dust2",
"de_cbble",
"de_inferno",
"de_mirage",
"de_nuke",
"de_overpass",
"de_train",
"de_vertigo",
]:
raise ValueError(
f'Invalid map name: got {map}, expected one of: "de_dust2", "de_cbble", "de_inferno", "de_mirage", "de_nuke", "de_overpass", "de_train", "de_vertigo"'
)
if frame["T"]["Players"] is None or frame["CT"]["Players"] is None:
raise ValueError("No players!")
encoder = Encoder()
players = frame["T"]["Players"] + frame["CT"]["Players"]
# Create player nodes
nodes = []
if len(frame["T"]["Players"]) > 0:
side_ind = 0
for p in frame["T"]["Players"]:
item = []
if full:
item = [
side_ind,
p["IsAlive"],
p["Hp"],
p["Armor"],
p["EquipmentValue"],
p["HasHelmet"],
p["HasDefuse"],
p["TotalUtility"],
p["DistToBombsiteA"],
p["DistToBombsiteB"],
]
if coordinates:
item.extend([p["X"], p["Y"], p["Z"]])
if places:
item.extend(encoder.encode("places", p["AreaName"]))
nodes.append(item)
elif full is False and p["IsAlive"] is True:
item = [
side_ind,
p["Hp"],
p["Armor"],
p["EquipmentValue"],
p["HasHelmet"],
p["HasDefuse"],
p["TotalUtility"],
p["DistToBombsiteA"],
p["DistToBombsiteB"],
]
if coordinates:
item.extend([p["X"], p["Y"], p["Z"]])
if places:
item.extend(encoder.encode("places", p["AreaName"]))
nodes.append(item)
if len(frame["CT"]["Players"]) > 0:
side_ind = 1
for p in frame["CT"]["Players"]:
item = []
if full:
item = [
side_ind,
p["IsAlive"],
p["Hp"],
p["Armor"],
p["EquipmentValue"],
p["HasHelmet"],
p["HasDefuse"],
p["TotalUtility"],
p["DistToBombsiteA"],
p["DistToBombsiteB"],
]
if coordinates:
item.extend([p["X"], p["Y"], p["Z"]])
if places:
item.extend(encoder.encode("places", p["AreaName"]))
nodes.append(item)
elif full is False and p["IsAlive"] is True:
item = [
side_ind,
p["Hp"],
p["Armor"],
p["EquipmentValue"],
p["HasHelmet"],
p["HasDefuse"],
p["TotalUtility"],
p["DistToBombsiteA"],
p["DistToBombsiteB"],
]
if coordinates:
item.extend([p["X"], p["Y"], p["Z"]])
if places:
item.extend(encoder.encode("places", p["AreaName"]))
nodes.append(item)
# Create adjacency matrix
adjacency = []
for p1 in players:
player_distances = []
if full:
for p2 in players:
if metric == "graph":
player_distances.append(
area_distance(area_one=p1["AreaId"],
area_two=p2["AreaId"],
map=map_name))
else:
player_distances.append(
point_distance(
point_a=[p1["X"], p1["Y"], p1["Z"]],
point_b=[p2["X"], p2["Y"], p2["Z"]],
type=metric,
map=map_name,
))
adjacency.append(player_distances)
else:
if p1["IsAlive"]:
for p2 in players:
if p2["IsAlive"]:
if metric == "graph":
player_distances.append(
area_distance(
area_one=p1["AreaId"],
area_two=p2["AreaId"],
map=map_name,
))
else:
player_distances.append(
point_distance(
point_a=[p1["X"], p1["Y"], p1["Z"]],
point_b=[p2["X"], p2["Y"], p2["Z"]],
type=metric,
map=map_name,
))
adjacency.append(player_distances)
X = np.array(nodes)
A = np.array(adjacency)
return X, A
def test_area_dist(self):
"""
Tests that area distance returns correct value.
"""
assert area_distance(26, 42, map="de_mirage") == 26
def area_dist_all(x, game_map):
return area_distance(
area_one=x.values[0][0],
area_two=x.values[0][1],
map=game_map,
)
def test_area_invalid_map(self):
"""
Test area distance function with an invalid map.
"""
with pytest.raises(ValueError):
area_distance(26, 42, map="dust2")