from ijutils import *
from ch.psi.pshell.imaging.Overlays import *
import ch.psi.pshell.imaging.Pen as Pen

def wait_next():
    camera.waitNext(250)

def get_centroid(source):
    bi = source.getImage()
    if bi is None:
        return None
    op = show_panel(bi, "Original")
    ip = load_image(bi)    
    grayscale(ip)    
    invert(ip)
    gaussian_blur(ip)    
    auto_threshold(ip)
    
    #binary_erode(ip)        
    show_panel(ip.getBufferedImage(), "Image")
    (results,output_img)=analyse_particles(ip, 1000,20000, print_table=True)    
    op.clearOverlays()
    show_panel(output_img.getBufferedImage(), "Outlines")
    if results.size()>0:
        print results        
        centroid  = (results.getValue("XM",0), results.getValue("YM",0))
        ov = Crosshairs(Pen(java.awt.Color.ORANGE),  java.awt.Point(int(centroid[0]),int(centroid[1])), java.awt.Dimension(15,15))
        op.addOverlay(ov)
        return centroid
              


class ImageStats(DeviceBase):
    def __init__(self, name, source):
        DeviceBase.__init__(self, name)
        self.source = source
        self.com_x_samples, self.com_y_samples = [], []
        class ComX(Readable):
            def read(self):     
                return mean(self.image_stats.com_x_samples)
        self.com_x_mean = ComX(); self.com_x_mean.image_stats = self
        class ComY(Readable):
            def read(self):     
                return mean(self.image_stats.com_y_samples)
        self.com_y_mean = ComY(); self.com_y_mean.image_stats = self
        class ComXVar(Readable):
            def read(self):     
                return stdev(self.image_stats.com_x_samples)
        self.com_x_stdev = ComXVar(); self.com_x_stdev.image_stats = self
        class ComXVar(Readable):
            def read(self):     
                return stdev(self.image_stats.com_y_samples)
        self.com_y_stdev = ComXVar(); self.com_y_stdev.image_stats = self
        set_device_alias(self.com_x_mean, name + " com x mean")
        set_device_alias(self.com_y_mean, name + " com y mean")
        set_device_alias(self.com_x_stdev, name + " com x stdev")
        set_device_alias(self.com_y_stdev, name + " com y stdev")
        self.bg_en = False
        self.num_images = 1
        self.initialize()        
    
    
    def doUpdate(self):
        self.com_x_samples, self.com_y_samples = [], []
        for i in range(self.num_images):
            self.source.waitNext(2000)
            centroid = get_centroid(self.source)
            if centroid is not None:
                self.com_x_samples.append(centroid[0])
                self.com_y_samples.append(centroid[1])

    def setNumberOfImages(self, value):
        self.num_images = value

    def enableBackground(self, value):
        self.bg_en = value
        
    def captureBackground(self, images):               
        self.doInitialize()
    
    def doClose(self):        
        pass    

if __name__ == "__builtin__":
    print get_centroid(image)    
    
    add_device(ImageStats("image_stats", image), True)
    image_stats.enableBackground(False)    
    for i in range (100):        
        image_stats.update()        
        print camera_tool.take(), camera_tool.com_x_mean.read(), camera_tool.com_y_mean.read()