session-py 0.1.3__py3-none-any.whl

session_py/__init__.py

@@ -0,0 +1,70 @@
+"""
+This module contains all the classes and functions that are exposed to the user.
+"""
+
+from .color import Color
+from .point import Point
+from .vector import Vector
+from .plane import Plane
+from .line import Line
+from .polyline import Polyline
+from .xform import Xform
+from .quaternion import Quaternion
+from .tree import Tree
+from .treenode import TreeNode
+from .graph import Graph
+from .vertex import Vertex
+from .edge import Edge
+from .objects import Objects
+from .session import Session
+from .mesh import Mesh, NormalWeighting
+from .boundingbox import BoundingBox
+from .pointcloud import PointCloud
+from .bvh import BVH, BVHNode
+from .tolerance import Tolerance
+from . import encoders
+from .obj import load_obj, save_obj
+from . import intersection
+from .nurbscurve import NurbsCurve
+from .nurbssurface import NurbsSurface
+from .primitives import Primitives
+from .trimmedsurface import TrimmedSurface
+from .brep import BRep
+from .ray_box_intersection import ray_box
+from .closest import Closest
+
+__all__ = [
+    "Color",
+    "Point",
+    "Vector",
+    "Plane",
+    "Line",
+    "Polyline",
+    "Xform",
+    "Quaternion",
+    "Tree",
+    "TreeNode",
+    "Graph",
+    "Vertex",
+    "Edge",
+    "Objects",
+    "Session",
+    "Mesh",
+    "NormalWeighting",
+    "BoundingBox",
+    "PointCloud",
+    "BVH",
+    "BVHNode",
+    "Tolerance",
+    "encoders",
+    "load_obj",
+    "save_obj",
+    "intersection",
+    "NurbsCurve",
+    "NurbsSurface",
+    "Primitives",
+    "TrimmedSurface",
+    "BRep",
+    "ray_box",
+    "Closest",
+]

session_py/aabb_test.py

@@ -0,0 +1,106 @@
+from .mini_test import MINI_TEST
+from .mini_test import MINI_CHECK
+from .mini_test import run_all
+from .tolerance import TOLERANCE
+
+
+@MINI_TEST("AABBTree", "Build Empty")
+def test_aabbtree_build_empty():
+    from session_py import Closest
+    from session_py import Mesh
+    from session_py import Point
+
+    m = Mesh()
+    cp, fk, d = Closest.mesh_point_aabb(m, Point(0.0, 0.0, 0.0))
+    MINI_CHECK(d == float('inf'))
+
+
+@MINI_TEST("AABBTree", "Build Single")
+def test_aabbtree_build_single():
+    from session_py import Closest
+    from session_py import Mesh
+    from session_py import Point
+
+    m = Mesh()
+    vk0 = m.add_vertex(Point(0.0, 0.0, 0.0))
+    vk1 = m.add_vertex(Point(1.0, 0.0, 0.0))
+    vk2 = m.add_vertex(Point(0.0, 1.0, 0.0))
+    m.add_face([vk0, vk1, vk2])
+    cp, fk, d = Closest.mesh_point_aabb(m, Point(0.0, 0.0, 1.0))
+    MINI_CHECK(d > 0.0)
+    MINI_CHECK(TOLERANCE.is_close(d, 1.0))
+
+
+@MINI_TEST("AABBTree", "Build Multiple")
+def test_aabbtree_build_multiple():
+    from session_py import Closest
+    from session_py import Mesh
+    from session_py import Point
+
+    m = Mesh()
+    vk0 = m.add_vertex(Point(0.0, 0.0, 0.0))
+    vk1 = m.add_vertex(Point(1.0, 0.0, 0.0))
+    vk2 = m.add_vertex(Point(0.0, 1.0, 0.0))
+    vk3 = m.add_vertex(Point(5.0, 0.0, 0.0))
+    vk4 = m.add_vertex(Point(6.0, 0.0, 0.0))
+    vk5 = m.add_vertex(Point(5.0, 1.0, 0.0))
+    vk6 = m.add_vertex(Point(10.0, 0.0, 0.0))
+    vk7 = m.add_vertex(Point(11.0, 0.0, 0.0))
+    vk8 = m.add_vertex(Point(10.0, 1.0, 0.0))
+    m.add_face([vk0, vk1, vk2])
+    m.add_face([vk3, vk4, vk5])
+    m.add_face([vk6, vk7, vk8])
+    cp, fk, d = Closest.mesh_point_aabb(m, Point(0.5, 0.0, 0.0))
+    MINI_CHECK(d < 0.5)
+
+
+@MINI_TEST("AABBTree", "Node Count")
+def test_aabbtree_node_count():
+    from session_py import Closest
+    from session_py import Mesh
+    from session_py import Point
+
+    m = Mesh()
+    vkeys = []
+    for i in range(100):
+        vkeys.append(m.add_vertex(Point(float(i), 0.0, 0.0)))
+        vkeys.append(m.add_vertex(Point(float(i) + 0.5, 0.5, 0.0)))
+        vkeys.append(m.add_vertex(Point(float(i), 0.5, 0.0)))
+    for i in range(100):
+        m.add_face([vkeys[i*3], vkeys[i*3+1], vkeys[i*3+2]])
+    cp, fk, d = Closest.mesh_point_aabb(m, Point(50.0, 0.0, 0.0))
+    MINI_CHECK(d < 0.5)
+
+
+@MINI_TEST("AABBTree", "Mesh Point Aabb")
+def test_aabbtree_mesh_point_aabb():
+    from session_py import Closest
+    from session_py import Primitives
+    from session_py import Point
+
+    m = Primitives.cube(2.0)
+    cp1, fk1, d1 = Closest.mesh_point_aabb(m, Point(0.0, 0.0, 2.0))
+    MINI_CHECK(TOLERANCE.is_close(cp1[2], 1.0))
+    MINI_CHECK(TOLERANCE.is_close(d1, 1.0))
+    cp2, fk2, d2 = Closest.mesh_point_aabb(m, Point(1.0, 1.0, 1.0))
+    MINI_CHECK(TOLERANCE.is_close(d2, 0.0))
+
+
+@MINI_TEST("AABBTree", "Mesh Point Aabb Matches Bvh")
+def test_aabbtree_mesh_point_aabb_matches_bvh():
+    from session_py import Closest
+    from session_py import Primitives
+    from session_py import Point
+
+    m = Primitives.cube(2.0)
+    tp = Point(0.3, 0.7, 1.5)
+    cp_bvh, fk_bvh, d_bvh = Closest.mesh_point(m, tp)
+    cp_aabb, fk_aabb, d_aabb = Closest.mesh_point_aabb(m, tp)
+    MINI_CHECK(TOLERANCE.is_close(d_bvh, d_aabb))
+    MINI_CHECK(TOLERANCE.is_close(cp_bvh[0], cp_aabb[0]))
+    MINI_CHECK(TOLERANCE.is_close(cp_bvh[1], cp_aabb[1]))
+    MINI_CHECK(TOLERANCE.is_close(cp_bvh[2], cp_aabb[2]))
+
+
+if __name__ == "__main__":
+    run_all(language="python")