def draw(self, draw_obj):
if isinstance(draw_obj, (list, tuple)):
draw = pgmagick.DrawableList()
for d in draw_obj:
draw.append(d)
elif isinstance(draw_obj, Draw):
draw = draw_obj.drawer
else:
draw = draw_obj
self.img.draw(draw)
def __init__(self):
self.drawer = pgmagick.DrawableList()
def render(self, out_path, size=10000, scale=5, bg_color='#0b131a'):
"""
Render a PNG.
"""
image = pgm.Image(
pgm.Geometry(size, size),
pgm.Color(bg_color),
)
# TODO: font
nodes = self.graph.nodes_iter(data=True)
count = len(self.graph)
for tid, n in progress.bar(nodes, expected_size=count):
# Get X/Y, radius.
x = (n['x'] * scale) + (size / 2)
y = -(n['y'] * scale) + (size / 2)
r = (n['size'] * scale) / 2
# Index the coordinates.
self.graph.node[tid]['pixel_x'] = x
self.graph.node[tid]['pixel_y'] = y
self.graph.node[tid]['pixel_r'] = r
# ** Node **
# Hex-ify color.
color = '#%02x%02x%02x' % (n['r'], n['g'], n['b'])
# Draw the node.
dl = pgm.DrawableList()
dl.append(pgm.DrawableFillColor(color))
dl.append(pgm.DrawableStrokeColor('black'))
dl.append(pgm.DrawableStrokeWidth(r / 15))
dl.append(pgm.DrawableStrokeOpacity(0.9))
dl.append(pgm.DrawableCircle(x, y, x + r, y + r))
image.draw(dl)
# ** Label **
label = ', '.join([
n['label'],
n['author'],
])
# Measure the width of the label.
image.fontPointsize(n['size'])
tm = pgm.TypeMetric()
image.fontTypeMetrics(label, tm)
tw = tm.textWidth()
# Draw the label.
dl = pgm.DrawableList()
dl.append(pgm.DrawablePointSize(n['size']))
dl.append(pgm.DrawableFillColor('white'))
dl.append(pgm.DrawableText(x - (tw / 2), y, label))
image.draw(dl)
image.write(os.path.abspath(out_path))