1 | // Copyright 2009-2021 Intel Corporation |
2 | // SPDX-License-Identifier: Apache-2.0 |
3 | |
4 | #pragma once |
5 | |
6 | #include "primitive.h" |
7 | |
8 | namespace embree |
9 | { |
10 | /* Precalculated representation for M triangles. Stores for each |
11 | triangle a base vertex, two edges, and the geometry normal to |
12 | speed up intersection calculations */ |
13 | template<int M> |
14 | struct TriangleM |
15 | { |
16 | public: |
17 | struct Type : public PrimitiveType |
18 | { |
19 | const char* name() const; |
20 | size_t sizeActive(const char* This) const; |
21 | size_t sizeTotal(const char* This) const; |
22 | size_t getBytes(const char* This) const; |
23 | }; |
24 | static Type type; |
25 | |
26 | public: |
27 | |
28 | /* Returns maximum number of stored triangles */ |
29 | static __forceinline size_t max_size() { return M; } |
30 | |
31 | /* Returns required number of primitive blocks for N primitives */ |
32 | static __forceinline size_t blocks(size_t N) { return (N+max_size()-1)/max_size(); } |
33 | |
34 | public: |
35 | |
36 | /* Default constructor */ |
37 | __forceinline TriangleM() {} |
38 | |
39 | /* Construction from vertices and IDs */ |
40 | __forceinline TriangleM(const Vec3vf<M>& v0, const Vec3vf<M>& v1, const Vec3vf<M>& v2, const vuint<M>& geomIDs, const vuint<M>& primIDs) |
41 | : v0(v0), e1(v0-v1), e2(v2-v0), geomIDs(geomIDs), primIDs(primIDs) {} |
42 | |
43 | /* Returns a mask that tells which triangles are valid */ |
44 | __forceinline vbool<M> valid() const { return geomIDs != vuint<M>(-1); } |
45 | |
46 | /* Returns true if the specified triangle is valid */ |
47 | __forceinline bool valid(const size_t i) const { assert(i<M); return geomIDs[i] != -1; } |
48 | |
49 | /* Returns the number of stored triangles */ |
50 | __forceinline size_t size() const { return bsf(~movemask(valid())); } |
51 | |
52 | /* Returns the geometry IDs */ |
53 | __forceinline vuint<M>& geomID() { return geomIDs; } |
54 | __forceinline const vuint<M>& geomID() const { return geomIDs; } |
55 | __forceinline unsigned int geomID(const size_t i) const { assert(i<M); return geomIDs[i]; } |
56 | |
57 | /* Returns the primitive IDs */ |
58 | __forceinline vuint<M>& primID() { return primIDs; } |
59 | __forceinline const vuint<M>& primID() const { return primIDs; } |
60 | __forceinline unsigned int primID(const size_t i) const { assert(i<M); return primIDs[i]; } |
61 | |
62 | /* Calculate the bounds of the triangle */ |
63 | __forceinline BBox3fa bounds() const |
64 | { |
65 | Vec3vf<M> p0 = v0; |
66 | Vec3vf<M> p1 = v0-e1; |
67 | Vec3vf<M> p2 = v0+e2; |
68 | Vec3vf<M> lower = min(p0,p1,p2); |
69 | Vec3vf<M> upper = max(p0,p1,p2); |
70 | vbool<M> mask = valid(); |
71 | lower.x = select(mask,lower.x,vfloat<M>(pos_inf)); |
72 | lower.y = select(mask,lower.y,vfloat<M>(pos_inf)); |
73 | lower.z = select(mask,lower.z,vfloat<M>(pos_inf)); |
74 | upper.x = select(mask,upper.x,vfloat<M>(neg_inf)); |
75 | upper.y = select(mask,upper.y,vfloat<M>(neg_inf)); |
76 | upper.z = select(mask,upper.z,vfloat<M>(neg_inf)); |
77 | return BBox3fa(Vec3fa(reduce_min(lower.x),reduce_min(lower.y),reduce_min(lower.z)), |
78 | Vec3fa(reduce_max(upper.x),reduce_max(upper.y),reduce_max(upper.z))); |
79 | } |
80 | |
81 | /* Non temporal store */ |
82 | __forceinline static void store_nt(TriangleM* dst, const TriangleM& src) |
83 | { |
84 | vfloat<M>::store_nt(&dst->v0.x,src.v0.x); |
85 | vfloat<M>::store_nt(&dst->v0.y,src.v0.y); |
86 | vfloat<M>::store_nt(&dst->v0.z,src.v0.z); |
87 | vfloat<M>::store_nt(&dst->e1.x,src.e1.x); |
88 | vfloat<M>::store_nt(&dst->e1.y,src.e1.y); |
89 | vfloat<M>::store_nt(&dst->e1.z,src.e1.z); |
90 | vfloat<M>::store_nt(&dst->e2.x,src.e2.x); |
91 | vfloat<M>::store_nt(&dst->e2.y,src.e2.y); |
92 | vfloat<M>::store_nt(&dst->e2.z,src.e2.z); |
93 | vuint<M>::store_nt(&dst->geomIDs,src.geomIDs); |
94 | vuint<M>::store_nt(&dst->primIDs,src.primIDs); |
95 | } |
96 | |
97 | /* Fill triangle from triangle list */ |
98 | __forceinline void fill(const PrimRef* prims, size_t& begin, size_t end, Scene* scene) |
99 | { |
100 | vuint<M> vgeomID = -1, vprimID = -1; |
101 | Vec3vf<M> v0 = zero, v1 = zero, v2 = zero; |
102 | |
103 | for (size_t i=0; i<M && begin<end; i++, begin++) |
104 | { |
105 | const PrimRef& prim = prims[begin]; |
106 | const unsigned geomID = prim.geomID(); |
107 | const unsigned primID = prim.primID(); |
108 | const TriangleMesh* __restrict__ const mesh = scene->get<TriangleMesh>(i: geomID); |
109 | const TriangleMesh::Triangle& tri = mesh->triangle(i: primID); |
110 | const Vec3fa& p0 = mesh->vertex(i: tri.v[0]); |
111 | const Vec3fa& p1 = mesh->vertex(i: tri.v[1]); |
112 | const Vec3fa& p2 = mesh->vertex(i: tri.v[2]); |
113 | vgeomID [i] = geomID; |
114 | vprimID [i] = primID; |
115 | v0.x[i] = p0.x; v0.y[i] = p0.y; v0.z[i] = p0.z; |
116 | v1.x[i] = p1.x; v1.y[i] = p1.y; v1.z[i] = p1.z; |
117 | v2.x[i] = p2.x; v2.y[i] = p2.y; v2.z[i] = p2.z; |
118 | } |
119 | TriangleM::store_nt(dst: this,src: TriangleM(v0,v1,v2,vgeomID,vprimID)); |
120 | } |
121 | |
122 | /* Updates the primitive */ |
123 | __forceinline BBox3fa update(TriangleMesh* mesh) |
124 | { |
125 | BBox3fa bounds = empty; |
126 | vuint<M> vgeomID = -1, vprimID = -1; |
127 | Vec3vf<M> v0 = zero, v1 = zero, v2 = zero; |
128 | |
129 | for (size_t i=0; i<M; i++) |
130 | { |
131 | if (unlikely(geomID(i) == -1)) break; |
132 | const unsigned geomId = geomID(i); |
133 | const unsigned primId = primID(i); |
134 | const TriangleMesh::Triangle& tri = mesh->triangle(i: primId); |
135 | const Vec3fa p0 = mesh->vertex(i: tri.v[0]); |
136 | const Vec3fa p1 = mesh->vertex(i: tri.v[1]); |
137 | const Vec3fa p2 = mesh->vertex(i: tri.v[2]); |
138 | bounds.extend(other: merge(a: BBox3fa(p0),b: BBox3fa(p1),c: BBox3fa(p2))); |
139 | vgeomID [i] = geomId; |
140 | vprimID [i] = primId; |
141 | v0.x[i] = p0.x; v0.y[i] = p0.y; v0.z[i] = p0.z; |
142 | v1.x[i] = p1.x; v1.y[i] = p1.y; v1.z[i] = p1.z; |
143 | v2.x[i] = p2.x; v2.y[i] = p2.y; v2.z[i] = p2.z; |
144 | } |
145 | TriangleM::store_nt(dst: this,src: TriangleM(v0,v1,v2,vgeomID,vprimID)); |
146 | return bounds; |
147 | } |
148 | |
149 | public: |
150 | Vec3vf<M> v0; // base vertex of the triangles |
151 | Vec3vf<M> e1; // 1st edge of the triangles (v0-v1) |
152 | Vec3vf<M> e2; // 2nd edge of the triangles (v2-v0) |
153 | private: |
154 | vuint<M> geomIDs; // geometry IDs |
155 | vuint<M> primIDs; // primitive IDs |
156 | }; |
157 | |
158 | template<int M> |
159 | typename TriangleM<M>::Type TriangleM<M>::type; |
160 | |
161 | typedef TriangleM<4> Triangle4; |
162 | } |
163 | |