back_and_forth

Author: ddkohler

WrightTools/artists/_animate.py

@@ -111,7 +111,11 @@ def updater(frame):
         fig.canvas.draw_idle()
         return art
 
+    # generate frame sequence
     frames = list(np.ndindex(data.shape[:-2]))
+    if back_and_forth:
+        frames += reversed(frames)
+
     return FuncAnimation(
         fig=fig,
         func=updater,
@@ -139,6 +143,9 @@ def update(frame):
             slider.set_val(ind)
 
     frames = list(np.ndindex(tuple([s.valmax + 1 for s in interact2D.sliders.values()])))
+    if back_and_forth:
+        frames += reversed(frames)
+
     return FuncAnimation(
         fig=interact2D.fig,
         func=update,

tests/artists/test_animate.py

@@ -15,7 +15,7 @@ def test_animate2D():
     from matplotlib.colors import CenteredNorm
     from functools import partial
 
-    ani = wt.artists.animate2D(d, interval=100, norm=partial(CenteredNorm, vcenter=0))
+    ani = wt.artists.animate2D(d, interval=50, back_and_forth=True, norm=partial(CenteredNorm, vcenter=0))
     return ani