def test_sequence():
"""Test function: track some cells in a small sequence, illustration is grabbed to .png files
"""
datazip = '../test/data/seq0_extract.zip'
reader = ZipSource(datazip)
g = reader.generator()
cellLocations = [(221,184),(408,158),(529,367)]
params = {'N':32,'radius_halo':30,'radius_soma':10}
cell_list = [ Cell(x0,y0,**params) for x0,y0 in cellLocations ]
fig = plt.figure(1)
ax = fig.add_subplot(111)
for frame,im in g:
ax.imshow(im, interpolation='nearest')
for c in cell_list:
c.update(im)
CellUi(c,ax).draw(ax)
ax.set_xlim([300,500])
ax.set_ylim([100,250])
plt.draw()
plt.savefig('../test/temp/snp%04d.png'%frame)
sleep(.01)
plt.cla()
make_movie('../test/temp/snp*.png',out='../test/temp/movie.avi')
def access():
datazip = '../test/data/seq0_extract.zip'
reader = ZipSource(datazip)
g = reader.generator()
for i,im in enumerate(g):
pass
print '#frames:',i,' frame size:',im[1].shape
def process():
datazip = '../test/data/seq0_extract.zip'
reader = ZipSource(datazip)
g = reader.generator()
cellLocations = [(221,184),(408,158),(529,367)]
params = {'N':16,'radius_halo':30,'radius_soma':15}
cell_list = [ Cell(x0,y0,**params) for x0,y0 in cellLocations ]
for i,im in enumerate(g):
for c in cell_list:
c.update(im[1])
print '#frames:',i,' #cells:',len(cell_list)
