84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
from startup import *
|
|
from ijutils import *
|
|
from ch.psi.pshell.imaging.Overlays import *
|
|
import ch.psi.pshell.imaging.Pen as Pen
|
|
import java.awt.Color as Color
|
|
import random
|
|
|
|
|
|
|
|
class BpmStats(DeviceBase):
|
|
def __init__(self, name, prefix, read_interval = 0.1):
|
|
DeviceBase.__init__(self, name)
|
|
self.read_interval = read_interval
|
|
self.prefix = prefix
|
|
self.bpm_x = Channel(prefix + ":X1", type = 'd', alias = name + " x")
|
|
self.bpm_y = Channel(prefix + ":Y1", type = 'd', alias = name + " y")
|
|
self.com_x_samples, self.com_y_samples = [], []
|
|
class ComX(Readable):
|
|
def read(self):
|
|
if len(self.image_stats.com_x_samples)==0: return None
|
|
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):
|
|
if len(self.image_stats.com_y_samples)==0: return None
|
|
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):
|
|
if len(self.image_stats.com_x_samples)==0: return None
|
|
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):
|
|
if len(self.image_stats.com_y_samples)==0: return None
|
|
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 + " x mean")
|
|
set_device_alias(self.com_y_mean, name + " y mean")
|
|
set_device_alias(self.com_x_stdev, name + " x stdev")
|
|
set_device_alias(self.com_y_stdev, name + " y stdev")
|
|
self.num_images = 1
|
|
self.initialize()
|
|
|
|
|
|
def doUpdate(self):
|
|
self.com_x_samples, self.com_y_samples = [], []
|
|
for i in range(self.num_images):
|
|
time.sleep(self.read_interval)
|
|
cx = self.bpm_x.read()
|
|
cy = self.bpm_y.read()
|
|
print "centroid ",[cx,cy]
|
|
self.com_x_samples.append(cx)
|
|
self.com_y_samples.append(cy)
|
|
|
|
def setNumberOfImages(self, value):
|
|
self.num_images = value
|
|
|
|
def enableBackground(self, value):
|
|
pass
|
|
|
|
def captureBackground(self, images):
|
|
pass
|
|
|
|
def doClose(self):
|
|
pass
|
|
|
|
def start(self):
|
|
pass
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
|
|
"""
|
|
if __name__ == "__builtin__":
|
|
add_device(BpmStats("bpm_com", "SINEG01-DBPM340"), True)
|
|
bpm_com.setNumberOfImages(5)
|
|
for i in range (10):
|
|
bpm_com.update()
|
|
print bpm_com.take(), bpm_com.com_x_mean.read(), bpm_com.com_y_mean.read()
|
|
time.sleep(1)
|
|
|
|
""" |