From 05f6d0e94e5603caa769137508cf005bea3c9adc Mon Sep 17 00:00:00 2001
From: Arun Isaac
Date: Fri, 26 Mar 2021 14:17:19 +0530
Subject: Move planar angle generation into random_on_cap.
* src/sambal/sambal.py (random_planar_angle_pdf): Move function into
random_on_cap.
---
src/sambal/sambal.py | 22 ++++++++++------------
1 file changed, 10 insertions(+), 12 deletions(-)
(limited to 'src')
diff --git a/src/sambal/sambal.py b/src/sambal/sambal.py
index 3a35438..c42ea44 100644
--- a/src/sambal/sambal.py
+++ b/src/sambal/sambal.py
@@ -39,17 +39,6 @@ def rotate_from_nth_canonical(x, axis):
- axisn*(xn*s + a*(axisn - 1))/b
return x
-def random_planar_angle_pdf(maximum_planar_angle, dim):
- """Return a random planar angle using rejection sampling."""
- # We apply the log function just to prevent the floats from
- # underflowing.
- box_height = (dim-2)*log(sin(min(maximum_planar_angle, pi/2)))
- while True:
- theta = maximum_planar_angle*random()
- f = box_height + log(random())
- if f < (dim-2)*log(sin(theta)):
- return theta
-
def random_on_disk(axis, planar_angle):
"""Return a random vector uniformly distributed on the periphery of a
disk."""
@@ -64,4 +53,13 @@ def random_on_cap(axis, maximum_planar_angle):
cap. The random planar angle is generated using rejection sampling.
"""
- return random_on_disk(axis, random_planar_angle_pdf(maximum_planar_angle, axis.size))
+ # We apply the log function just to prevent the floats from
+ # underflowing.
+ dim = axis.size
+ box_height = (dim-2)*log(sin(min(maximum_planar_angle, pi/2)))
+ while True:
+ theta = maximum_planar_angle*random()
+ f = box_height + log(random())
+ if f < (dim-2)*log(sin(theta)):
+ break
+ return random_on_disk(axis, theta)
--
cgit v1.2.3