pzstack.h

Go to the documentation of this file.

#ifndef PZSTACK_H
#define PZSTACK_H

#include <iostream>
#include <cstdlib>
#include "pzerror.h"
#include "pzmanvector.h"

template< class T, int NumExtAlloc = DEFAULTVEC_ALLOC >
class TPZStack : public TPZManVector< T, NumExtAlloc >
{
public:
                T & Peek() const;
                
                TPZStack();
    
    TPZStack(int nel, const T Object) : TPZManVector<T,NumExtAlloc>(nel,Object)
    {
        
    }
                
                void Push(const T object);
    void push_back(const T object)
    {
        Push(object);
    }
                
                T Pop();
                
                operator T*() const { return this->fStore; }
};

//--| IMPLEMENTATION |----------------------------------------------------------

template<class T, int NumExtAlloc >
TPZStack<T, NumExtAlloc>::TPZStack() : TPZManVector<T, NumExtAlloc>(0) {
                this->Expand(NumExtAlloc);
}

// Puts an object on the stack
template<class T, int NumExtAlloc >
void TPZStack<T, NumExtAlloc>::Push(const T object) {
                this->Resize(this->NElements()+1);
                this->operator[](this->NElements()-1) = object;
}

// Retrieve an object from the stack
template<class T, int NumExtAlloc >
T TPZStack<T, NumExtAlloc>::Pop() {
                this->fNElements--;
                if(this->fNElements <0){
                                this->fNElements = 0;
                                PZError << "TPZStack popping beyond the stack object" << std::endl;
                                PZError.flush();
                                T temp(0);
                                return temp;
                }
                return this->fStore[this->fNElements];
}

template <class T, int NumExtAlloc >
T & TPZStack<T, NumExtAlloc>::Peek() const {
                if(this->NElements() <= 0) {
                                PZError << "TPZStack peek beyond the stack object" << std::endl;
                                PZError.flush();
                                exit(-1);
                }
                return this->operator[](this->NElements()-1);
}

#endif // PZSTACK_H