BTElement.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 BTElement : public TreeElement<E> {
            private:
                int leftPos = 0, rightPos = 1;
            public:
                BTElement(BTElement* l, BTElement* r, const E& e = E(),
                    const string& lab = string()) :
                    TreeElement<E>(e, lab) {
                    this->addChild(l);
                    this->addChild(r);
                }
                BTElement(const E& e = E(), const string& lab = string())
                    : BTElement(nullptr, nullptr, e, lab) {}

                virtual const string getDStype() const override {
                    return "BinaryTree";
                }

                virtual BTElement* getLeft() {
                    return static_cast<BTElement*>(this->getChild(leftPos));
                }
                virtual const BTElement* getLeft() const {
                    return static_cast<const BTElement*>(this->getChild(leftPos));
                }
                void setLeft(BTElement* l) {
                    this->setChild(leftPos, l);
                }

                virtual BTElement* getRight() {
                    return static_cast<BTElement*>(this->getChild(rightPos));
                }

                virtual const BTElement* getRight() const {
                    return static_cast<const BTElement*>(this->getChild(rightPos));
                }
                void setRight(BTElement* r) {
                    this->setChild(rightPos, r);
                }
        }; //end of BTElement class
    }
}//end of bridges namespace
#endif