1// Copyright 2009-2021 Intel Corporation
2// SPDX-License-Identifier: Apache-2.0
3
4#include "acceln.h"
5#include "ray.h"
6#include "../../include/embree3/rtcore_ray.h"
7#include "../../common/algorithms/parallel_for.h"
8
9namespace embree
10{
11 AccelN::AccelN()
12 : Accel(AccelData::TY_ACCELN), accels() {}
13
14 AccelN::~AccelN()
15 {
16 for (size_t i=0; i<accels.size(); i++)
17 delete accels[i];
18 }
19
20 void AccelN::accels_add(Accel* accel)
21 {
22 assert(accel);
23 accels.push_back(accel);
24 }
25
26 void AccelN::accels_init()
27 {
28 for (size_t i=0; i<accels.size(); i++)
29 delete accels[i];
30
31 accels.clear();
32 }
33
34 bool AccelN::pointQuery (Accel::Intersectors* This_in, PointQuery* query, PointQueryContext* context)
35 {
36 bool changed = false;
37 AccelN* This = (AccelN*)This_in->ptr;
38 for (size_t i=0; i<This->accels.size(); i++)
39 if (!This->accels[i]->isEmpty())
40 changed |= This->accels[i]->intersectors.pointQuery(query,context);
41 return changed;
42 }
43
44 void AccelN::intersect (Accel::Intersectors* This_in, RTCRayHit& ray, IntersectContext* context)
45 {
46 AccelN* This = (AccelN*)This_in->ptr;
47 for (size_t i=0; i<This->accels.size(); i++)
48 if (!This->accels[i]->isEmpty())
49 This->accels[i]->intersectors.intersect(ray,context);
50 }
51
52 void AccelN::intersect4 (const void* valid, Accel::Intersectors* This_in, RTCRayHit4& ray, IntersectContext* context)
53 {
54 AccelN* This = (AccelN*)This_in->ptr;
55 for (size_t i=0; i<This->accels.size(); i++)
56 if (!This->accels[i]->isEmpty())
57 This->accels[i]->intersectors.intersect4(valid,ray,context);
58 }
59
60 void AccelN::intersect8 (const void* valid, Accel::Intersectors* This_in, RTCRayHit8& ray, IntersectContext* context)
61 {
62 AccelN* This = (AccelN*)This_in->ptr;
63 for (size_t i=0; i<This->accels.size(); i++)
64 if (!This->accels[i]->isEmpty())
65 This->accels[i]->intersectors.intersect8(valid,ray,context);
66 }
67
68 void AccelN::intersect16 (const void* valid, Accel::Intersectors* This_in, RTCRayHit16& ray, IntersectContext* context)
69 {
70 AccelN* This = (AccelN*)This_in->ptr;
71 for (size_t i=0; i<This->accels.size(); i++)
72 if (!This->accels[i]->isEmpty())
73 This->accels[i]->intersectors.intersect16(valid,ray,context);
74 }
75
76 void AccelN::intersectN (Accel::Intersectors* This_in, RTCRayHitN** ray, const size_t N, IntersectContext* context)
77 {
78 AccelN* This = (AccelN*)This_in->ptr;
79 for (size_t i=0; i<This->accels.size(); i++)
80 if (!This->accels[i]->isEmpty())
81 This->accels[i]->intersectors.intersectN(ray,N,context);
82 }
83
84 void AccelN::occluded (Accel::Intersectors* This_in, RTCRay& ray, IntersectContext* context)
85 {
86 AccelN* This = (AccelN*)This_in->ptr;
87 for (size_t i=0; i<This->accels.size(); i++) {
88 if (This->accels[i]->isEmpty()) continue;
89 This->accels[i]->intersectors.occluded(ray,context);
90 if (ray.tfar < 0.0f) break;
91 }
92 }
93
94 void AccelN::occluded4 (const void* valid, Accel::Intersectors* This_in, RTCRay4& ray, IntersectContext* context)
95 {
96 AccelN* This = (AccelN*)This_in->ptr;
97 for (size_t i=0; i<This->accels.size(); i++) {
98 if (This->accels[i]->isEmpty()) continue;
99 This->accels[i]->intersectors.occluded4(valid,ray,context);
100#if defined(__SSE2__) || defined(__ARM_NEON)
101 vbool4 valid0 = asBool(((vint4*)valid)[0]);
102 vbool4 hit0 = ((vfloat4*)ray.tfar)[0] >= vfloat4(zero);
103 if (unlikely(none(valid0 & hit0))) break;
104#endif
105 }
106 }
107
108 void AccelN::occluded8 (const void* valid, Accel::Intersectors* This_in, RTCRay8& ray, IntersectContext* context)
109 {
110 AccelN* This = (AccelN*)This_in->ptr;
111 for (size_t i=0; i<This->accels.size(); i++) {
112 if (This->accels[i]->isEmpty()) continue;
113 This->accels[i]->intersectors.occluded8(valid,ray,context);
114#if defined(__SSE2__) || defined(__ARM_NEON) // FIXME: use higher ISA
115 vbool4 valid0 = asBool(((vint4*)valid)[0]);
116 vbool4 hit0 = ((vfloat4*)ray.tfar)[0] >= vfloat4(zero);
117 vbool4 valid1 = asBool(((vint4*)valid)[1]);
118 vbool4 hit1 = ((vfloat4*)ray.tfar)[1] >= vfloat4(zero);
119 if (unlikely((none((valid0 & hit0) | (valid1 & hit1))))) break;
120#endif
121 }
122 }
123
124 void AccelN::occluded16 (const void* valid, Accel::Intersectors* This_in, RTCRay16& ray, IntersectContext* context)
125 {
126 AccelN* This = (AccelN*)This_in->ptr;
127 for (size_t i=0; i<This->accels.size(); i++) {
128 if (This->accels[i]->isEmpty()) continue;
129 This->accels[i]->intersectors.occluded16(valid,ray,context);
130#if defined(__SSE2__) || defined(__ARM_NEON) // FIXME: use higher ISA
131 vbool4 valid0 = asBool(((vint4*)valid)[0]);
132 vbool4 hit0 = ((vfloat4*)ray.tfar)[0] >= vfloat4(zero);
133 vbool4 valid1 = asBool(((vint4*)valid)[1]);
134 vbool4 hit1 = ((vfloat4*)ray.tfar)[1] >= vfloat4(zero);
135 vbool4 valid2 = asBool(((vint4*)valid)[2]);
136 vbool4 hit2 = ((vfloat4*)ray.tfar)[2] >= vfloat4(zero);
137 vbool4 valid3 = asBool(((vint4*)valid)[3]);
138 vbool4 hit3 = ((vfloat4*)ray.tfar)[3] >= vfloat4(zero);
139 if (unlikely((none((valid0 & hit0) | (valid1 & hit1) | (valid2 & hit2) | (valid3 & hit3))))) break;
140#endif
141 }
142 }
143
144 void AccelN::occludedN (Accel::Intersectors* This_in, RTCRayN** ray, const size_t N, IntersectContext* context)
145 {
146 AccelN* This = (AccelN*)This_in->ptr;
147 size_t M = N;
148 for (size_t i=0; i<This->accels.size(); i++)
149 if (!This->accels[i]->isEmpty())
150 This->accels[i]->intersectors.occludedN(ray,M,context);
151 }
152
153 void AccelN::accels_print(size_t ident)
154 {
155 for (size_t i=0; i<accels.size(); i++)
156 {
157 for (size_t j=0; j<ident; j++) std::cout << " ";
158 std::cout << "accels[" << i << "]" << std::endl;
159 accels[i]->intersectors.print(ident+2);
160 }
161 }
162
163 void AccelN::accels_immutable()
164 {
165 for (size_t i=0; i<accels.size(); i++)
166 accels[i]->immutable();
167 }
168
169 void AccelN::accels_build ()
170 {
171 /* reduce memory consumption */
172 accels.shrink_to_fit();
173
174 /* build all acceleration structures in parallel */
175 parallel_for (accels.size(), [&] (size_t i) {
176 accels[i]->build();
177 });
178
179 /* create list of non-empty acceleration structures */
180 bool valid1 = true;
181 bool valid4 = true;
182 bool valid8 = true;
183 bool valid16 = true;
184 for (size_t i=0; i<accels.size(); i++) {
185 valid1 &= (bool) accels[i]->intersectors.intersector1;
186 valid4 &= (bool) accels[i]->intersectors.intersector4;
187 valid8 &= (bool) accels[i]->intersectors.intersector8;
188 valid16 &= (bool) accels[i]->intersectors.intersector16;
189 }
190
191 if (accels.size() == 1) {
192 type = accels[0]->type; // FIXME: should just assign entire Accel
193 bounds = accels[0]->bounds;
194 intersectors = accels[0]->intersectors;
195 }
196 else
197 {
198 type = AccelData::TY_ACCELN;
199 intersectors.ptr = this;
200 intersectors.intersector1 = Intersector1(&intersect,&occluded,&pointQuery,valid1 ? "AccelN::intersector1": nullptr);
201 intersectors.intersector4 = Intersector4(&intersect4,&occluded4,valid4 ? "AccelN::intersector4" : nullptr);
202 intersectors.intersector8 = Intersector8(&intersect8,&occluded8,valid8 ? "AccelN::intersector8" : nullptr);
203 intersectors.intersector16 = Intersector16(&intersect16,&occluded16,valid16 ? "AccelN::intersector16": nullptr);
204 intersectors.intersectorN = IntersectorN(&intersectN,&occludedN,"AccelN::intersectorN");
205
206 /*! calculate bounds */
207 bounds = empty;
208 for (size_t i=0; i<accels.size(); i++)
209 bounds.extend(accels[i]->bounds);
210 }
211 }
212
213 void AccelN::accels_select(bool filter)
214 {
215 for (size_t i=0; i<accels.size(); i++)
216 accels[i]->intersectors.select(filter);
217 }
218
219 void AccelN::accels_deleteGeometry(size_t geomID)
220 {
221 for (size_t i=0; i<accels.size(); i++)
222 accels[i]->deleteGeometry(geomID);
223 }
224
225 void AccelN::accels_clear()
226 {
227 for (size_t i=0; i<accels.size(); i++) {
228 accels[i]->clear();
229 }
230 }
231}
232
233