def test_ray2_to_from_dict():
"""Test the to/from dict of Ray2D objects."""
pt = Point2D(2, 0)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
ray_dict = ray.to_dict()
new_ray = Ray2D.from_dict(ray_dict)
assert isinstance(new_ray, Ray2D)
assert new_ray.to_dict() == ray_dict
def test_to_from_array():
"""Test to/from array method."""
test_ray = Ray2D(Point2D(2, 0), Vector2D(2, 2))
ray_array = ((2, 0), (2, 2))
assert test_ray == Ray2D.from_array(ray_array)
ray_array = ((2, 0), (2, 2))
test_ray = Ray2D(Point2D(2, 0), Vector2D(2, 2))
assert test_ray.to_array() == ray_array
test_ray_2 = Ray2D.from_array(test_ray.to_array())
assert test_ray == test_ray_2
def test_equality():
"""Test the equality of Ray2D objects."""
pt = Point2D(2, 0)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
ray_dup = ray.duplicate()
ray_alt = Ray2D(Point2D(2, 0.1), vec)
assert ray is ray
assert ray is not ray_dup
assert ray == ray_dup
assert hash(ray) == hash(ray_dup)
assert ray != ray_alt
assert hash(ray) != hash(ray_alt)
def test_intersect_line_ray():
"""Test the Ray2D distance_to_point method."""
pt_1 = Point2D(2, 2)
vec_1 = Vector2D(0, 2)
ray_1 = Ray2D(pt_1, vec_1)
pt_2 = Point2D(0, 3)
vec_2 = Vector2D(4, 0)
ray_2 = Ray2D(pt_2, vec_2)
pt_3 = Point2D(0, 0)
vec_3 = Vector2D(1, 1)
ray_3 = Ray2D(pt_3, vec_3)
assert ray_1.intersect_line_ray(ray_2) == Point2D(2, 3)
assert ray_1.intersect_line_ray(ray_3) == Point2D(2, 2)
def __init__(self,
point,
edge_left,
edge_right,
direction_vectors=None,
tol=1e-10):
self.point = point
self.edge_left = edge_left
self.edge_right = edge_right
self.prev = None
self.next = None
self.lav = None
self.tol = tol
self.id = None
# this should be handled better. Maybe membership in lav implies validity?
self._valid = True
creator_vectors = (edge_left.v.normalize() * -1,
edge_right.v.normalize())
if direction_vectors is None:
direction_vectors = creator_vectors
# The determinant of two 2d vectors equals the sign of their cross product
# If second vector is to left, then sign of det is pos, else neg
# So if cw polygon, convex angle will be neg (since second vector to right)
# In this case, since we flip first vector, concave will be neg
self._is_reflex = direction_vectors[0].determinant(
direction_vectors[1]) < 0.0
self._bisector = Ray2D(
self.point,
operator.add(*creator_vectors) * (-1 if self.is_reflex else 1))
log.info("Created vertex %s", self.__repr__())
_debug.line((self.bisector.p.x, self.bisector.p.y,
self.bisector.p.x + self.bisector.v.x * 100,
self.bisector.p.y + self.bisector.v.y * 100),
fill="blue")
def test_reflect():
"""Test the Ray2D reflect method."""
pt = Point2D(2, 2)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
origin_1 = Point2D(0, 1)
origin_2 = Point2D(1, 1)
normal_1 = Vector2D(0, 1)
normal_2 = Vector2D(-1, 1).normalize()
assert ray.reflect(normal_1, origin_1).p == Point2D(2, 0)
assert ray.reflect(normal_1, origin_1).v == Vector2D(0, -2)
assert ray.reflect(normal_1, origin_2).p == Point2D(2, 0)
assert ray.reflect(normal_1, origin_2).v == Vector2D(0, -2)
test_1 = ray.reflect(normal_2, origin_2)
assert test_1.p == Point2D(2, 2)
assert test_1.v.x == pytest.approx(2, rel=1e-3)
assert test_1.v.y == pytest.approx(0, rel=1e-3)
test_2 = ray.reflect(normal_2, origin_1)
assert test_2.p.x == pytest.approx(1, rel=1e-3)
assert test_2.p.y == pytest.approx(3, rel=1e-3)
assert test_1.v.x == pytest.approx(2, rel=1e-3)
assert test_1.v.y == pytest.approx(0, rel=1e-3)
def test_arc2_split_line_infinite():
"""Test the Arc2D split_line_ray_infinite method."""
pt = Point2D(2, 0)
arc = Arc2D(pt, 1, 0, math.pi)
circle = Arc2D(pt, 1)
int1 = circle.split_line_infinite(Ray2D(Point2D(), Vector2D(1, 0)))
assert len(int1) == 2
assert int1[0].p1 == Point2D(3, 0)
assert int1[0].p2.x == pytest.approx(1, rel=1e-2)
assert int1[0].p2.y == pytest.approx(0, rel=1e-2)
assert int1[1].p2 == Point2D(3, 0)
assert int1[1].p1.x == pytest.approx(1, rel=1e-2)
assert int1[1].p1.y == pytest.approx(0, rel=1e-2)
int2 = arc.split_line_infinite(Ray2D(Point2D(0, 0.5), Vector2D(1, 0)))
assert len(int2) == 3
def test_scale_world_origin():
"""Test the Ray2D scale method with None origin."""
pt = Point2D(2, 2)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
new_ray = ray.scale(2)
assert new_ray.p == Point2D(4, 4)
assert new_ray.v == Point2D(0, 4)
assert new_ray.v.magnitude == 4
def test_move():
"""Test the Ray2D move method."""
pt = Point2D(2, 0)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
vec_1 = Vector2D(2, 2)
new_ray = ray.move(vec_1)
assert new_ray.p == Point2D(4, 2)
assert new_ray.v == vec
def test_closest_point():
"""Test the Ray2D closest_point method."""
pt = Point2D(2, 2)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
near_pt = Point2D(3, 3)
assert ray.closest_point(near_pt) == Point2D(2, 3)
near_pt = Point2D(2, 0)
assert ray.closest_point(near_pt) == Point2D(2, 2)
near_pt = Point2D(1, 5)
assert ray.closest_point(near_pt) == Point2D(2, 5)
def test_distance_to_point():
"""Test the Ray2D distance_to_point method."""
pt = Point2D(2, 2)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
near_pt = Point2D(3, 3)
assert ray.distance_to_point(near_pt) == 1
near_pt = Point2D(2, 0)
assert ray.distance_to_point(near_pt) == 2
near_pt = Point2D(1, 5)
assert ray.distance_to_point(near_pt) == 1
def _point_height_rights(point, vector_2d, altitude, obstacle_poly,
obstacle_height, default):
"""Get the height of a given 2D point and sun vector for a rights boundary.
This will return None if the obstacle has no effect on the height.
"""
int_pts = obstacle_poly.intersect_line_ray(Ray2D(point, vector_2d))
if len(int_pts) != 0:
pt_dist = min([point.distance_to_point(pt) for pt in int_pts])
new_height = pt_dist * math.tan(altitude) + obstacle_height
if new_height < default:
return new_height
return default
def test_intersect_line_ray():
"""Test the Polygon2D intersect_line_ray method."""
pts = (Point2D(0, 0), Point2D(2, 0), Point2D(2, 2), Point2D(0, 2))
polygon = Polygon2D(pts)
ray_1 = Ray2D(Point2D(-1, 1), Vector2D(1, 0))
ray_2 = Ray2D(Point2D(1, 1), Vector2D(1, 0))
ray_3 = Ray2D(Point2D(1, 1), Vector2D(11, 0))
ray_4 = Ray2D(Point2D(-1, 1), Vector2D(-1, 0))
assert len(polygon.intersect_line_ray(ray_1)) == 2
assert len(polygon.intersect_line_ray(ray_2)) == 1
assert len(polygon.intersect_line_ray(ray_3)) == 1
assert len(polygon.intersect_line_ray(ray_4)) == 0
line_1 = LineSegment2D(Point2D(-1, 1), Vector2D(0.5, 0))
line_2 = LineSegment2D(Point2D(-1, 1), Vector2D(2, 0))
line_3 = LineSegment2D(Point2D(-1, 1), Vector2D(3, 0))
assert len(polygon.intersect_line_ray(line_1)) == 0
assert len(polygon.intersect_line_ray(line_2)) == 1
assert len(polygon.intersect_line_ray(line_3)) == 2
def test_ray2d_init():
"""Test the initalization of Ray2D objects and basic properties."""
pt = Point2D(2, 0)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
str(ray) # test the string representation of the ray
assert ray.p == Point2D(2, 0)
assert ray.v == Vector2D(0, 2)
flip_ray = ray.reverse()
assert flip_ray.p == Point2D(2, 0)
assert flip_ray.v == Vector2D(0, -2)
def test_scale():
"""Test the Ray2D scale method."""
pt = Point2D(2, 2)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
origin_1 = Point2D(0, 2)
origin_2 = Point2D(1, 1)
new_ray = ray.scale(2, origin_1)
assert new_ray.p == Point2D(4, 2)
assert new_ray.v == Point2D(0, 4)
assert new_ray.v.magnitude == 4
new_ray = ray.scale(2, origin_2)
assert new_ray.p == Point2D(3, 3)
assert new_ray.v == Point2D(0, 4)
def test_rotate():
"""Test the Ray2D rotate method."""
pt = Point2D(2, 2)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
origin_1 = Point2D(0, 2)
test_1 = ray.rotate(math.pi, origin_1)
assert test_1.p.x == pytest.approx(-2, rel=1e-3)
assert test_1.p.y == pytest.approx(2, rel=1e-3)
assert test_1.v.x == pytest.approx(0, rel=1e-3)
assert test_1.v.y == pytest.approx(-2, rel=1e-3)
test_2 = ray.rotate(math.pi / 2, origin_1)
assert test_2.p.x == pytest.approx(0, rel=1e-3)
assert test_2.p.y == pytest.approx(4, rel=1e-3)
assert test_2.v.x == pytest.approx(-2, rel=1e-3)
assert test_2.v.y == pytest.approx(0, rel=1e-3)
def test_ray2d_immutability():
"""Test the immutability of Ray2D objects."""
pt = Point2D(2, 0)
vec = Vector2D(0, 2)
ray = Ray2D(pt, vec)
assert isinstance(ray, Ray2D)
with pytest.raises(AttributeError):
ray.p.x = 3
with pytest.raises(AttributeError):
ray.v.x = 3
with pytest.raises(AttributeError):
ray.p = Point2D(0, 0)
with pytest.raises(AttributeError):
ray.v = Vector2D(2, 2)
ray_copy = ray.duplicate()
assert ray.p == ray_copy.p
assert ray.v == ray_copy.v
def test_intersect_line_infinite():
"""Test the Polygon2D intersect_line_infinite method."""
pts = (Point2D(0, 0), Point2D(2, 0), Point2D(2, 2), Point2D(0, 2))
polygon = Polygon2D(pts)
ray_1 = Ray2D(Point2D(-1, 1), Vector2D(1, 0))
ray_2 = Ray2D(Point2D(1, 1), Vector2D(1, 0))
ray_3 = Ray2D(Point2D(1, 1), Vector2D(11, 0))
ray_4 = Ray2D(Point2D(-1, 1), Vector2D(-1, 0))
ray_5 = Ray2D(Point2D(-1, 3), Vector2D(-1, 0))
ray_6 = Ray2D(Point2D(0, 2), Vector2D(-1, -1))
assert len(polygon.intersect_line_infinite(ray_1)) == 2
assert len(polygon.intersect_line_infinite(ray_2)) == 2
assert len(polygon.intersect_line_infinite(ray_3)) == 2
assert len(polygon.intersect_line_infinite(ray_4)) == 2
assert len(polygon.intersect_line_infinite(ray_5)) == 0
assert len(polygon.intersect_line_infinite(ray_6)) > 0
def to_ray2d(ray):
"""Ladybug Ray2D from Rhino Ray3d."""
return Ray2D(to_point2d(ray.Position), to_vector2d(ray.Direction))
