# Part of the Fifteen-Thieves Project
# Chris Xiong 2020
# License: Expat (MIT)
import sys
import xv.util
import xv.amap

wavenames = ['None']

def dfs_checkpopulated(node):
    node.populated = False
    for ch in node.children_nodes:
        if node.template.is_aggregate_node:
            if type(ch) == list:
                for el in ch:
                    node.populated |= dfs_checkpopulated(el)
            else:
                node.populated |= dfs_checkpopulated(ch)
        else:
            for a in node.get_data():
                if a != 0xff:
                    node.populated = True
                    break
    return node.populated

def dfs_print(node, level=0, instance=None):
    if not node.populated:
        return
    ba_rep = ''.join([format(x, "02x") for x in xv.util.padbelist(xv.util.int2b7belist(node.base_address))])
    print(f"{ba_rep}{level*'  '} {node.template.name}{'['+str(instance)+']' if instance is not None else ''}")
    for child in node.children_nodes:
        if node.template.is_aggregate_node:
            if type(child) == list:
                for i, element in enumerate(child):
                    dfs_print(element, level + 1, i)
            else:
                dfs_print(child, level + 1)
        else:
            if type(child) == list:
                if child[0].name == "name":
                    print(f"{ba_rep}{(level + 1)*'  '} {child[0].name} {node.data[child[0].offset:child[0].offset + 12].decode()}")
                else:
                    for i, element in enumerate(child):
                        ba_rep = ''.join([format(x, "02x") for x in xv.util.padbelist(xv.util.int2b7belist(element.offset))])
                        print(f"{ba_rep}{(level + 1)*'  '} {element.name}[{i}] {xv.amap.get_leaf_data(node.data, element)}")
            else:
                ba_rep = ''.join([format(x, "02x") for x in xv.util.padbelist(xv.util.int2b7belist(child.offset))])
                if child.name.startswith("wave_number"):
                    try:
                        print(f"{ba_rep}{(level + 1)*'  '} {child.name} {xv.amap.get_leaf_data(node.data, child)}: {wavenames[xv.amap.get_leaf_data(node.data, child)]}")
                    except IndexError:
                        print(f"{ba_rep}{(level + 1)*'  '} {child.name} {xv.amap.get_leaf_data(node.data, child)}: unknown")
                else:
                    print(f"{ba_rep}{(level + 1)*'  '} {child.name} {xv.amap.get_leaf_data(node.data, child)}")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print(f"Usage: {sys.argv[0]} <memoryregion file>")
    print("loading address mapping...", flush=True, file=sys.stderr)
    m = xv.amap.AddrMapTemplate("amaps/sd-80.amap")
    t = xv.amap.create_addr_map_tree(m)
    t.data[:] = [0xff for i in range(0, len(t.data))]

    print("loading wave names...", file=sys.stderr)
    with open("data/sdwavelist","r") as f:
        for l in f:
            wavenames += [l.split('\t')[1].strip()]

    print("loading memoryregion...", file=sys.stderr)
    mr = xv.util.load_memoryregion(sys.argv[1])
    for addr, data in mr:
        ra = xv.util.b7belist2int(addr)
        t.data[ra:ra + len(data)] = data

    dfs_checkpopulated(t)
    dfs_print(t)