KdTreeElement.h

Go to the documentation of this file.

#ifndef KD_TREE_ELEMENT_H
#define KD_TREE_ELEMENT_H

#include <sstream>

#include "BSTElement.h" //string, sstream, using std

#include <JSONutil.h>

namespace bridges {
    namespace datastructure {
        template <typename K, typename E>
        class KdTreeElement : public BSTElement<K, E> {
            private:
                int dimension = 0;      // dimension, in the range [0..k-1]
                float thickness = 0;        // thickness of partitioning lines
            public:
                KdTreeElement(const K& k, int dim, int th,
                    const KdTreeElement* l, const KdTreeElement* r,
                    const E& val = E(), const string& lab = string())
                    : BSTElement<K, E>(k, l, r, val, lab),
                      thickness(th), dimension(dim) {
                }
                KdTreeElement(const K& k, int dim, int th = 0,
                    const E& val = E(), const string& lab = string())
                    : BSTElement<K, E>(k, nullptr, nullptr, val, lab),
                      dimension(dim), thickness(th) {
                }
                virtual const string getDStype() const override {
                    return "KdTree";
                }
                int getDimension() const {
                    return dimension;
                }
                void setDimension(const int& dim) {
                    dimension = dim;
                }
                float getThickness() const {
                    return thickness;
                }
                void setThickness(const float& th) {
                    thickness = th;
                }
                K getPartitioner() const {
                    return this->getKey();
                }
                void setPartitioner(const K& p) {
                    this->setKey();
                }
                virtual KdTreeElement* getLeft() override {
                    return static_cast<KdTreeElement*>(BSTElement<K, E>::getLeft());
                }
                virtual const KdTreeElement* getLeft() const override {
                    return static_cast<const KdTreeElement*>(BSTElement<K, E>::getLeft());
                }
                void setLeft(KdTreeElement* l) {
                    BSTElement<K, E>::setLeft(l);
                }
                virtual KdTreeElement* getRight() override {
                    return static_cast<KdTreeElement*>(BSTElement<K, E>::getRight());
                }
                virtual const KdTreeElement* getRight() const override {
                    return static_cast<const KdTreeElement*>(BSTElement<K, E>::getRight());
                }
                void setRight(KdTreeElement* r) {
                    BSTElement<K, E>::setRight(r);
                }
            protected:
                virtual const string getElementRepresentation() const override {
                    using bridges::JSONUtil::JSONencode;
                    string json =  BSTElement<K, E>::getElementRepresentation();
                    stringstream dim_str, thck_str;
                    dim_str << dimension;
                    thck_str << thickness;
                    return json.insert(
                            json.size() - 1,
                            COMMA + QUOTE + "dimension" + QUOTE + COLON +
                            JSONencode(dim_str.str()) +
                            COMMA + QUOTE + "thickness" + QUOTE + COLON +
                            JSONencode(thck_str.str())
                        );
                }
        }; //end of KdtreeElement class
    }
}//end of bridges namespace
#endif