summaryrefslogtreecommitdiff
path: root/thirdparty/embree-aarch64/kernels/bvh/bvh_node_obb.h
blob: fa7cc0821126e9f4ea025eff2f6fdd86e266d7b8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
// Copyright 2009-2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0

#pragma once

#include "bvh_node_base.h"

namespace embree
{
  /*! Node with unaligned bounds */
  template<typename NodeRef, int N>
    struct OBBNode_t : public BaseNode_t<NodeRef, N>
  {
    using BaseNode_t<NodeRef,N>::children;
    
    struct Create
    {
      __forceinline NodeRef operator() (const FastAllocator::CachedAllocator& alloc) const
      {
        OBBNode_t* node = (OBBNode_t*) alloc.malloc0(sizeof(OBBNode_t),NodeRef::byteNodeAlignment); node->clear();
        return NodeRef::encodeNode(node);
      }
    };
    
    struct Set
    {
      __forceinline void operator() (NodeRef node, size_t i, NodeRef child, const OBBox3fa& bounds) const {
        node.ungetAABBNode()->setRef(i,child);
        node.ungetAABBNode()->setBounds(i,bounds);
      }
    };
    
    /*! Clears the node. */
    __forceinline void clear()
    {
      naabb.l.vx = Vec3fa(nan);
      naabb.l.vy = Vec3fa(nan);
      naabb.l.vz = Vec3fa(nan);
      naabb.p    = Vec3fa(nan);
      BaseNode_t<NodeRef,N>::clear();
    }
    
    /*! Sets bounding box. */
    __forceinline void setBounds(size_t i, const OBBox3fa& b)
    {
      assert(i < N);
      
      AffineSpace3fa space = b.space;
      space.p -= b.bounds.lower;
      space = AffineSpace3fa::scale(1.0f/max(Vec3fa(1E-19f),b.bounds.upper-b.bounds.lower))*space;
      
      naabb.l.vx.x[i] = space.l.vx.x;
      naabb.l.vx.y[i] = space.l.vx.y;
      naabb.l.vx.z[i] = space.l.vx.z;
      
      naabb.l.vy.x[i] = space.l.vy.x;
      naabb.l.vy.y[i] = space.l.vy.y;
      naabb.l.vy.z[i] = space.l.vy.z;
      
      naabb.l.vz.x[i] = space.l.vz.x;
      naabb.l.vz.y[i] = space.l.vz.y;
      naabb.l.vz.z[i] = space.l.vz.z;
      
      naabb.p.x[i] = space.p.x;
      naabb.p.y[i] = space.p.y;
      naabb.p.z[i] = space.p.z;
    }
    
    /*! Sets ID of child. */
    __forceinline void setRef(size_t i, const NodeRef& ref) {
      assert(i < N);
      children[i] = ref;
    }
    
    /*! Returns the extent of the bounds of the ith child */
    __forceinline Vec3fa extent(size_t i) const {
      assert(i<N);
      const Vec3fa vx(naabb.l.vx.x[i],naabb.l.vx.y[i],naabb.l.vx.z[i]);
      const Vec3fa vy(naabb.l.vy.x[i],naabb.l.vy.y[i],naabb.l.vy.z[i]);
      const Vec3fa vz(naabb.l.vz.x[i],naabb.l.vz.y[i],naabb.l.vz.z[i]);
      return rsqrt(vx*vx + vy*vy + vz*vz);
    }
    
    /*! Returns reference to specified child */
    __forceinline       NodeRef& child(size_t i)       { assert(i<N); return children[i]; }
    __forceinline const NodeRef& child(size_t i) const { assert(i<N); return children[i]; }
    
    /*! output operator */
    friend embree_ostream operator<<(embree_ostream o, const OBBNode_t& n)
    {
      o << "UnAABBNode { " << n.naabb << " } " << embree_endl;
      return o;
    }
    
  public:
    AffineSpace3vf<N> naabb;   //!< non-axis aligned bounding boxes (bounds are [0,1] in specified space)
  };
}