Example #1
0
    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)

        mods_size = reader.read_int()
        self.mods = reader.read_lua()
        scenario_size = reader.read_int()
        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_int()
            self.players[name] = str(player_id)

        self.cheats_enabled = reader.read_bool()
        self.numbers_of_armies = reader.read_byte()

        self.armies = {}
        for i in range(self.numbers_of_armies):
            player_data_size = reader.read_int()
            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_float()
def command_set_command_type(
        reader: ReplayReader) -> Dict[str, Union[str, int]]:
    return {
        "type": "set_command_type",
        "command_id": reader.read_uint(),
        "target_id": reader.read_int()
    }
def command_warp_entity(
        reader: ReplayReader) -> Dict[str, Union[str, TYPE_VECTOR]]:
    return {
        "type": "warp_entity",
        "entity_id": reader.read_int(),
        "vector": _read_vector(reader)
    }
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 command_command_count_decrease(
        reader: ReplayReader) -> Dict[str, Union[str, int]]:
    return {
        "type": "command_count_decrease",
        "command_id": reader.read_uint(),
        "delta": reader.read_int()
    }
def _parse_formation(reader: ReplayReader) -> TYPE_FORMATION:
    formation = reader.read_int()
    if formation != -1:
        w = reader.read_float()
        position = _read_vector(reader)
        scale = reader.read_float()
        return {"w": w, "position": position, "scale": scale}
    return None
Example #7
0
def command_process_info_pair(reader: ReplayReader) -> Dict[str, Union[int, str]]:
    entity_id = reader.read_int()
    arg1 = reader.read_string()
    arg2 = reader.read_string()
    return {"type": "process_info_pair",
            "entity_id": entity_id,
            "arg1": arg1,
            "arg2": arg2}
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}
Example #9
0
def command_set_command_cells(reader: ReplayReader) -> Dict[str, Union[str, int, TYPE_LUA]]:
    command_id = reader.read_int()
    cells = reader.read_lua()
    if cells:
        reader.read(1)
    vector = (reader.read_float(), reader.read_float(), reader.read_float())
    return {"type": "set_command_cells",
            "command_id": command_id,
            "cells": cells,
            "vector": vector}
Example #10
0
def command_lua_sim_callback(reader: ReplayReader) -> Dict[str, Union[str, TYPE_LUA, bytes]]:
    lua_name = reader.read_string()
    lua = reader.read_lua()
    size = None
    data = None
    if lua:
        size = reader.read_int()
        data = reader.read(4 * size)
    else:
        data = reader.read(4 + 3)

    return {"type": "lua_sim_callback",
            "lua_name": lua_name,
            "lua": lua,
            "size": size,
            "data": data}
def command_destroy_entity(reader: ReplayReader) -> Dict[str, int]:
    return {"type": "destroy_entity", "entity_id": reader.read_int()}
def command_remove_from_queue(reader: ReplayReader) -> Dict[str, int]:
    return {
        "type": "remove_from_queue",
        "command_id": reader.read_uint(),
        "unit_id": reader.read_int()
    }
Example #13
0
def _parse_entity_ids_set(reader: ReplayReader) -> TYPE_ENTITY_IDS_SET:
    units_number = reader.read_int()
    unit_ids = []
    for i in range(units_number):
        unit_ids.append(reader.read_int())
    return {"units_number": units_number, "unit_ids": unit_ids}
Example #14
0
def command_verify_checksum(reader: ReplayReader) -> Dict[str, Union[str, int]]:
    checksum = "".join("{:02X}".format(ord(reader.read(1))) for _ in range(16))
    return {"type": "verify_checksum",
            "checksum": checksum,
            "tick": reader.read_int()}
Example #15
0
def command_advance(reader: ReplayReader) -> Dict[str, int]:
    return {"type": "advance",
            "advance": reader.read_int()}
Example #16
0
def command_set_command_target(reader: ReplayReader) -> Dict[str, Union[int, TYPE_TARGET]]:
    return {"type": "set_command_target",
            "command_id": reader.read_int(),
            "target": _parse_target(reader)}