script/imgproc/CreateMask.py

@@ -0,0 +1,29 @@
+
+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)

script/imgproc/LedDetectionProc.py

@@ -4,6 +4,7 @@
 
 COVER_PRESENT = True
 ROOM_TEMP = is_room_temp()
+USE_MASK = True
 
 if get_exec_pars().source == CommandSource.ui:
     PLOT = None
@@ -23,9 +24,9 @@ print "Block id: ", block_id
 
 
 number_frames = 5 if ROOM_TEMP else 10
-number_backgrounds = 5 if ROOM_TEMP else 10
-minimum_size = 150
-maximum_size = 1500
+number_backgrounds = 5 if ROOM_TEMP else 5
+minimum_size = 78     # r = 5 # 150   
+maximum_size = 750    # r = 15 #1500
 min_circ = 0.2
 
 threshold_method = "MaxEntropy" if ROOM_TEMP else "Default" #Apparently good for LN2: Default, Intermodes, IsoData, Otsu
@@ -52,13 +53,21 @@ image = average_frames(number_frames)
 set_led_state(False)
 
 op_image(image, background, "subtract", float_result=True, in_place=True)
+image=grayscale(image)
 
-if RENDERER is not None:
+if RENDERER is not None and RENDERER.isShowing():
     RENDERER.setImage(None, image.getBufferedImage(), None)    
 else:    
     RENDERER = show_panel(image.getBufferedImage())
 RENDERER.clearOverlays()
 
+if USE_MASK:
+    mask_img = run("imgproc/CreateMask")    
+    #mask_img=grayscale(mask_img)
+    #show_panel( mask_img.getBufferedImage())
+    op_image(image, mask_img, "and", float_result=False, in_place=True)
+    RENDERER.setImage(None, image.getBufferedImage(), None)    
+
 invert(image)
 if threshold_method == "Manual":
     threshold(image, threshold_range[0], threshold_range[1])