TreeElement.h

Go to the documentation of this file.

#ifndef TREE_ELEMENT_H
#define TREE_ELEMENT_H

#include <vector>

#include "Element.h" //string, using std

namespace bridges {
    namespace datastructure {
        template <typename E>
        class TreeElement : public Element<E>, public DataStructure {
            private:
                vector<TreeElement*> children;
            public:
                TreeElement(const E& e = E(), const string& lab = string())
                    : Element<E>(e, lab) {
                }
                virtual const string getDStype() const override {
                    return "Tree";
                }
                vector<TreeElement*>& getChildren() {
                    return children;
                }
                const vector<TreeElement*>& getChildren() const {
                    return children;
                }
                TreeElement* getChild(const int& n) {
                    return (n >= children.size() || n < 0 ) ? nullptr : children.at(n);
                }
                const TreeElement* getChild(const int& n) const {
                    return (n >= children.size() || n < 0) ? nullptr : children.at(n);
                }
                void addChild(TreeElement* child) {
                    children.push_back(child);
                    if (child) {
                        this->links[child];
                    }
                }
                void setChild(const size_t& index, TreeElement* kid) {
                    if ((index < children.size()) && index >= 0) {
                        children[index] = kid;
                        this->links.erase(children[index]); // remove any existing links
                        this->links[kid];           // initialize this element in the map
                    }
                }
            private:
                virtual const string getDataStructureRepresentation() const override final {
                    //TODO: Check for exceeding max node
                    string tree_json =
                        QUOTE + "nodes"  + QUOTE + COLON +
                        (OPEN_CURLY +
                            this->preOrder((TreeElement<E>*)this)
                            + CLOSE_CURLY + CLOSE_CURLY);

                    return tree_json;
                }

                string  preOrder(TreeElement<E>* root) const {
                    using bridges::JSONUtil::JSONencode;
                    string json_str = "", children = "", link_props = "", elem_rep = "";
                    string t_str;
                    if (root != NULL) {
                        // first get the node representation
                        elem_rep = root->getElementRepresentation();
                        // remove surrounding curly braces
                        t_str = elem_rep.substr(1, elem_rep.size() - 2);
                        json_str += t_str + COMMA;
                        // now get the children
                        if (root->children.size())
                            json_str += QUOTE + "children" + QUOTE + COLON + OPEN_BOX ;
                        for (int k = 0; k < root->children.size(); k++) {
                            if (root->children[k] == NULL) {
                                json_str += OPEN_CURLY + QUOTE + "name" + QUOTE + COLON +
                                    QUOTE + "NULL" + QUOTE + CLOSE_CURLY + COMMA;
                            }
                            else {
                                LinkVisualizer *lv =
                                    root->getLinkVisualizer(root->children[k]);
                                json_str += OPEN_CURLY;
                                if (lv) {
                                    Color c = lv->getColor();
                                    json_str += QUOTE + "linkProperties" + QUOTE + COLON + OPEN_CURLY +
                                        QUOTE + "color" + QUOTE + COLON +
                                        c.getCSSRepresentation() + COMMA +
                                        QUOTE + "thickness" + QUOTE + COLON +
                                        JSONencode(lv->getThickness()) +
                                        //                                  QUOTE + "weight" + QUOTE + COLON +
                                        //                                  JSONencode(lv->getWeight()) +
                                        CLOSE_CURLY + COMMA;
                                }
                                else
                                    json_str += "linkProperties" + COLON + "{}" + COMMA;
                                // process its children
                                json_str += preOrder(root->children[k]);
                                json_str += CLOSE_CURLY + COMMA;
                            }
                        }
                        // remove last comma
                        json_str = json_str.substr(0, json_str.size() - 1);
                        // end of children
                        if (root->children.size())
                            json_str += CLOSE_BOX;
                    }

                    return json_str;
                }
        }; //end of TreeElement class

    }
}//end of bridges namespace
#endif