guilhem_lavaux/cosmotool
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow conditioned walking

Browse Source
This commit is contained in:
Guilhem Lavaux 2009-11-06 22:05:00 -06:00
parent b8a67a50eb
commit 145b298730
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/octTree.hpp
View File

@ -49,10 +49,16 @@ namespace CosmoTool
template<typename FunT>
void walkTree(FunT f)
{
walkTree(f, unconditioned);
}
template<typename FunT, typename CondT>
void walkTree(FunT f, CondT condition)
{
OctCoords rootCenter = { octCoordCenter, octCoordCenter, octCoordCenter };
walkTreeElements(f, 0, rootCenter, octCoordCenter);
walkTreeElements(f, condition, 0, rootCenter, octCoordCenter);
}
protected:
@ -66,8 +72,14 @@ namespace CosmoTool
float xMin[3];
template<typename FunT>
void walkTreeElements(FunT f, octPtr node,
static bool unconditioned()
{
return true;
}
template<typename FunT, typename CondT>
void walkTreeElements(FunT f, CondT condition,
octPtr node,
const OctCoords& icoord,
octCoordType halfNodeLength)
{
@ -83,6 +95,11 @@ namespace CosmoTool
f(realCenter, cells[node].numberLeaves, lenNorm*halfNodeLength/(float)octCoordCenter,
cells[node].children[0] == invalidOctCell, // True if this is a meta-node
false);
if (!condition(realCenter, cells[node].numberLeaves,
lenNorm*halfNodeLength/(float)octCoordCenter,
cells[node].children[0] == invalidOctCell))
return;
for (int i = 0; i < 8; i++)
{
@ -108,7 +125,8 @@ namespace CosmoTool
false, true);
continue;
}
walkTreeElements(f, cells[node].children[i], newCoord, halfNodeLength/2);
walkTreeElements(f, condition,
cells[node].children[i], newCoord, halfNodeLength/2);
}
}