extract_stream_names.py

import msgpack


def summarize_logfile(filename):
    n_samples = {}
    typenames = {}
    with open(filename, "r") as f:
        unpacker = msgpack.Unpacker(file_like=f)
        n_keys = unpacker.read_map_header()
        stream_names = []
        for i in range(n_keys):
            key = unpacker.unpack()
            if key.endswith(".meta"):
                stream_name = key[:-5]
                n_meta_keys = unpacker.read_map_header()
                for j in range(n_meta_keys):
                    meta_key = unpacker.unpack()
                    if meta_key == "type":
                        typenames[stream_name] = unpacker.unpack()
                    elif meta_key == "timestamps":
                        n_samples[stream_name] = unpacker.read_array_header()
                        for k in range(n_samples[stream_name]):
                            unpacker.skip()
                    else:
                        unpacker.skip()
            else:
                unpacker.skip()
    return n_samples, typenames