def gradient_angle_list(self):
""" Gets the gradient angles between neighboring linear regions """
angle_list = {}
for (u, v) in self.return_obj.polytope_graph.keys():
u_grad = self.collection[u].get_gradient()
v_grad = self.collection[v].get_gradient()
angle_list[(u, v)] = utils.angle(u_grad, v_grad)
return angle_list
def theta(rvec, chi):
"""
Return the angle between chi and rvec.
"""
chi_ref = stride_tricks.as_strided(np.array(chi),
strides=(0, 1 * 8),
shape=rvec.shape)
return bbh.angle(rvec, chi_ref)
def phi(rvec):
rhovec, ref = rhoref(rvec)
return bbh.angle(rhovec, ref)
def test_3(self):
assert round(utilities.angle((0, 0), (90, 0), (0, 90)), 3) == 90.0
def test_2(self):
assert round(utilities.angle((0, 0), (90, 0), (90, 0)), 3) == 0.0
def test_1(self):
round(utilities.angle((0, 0), (0, 90), (90, 0)), 3) == 90.0
def test_3(self):
assert round(utilities.angle((0, 0), (90, 0), (0, 90)), 3) == 90.0
def test_2(self):
assert round(utilities.angle((0, 0), (90, 0), (90, 0)), 3) == 0.0
def test_1(self):
round(utilities.angle((0, 0), (0, 90), (90, 0)), 3) == 90.0
def bounce(self, wall):
perpVector = wall.getPerpVector()
lumVector = [cos(self.direction), sin(self.direction)]
angl = angle(lumVector, wall.getVector())
self.direction = self.direction + 2*angl
