SF.net SVN: matplotlib: [3980] trunk/py4science/examples/glass_dots2.py

[<jd...@us...>]

Revision: 3980
          http://matplotlib.svn.sourceforge.net/matplotlib/?rev=3980&view=rev
Author:   jdh2358
Date:     2007-10-21 14:19:51 -0700 (Sun, 21 Oct 2007)

Log Message:
-----------
added glass dots2

Added Paths:
-----------
    trunk/py4science/examples/glass_dots2.py

Added: trunk/py4science/examples/glass_dots2.py
===================================================================
--- trunk/py4science/examples/glass_dots2.py	                        (rev 0)
+++ trunk/py4science/examples/glass_dots2.py	2007-10-21 21:19:51 UTC (rev 3980)
@@ -0,0 +1,83 @@
+import numpy as npy
+import numpy.linalg as linalg
+from pylab import figure, show
+
+class Transformer:
+    def __init__(self, axes):
+        self.theta = 0
+        self.dtheta = 0.01
+        self.sx = 1.
+        self.sy = 1.
+        self.dx = 0.001
+        self.axes = axes
+        self.canvas = axes.figure.canvas
+        self.canvas.mpl_connect('button_press_event', self.press)
+        self.canvas.mpl_connect('button_release_event', self.release)        
+        self.canvas.mpl_connect('motion_notify_event', self.move)        
+
+        X1 = self.X1 = npy.matrix(npy.random.rand(2,2000))-0.5
+
+        x1 = X1[0].flat
+        y1 = X1[1].flat
+        x2 = X1[0].flat # no transformation yet
+        y2 = X1[1].flat
+        self.line1, self.line2 = ax.plot(x1, y1,'go', x2, y2, 'ro', markersize=2)
+        self.xlast = None
+        self.ylast = None
+        self.title = ax.set_title('drag the left or right mouse to stretch and rotate', fontweight='bold')
+
+    def press(self, event):
+        'mouse press, save the x and y locations'
+        self.xlast = event.xdata
+        self.ylast = event.ydata
+
+
+    def release(self, event):
+        'release the mouse'
+        self.xlast = None
+        self.ylast = None
+        self.draw()
+        
+    def draw(self):
+        sx, sy, theta = self.sx, self.sy, self.theta
+        a =  sx*npy.cos(theta)
+        b = -sx*npy.sin(theta)
+        c =  sy*npy.sin(theta)
+        d =  sy*npy.cos(theta)
+        M =  npy.matrix([[a,b],[c,d]])
+
+        X2 = M*self.X1        
+        x2 = X2[0].flat
+        y2 = X2[1].flat
+        self.line2.set_data(x2,y2)
+
+        self.canvas.draw()
+        
+    def move(self, event):
+
+        if not event.inaxes: return    # not over axes
+        if self.xlast is None: return  # no initial data
+        if not event.button: return    # no button press
+        
+        dx = event.xdata - self.xlast
+        dy = event.ydata - self.ylast        
+
+        if event.button==1:
+            self.theta += dx
+        elif event.button==3:
+            self.sx += dx
+            self.sy += dy
+            
+        self.title.set_text('sx=%1.2f, sy=%1.2f, theta=%1.2f'%(self.sx, self.sy, self.theta))
+        self.draw()
+        self.xlast = event.xdata
+        self.ylast = event.ydata
+
+
+
+        
+from pylab import subplot, show        
+fig = figure()
+ax = fig.add_subplot(111)
+t = Transformer(ax)
+show()


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.