ClusterTreeNode.h

Go to the documentation of this file.

#ifndef GRT_CLUSTER_TREE_NODE_HEADER
#define GRT_CLUSTER_TREE_NODE_HEADER

#include "../../CoreAlgorithms/Tree/Node.h"

GRT_BEGIN_NAMESPACE
    
class ClusterTreeNode : public Node{
public:
    ClusterTreeNode(){
        nodeType = "ClusterTreeNode";
        parent = NULL;
        leftChild = NULL;
        rightChild = NULL;
        clear();
    }
    
    virtual ~ClusterTreeNode(){
        clear();
    }
    
    virtual bool predict(const VectorFloat &x){
        if( x[ featureIndex ] >= threshold ) return true;
        return false;
    }
    
    virtual bool predict(const VectorFloat &x,VectorFloat &y){
        
        if( isLeafNode ){
            if( y.size() != 1 ) y.resize( 1 );
            y[0] = clusterLabel;
            return true;
        }
        
        if( leftChild == NULL && rightChild == NULL )
            return false;
        
        if( predict( x ) ){
            if( rightChild )
                return rightChild->predict( x, y );
        }else{
            if( leftChild )
                return leftChild->predict( x, y );
        }
        
        return false;
    }
    
    virtual bool clear(){
        
        //Call the base class clear function
        Node::clear();
        
        nodeSize = 0;
        featureIndex = 0;
        threshold = 0;
        clusterLabel = 0;
        
        return true;
    }
    
    virtual bool print() const{
        
        std::string tab = "";
        for(UINT i=0; i<depth; i++) tab += "\t";
        
        std::cout << tab << "depth: " << depth << " nodeSize: " << nodeSize << " featureIndex: " << featureIndex << " threshold " << threshold << " isLeafNode: " << isLeafNode << std::endl;
        std::cout << tab << "ClusterLabel: " << clusterLabel << std::endl;
        
        if( leftChild != NULL ){
            std::cout << tab << "LeftChild: " << std::endl;
            leftChild->print();
        }
        
        if( rightChild != NULL ){
            std::cout << tab << "RightChild: " << std::endl;
            rightChild->print();
        }
        
        return true;
    }
    
    virtual Node* deepCopyNode() const{
        
        ClusterTreeNode *node = new ClusterTreeNode;
        
        if( node == NULL ){
            return NULL;
        }
        
        //Copy this node into the node
        node->depth = depth;
        node->isLeafNode = isLeafNode;
        node->nodeSize = nodeSize;
        node->featureIndex = featureIndex;
        node->threshold = threshold;
        node->clusterLabel = clusterLabel;
        
        //Recursively deep copy the left child
        if( leftChild ){
            node->leftChild = leftChild->deepCopyNode();
            node->leftChild->setParent( node );
        }
        
        //Recursively deep copy the right child
        if( rightChild ){
            node->rightChild = rightChild->deepCopyNode();
            node->rightChild->setParent( node );
        }
        
        return (Node*)node;
    }
    
    ClusterTreeNode* deepCopyTree() const{
        ClusterTreeNode *node = (ClusterTreeNode*)deepCopyNode();
        return node;
    }
    
    UINT getNodeSize() const{
        return nodeSize;
    }
    
    UINT getFeatureIndex() const{
        return featureIndex;
    }
    
    Float getThreshold() const{
        return threshold;
    }
    
    UINT getClusterLabel() const{
        return clusterLabel;
    }
    
    bool set(const UINT nodeSize,const UINT featureIndex,const Float threshold,const UINT clusterLabel){
        this->nodeSize = nodeSize;
        this->featureIndex = featureIndex;
        this->threshold = threshold;
        this->clusterLabel = clusterLabel;
        return true;
    }
    
protected:
    virtual bool saveParametersToFile(std::fstream &file) const{
        
        if(!file.is_open())
        {
            errorLog << "saveParametersToFile(fstream &file) - File is not open!" << std::endl;
            return false;
        }
        
        //Save the custom ClusterTreeNode parameters
        file << "NodeSize: " << nodeSize << std::endl;
        file << "FeatureIndex: " << featureIndex << std::endl;
        file << "Threshold: " << threshold << std::endl;
        file << "ClusterLabel: " << clusterLabel << std::endl;
        
        return true;
    }
    
    virtual bool loadParametersFromFile(std::fstream &file){
        
        if(!file.is_open())
        {
            errorLog << "loadFromFile(fstream &file) - File is not open!" << std::endl;
            return false;
        }
        
        std::string word;
        
        //Load the custom ClusterTreeNode Parameters
        file >> word;
        if( word != "NodeSize:" ){
            errorLog << "loadParametersFromFile(fstream &file) - Failed to find NodeSize header!" << std::endl;
            return false;
        }
        file >> nodeSize;
        
        file >> word;
        if( word != "FeatureIndex:" ){
            errorLog << "loadParametersFromFile(fstream &file) - Failed to find FeatureIndex header!" << std::endl;
            return false;
        }
        file >> featureIndex;
        
        file >> word;
        if( word != "Threshold:" ){
            errorLog << "loadParametersFromFile(fstream &file) - Failed to find Threshold header!" << std::endl;
            return false;
        }
        file >> threshold;
        
        file >> word;
        if( word != "ClusterLabel:" ){
            errorLog << "loadParametersFromFile(fstream &file) - Failed to find ClusterLabel header!" << std::endl;
            return false;
        }
        file >> clusterLabel;
        
        return true;
    }
    
    UINT clusterLabel;
    UINT nodeSize;
    UINT featureIndex;
    Float threshold;
    
    static RegisterNode< ClusterTreeNode > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_CLUSTER_TREE_NODE_HEADER