Skip to content

GitLab

Commit 45a79f16 authored by BRAMAS Berenger's avatar BRAMAS Berenger
Browse files

use allocator in tree

parent 022d056c
Hide whitespace changes
Inline Side-by-side
Src/Containers/FOctree.hpp
View file @ 45a79f16
......@@ -20,6 +20,7 @@
#include "FSubOctree.hpp"
#include "FTreeCoordinate.hpp"
#include "FBlockAllocator.hpp"
#include "../Utils/FDebug.hpp"
#include "../Utils/FGlobal.hpp"
......@@ -49,13 +50,14 @@
*
* If the octree as an height H, then it goes from 0 to H-1
* at level 0 the space is not split
* CellAllocator can be FListBlockAllocator<CellClass, 10> or FBasicBlockAllocator<CellClass>
*/
template< class CellClass, class ContainerClass, class LeafClass>
template< class CellClass, class ContainerClass, class LeafClass, class CellAllocatorClass = FListBlockAllocator<CellClass, 15> >
class FOctree : protected FAssertable, public FNoCopyable {
typedef FSubOctreeWithLeafs< CellClass , ContainerClass, LeafClass> SubOctreeWithLeaves;
typedef FSubOctree< CellClass , ContainerClass, LeafClass> SubOctree;
typedef FSubOctreeWithLeafs< CellClass , ContainerClass, LeafClass, CellAllocatorClass> SubOctreeWithLeaves;
typedef FSubOctree< CellClass , ContainerClass, LeafClass, CellAllocatorClass> SubOctree;
FAbstractSubOctree< CellClass , ContainerClass, LeafClass>* root; //< root suboctree
FAbstractSubOctree< CellClass , ContainerClass, LeafClass, CellAllocatorClass>* root; //< root suboctree
FReal*const boxWidthAtLevel; //< to store the width of each boxs at all levels
......@@ -118,10 +120,10 @@ public:
fassert(subHeight <= height - 1, "Subheight cannot be greater than height", __LINE__, __FILE__ );
// Does we only need one suboctree?
if(subHeight == height - 1){
root = new FSubOctreeWithLeafs< CellClass , ContainerClass, LeafClass>(0, 0, this->subHeight, 1);
root = new FSubOctreeWithLeafs< CellClass , ContainerClass, LeafClass,CellAllocatorClass>(0, 0, this->subHeight, 1);
}
else {// if(subHeight < height - 1)
root = new FSubOctree< CellClass , ContainerClass, LeafClass>(0, 0, this->subHeight, 1);
root = new FSubOctree< CellClass , ContainerClass, LeafClass,CellAllocatorClass>(0, 0, this->subHeight, 1);
}
FReal tempWidth = this->boxWidth;
......@@ -235,18 +237,18 @@ public:
* depending if we are working on the bottom of the tree.
*/
union SubOctreeTypes {
FAbstractSubOctree<CellClass,ContainerClass,LeafClass>* tree; //< Usual pointer to work
FSubOctree<CellClass,ContainerClass,LeafClass>* middleTree; //< To access to sub-octree under
FSubOctreeWithLeafs<CellClass,ContainerClass,LeafClass>* leafTree;//< To access to particles lists
FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* tree; //< Usual pointer to work
FSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* middleTree; //< To access to sub-octree under
FSubOctreeWithLeafs<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* leafTree;//< To access to particles lists
};
/**
* This class is a const SubOctreeTypes
*/
union SubOctreeTypesConst {
const FAbstractSubOctree<CellClass,ContainerClass,LeafClass>* tree; //< Usual pointer to work
const FSubOctree<CellClass,ContainerClass,LeafClass>* middleTree; //< To access to sub-octree under
const FSubOctreeWithLeafs<CellClass,ContainerClass,LeafClass>* leafTree;//< To access to particles lists
const FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* tree; //< Usual pointer to work
const FSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* middleTree; //< To access to sub-octree under
const FSubOctreeWithLeafs<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* leafTree;//< To access to particles lists
};
/**
......
Src/Containers/FSubOctree.hpp
View file @ 45a79f16
......@@ -47,7 +47,7 @@
* Please refere to testOctree.cpp to see an example
* @warning Give the particleClass & cellClass
*/
template< class CellClass , class ContainerClass, class LeafClass>
template< class CellClass , class ContainerClass, class LeafClass, class CellAllocatorClass>
class FAbstractSubOctree : protected FAssertable{
protected:
......@@ -64,6 +64,7 @@ protected:
const bool isLeafSubtree; //< To know if a subtree is leaf or not (we prefere that to a virtual method)
CellAllocatorClass cellAllocator;
/**
* This function compute the morton index for the last level of this suboctree.
......@@ -93,7 +94,7 @@ protected:
int indexLevel = this->subOctreeHeight - 1;
int bottomToTop = 0;
while(indexLevel >= 0 && !this->cells[indexLevel][arrayIndex]){
CellClass* const newNode = new CellClass();
CellClass* const newNode = cellAllocator.newObject();//new CellClass();
newNode->setMortonIndex(inLeafCellIndex);
newNode->setCoordinate(treePosition.getX() >> bottomToTop,
......@@ -145,7 +146,7 @@ protected:
}
// remove the last cells
delete this->cells[indexLevel][arrayIndex];
cellAllocator.deleteObject(this->cells[indexLevel][arrayIndex]);
this->cells[indexLevel][arrayIndex] = 0;
// progress upper
--indexLevel;
......@@ -157,7 +158,7 @@ protected:
// continue while we are not in the last level and our child are empty
while(indexLevel >= 0 && memcmp(&this->cells[indexLevel+1][arrayIndex<<3], emptyArray, 8 * sizeof(CellClass*)) == 0 ){
delete this->cells[indexLevel][arrayIndex];
cellAllocator.deleteObject( this->cells[indexLevel][arrayIndex] );
this->cells[indexLevel][arrayIndex] = 0;
--indexLevel;
......@@ -213,7 +214,7 @@ public:
for( int indexLevel = this->subOctreeHeight - 1 ; indexLevel >= 0 ; --indexLevel ){
for( int indexCells = mostLeft ; indexCells <= mostRight ; ++indexCells ){
if(this->cells[indexLevel][indexCells]){
delete this->cells[indexLevel][indexCells];
cellAllocator.deleteObject( this->cells[indexLevel][indexCells] );
}
}
......@@ -334,10 +335,10 @@ public:
* Please refere to testOctree.cpp to see an example.
* @warning Give the particleClass & cellClass
*/
template< class CellClass , class ContainerClass, class LeafClass>
class FSubOctreeWithLeafs : public FAbstractSubOctree<CellClass,ContainerClass,LeafClass> {
template< class CellClass , class ContainerClass, class LeafClass, class CellAllocatorClass>
class FSubOctreeWithLeafs : public FAbstractSubOctree<CellClass,ContainerClass,LeafClass, CellAllocatorClass> {
private:
typedef FAbstractSubOctree<CellClass,ContainerClass,LeafClass> Parent;
typedef FAbstractSubOctree<CellClass,ContainerClass,LeafClass, CellAllocatorClass> Parent;
LeafClass** leafs; //< Leafs array
......@@ -352,9 +353,9 @@ public:
* @param inSubOctreeHeight Height of this suboctree
* @param inSubOctreePosition Level of the current suboctree in the global tree (1 if upper tree)
*/
FSubOctreeWithLeafs(FAbstractSubOctree<CellClass,ContainerClass,LeafClass>* const inParent, const int inIndexInParent,
FSubOctreeWithLeafs(FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* const inParent, const int inIndexInParent,
const int inSubOctreeHeight, const int inSubOctreePosition) :
FAbstractSubOctree<CellClass,ContainerClass,LeafClass>(inParent, inIndexInParent, inSubOctreeHeight, inSubOctreePosition, true) {
FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>(inParent, inIndexInParent, inSubOctreeHeight, inSubOctreePosition, true) {
const int cellsAtLeafLevel = 1 << (3 * inSubOctreeHeight);
......@@ -479,11 +480,11 @@ public:
*
* @warning Give the particleClass & cellClass
*/
template< class CellClass , class ContainerClass, class LeafClass>
class FSubOctree : public FAbstractSubOctree<CellClass,ContainerClass,LeafClass> {
template< class CellClass , class ContainerClass, class LeafClass, class CellAllocatorClass>
class FSubOctree : public FAbstractSubOctree<CellClass,ContainerClass,LeafClass, CellAllocatorClass> {
private:
typedef FAbstractSubOctree<CellClass,ContainerClass,LeafClass> Parent;
typedef FSubOctreeWithLeafs<CellClass,ContainerClass,LeafClass> SubOctreeWithLeaf;
typedef FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass> Parent;
typedef FSubOctreeWithLeafs<CellClass,ContainerClass,LeafClass,CellAllocatorClass> SubOctreeWithLeaf;
Parent** subleafs; //< Last levels is composed of suboctree
......@@ -499,9 +500,9 @@ public:
* @param inSubOctreeHeight Height of this suboctree
* @param inSubOctreePosition Level of the current suboctree in the global tree (0 if upper tree)
*/
FSubOctree(FAbstractSubOctree<CellClass,ContainerClass,LeafClass>* const inParent, const int inIndexInParent,
FSubOctree(FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* const inParent, const int inIndexInParent,
const int inSubOctreeHeight, const int inSubOctreePosition) :
FAbstractSubOctree<CellClass,ContainerClass,LeafClass>(inParent, inIndexInParent, inSubOctreeHeight, inSubOctreePosition, false) {
FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>(inParent, inIndexInParent, inSubOctreeHeight, inSubOctreePosition, false) {
const int cellsAtLeafLevel = 1 << (3 * inSubOctreeHeight);
......@@ -615,14 +616,14 @@ public:
/** To get access to leafs elements (child suboctree)
* @param index the position of the leaf/child suboctree
* @return child at this index */
FAbstractSubOctree<CellClass,ContainerClass,LeafClass>* leafs(const int index) {
FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* leafs(const int index) {
return this->subleafs[index];
}
/** To get access to leafs elements (child suboctree)
* @param index the position of the leaf/child suboctree
* @return child at this index */
const FAbstractSubOctree<CellClass,ContainerClass,LeafClass>* leafs(const int index) const {
const FAbstractSubOctree<CellClass,ContainerClass,LeafClass,CellAllocatorClass>* leafs(const int index) const {
return this->subleafs[index];
}
};
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment