def __init__(self, reader: ReplayReader) -> None:
self.version = reader.read_string()
reader.read(3)
self.replay_version, self.map_name = reader.read_string().split(
"\r\n", 1)
reader.read(4)
reader.read_uint() # mods_size
self.mods = reader.read_lua()
reader.read_uint() # scenario_size
self.scenario = reader.read_lua()
sources_number = reader.read_byte()
self.players = {}
for i in range(sources_number):
name = reader.read_string()
player_id = reader.read_uint()
self.players[name] = str(player_id)
self.cheats_enabled = reader.read_bool()
self.numbers_of_armies = reader.read_byte()
self.armies = {}
for _ in range(self.numbers_of_armies):
reader.read_uint() # player_data_size
player_data = reader.read_lua()
player_source = reader.read_byte()
self.armies[player_source] = player_data
if player_source != 255:
reader.read(1)
self.random_seed = reader.read_uint()
def _parse_command_data(reader: ReplayReader) -> TYPE_COMMAND_DATA:
command_id = reader.read_int()
arg1 = reader.read(4)
command_type = reader.read_byte()
arg2 = reader.read(4)
target = _parse_target(reader)
arg3 = reader.read(1)
formation = _parse_formation(reader)
blueprint_id = reader.read_string()
arg4 = reader.read(12)
arg5 = None
cells = reader.read_lua()
if cells:
arg5 = reader.read(1)
return {
"command_id": command_id,
"command_type": command_type,
"target": target,
"formation": formation,
"blueprint_id": blueprint_id,
"cells": cells,
"arg1": arg1,
"arg2": arg2,
"arg3": arg3,
"arg4": arg4,
"arg5": arg5
}
def _parse_target(reader: ReplayReader) -> TYPE_TARGET:
target = reader.read_byte()
entity_id = None
position = None
if target == TargetType.Entity:
entity_id = reader.read_int()
elif target == TargetType.Position:
position = _read_vector(reader)
return {"target": target, "entity_id": entity_id, "position": position}
def command_create_unit(
reader: ReplayReader) -> Dict[str, Union[int, str, TYPE_VECTOR]]:
army_index = reader.read_byte()
blueprint_id = reader.read_string()
x, y, heading = _read_vector(reader)
return {
"type": "create_unit",
"army_index": army_index,
"blueprint_id": blueprint_id,
"vector": (x, y, heading)
}
def command_debug_command(
reader: ReplayReader
) -> Dict[str, Union[str, TYPE_VECTOR, int, TYPE_ENTITY_IDS_SET]]:
debug_command = reader.read_string()
vector = _read_vector(reader)
focus_army_index = reader.read_byte()
unit_ids = _parse_entity_ids_set(reader)
return {
"type": "debug_command",
"debug_command": debug_command,
"vector": vector,
"focus_army_index": focus_army_index,
"entity_ids_set": unit_ids
}
def command_set_command_source(reader: ReplayReader) -> Dict[str, int]:
return {"type": "set_command_source", "player_id": reader.read_byte()}
