def plotFishpathandTraces(dataArr, colorArr):
fig = pyplot.figure(1, figsize=(11, 7))
for i in range(2):
if i == 0:
for k in range(3):
ax = pyplot.subplot(4, 3, k + 1)
aba.plotFishPath(dataArr[k], color=colorArr[k], smoothWinLen=5)
# pyplot.title('Fish Path in Tank')
pyplot.xlabel("46 mm")
pyplot.ylabel("22 mm")
ax.set_xticks([20, 40])
ax.set_yticks([10, 20])
ax2 = pyplot.subplot(4, 3, k + 4)
aba.plotFishXPosition(dataArr[k], fmt=colorArr[k], smooth=5)
ax2.set_yticks([20, 40])
pyplot.ylabel("x position")
if k == 1:
patch5 = mpl.patches.Rectangle((900, 0), 1844, 50, color=[1, 0.5, 0.5], fill=True)
pyplot.gca().add_patch(patch5)
if i == 1:
for k in range(3):
ax = pyplot.subplot(4, 3, i + k + 6)
aba.plotFishPath(dataArr[k + i + 2], color=colorArr[k + i + 2], smoothWinLen=5)
# pyplot.title('Fish Path in Tank')
pyplot.xlabel("46 mm")
pyplot.ylabel("22 mm")
ax.set_xticks([20, 40])
ax.set_yticks([10, 20])
ax2 = pyplot.subplot(4, 3, i + k + 9)
aba.plotFishXPosition(dataArr[k + i + 2], fmt=colorArr[k + i + 2], smooth=5)
pyplot.ylabel("x position")
ax2.set_yticks([20, 40])
if k == 1:
patch5 = mpl.patches.Rectangle((900, 0), 1844, 50, color=[1, 0.5, 0.5], fill=True)
pyplot.gca().add_patch(patch5)
aba.plotFishXPosition(dataArr[k + i + 2], fmt=colorArr[k + i + 2], smooth=5)
pyplot.ylabel("x position")
ax2.set_yticks([20, 40])
if k == 1:
patch5 = mpl.patches.Rectangle((900, 0), 1844, 50, color=[1, 0.5, 0.5], fill=True)
pyplot.gca().add_patch(patch5)
# pyplot.title('Fish Y Position')
plotFishpathandTraces(
[e_safe_same, e_shock_same, e_safe_same_sec, e_safe5_first, e_shock5, e_safe5_sec], ["b", "b", "b", "b", "b", "b"]
)
pyplot.gcf()
pyplot.subplot(432).clear()
aba.plotFishPath(e_shock5, trange=[0, 900], color="b", smoothWinLen=5)
# aba.plotFishPath(e_shock5,trange=[900,2744], color=[1,.5,.5], smoothWinLen=5)
aba.plotFishPath(e_shock5, trange=[2744, 3105], color="r", smoothWinLen=5)
pyplot.gcf()
pyplot.subplot(438).clear()
aba.plotFishPath(e_shock_same, trange=[0, 900], color="b", smoothWinLen=5)
# aba.plotFishPath(e_shock5,trange=[900,2744], color=[1,.5,.5], smoothWinLen=5)
aba.plotFishPath(e_shock_same, trange=[2744, 3105], color="r", smoothWinLen=5)
b = [False, False, False, False, False, False, False, False, False, True, False, False]
for i in range(len(b)):
if not b[i]:
ax = pyplot.subplot(4, 3, i + 1)
ax.set_xticks([])
ax.set_yticks([])
pyplot.ylabel("")
