1 | // Copyright 2009-2021 Intel Corporation |
2 | // SPDX-License-Identifier: Apache-2.0 |
3 | |
4 | #pragma once |
5 | |
6 | #include "bvh_node_base.h" |
7 | |
8 | namespace embree |
9 | { |
10 | /*! Node with unaligned bounds */ |
11 | template<typename NodeRef, int N> |
12 | struct OBBNode_t : public BaseNode_t<NodeRef, N> |
13 | { |
14 | using BaseNode_t<NodeRef,N>::children; |
15 | |
16 | struct Create |
17 | { |
18 | __forceinline NodeRef operator() (const FastAllocator::CachedAllocator& alloc) const |
19 | { |
20 | OBBNode_t* node = (OBBNode_t*) alloc.malloc0(sizeof(OBBNode_t),NodeRef::byteNodeAlignment); node->clear(); |
21 | return NodeRef::encodeNode(node); |
22 | } |
23 | }; |
24 | |
25 | struct Set |
26 | { |
27 | __forceinline void operator() (NodeRef node, size_t i, NodeRef child, const OBBox3fa& bounds) const { |
28 | node.ungetAABBNode()->setRef(i,child); |
29 | node.ungetAABBNode()->setBounds(i,bounds); |
30 | } |
31 | }; |
32 | |
33 | /*! Clears the node. */ |
34 | __forceinline void clear() |
35 | { |
36 | naabb.l.vx = Vec3fa(nan); |
37 | naabb.l.vy = Vec3fa(nan); |
38 | naabb.l.vz = Vec3fa(nan); |
39 | naabb.p = Vec3fa(nan); |
40 | BaseNode_t<NodeRef,N>::clear(); |
41 | } |
42 | |
43 | /*! Sets bounding box. */ |
44 | __forceinline void setBounds(size_t i, const OBBox3fa& b) |
45 | { |
46 | assert(i < N); |
47 | |
48 | AffineSpace3fa space = b.space; |
49 | space.p -= b.bounds.lower; |
50 | space = AffineSpace3fa::scale(1.0f/max(Vec3fa(1E-19f),b.bounds.upper-b.bounds.lower))*space; |
51 | |
52 | naabb.l.vx.x[i] = space.l.vx.x; |
53 | naabb.l.vx.y[i] = space.l.vx.y; |
54 | naabb.l.vx.z[i] = space.l.vx.z; |
55 | |
56 | naabb.l.vy.x[i] = space.l.vy.x; |
57 | naabb.l.vy.y[i] = space.l.vy.y; |
58 | naabb.l.vy.z[i] = space.l.vy.z; |
59 | |
60 | naabb.l.vz.x[i] = space.l.vz.x; |
61 | naabb.l.vz.y[i] = space.l.vz.y; |
62 | naabb.l.vz.z[i] = space.l.vz.z; |
63 | |
64 | naabb.p.x[i] = space.p.x; |
65 | naabb.p.y[i] = space.p.y; |
66 | naabb.p.z[i] = space.p.z; |
67 | } |
68 | |
69 | /*! Sets ID of child. */ |
70 | __forceinline void setRef(size_t i, const NodeRef& ref) { |
71 | assert(i < N); |
72 | children[i] = ref; |
73 | } |
74 | |
75 | /*! Returns the extent of the bounds of the ith child */ |
76 | __forceinline Vec3fa extent(size_t i) const { |
77 | assert(i<N); |
78 | const Vec3fa vx(naabb.l.vx.x[i],naabb.l.vx.y[i],naabb.l.vx.z[i]); |
79 | const Vec3fa vy(naabb.l.vy.x[i],naabb.l.vy.y[i],naabb.l.vy.z[i]); |
80 | const Vec3fa vz(naabb.l.vz.x[i],naabb.l.vz.y[i],naabb.l.vz.z[i]); |
81 | return rsqrt(vx*vx + vy*vy + vz*vz); |
82 | } |
83 | |
84 | /*! Returns reference to specified child */ |
85 | __forceinline NodeRef& child(size_t i) { assert(i<N); return children[i]; } |
86 | __forceinline const NodeRef& child(size_t i) const { assert(i<N); return children[i]; } |
87 | |
88 | /*! output operator */ |
89 | friend embree_ostream operator<<(embree_ostream o, const OBBNode_t& n) |
90 | { |
91 | o << "UnAABBNode { " << n.naabb << " } " << embree_endl; |
92 | return o; |
93 | } |
94 | |
95 | public: |
96 | AffineSpace3vf<N> naabb; //!< non-axis aligned bounding boxes (bounds are [0,1] in specified space) |
97 | }; |
98 | } |
99 | |