figure5.py

from gen import Tree
from demo_trees import trees
from buchheim import buchheim as layout
from PIL import Image, ImageDraw

t = layout(trees[3][1])
t2 = layout(trees[3][0])

DIAMETER = 30
SPACING_VERTICAL = DIAMETER * 1.5
SPACING_HORIZONTAL = DIAMETER * 1.5


def drawt(draw, root, depth, offset=0):
    draw.ellipse(
        [
            root.x * SPACING_HORIZONTAL + offset,
            depth * SPACING_VERTICAL,
            root.x * SPACING_HORIZONTAL + DIAMETER + offset,
            depth * SPACING_VERTICAL + DIAMETER,
        ],
        fill=(225),
        outline=(0),
    )
    for child in root.children:
        drawt(draw, child, depth + 1, offset)


def drawconn(draw, root, depth, offset=0):
    for child in root.children:
        draw.line(
            [
                root.x * SPACING_HORIZONTAL + (DIAMETER / 2) + offset,
                depth * SPACING_VERTICAL + (DIAMETER / 2),
                child.x * SPACING_HORIZONTAL + (DIAMETER / 2) + offset,
                (depth + 1) * SPACING_VERTICAL + (DIAMETER / 2),
            ],
            fill=(0),
        )
        drawconn(draw, child, depth + 1, offset)


im = Image.new("L", (1000, 500), (255))
draw = ImageDraw.Draw(im)
drawconn(draw, t2, 0)
drawconn(draw, t, 0, 80)
drawt(draw, t2, 0)
drawt(draw, t, 0, 80)

im.save("figure5.png")