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)