diff --git a/output_html.py b/output_html.py index 90d62bb..0e980f9 100644 --- a/output_html.py +++ b/output_html.py @@ -8,14 +8,26 @@ def build_rows_depth_first(node, rows, leaf_count=0): for child in sorted(node.children, key=lambda x: x.height, reverse=True): leaf_count = build_rows_depth_first(child, rows, leaf_count) + colspan = 0 + if node.siblings: + max_sibling_height = max([s.height for s in node.siblings]) + + if max_sibling_height > node.height: + colspan = max_sibling_height - node.height + 1 + if node.is_leaf: cur_row = leaf_count - rows[cur_row].append(f'