NodeTemplate.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: The Bio++ Development Group
//
// SPDX-License-Identifier: CECILL-2.1
 
#ifndef BPP_PHYL_TREE_NODETEMPLATE_H
#define BPP_PHYL_TREE_NODETEMPLATE_H
 
 
#include "Node.h"
 
namespace bpp
{
template<class NodeInfos>
class NodeTemplate :
  public Node
{
  friend class TreeTemplateTools;
 
private:
  NodeInfos infos_;
 
public:
  NodeTemplate() : Node(), infos_() {}
 
  NodeTemplate(int id) : Node(id), infos_() {}
 
  NodeTemplate(const std::string& name) : Node(name), infos_() {}
 
  NodeTemplate(int id, const std::string& name) : Node(id, name), infos_() {}
 
protected:
  NodeTemplate(const Node& node) : Node(node), infos_() {}
 
  NodeTemplate(const NodeTemplate<NodeInfos>& node) :
    Node(node), infos_(node.infos_)
  {}
 
  NodeTemplate<NodeInfos>& operator=(const NodeTemplate<NodeInfos>& node)
  {
    Node::operator=(node);
    infos_ = node.infos_;
    return *this;
  }
 
  NodeTemplate<NodeInfos>* clone() const { return new NodeTemplate<NodeInfos>(*this); }
 
public:
  virtual ~NodeTemplate() {}
 
public:
  const NodeTemplate<NodeInfos>* getFather() const { return dynamic_cast<const NodeTemplate<NodeInfos>*>(father_); }
 
  NodeTemplate<NodeInfos>* getFather() { return dynamic_cast<NodeTemplate<NodeInfos>*>(father_); }
 
  NodeTemplate<NodeInfos>* removeFather() { NodeTemplate<NodeInfos>* f = dynamic_cast<NodeTemplate<NodeInfos>*>(father_); father_ = 0; return f; }
 
  const NodeTemplate<NodeInfos>* getSon(size_t i) const { return dynamic_cast<NodeTemplate<NodeInfos>*>(sons_[i]); }
 
  NodeTemplate<NodeInfos>* getSon(size_t i) { return dynamic_cast<NodeTemplate<NodeInfos>*>(sons_[i]); }
 
  std::vector<const NodeTemplate<NodeInfos>*> getNeighbors() const
  {
    std::vector<const Node*> neighbors = Node::getNeighbors();
    std::vector<const NodeTemplate<NodeInfos>*> neighbors2(neighbors.size());
    for (size_t i = 0; i < neighbors.size(); i++)
    {
      neighbors2[i] = dynamic_cast<const NodeTemplate<NodeInfos>*>(neighbors[i]);
    }
    return neighbors2;
  }
 
  std::vector<NodeTemplate<NodeInfos>*> getNeighbors()
  {
    std::vector<Node*> neighbors = Node::getNeighbors();
    std::vector<NodeTemplate<NodeInfos>*> neighbors2(neighbors.size());
    for (size_t i = 0; i < neighbors.size(); i++)
    {
      neighbors2[i] = dynamic_cast<NodeTemplate<NodeInfos>*>(neighbors[i]);
    }
    return neighbors2;
  }
 
  NodeTemplate<NodeInfos>* operator[](int i) { return dynamic_cast<NodeTemplate<NodeInfos>*>((i < 0) ? father_ : sons_[i]); }
 
  const NodeTemplate<NodeInfos>* operator[](int i) const { return dynamic_cast<const NodeTemplate<NodeInfos>*>((i < 0) ? father_ : sons_[i]); }
 
 
  // Specific methods:
 
  virtual const NodeInfos& getInfos() const { return infos_; }
 
  virtual NodeInfos& getInfos() { return infos_; }
 
  virtual void setInfos(const NodeInfos& infos) { infos_ = infos; }
};
} // end of namespace bpp.
#endif // BPP_PHYL_TREE_NODETEMPLATE_H