cmsgheap.h

//==========================================================================
//  CMSGHEAP.H - part of
//                     OMNeT++/OMNEST
//            Discrete System Simulation in C++
//
//
//  Declaration of the following classes:
//    cMessageHeap : future event set, implemented as heap
//
//==========================================================================

/*--------------------------------------------------------------*
  Copyright (C) 1992-2005 Andras Varga

  This file is distributed WITHOUT ANY WARRANTY. See the file
  `license' for details on this and other legal matters.
*--------------------------------------------------------------*/

#ifndef __CMSGHEAP_H
#define __CMSGHEAP_H

#include "cobject.h"

class cMessage;

//==========================================================================

class SIM_API cMessageHeap : public cObject
{
  public:
    class Iterator
    {
      private:
        cMessageHeap *q;
        int pos;

      public:
        Iterator(const cMessageHeap& mh)  {q=const_cast<cMessageHeap*>(&mh);pos=1;}

        void init(const cMessageHeap& mh) {q=const_cast<cMessageHeap*>(&mh);pos=1;}

        cMessage *operator()()      {return q->h[pos];}

        cMessage *operator++(int)   {return pos<=q->n ? q->h[++pos] : NULL;}

        bool end() const   {return (bool)(pos>q->n);}
    };

    friend class Iterator;

  private:
    cMessage **h;             // pointer to the 'heap'  (h[0] always empty)
    int n;                    // number of elements in the heap
    int size;                 // size of allocated h array
    unsigned long insertcntr; // counts insertions

    // internal
    void shiftup(int from=1);

  public:

    cMessageHeap(const cMessageHeap& msgq);

    cMessageHeap(const char *name=NULL, int size=128);

    virtual ~cMessageHeap();

    cMessageHeap& operator=(const cMessageHeap& msgqueue);


    virtual cObject *dup() const  {return new cMessageHeap(*this);}

    virtual std::string info() const;

    virtual void forEachChild(cVisitor *v);

    // no netPack() and netUnpack()


    void insert(cMessage *event);

    cMessage *peekFirst() const;

    cMessage *getFirst();

    cMessage *get(cMessage *event);

    void sort();

    void clear();

    int length() const {return n;}

    bool empty() const {return n==0;}
};

#endif

---