4 using double4 = Point4D;
5 using double2 = Point2D;
6 using int2 = std::pair<int, int>;
29 const int nnodes = 2 * (int)
object.size() - 1;
30 const int nbodies = (int)
object.size();
31#pragma omp for schedule(dynamic,5)
32 for (
int k = nbodies; k < nnodes; ++k)
52 int srt = indexSort[(nnodes - 1) - k];
53 int2 chdPair = child[srt];
55 for (i = 0; i < 2; i++) {
56 int chd = i * chdPair.second + (1 - i) * chdPair.first;
58 ch = (chd >= nbodies) ? (chd - nbodies) : ((nnodes - 1) - indexSortT[chd]);
60 if ((chd >= nbodies) || (mass[nnodes - 1 - ch] >= 0))
71 for (i = 0; i < 2; i++)
73 const int chd = i * chdPair.second + (1 - i) * chdPair.first;
77 const int sortedBody = mortonCodesIdx[ch];
79 double4 xyAB = gabForLeaves[sortedBody];
81 ::fmin(xyAB[0], xyAB[2]), ::fmin(xyAB[1], xyAB[3]),
82 ::fmax(xyAB[0], xyAB[2]), ::fmax(xyAB[1], xyAB[3])
89 lu[i] = lowerupper[chd];
94 lowerupper[srt] = double4{
95 ::fmin(lu[0][0], lu[1][0]), ::fmin(lu[0][1], lu[1][1]),
96 ::fmax(lu[0][2], lu[1][2]), ::fmax(lu[0][3], lu[1][3])
99 cen = center[srt] = double2{ 0.5 * (lowerupper[srt][0] + lowerupper[srt][2]),
100 0.5 * (lowerupper[srt][1] + lowerupper[srt][3]) };
102 const double2 zero = { 0.0, 0.0 };
103 double2 momh0 = zero;
104 double2 momh1 = zero;
105 double2 momh2 = zero;
106 double2 momh3 = zero;
107 double2 momh4 = zero;
108 double2 momh5 = zero;
109 double2 momh6 = zero;
110 double2 momh7 = zero;
112 for (i = 0; i < 2; i++)
114 const int chd = i * chdPair.second + (1 - i) * chdPair.first;
118 const int sortedBody = mortonCodesIdx[ch];
122 mom0 = double2{ gamma[sortedBody], 0.0 };
124mom1 = mom2 = mom3 = mom4 = mom5 = mom6 = mom7 = double2{ 0.0, 0.0 };
125 double2 pos =
object[sortedBody];
138 const int srtT = indexSortT[chd];
139 ch = (nnodes - 1) - srtT;
143 mom0 = double2{ moms[nch + 0][0], (double)0 };
144 mom1 = moms[nch + 1];
145 mom2 = moms[nch + 2];
146 mom3 = moms[nch + 3];
147 mom4 = moms[nch + 4];
148 mom5 = moms[nch + 5];
149 mom6 = moms[nch + 6];
150 mom7 = moms[nch + 7];
151 dr = center[chd] - cen;
167 momh1 += multz(mom0, z);
168 momh2 += 2 * multz(mom1, z);
169 momh3 += 3 * multz(mom2, z);
170 momh4 += 4 * multz(mom3, z);
171 momh5 += 5 * multz(mom4, z);
172 momh6 += 6 * multz(mom5, z);
173 momh7 += 7 * multz(mom6, z);
177 momh2 += multz(mom0, z);
178 momh3 += 3 * multz(mom1, z);
179 momh4 += 6 * multz(mom2, z);
180 momh5 += 10 * multz(mom3, z);
181 momh6 += 15 * multz(mom4, z);
182 momh7 += 21 * multz(mom5, z);
186 momh3 += multz(mom0, z);
187 momh4 += 4 * multz(mom1, z);
188 momh5 += 10 * multz(mom2, z);
189 momh6 += 20 * multz(mom3, z);
190 momh7 += 35 * multz(mom4, z);
194 momh4 += multz(mom0, z);
195 momh5 += 5 * multz(mom1, z);
196 momh6 += 15 * multz(mom2, z);
197 momh7 += 35 * multz(mom3, z);
201 momh5 += multz(mom0, z);
202 momh6 += 6 * multz(mom1, z);
203 momh7 += 21 * multz(mom2, z);
207 momh6 += multz(mom0, z);
208 momh7 += 7 * multz(mom1, z);
212 momh7 += multz(mom0, z);
217 moms[kch + 0] = momh0;
218 moms[kch + 1] = momh1;
219 moms[kch + 2] = momh2;
220 moms[kch + 3] = momh3;
221 moms[kch + 4] = momh4;
222 moms[kch + 5] = momh5;
223 moms[kch + 6] = momh6;
224 moms[kch + 7] = momh7;
232 mass[nnodes - 1 - k] = cm;