114 lines
2.6 KiB
C++
114 lines
2.6 KiB
C++
|
#ifndef __COSMOTOOL_AMR_HPP
|
||
|
#define __COSMOTOOL_AMR_HPP
|
||
|
|
||
|
#include "config.hpp"
|
||
|
|
||
|
namespace CosmoTool
|
||
|
{
|
||
|
|
||
|
typedef uint32_t octPtr;
|
||
|
typedef uint16_t octCoordType;
|
||
|
static const uint16_t octCoordTypeNorm = 0xffff;
|
||
|
static const uint16_t octCoordCenter = 0x8000;
|
||
|
|
||
|
// This is also the root cell, but this one
|
||
|
// is never referenced (really ??).
|
||
|
static const octPtr invalidOctCell = 0xffffffff;
|
||
|
static const octPtr emptyOctCell = 0;
|
||
|
static const octPtr octParticleMarker = 0x80000000;
|
||
|
static const octPtr octParticleMask = 0x7fffffff;
|
||
|
|
||
|
typedef octCoordType OctCoords[3];
|
||
|
|
||
|
struct OctCell
|
||
|
{
|
||
|
octPtr numberLeaves;
|
||
|
octPtr children[8];
|
||
|
};
|
||
|
|
||
|
class OctTree
|
||
|
{
|
||
|
public:
|
||
|
//Coordinates of particles must be in the [0:1] range
|
||
|
OctTree(const FCoordinates *particles, octPtr numParticles,
|
||
|
uint32_t maxTreeDepth, uint32_t maxAbsoluteDepth,
|
||
|
uint32_t threshold = 1);
|
||
|
~OctTree();
|
||
|
|
||
|
void buildTree(uint32_t maxAbsoluteDepth);
|
||
|
void insertParticle(octPtr node,
|
||
|
const OctCoords& icoord,
|
||
|
octCoordType halfNodeLength,
|
||
|
octPtr particleId,
|
||
|
uint32_t maxAbsoluteDepth);
|
||
|
|
||
|
|
||
|
octPtr getNumberLeaves() const {
|
||
|
return cells[0].numberLeaves;
|
||
|
}
|
||
|
|
||
|
template<typename FunT>
|
||
|
void walkTree(FunT f)
|
||
|
{
|
||
|
OctCoords rootCenter = { octCoordCenter, octCoordCenter, octCoordCenter };
|
||
|
|
||
|
walkTreeElements(f, 0, rootCenter, octCoordCenter);
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
const FCoordinates *particles;
|
||
|
octPtr numParticles;
|
||
|
OctCell *cells;
|
||
|
float Lbox;
|
||
|
octPtr lastNode;
|
||
|
octPtr numCells;
|
||
|
|
||
|
|
||
|
template<typename FunT>
|
||
|
void walkTreeElements(FunT f, octPtr node,
|
||
|
const OctCoords& icoord,
|
||
|
octCoordType halfNodeLength)
|
||
|
{
|
||
|
OctCoords newCoord;
|
||
|
FCoordinates realCenter;
|
||
|
|
||
|
for (int j = 0; j < 3; j++)
|
||
|
realCenter[j] = icoord[j]/(2.*octCoordCenter);
|
||
|
|
||
|
f(realCenter, cells[node].numberLeaves, halfNodeLength/(float)octCoordCenter,
|
||
|
cells[node].children[0] == invalidOctCell, // True if this is a meta-node
|
||
|
false);
|
||
|
|
||
|
for (int i = 0; i < 8; i++)
|
||
|
{
|
||
|
octPtr newNode = cells[node].children[i];
|
||
|
int ipos[3] = { (i&1), (i&2)>>1, (i&4)>>2 };
|
||
|
|
||
|
if (newNode == emptyOctCell || newNode == invalidOctCell)
|
||
|
continue;
|
||
|
|
||
|
for (int j = 0; j < 3; j++)
|
||
|
newCoord[j] = icoord[j]+(2*ipos[j]-1)*halfNodeLength/2;
|
||
|
|
||
|
if (newNode & octParticleMarker)
|
||
|
{
|
||
|
for (int j = 0; j < 3; j++)
|
||
|
realCenter[j] = newCoord[j]/(2.*octCoordCenter);
|
||
|
|
||
|
f(realCenter,
|
||
|
1, halfNodeLength/(2.*octCoordCenter),
|
||
|
false, true);
|
||
|
continue;
|
||
|
}
|
||
|
walkTreeElements(f, cells[node].children[i], newCoord, halfNodeLength/2);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
};
|
||
|
|
||
|
};
|
||
|
|
||
|
|
||
|
#endif
|