TreeNode.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <queue>
#include <vector>

// Package includes
#include <boost/foreach.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>

// Local includes
#include "trainer/source/ObjectSet.h"
#include <ISM/common_type/ObjectSet.hpp>
namespace SceneModel {
  
  class TreeNode : public boost::enable_shared_from_this<TreeNode>  {
  public:

    TreeNode(boost::shared_ptr<ISM::ObjectSet> pObjectSet);
    
    TreeNode(boost::shared_ptr<TreeNode> pRoot);
    
    ~TreeNode();
    
    boost::shared_ptr<TreeNode> setNewRootNodeByType(std::string pType);
    
    boost::shared_ptr<ISM::ObjectSet> getObjectSet();
    
    std::vector<boost::shared_ptr<TreeNode> > getChildren();
    
    void addChild(boost::shared_ptr<TreeNode> pChild);
    
    unsigned int getNumberOfNodes();
    
    boost::shared_ptr<TreeNode> f();
    
    void printTreeToConsole(unsigned int space);

    void setIDs();
    
  public:
    
    boost::shared_ptr<ISM::ObjectSet> mObjectSet;
    
    boost::shared_ptr<TreeNode> mParent;
    
    std::vector<boost::shared_ptr<TreeNode> > mChildren;

    bool mIsReference = false;

    boost::shared_ptr<TreeNode> mReferenceTo;

    unsigned int mID;
    
  private:
    void reassignNewParentNode(boost::shared_ptr<TreeNode> pParent);

    void updateIDs(unsigned int& pID, bool pUpdateReferencIDs);

    std::vector<boost::shared_ptr<TreeNode>> updateReferences(boost::shared_ptr<TreeNode> pRoot);

  };
}