from plotutils import *

p = plot(None)[0]
p.setLegendVisible(True)
for block in get_blocks():
    print block.id
    print block
    (xmin, xmax) = block.x_range
    (ymin, ymax) = block.y_range
    r = plot_rectangle(p, xmin, ymin, xmax, ymax, name = block.id)

    #In the first time the plot shows, it takes some time for the color to be assigned
    while r.color is None:
        time.sleep(0.001)
            
    for puck in get_pucks(block.id):
        print puck.number
        (xu, yu) = puck.led_uni
        (xm, ym) = puck.led_mini
        plot_point(p, xu, yu, size = 3, color = r.color, name = str(puck.number)+"u")
        plot_point(p, xm, ym, size = 7, color = r.color, name = str(puck.number)+"m")
        p.addText((xu+xm)/2, (yu+ym)/2, str(puck.number))