priorityarray.h

Go to the documentation of this file.

#ifndef N_PRIORITYARRAY_H
#define N_PRIORITYARRAY_H
//------------------------------------------------------------------------------
#include "core/types.h"
 
namespace Util
{
 
//------------------------------------------------------------------------------
template<class TYPE> class PriorityArray
{
public:
    PriorityArray(int size);
    PriorityArray(const PriorityArray<TYPE>& rhs);
    ~PriorityArray();
    PriorityArray<TYPE>& operator=(const PriorityArray<TYPE>& rhs);
    TYPE& operator[](int index) const;
    void Clear();
    void Add(const TYPE& elm, float pri);
    int Size() const;
    TYPE& At(int index);
    bool IsEmpty() const;
 
private:
    void UpdateMinPriElementIndex();
    void Copy(const PriorityArray<TYPE>& src);
    void Delete();
    void Destroy(TYPE* elm);
 
    struct Element
    {
        TYPE element;
        float priority;
    };
 
    int numElements;
    int maxElements;
    int minPriElementIndex;
    Element* elements;
};
 
//------------------------------------------------------------------------------
template<class TYPE>
PriorityArray<TYPE>::PriorityArray(int size) :
    numElements(0),
    maxElements(size),
    minPriElementIndex(0)
{
    n_assert(size > 0);
    this->elements = n_new_array(Element, size);
}
 
//------------------------------------------------------------------------------
template<class TYPE>
void
PriorityArray<TYPE>::Copy(const PriorityArray<TYPE>& src) 
{    
    n_assert(0 == this->elements);    
    this->numElements = src.numElements;
    this->maxElements = src.maxElements;
    this->minPriElementIndex = src.minPriElementIndex;
 
    this->elements = n_new_array(Element, this->maxElements);
    int i;
    for (i = 0; i < this->numElements; i++)
    {
        this->elements[i] = src.elements[i];
    }
}
 
//------------------------------------------------------------------------------
template<class TYPE>
void
PriorityArray<TYPE>::Delete()
{
    this->numElements = 0;
    this->maxElements = 0;
    this->minPriElementIndex = 0;
    if (this->elements)
    {
        n_delete_array(this->elements);
        this->elements = 0;
    }
}
 
//------------------------------------------------------------------------------
template<class TYPE>
PriorityArray<TYPE>::PriorityArray(const PriorityArray<TYPE>& rhs) :
    numElements(0),
    maxElements(0),
    minPriElementIndex(0),
    elements(0)
{
    this->Copy(rhs);
}
 
//------------------------------------------------------------------------------
template<class TYPE>
PriorityArray<TYPE>::~PriorityArray()
{
    this->Delete();
}
 
//------------------------------------------------------------------------------
template<class TYPE>
void
PriorityArray<TYPE>::Destroy(TYPE* elm)
{
    elm->~TYPE();
}
 
//------------------------------------------------------------------------------
template<class TYPE>
void
PriorityArray<TYPE>::Clear()
{
    n_assert(this->elements);
 
    // call element destructors
    int i;
    for (i = 0; i < this->numElements; i++)
    {
        this->Destroy(&(this->elements[i].element));
    }
    this->numElements = 0;
    this->minPriElementIndex = 0;
}
 
//------------------------------------------------------------------------------
template<class TYPE>
void
PriorityArray<TYPE>::UpdateMinPriElementIndex()
{
    int i;
    this->minPriElementIndex = 0;
    float minPri = this->elements[0].priority;
    for (i = 1; i < this->numElements; i++)
    {
        if (this->elements[i].priority < minPri)
        {
            minPri = this->elements[i].priority;
            this->minPriElementIndex = i;
        }
    }
}
 
//------------------------------------------------------------------------------
template<class TYPE>
void
PriorityArray<TYPE>::Add(const TYPE& elm, float pri)
{
    if (this->numElements < this->maxElements)
    {
        this->elements[this->numElements].element = elm;
        this->elements[this->numElements].priority = pri;
        this->numElements++;
        if (this->numElements == this->maxElements)
        {
            this->UpdateMinPriElementIndex();
        }
    }
    else
    {
        if (pri > this->elements[this->minPriElementIndex].priority)
        {
            this->elements[this->minPriElementIndex].element = elm;
            this->elements[this->minPriElementIndex].priority = pri;
            this->UpdateMinPriElementIndex();
        }
    }
}
 
//------------------------------------------------------------------------------
template<class TYPE>
int
PriorityArray<TYPE>::Size() const
{
    return this->numElements;
}
 
//------------------------------------------------------------------------------
template<class TYPE>
TYPE&
PriorityArray<TYPE>::At(int index)
{
    n_assert((index >= 0) && (index < this->numElements));
    return this->elements[index].element;
}
 
//------------------------------------------------------------------------------
template<class TYPE>
TYPE&
PriorityArray<TYPE>::operator[](int index) const
{
    n_assert((index >= 0) && (index < this->numElements));
    return this->elements[index].element;
}
 
//------------------------------------------------------------------------------
template<class TYPE>
PriorityArray<TYPE>& 
PriorityArray<TYPE>::operator=(const PriorityArray<TYPE>& rhs)
{
    this->Delete();
    this->Copy(rhs);
    return *this;
}
 
//------------------------------------------------------------------------------
template<class TYPE>
bool
PriorityArray<TYPE>::IsEmpty() const
{
    return (0 == this->numElements);
}
 
} // namespace Util
//------------------------------------------------------------------------------
#endif