From 084dfd08b714faefaea77cb7dc04d2e93dc04b1d Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Fri, 16 Sep 2011 14:27:46 -0400 Subject: File reorganization to move toward standard python package layout --- transform.py | 38 -------------------------------------- 1 file changed, 38 deletions(-) delete mode 100644 transform.py (limited to 'transform.py') diff --git a/transform.py b/transform.py deleted file mode 100644 index 299da46..0000000 --- a/transform.py +++ /dev/null @@ -1,38 +0,0 @@ -import numpy as np - -def make_rotation_matrix(phi, n): - """ - Make the rotation matrix to rotate points through an angle `phi` - counter-clockwise around the axis `n` (when looking towards +infinity). - - Source: Weissten, Eric W. "Rotation Formula." Mathworld. - """ - n = np.asarray(n)/np.linalg.norm(n) - - return np.cos(phi)*np.identity(3) + (1-np.cos(phi))*np.outer(n,n) + \ - np.sin(phi)*np.array([[0,n[2],-n[1]],[-n[2],0,n[0]],[n[1],-n[0],0]]) - -def rotate(x, phi, n): - """ - Rotate an array of points `x` through an angle phi counter-clockwise - around the axis `n` (when looking towards +infinity). - """ - return np.inner(np.asarray(x),make_rotation_matrix(phi, n)) - -def normalize(x): - "Returns unit vectors in the direction of `x`." - x = np.asarray(x) - - if x.shape[-1] != 3: - raise ValueError('dimension of last axis must be 3.') - - d = len(x.shape) - - if d == 1: - norm = np.sqrt(x.dot(x)) - elif d == 2: - norm = np.sqrt(np.sum(x*x, axis=1))[:,np.newaxis] - else: - raise ValueError('len(`x`.shape) must be zero or one.') - - return x/norm -- cgit