RBtreeNode.h

#pragma once


#include <libutl/BinTreeNode.h>


UTL_NS_BEGIN;



class RBtreeNode : public BinTreeNode
{
public:
    RBtreeNode()
    {
        init();
    }

    RBtreeNode(const Object* object, uint_t color = RED)
        : BinTreeNode(object)
    {
        init(color);
    }

    RBtreeNode*
    left() const
    {
        return static_cast<RBtreeNode*>(_left);
    }
    RBtreeNode*
    next() const
    {
        return static_cast<RBtreeNode*>(BinTreeNode::next());
    }
    RBtreeNode*
    parent() const
    {
        return static_cast<RBtreeNode*>(_parent);
    }
    RBtreeNode*
    prev() const
    {
        return static_cast<RBtreeNode*>(BinTreeNode::prev());
    }
    RBtreeNode*
    right() const
    {
        return static_cast<RBtreeNode*>(_right);
    }
    virtual void addFixup();
    virtual void removeFixup(const BinTreeNode* removedNode);
    virtual void swapWith(BinTreeNode* rhs);

    uint_t
    getColor() const
    {
        return (uint_t)_color;
    }

    void
    setColor(uint_t color)
    {
        _color = color;
    }

    RBtreeNode*
    sibling() const
    {
        return (RBtreeNode*)(BinTreeNode::sibling());
    }

    enum color_t
    {
        BLACK, 
        RED    
    };

private:
    void init(uint_t color = RED);
    void
    deInit()
    {
    }

private:
    enum flg_t
    {
        flg_color
    };
    byte_t _color;
};


UTL_NS_END;