Added missing splitter
This commit is contained in:
parent
5fa5a73104
commit
1ec0ac8d63
71
src/kdtree_splitters.hpp
Normal file
71
src/kdtree_splitters.hpp
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
#ifndef __KDTREE_SPLITTERS_HPP
|
||||||
|
#define __KDTREE_SPLITTERS_HPP
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
|
||||||
|
namespace CosmoTool
|
||||||
|
{
|
||||||
|
|
||||||
|
template<int N, typename ValType, typename CType = ComputePrecision>
|
||||||
|
struct KD_homogeneous_cell_splitter
|
||||||
|
{
|
||||||
|
typedef typename KDDef<N,CType>::KDCoordinates coords;
|
||||||
|
typedef typename KDDef<N,CType>::CoordType ctype;
|
||||||
|
|
||||||
|
void operator()(KDCell<N,ValType,CType> **cells, uint32_t Ncells, uint32_t& split_index, int axis, coords minBound, coords maxBound)
|
||||||
|
{
|
||||||
|
ctype midCoord = 0.5*(maxBound[axis]+minBound[axis]);
|
||||||
|
uint32_t below = 0, above = Ncells-1;
|
||||||
|
ctype delta_max = std::abs(cells[0]->coord[axis]-midCoord);
|
||||||
|
uint32_t idx_max = 0;
|
||||||
|
|
||||||
|
while (below < above)
|
||||||
|
{
|
||||||
|
ctype delta = cells[below]->coord[axis]-midCoord;
|
||||||
|
if (delta > 0)
|
||||||
|
{
|
||||||
|
if (delta < delta_max)
|
||||||
|
{
|
||||||
|
delta_max = delta;
|
||||||
|
idx_max = above;
|
||||||
|
}
|
||||||
|
std::swap(cells[below], cells[above--]);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (-delta < delta_max)
|
||||||
|
{
|
||||||
|
delta_max = -delta;
|
||||||
|
idx_max = below;
|
||||||
|
}
|
||||||
|
below++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (idx_max != above)
|
||||||
|
{
|
||||||
|
bool cond1 = cells[idx_max]->coord[axis] > midCoord;
|
||||||
|
bool cond2 = cells[above]->coord[axis] > midCoord;
|
||||||
|
if ((cond1 && cond2) || (!cond1 && !cond2))
|
||||||
|
{
|
||||||
|
split_index = above;
|
||||||
|
std::swap(cells[above], cells[idx_max]);
|
||||||
|
}
|
||||||
|
else if (cond2)
|
||||||
|
{
|
||||||
|
split_index = above-1;
|
||||||
|
std::swap(cells[above-1], cells[idx_max]);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
split_index = above+1;
|
||||||
|
std::swap(cells[above+1], cells[idx_max]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else split_index = above;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
Loading…
Reference in New Issue
Block a user