def draw_symbol(grid, skel, color):
length = skel.sourceline.length
if length < 1:
return
# draw white "grain of rice" at start of symbol
pos = skel.position
grid.moveto(get_xy(pos))
grid.drawto(get_xy(pos + 1), "#fff")
for offset in xrange(length):
grid.drawto(get_xy(pos + offset + 1), color)
def draw_highlight(grid, diagram):
folder_pos = [pos for pos, symbol, _, _ in diagram if symbol.path.endswith("/setup.py")]
folder_range = xrange(min(folder_pos), max(folder_pos))
grid.draw_many((get_xy(pos) for pos in folder_range), ImageColor.getrgb("white"))
