################################################################################################### 
# Procedure to detect the puck light spots.
################################################################################################### 
assert_imaging_enabled()


COVER_PRESENT = True
ROOM_TEMP = is_room_temp()
USE_MASK = True

if get_exec_pars().source == CommandSource.ui:
    PLOT = None
    RENDERER = None
    TEXT = None

if COVER_PRESENT:
    cover_position =  hexiposi.readback.take()
    if (cover_position is None) or (cover_position == "Unknown"):    
        raise Exception("Unknown cover position")
    else:
        block_id = cover_position.upper()[0]
else:
    block_id = None    
print "Block id: ", block_id



number_frames = 5 if ROOM_TEMP else 10
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
threshold_method,threshold_range = "Manual",  (0, 215)

exclude_edges = True
led_latency = 0.5 #0.1



set_led_state(False)
time.sleep(led_latency)
img.waitNext(2000)

background = average_frames(number_backgrounds)
#background = integrate_frames(number_backgrounds)

set_led_state(True)
time.sleep(led_latency)
img.waitNext(2000)
image = average_frames(number_frames)
#image = integrate_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 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])
else:
    auto_threshold(image, method = threshold_method) #Tested ok:  MaxEntropy, Triangle, Yen
(r,output) =  analyse_particles(image, minimum_size,maximum_size, 
        fill_holes = True, exclude_edges = exclude_edges, print_table=False,
        output_image = "outlines", minCirc = min_circ
        , maxCirc = 1.0)

points = []
for row in range (r.counter):
    if in_roi(r.getValue("XM",row), r.getValue("YM",row)):
        x, y = int(r.getValue("XM", row)), int(r.getValue("YM", row))
        cx, cy = img.getCalibration().convertToAbsoluteX(x), img.getCalibration().convertToAbsoluteY(y)
        points.append([cx,cy])
        if RENDERER is not None:
            RENDERER.addOverlay(Crosshairs(Pen(java.awt.Color.MAGENTA),  java.awt.Point(x,y), java.awt.Dimension(15,15)))


clear_detection(block_id)
detect_pucks(points, block_id)
if PLOT is not None:
    plot_base_plate(points, p=PLOT)    

ret = get_puck_detection_dict(block_id)


if TEXT is not None:
    TEXT.setText(str(ret)) 


set_return(ret)