From 870236b3c4950762a73247c68023a8dee6e14a7b Mon Sep 17 00:00:00 2001 From: Anthony LaTorre Date: Sun, 12 Jun 2011 21:31:22 -0400 Subject: added some fun models; added some untested code to implement absorption, scattering, reflection, and refraction --- sample.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 sample.py (limited to 'sample.py') diff --git a/sample.py b/sample.py new file mode 100644 index 0000000..b494384 --- /dev/null +++ b/sample.py @@ -0,0 +1,29 @@ +import numpy as np + +def uniform_sphere(size=None, dtype=np.double): + """ + Generate random points isotropically distributed across the unit sphere. + + Args: + - size: int, *optional* + Number of points to generate. If no size is specified, a single + point is returned. + + Source: Weisstein, Eric W. "Sphere Point Picking." Mathworld. + """ + + theta, u = np.random.uniform(0.0, 2*np.pi, size), \ + np.random.uniform(-1.0, 1.0, size) + + c = np.sqrt(1-u**2) + + if size is None: + return np.array([c*np.cos(theta), c*np.sin(theta), u]) + + points = np.empty((size, 3), dtype) + + points[:,0] = c*np.cos(theta) + points[:,1] = c*np.sin(theta) + points[:,2] = u + + return points -- cgit