from anytree import LevelOrderGroupIter
from anytree.importer import JsonImporter
from sys import argv

if len(argv) < 2:
    print(f"usage: {argv[0]} <json file>")
    exit()

with open(argv[1], 'r') as f:
    root = JsonImporter().read(f)

rows = []
leaves = []

for i in range(len(root.leaves)):
    rows.append([])
    leaves.append({"name":"wrong"})

cur_row = 0
for level in LevelOrderGroupIter(root):
    cur_row = 0

    for node in sorted(level, key=lambda x: x.size, reverse=True):
        leaf_count = len(node.leaves)
        if False: #node in root.leaves:
            leaves[cur_row] = node
        else:
            rows[cur_row].append(f'<td rowspan="{leaf_count}">{node.name}</td>')

        cur_row += max(leaf_count, 1)
        last_depth = node.depth

print(leaves)

for i, leaf in enumerate(leaves):
    #rows[i].append(f"<td>{leaf.name}</td>")
    rows[i].reverse()

    print("<tr>")
    for cell in rows[i]:
        print(f"  {cell}")
    print("</tr>")



output_rows = []
rows = []

for i in range(len(root.leaves)):
    output_rows.append([])

def depth_first_search(node, rows, leaf_count):

    for child in sorted(node.children, key=lambda x: x.size, reverse=True):
        leaf_count = depth_first_search(child, rows, leaf_count)

    if node.is_leaf:
        cur_row = leaf_count
        rows[cur_row].append(f'<td>{node.name}</td>')
        leaf_count += 1
    else:
        cur_row = leaf_count - len(node.leaves)
        rows[cur_row].append(f'<td rowspan="{len(node.leaves)}">{node.name}</td>')

    return leaf_count

depth_first_search(root, output_rows, 0)

for row in output_rows:
    print("<tr>")
    for cell in row:
        print(f"  {cell}")
    print("</tr>")