from wood_nano import int2 as wood_nano_int2
from wood_nano import point2 as wood_nano_point2
from wood_nano import rtree as wood_nano_rtree
from wood_nano.conversions_python import from_int2
from compas_wood.conversions_compas import to_point2
from compas_wood.conversions_compas import from_point2
[docs]def rtree(polylines):
"""Given a list of polylines-pairs find neighbors between them via aabb and obb collisions.
Polyline pairs means the top and bottom polylines of plates.
Parameters
----------
polylines : list[compas.geometry.Polyline]
List of polylines.
Returns
-------
list[int]
List of neighbors.
list[compas.geometry.Box]
List of aabb boxes.
list[compas.geometry.Box]
List of obb boxes.
"""
w_polylines = to_point2(polylines)
w_neighbors = wood_nano_int2()
w_aabb = wood_nano_point2()
w_obb = wood_nano_point2()
wood_nano_rtree(w_polylines, w_neighbors, w_aabb, w_obb)
return from_int2(w_neighbors), from_point2(w_aabb, "Box"), from_point2(w_obb, "Box")
if __name__ == "__main__":
from compas.geometry import Polyline, Point
# Example
polyline0 = Polyline([Point(0, 0, 0), Point(1, 0, 0), Point(1, 1, 0), Point(0, 1, 0), Point(0, 0, 0)])
polyline1 = Polyline([Point(0, 0, 1), Point(1, 0, 1), Point(1, 1, 1), Point(0, 1, 1), Point(0, 0, 1)])
polyline2 = Polyline([Point(0, 0, 1), Point(1, 0, 1), Point(1, 1, 1), Point(0, 1, 1), Point(0, 0, 1)])
polyline3 = Polyline([Point(0, 0, 2), Point(1, 0, 2), Point(1, 1, 2), Point(0, 1, 2), Point(0, 0, 2)])
polylines = [polyline0, polyline1, polyline2, polyline3]
neighbors, aabb, obb = rtree(polylines)
print(neighbors)
print(aabb)
print(obb)
import sys
if sys.version_info >= (3, 9):
from compas_viewer import Viewer
viewer = Viewer()
for polyline in polylines:
viewer.scene.add(polyline, show_points=False)
for box in aabb:
viewer.scene.add(box, show_points=False)
for box in obb:
viewer.scene.add(box, show_points=False)
viewer.show()