fixed anim.py to use tk idiom

jdh2358 · jdh2358 · commit 8db1a39af653 · 2008-07-24T22:34:32.000Z
svn path=/trunk/matplotlib/; revision=5854
diff --git a/examples/animation/anim.py b/examples/animation/anim.py
@@ -1,32 +1,24 @@
 #!/usr/bin/env python
 """
-A simple example of an animated plot in matplotlib.  You can test the
-speed of animation of various backends by running the script with the
-'-dSomeBackend' flag
-
-SC Aug 31 2005 mpl 0.83.2:
-Here are some numbers from my system, where FPS is the frames rendered
-per second
-
-  GTK       29 FPS
-  GTKAgg    18 FPS
-  GTKCairo  15 FPS
-  TkAgg     13 FPS
-  QkAgg     13 FPS
+A simple example of an animated plot in tkagg
 """
-import time
+import matplotlib
+matplotlib.use('TkAgg') # do this before importing pylab
 
-import pylab as p
+import matplotlib.pyplot as plt
+fig = plt.figure()
+ax = fig.add_subplot(111)
 
-# turn interactive mode on for dynamic updates.  If you aren't in
-# interactive mode, you'll need to use a GUI event handler/timer.
-p.ion()
+def animate():
+    tstart = time.time()                   # for profiling
+    x = np.arange(0, 2*np.pi, 0.01)        # x-array
+    line, = ax.plot(x, np.sin(x))
 
-tstart = time.time()                 # for profiling
-x = p.arange(0, 2*p.pi, 0.01)        # x-array
-line, = p.plot(x, p.sin(x))
-for i in p.arange(1,200):
-    line.set_ydata(p.sin(x+i/10.0))  # update the data
-    p.draw()                         # redraw the canvas
+    for i in np.arange(1,200):
+        line.set_ydata(np.sin(x+i/10.0))          # update the data
+        fig.canvas.draw()                         # redraw the canvas
+    print 'FPS:' , 200/(time.time()-tstart)
 
-print 'FPS:' , 200/(time.time()-tstart)
+win = fig.canvas.manager.window
+fig.canvas.manager.window.after(100, animate)
+plt.show()
