91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
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.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.waitNext(2000)
|
|
self.com_x_samples.append(self.com_x.read())
|
|
self.com_y_samples.append(self.com_y.read())
|
|
|
|
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):
|
|
imagestats.update()
|
|
print camera_tool.take(), camera_tool.com_x.read(), camera_tool.com_y.read()
|
|
|
|
|