273f7d6647
modifié : include/ArrayTree/ArrayTreeBuilder.tpp modifié : include/QuadTree/QuadTreeBuilder.hpp modifié : include/Tree.hpp modifié : include/TreeBuilder.hpp modifié : include/XMLTree/XMLTreeBuilder.hpp modifié : include/triskeleDealThreads.hpp modifié : src/Attributes/AreaAttribute.cpp modifié : src/QuadTree/QuadTreeBuilder.cpp modifié : src/Tree.cpp modifié : src/XMLTree/XMLTreeBuilder.cpp modifié : src/testMain.cpp
127 lines
3.1 KiB
C++
127 lines
3.1 KiB
C++
#ifndef _OTB_TRISKELE_ARRAY_TREE_BUILDER_HPP
|
|
#define _OTB_TRISKELE_ARRAY_TREE_BUILDER_HPP
|
|
|
|
//XXX #include "triskeleDebug.hpp"
|
|
|
|
#ifdef SMART_LOG
|
|
#include <iostream>
|
|
#endif // SMART_LOG
|
|
|
|
#include "triskeleSort.hpp"
|
|
#include "triskeleDealThreads.hpp"
|
|
#include "TreeBuilder.hpp"
|
|
#include "IImage.hpp"
|
|
#include "GraphWalker.hpp"
|
|
#include "Weight.hpp"
|
|
#include "Leader.hpp"
|
|
|
|
namespace otb {
|
|
namespace triskele {
|
|
namespace arrayTree {
|
|
|
|
template <typename WeightT, typename PixelT>
|
|
class ArrayTreeBuilder : public TreeBuilder {
|
|
const unsigned int coreCount;
|
|
const IImage<PixelT> ℑ
|
|
const GraphWalker &graphWalker;
|
|
TreeType type;
|
|
bool countingFlag;
|
|
Leader leaders;
|
|
DimNodeId *childCountRec;
|
|
DimImg *newCompIdx;
|
|
WeightT *compWeights; // [leafCount]
|
|
|
|
|
|
|
|
public:
|
|
ArrayTreeBuilder (const unsigned int &coreCount,
|
|
IImage<PixelT> &image, const GraphWalker &graphWalker,
|
|
const TreeType &treeType, const bool &countingSort = true);
|
|
~ArrayTreeBuilder () {}
|
|
|
|
inline void
|
|
buildTree (Tree &tree);
|
|
|
|
private:
|
|
template<typename WeightFunct>
|
|
inline void
|
|
buildTree (Tree &tree, const WeightFunct &weightFunct);
|
|
|
|
template<typename WeightFunct>
|
|
inline void
|
|
fillAPTree (PixelT *leafAPTree, const WeightFunct &weightFunct);
|
|
|
|
template<typename WeightFunct>
|
|
inline void
|
|
buildTree (Edge<WeightT> *edges, const WeightFunct &weightFunct, const Rect &tile, DimImg &topParent);
|
|
|
|
template<typename WeightFunct>
|
|
inline void
|
|
connectLeaf (DimImg a, DimImg b, const WeightT &weight, DimImg &parCount, const WeightFunct &weightFunct);
|
|
|
|
template<typename WeightFunct>
|
|
inline void
|
|
connectComp (DimImg newComp, DimImg topA, DimImg topB, const WeightFunct &weightFunct);
|
|
|
|
template<typename WeightFunct>
|
|
inline void
|
|
connectComp (DimImg topA, DimImg topB, const WeightFunct &weightFunct);
|
|
|
|
template<typename WeightFunct>
|
|
inline DimImg
|
|
updateNewIdx (const vector<DimImg> &compBases, const vector<DimImg> &compTops, const WeightFunct &weightFunct);
|
|
|
|
inline void
|
|
updateNewIdx (const DimImg curComp, DimImg &compCount);
|
|
|
|
inline void
|
|
compress (const DimImg &compTop);
|
|
|
|
inline void
|
|
createParent (DimImg &topParent, const WeightT &weight, DimImg &childA, DimImg &childB);
|
|
|
|
inline void
|
|
addChild (const DimImg &parent, DimImg &child);
|
|
|
|
inline void
|
|
addChildren (const DimImg &parent, const DimImg &sibling);
|
|
|
|
inline DimImg
|
|
findRoot (DimImg comp);
|
|
|
|
template<typename WeightFunct>
|
|
inline DimImg
|
|
findTopComp (const DimImg &comp, const WeightFunct &weightFunct);
|
|
|
|
template<typename WeightFunct>
|
|
inline DimImg
|
|
findTopComp (DimImg comp, const WeightT &weight, const WeightFunct &weightFunct);
|
|
|
|
inline DimImg
|
|
findCompMultiChild (DimImg comp);
|
|
|
|
inline void
|
|
buildChildren ();
|
|
|
|
template<typename WeightFunct>
|
|
inline void
|
|
initWeights (const GraphWalker &graphWalker, const WeightFunct &weightFunct);
|
|
|
|
#ifdef SMART_LOG
|
|
inline void
|
|
printTree (const Size &size, const bool &rec);
|
|
inline void
|
|
printLeaders (const Size &size);
|
|
inline void
|
|
printNewCompIdx (const Size &size);
|
|
#endif
|
|
};
|
|
|
|
#include "ArrayTreeBuilder.tpp"
|
|
|
|
} // arrayTree
|
|
} // triskele
|
|
} // otb
|
|
|
|
#endif // _OTB_TRISKELE_ARRAY_TREE_BUILDER_HPP
|