def flatten(class_info, send_tables):
flattener = Flattener(send_tables)
recv_tables = collections.OrderedDict()
for st in filter(test_needs_decoder, send_tables.values()):
props = flattener.flatten(st)
cls = next(_id for _id, (dt, _) in class_info.items() if dt == st.dt)
recv_tables[cls] = dt_recv.construct(st.dt, props)
return recv_tables
def flatten(class_info, send_tables):
flattener = Flattener(send_tables)
recv_tables = collections.OrderedDict()
for st in filter(test_needs_decoder, send_tables.values()):
props = flattener.flatten(st)
cls = next(_id for _id, (dt, _) in class_info.items() if dt == st.dt)
recv_tables[cls] = dt_recv.construct(st.dt, props)
return recv_tables
def load(io, tick=0):
demo_io = d_io.construct(io)
prologue = id_prologue.construct(demo_io)
# mash all packet svc messages together, then index them
signon_packets = list(prologue.all_dem_signon_packet)
data = ''.join([pb.data for _, pb in signon_packets])
packet = ie_packet.construct(p_io.construct(data))
# meta: file header
_, pbmsg = prologue.dem_file_header
file_header = FileHeader(*[getattr(pbmsg, a) for a in FileHeader._fields])
# meta: server info
_, pbmsg = packet.svc_server_info
server_info = ServerInfo(*[getattr(pbmsg, a) for a in ServerInfo._fields])
# meta: voice init
_, pbmsg = packet.svc_voice_init
voice_init = VoiceInit(*[getattr(pbmsg, a) for a in VoiceInit._fields])
# prologue: meta
meta = Meta(file_header, server_info, voice_init)
# prologue: send tables
_, pbmsg = prologue.dem_send_tables
_send_tables = ie_send_tables.construct(p_io.construct(pbmsg.data))
send_tables = c.OrderedDict()
for pbmsg in [pb for _, pb in _send_tables.all_svc_send_table]:
if pbmsg.is_end:
break
send_table = _parse_cdemo_send_table(pbmsg)
send_tables[send_table.dt] = send_table
# prologue: recv tables
flattener = Flattener(send_tables)
recv_tables = c.OrderedDict()
_, pbmsg = prologue.dem_class_info
class_info = c.OrderedDict()
for cls in pbmsg.classes:
_id, dt, name = str(cls.class_id), cls.table_name, cls.network_name
class_info[_id] = (dt, name)
for st in filter(test_needs_decoder, send_tables.values()):
props = flattener.flatten(st)
cls = next(_id for _id, (dt, _) in class_info.items() if dt == st.dt)
recv_tables[cls] = dt_r.construct(st.dt, props)
# prologue: string tables
pbmsgs = [pb for _, pb in packet.all_svc_create_string_table]
string_tables = _parse_all_csvc_create_string_tables(pbmsgs)
# prologue: game event list
_, pbmsg = packet.svc_game_event_list
game_event_list = c.OrderedDict()
for desc in pbmsg.descriptors:
_id, name = desc.eventid, desc.name
keys = [(k.type, k.name) for k in desc.keys]
game_event_list[_id] = (name, keys)
# prologue: class bits
class_bits = server_info.max_classes.bit_length()
return Prologue(meta, recv_tables, string_tables, game_event_list, class_bits)
def load(io, tick=0):
demo_io = d_io.construct(io)
prologue = i_p.construct(demo_io)
# mash all packet svc messages together, then index them
signon_packets = list(prologue.signon_packets)
pbmsgs = [d_io.parse(p.kind, p.compressed, m) for p, m in signon_packets]
data = ''.join([p.data for p in pbmsgs])
packet_io = p_io.construct(data)
packet = i.construct(packet_io)
# class info
peek, message = prologue.class_info
pbmsg = d_io.parse(peek.kind, peek.compressed, message)
class_info = c.OrderedDict()
for _c in pbmsg.classes:
_id, dt, name = str(_c.class_id), _c.table_name, _c.network_name
class_info[_id] = (dt, name)
# send tables
peek, message = prologue.send_tables
pbmsg = d_io.parse(peek.kind, peek.compressed, message)
send_tables = c.OrderedDict()
for peek, message in p_io.construct(pbmsg.data):
pbmsg = p_io.parse(peek.kind, message)
if pbmsg.is_end:
break
send_table = _parse_cdemo_send_table(pbmsg)
send_tables[send_table.dt] = send_table
# recv tables
flattener = Flattener(send_tables)
recv_tables = c.OrderedDict()
for st in filter(test_needs_decoder, send_tables.values()):
props = flattener.flatten(st)
cls = next(_id for _id, (dt, _) in class_info.items() if dt == st.dt)
recv_tables[cls] = dt_r.construct(st.dt, props)
# game event list
peek, message = packet.find(pb_n.svc_GameEventList)
pbmsg = p_io.parse(peek.kind, message)
game_event_list = c.OrderedDict()
for desc in pbmsg.descriptors:
_id, name = desc.eventid, desc.name
keys = [(k.type, k.name) for k in desc.keys]
game_event_list[_id] = (name, keys)
# string tables
entries = packet.find_all(pb_n.svc_CreateStringTable)
pbmsgs = [p_io.parse(p.kind, m) for p, m in entries]
string_tables = _parse_all_csvc_create_string_tables(pbmsgs)
# meta: file header
peek, message = prologue.file_header
pbmsg = d_io.parse(peek.kind, peek.compressed, message)
file_header = FileHeader(*[getattr(pbmsg, a) for a in FileHeader._fields])
# meta: server_info
peek, message = packet.find(pb_n.svc_ServerInfo)
pbmsg = p_io.parse(peek.kind, message)
server_info = ServerInfo(*[getattr(pbmsg, a) for a in ServerInfo._fields])
# meta: class bits
class_bits = server_info.max_classes.bit_length()
# meta: voice init
peek, message = packet.find(pb_n.svc_VoiceInit)
pbmsg = p_io.parse(peek.kind, message)
voice_init = VoiceInit(*[getattr(pbmsg, a) for a in VoiceInit._fields])
meta = Meta(file_header, server_info, voice_init)
return Prologue(meta, recv_tables, string_tables, game_event_list,
class_bits)
def load(io, tick=0):
demo_io = d_io.construct(io)
prologue = id_prologue.construct(demo_io)
# mash all packet svc messages together, then index them
signon_packets = list(prologue.all_dem_signon_packet)
data = ''.join([pb.data for _, pb in signon_packets])
packet = ie_packet.construct(p_io.construct(data))
# meta: file header
_, pbmsg = prologue.dem_file_header
file_header = FileHeader(*[getattr(pbmsg, a) for a in FileHeader._fields])
# meta: server info
_, pbmsg = packet.svc_server_info
server_info = ServerInfo(*[getattr(pbmsg, a) for a in ServerInfo._fields])
# meta: voice init
_, pbmsg = packet.svc_voice_init
voice_init = VoiceInit(*[getattr(pbmsg, a) for a in VoiceInit._fields])
# prologue: meta
meta = Meta(file_header, server_info, voice_init)
# prologue: send tables
_, pbmsg = prologue.dem_send_tables
_send_tables = ie_send_tables.construct(p_io.construct(pbmsg.data))
send_tables = c.OrderedDict()
for pbmsg in [pb for _, pb in _send_tables.all_svc_send_table]:
if pbmsg.is_end:
break
send_table = _parse_cdemo_send_table(pbmsg)
send_tables[send_table.dt] = send_table
# prologue: recv tables
flattener = Flattener(send_tables)
recv_tables = c.OrderedDict()
_, pbmsg = prologue.dem_class_info
class_info = c.OrderedDict()
for cls in pbmsg.classes:
_id, dt, name = str(cls.class_id), cls.table_name, cls.network_name
class_info[_id] = (dt, name)
for st in filter(test_needs_decoder, send_tables.values()):
props = flattener.flatten(st)
cls = next(_id for _id, (dt, _) in class_info.items() if dt == st.dt)
recv_tables[cls] = dt_r.construct(st.dt, props)
# prologue: string tables
pbmsgs = [pb for _, pb in packet.all_svc_create_string_table]
string_tables = _parse_all_csvc_create_string_tables(pbmsgs)
# prologue: game event list
_, pbmsg = packet.svc_game_event_list
game_event_list = c.OrderedDict()
for desc in pbmsg.descriptors:
_id, name = desc.eventid, desc.name
keys = [(k.type, k.name) for k in desc.keys]
game_event_list[_id] = (name, keys)
# prologue: class bits
class_bits = server_info.max_classes.bit_length()
return Prologue(meta, recv_tables, string_tables, game_event_list, class_bits)
