29 lines
871 B
Python
29 lines
871 B
Python
|
|
mask_img = new_image(img.getOutput().getWidth(), img.getOutput().getHeight(), image_type="byte", title = "mask_img", fill_color = Color.BLACK)
|
|
|
|
mask_radius = 14
|
|
mask_points = []
|
|
|
|
|
|
def to_img_coords(absolute_coords):
|
|
return [img.getCalibration().convertToImageX(absolute_coords[0]), img.getCalibration().convertToImageY(absolute_coords[1])]
|
|
|
|
for p in _puck_list:
|
|
mask_points.append(to_img_coords(p.led_mini))
|
|
mask_points.append(to_img_coords(p.led_uni))
|
|
|
|
|
|
|
|
i = mask_img.getBufferedImage()
|
|
for p in mask_points:
|
|
#i.setRGB(p[0], p[1], 0xFFFFFF)
|
|
for x in range (p[0]-mask_radius, p[0]+mask_radius):
|
|
for y in range (p[1]-mask_radius, p[1]+mask_radius):
|
|
if math.hypot(x-p[0], y-p[1]) <= mask_radius:
|
|
i.setRGB(x,y, 0xFFFFFF)
|
|
|
|
|
|
mask_img = load_image(i)
|
|
#show_panel( mask_img.getBufferedImage())
|
|
|
|
set_return(mask_img) |