######## PLOT PATHS
whichndx = [1,2,3,5,6,7]
trange = [[0,1700],[0,1700],[0,1700]]
f = pyplot.figure(1)
f.set_facecolor('w')
titles = ['Baseline','Vehicle','Schreckstoff']
for i,d in enumerate([d_pre, d_veh, d_sch]):
pyplot.subplot(1,3,i+1)
map(aba.plotFishPath, d[whichndx], [trange[i]]*len(whichndx))
pyplot.title(titles[i])
######## PLOT HEATMAPS
whichndx = [0,1,2,3,4,5,6,7]
trange = [[1000,1700],[0,360],[1000,1700],[0,360]]
titles = ['Baseline','Vehicle_Start','Vehicle_End','Schreckstoff_Start']
bins = np.linspace(0,300,50)
for i,d in enumerate([d_pre[whichndx],d_veh[whichndx],d_veh[whichndx], d_sch[whichndx]]):
f = pyplot.figure(2+i, figsize=(10,4))
f.set_facecolor('w')
f.suptitle(titles[i])
for j,f in enumerate(d):
pyplot.subplot(2,math.ceil(len(whichndx)/2.0),j+1)
tracking = aba.getTracking(f)
tracking[:,1]-=tracking[:,1].min()
tracking[:,2]-=tracking[:,2].min()
aba.plotFishPathHeatmap(f,trange[i],bins=bins,tracking=tracking)
pyplot.savefig('/Users/andalman/Dropbox/ConchisData/Schreckstoff/20130710_%s_heatmap_%dto%d.png'%
(titles[i],trange[i][0],trange[i][1]), dpi=300)
pyplot.show()
