AVLRedBlackNode.h

Go to the documentation of this file.

#ifndef MMANN_AVLREDBLACKNODE_H
#define MMANN_AVLREDBLACKNODE_H


#include <assert.h>
#include "AGenericTree.h"

enum RedBlackNodeColor {BLACK, RED};

template <class Data>
class AVLRedBlackNode
{
public:
    AVLRedBlackNode(DeleteTreeDataEnum DeleteTreeData);

    virtual ~AVLRedBlackNode();

    AVLRedBlackNode<Data>* m_Left;

    AVLRedBlackNode<Data>* m_Right;

    RedBlackNodeColor m_Color;

    Data* m_Data;

    DeleteTreeDataEnum m_DeleteTreeData;
};


template <class Data>
AVLRedBlackNode<Data>::AVLRedBlackNode(DeleteTreeDataEnum DeleteTreeData)
{
    m_Left = m_Right = NULL;
    m_Color = BLACK;
    m_Data = NULL;
    m_DeleteTreeData = DeleteTreeData;
}

template <class Data>
AVLRedBlackNode<Data>::~AVLRedBlackNode()
{
    switch(m_DeleteTreeData)
    {
    case NO_DELETE_TREE_ITEM:
        break;
    case DELETE_TREE_ITEM:
        delete m_Data;
        break;
    case DELETE_TREE_ITEM_ARRAY:
        delete [] m_Data;
        break;
    default:
        //This shouldn't happen
        assert(false);
    }
}

#endif

---