BinTreeElement.h

Go to the documentation of this file.

#ifndef BT_ELEMENT_H
#define BT_ELEMENT_H
 
#include "TreeElement.h" //string, using std
 
namespace bridges {
    namespace datastructure {
        template <typename E>
        class BinTreeElement : public TreeElement<E> {
            private:
                int leftPos = 0, rightPos = 1;
            public:
                BinTreeElement(BinTreeElement* l, BinTreeElement* r, const E& e = E(),
                    const string& lab = string()) : TreeElement<E>(e, lab) {
                    this->addChild(l);
                    this->addChild(r);
                }
 
                BinTreeElement(const E& e = E(), const string& lab = string())
                    : BinTreeElement(nullptr, nullptr, e, lab) {
                }
 
                virtual const string getDStype() const override {
                    return "BinaryTree";
                }
 
                virtual BinTreeElement* getLeft() {
                    return static_cast<BinTreeElement*>(this->getChild(leftPos));
                }
 
                virtual const BinTreeElement* getLeft() const {
                    return static_cast<const BinTreeElement*>(this->getChild(leftPos));
                }
 
                void setLeft(BinTreeElement* l) {
                    this->setChild(leftPos, l);
                }
 
                virtual BinTreeElement* getRight() {
                    return static_cast<BinTreeElement*>(this->getChild(rightPos));
                }
 
                virtual const BinTreeElement* getRight() const {
                    return static_cast<const BinTreeElement*>(this->getChild(rightPos));
                }
 
                void setRight(BinTreeElement* r) {
                    this->setChild(rightPos, r);
                }
        }; //end of BinTreeElement class
    }
}//end of bridges namespace
#endif