GiaNeighbors.h

Go to the documentation of this file.

//
// Copyright (C) 2006 Institut fuer Telematik, Universitaet Karlsruhe (TH)
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
//

#ifndef __GIANEIGHBORS_H_
#define __GIANEIGHBORS_H_

#include <vector>

#include <omnetpp.h>

#include <OverlayKey.h>
#include <InitStages.h>

#include "GiaNode.h"
#include "GiaKeyList.h"


struct GiaNeighborInfo
{
    unsigned int connectionDegree;
    unsigned int receivedTokens;
    unsigned int sentTokens;
    simtime_t timestamp;
    GiaKeyList keyList;
};

struct FullGiaNodeInfo
{
    GiaNode node;
    GiaNeighborInfo* info;
};


std::ostream& operator<<(std::ostream& os, const GiaNeighborInfo& info);

class GiaNeighbors : public cSimpleModule
{
  public:
    // OMNeT++ methodes
    virtual int numInitStages() const
    {
        return MAX_STAGE_OVERLAY + 1;
    }

    virtual void initialize( int stage );

    virtual void handleMessages( cMessage* msg );

    // class methodes
    virtual unsigned int getSize() const;

    virtual bool contains(const GiaNode& node) const;
    //virtual bool contains(const NodeHandle& node) const;

    virtual bool contains(const OverlayKey& key) const;

    virtual void add(const GiaNode& node, unsigned int degree);
    //virtual void add(const NodeHandle& node);

    virtual void remove(const GiaNode& node);

    virtual const GiaNode& get(unsigned int position);

    virtual const GiaNode& get(const OverlayKey& key);
    
    //bullshit
    GiaNeighborInfo* get(const GiaNode& node);

    void updateTimestamp(const GiaNode& node);

    void removeTimedoutNodes();

    void setNeighborKeyList(const GiaNode& node, const GiaKeyList& keyList);
    GiaKeyList* getNeighborKeyList(const GiaNode& node);

    double getCapacity(const GiaNode& node) const;
    //void setCapacity(const GiaNode& node, double capacity);

    void setConnectionDegree(const GiaNode& node, unsigned int degree);
    unsigned int getConnectionDegree(const GiaNode& node) const;

    void setReceivedTokens(const GiaNode& node, unsigned int tokens);
    void increaseReceivedTokens(const GiaNode& node);
    void decreaseReceivedTokens(const GiaNode& node);
    unsigned int getReceivedTokens(const GiaNode& node) const;

    void setSentTokens(const GiaNode& node, unsigned int tokens);
    void increaseSentTokens(const GiaNode& node);
    unsigned int getSentTokens(const GiaNode& node) const;

    const GiaNode& getDropCandidate(double capacity,
                                       unsigned int degree) const;

  protected:
    std::map<GiaNode, GiaNeighborInfo> neighbors; 
    typedef std::map<GiaNode, GiaNeighborInfo>::iterator NeighborsIterator;
    typedef std::map<GiaNode, GiaNeighborInfo>::const_iterator NeighborsConstIterator;
    GiaNode thisNode; 
    simtime_t timeout; 
};

#endif