4 using double4 = Point4D;
5 using double2 = Point2D;
6 using int2 = std::pair<int, int>;
33 const int nnodes = 2 * (int)
object.size() - 1;
34 const int nbodies = (int)
object.size();
35#pragma omp for schedule(dynamic,5)
36 for (
int k = nbodies; k < nnodes; ++k)
56 int srt = indexSort[(nnodes - 1) - k];
57 int2 chdPair = child[srt];
59 for (i = 0; i < 2; i++) {
60 int chd = i * chdPair.second + (1 - i) * chdPair.first;
62 ch = (chd >= nbodies) ? (chd - nbodies) : ((nnodes - 1) - indexSortT[chd]);
64 if ((chd >= nbodies) || (mass[nnodes - 1 - ch] >= 0))
75 for (i = 0; i < 2; i++)
77 const int chd = i * chdPair.second + (1 - i) * chdPair.first;
81 const int sortedBody = mortonCodesIdx[ch];
83 double4 xyAB = gabForLeaves[sortedBody];
85 ::fmin(xyAB[0], xyAB[2]), ::fmin(xyAB[1], xyAB[3]),
86 ::fmax(xyAB[0], xyAB[2]), ::fmax(xyAB[1], xyAB[3])
93 lu[i] = lowerupper[chd];
98 lowerupper[srt] = double4{
99 ::fmin(lu[0][0], lu[1][0]), ::fmin(lu[0][1], lu[1][1]),
100 ::fmax(lu[0][2], lu[1][2]), ::fmax(lu[0][3], lu[1][3])
103 cen = center[srt] = double2{ 0.5 * (lowerupper[srt][0] + lowerupper[srt][2]),
104 0.5 * (lowerupper[srt][1] + lowerupper[srt][3]) };
106 const double2 zero = { 0.0, 0.0 };
107 double2 momh0 = zero;
108 double2 momh1 = zero;
109 double2 momh2 = zero;
110 double2 momh3 = zero;
111 double2 momh4 = zero;
112 double2 momh5 = zero;
113 double2 momh6 = zero;
114 double2 momh7 = zero;
115 double2 momh8 = zero;
116 double2 momh9 = zero;
117 double2 momh10 = zero;
118 double2 momh11 = zero;
120 for (i = 0; i < 2; i++)
122 const int chd = i * chdPair.second + (1 - i) * chdPair.first;
126 const int sortedBody = mortonCodesIdx[ch];
130 mom0 = double2{ gamma[sortedBody], 0.0 };
132mom1 = mom2 = mom3 = mom4 = mom5 = mom6 = mom7 = mom8 = mom9 = mom10 = mom11 = double2{ 0.0, 0.0 };
133 double2 pos =
object[sortedBody];
146 const int srtT = indexSortT[chd];
147 ch = (nnodes - 1) - srtT;
151 mom0 = double2{ moms[nch + 0][0], (double)0 };
152 mom1 = moms[nch + 1];
153 mom2 = moms[nch + 2];
154 mom3 = moms[nch + 3];
155 mom4 = moms[nch + 4];
156 mom5 = moms[nch + 5];
157 mom6 = moms[nch + 6];
158 mom7 = moms[nch + 7];
159 mom8 = moms[nch + 8];
160 mom9 = moms[nch + 9];
161 mom10 = moms[nch + 10];
162 mom11 = moms[nch + 11];
163 dr = center[chd] - cen;
183 momh1 += multz(mom0, z);
184 momh2 += 2 * multz(mom1, z);
185 momh3 += 3 * multz(mom2, z);
186 momh4 += 4 * multz(mom3, z);
187 momh5 += 5 * multz(mom4, z);
188 momh6 += 6 * multz(mom5, z);
189 momh7 += 7 * multz(mom6, z);
190 momh8 += 8 * multz(mom7, z);
191 momh9 += 9 * multz(mom8, z);
192 momh10 += 10 * multz(mom9, z);
193 momh11 += 11 * multz(mom10, z);
197 momh2 += multz(mom0, z);
198 momh3 += 3 * multz(mom1, z);
199 momh4 += 6 * multz(mom2, z);
200 momh5 += 10 * multz(mom3, z);
201 momh6 += 15 * multz(mom4, z);
202 momh7 += 21 * multz(mom5, z);
203 momh8 += 28 * multz(mom6, z);
204 momh9 += 36 * multz(mom7, z);
205 momh10 += 45 * multz(mom8, z);
206 momh11 += 55 * multz(mom9, z);
210 momh3 += multz(mom0, z);
211 momh4 += 4 * multz(mom1, z);
212 momh5 += 10 * multz(mom2, z);
213 momh6 += 20 * multz(mom3, z);
214 momh7 += 35 * multz(mom4, z);
215 momh8 += 56 * multz(mom5, z);
216 momh9 += 84 * multz(mom6, z);
217 momh10 += 120 * multz(mom7, z);
218 momh11 += 165 * multz(mom8, z);
222 momh4 += multz(mom0, z);
223 momh5 += 5 * multz(mom1, z);
224 momh6 += 15 * multz(mom2, z);
225 momh7 += 35 * multz(mom3, z);
226 momh8 += 70 * multz(mom4, z);
227 momh9 += 126 * multz(mom5, z);
228 momh10 += 210 * multz(mom6, z);
229 momh11 += 330 * multz(mom7, z);
233 momh5 += multz(mom0, z);
234 momh6 += 6 * multz(mom1, z);
235 momh7 += 21 * multz(mom2, z);
236 momh8 += 56 * multz(mom3, z);
237 momh9 += 126 * multz(mom4, z);
238 momh10 += 252 * multz(mom5, z);
239 momh11 += 462 * multz(mom6, z);
243 momh6 += multz(mom0, z);
244 momh7 += 7 * multz(mom1, z);
245 momh8 += 28 * multz(mom2, z);
246 momh9 += 84 * multz(mom3, z);
247 momh10 += 210 * multz(mom4, z);
248 momh11 += 462 * multz(mom5, z);
252 momh7 += multz(mom0, z);
253 momh8 += 8 * multz(mom1, z);
254 momh9 += 36 * multz(mom2, z);
255 momh10 += 120 * multz(mom3, z);
256 momh11 += 330 * multz(mom4, z);
260 momh8 += multz(mom0, z);
261 momh9 += 9 * multz(mom1, z);
262 momh10 += 45 * multz(mom2, z);
263 momh11 += 165 * multz(mom3, z);
267 momh9 += multz(mom0, z);
268 momh10 += 10 * multz(mom1, z);
269 momh11 += 55 * multz(mom2, z);
273 momh10 += multz(mom0, z);
274 momh11 += 11 * multz(mom1, z);
278 momh11 += multz(mom0, z);
283 moms[kch + 0] = momh0;
284 moms[kch + 1] = momh1;
285 moms[kch + 2] = momh2;
286 moms[kch + 3] = momh3;
287 moms[kch + 4] = momh4;
288 moms[kch + 5] = momh5;
289 moms[kch + 6] = momh6;
290 moms[kch + 7] = momh7;
291 moms[kch + 8] = momh8;
292 moms[kch + 9] = momh9;
293 moms[kch + 10] = momh10;
294 moms[kch + 11] = momh11;
302 mass[nnodes - 1 - k] = cm;