Home Research Papers Posters Manuals Talks Download People Help Internal

conv-com/graphrouter.h

Go to the documentation of this file.

#ifndef _GRAPHROUTER_H
#define _GRAPHROUTER_H

#include <math.h>
//#include <converse.h>
#include "petable.h"

#include "persistent.h"

class TopologyDescriptor {
 public:
    //TopologyDescriptor(int npes, int mype)=0;
    //Entry methods which will define the specific graph.
    virtual void getNeighbors(int &npes, int *pelist) = 0;
    virtual int getNumStages() = 0;
    virtual int getNumSteps(int stage) = 0;
    virtual void getPesToSend(int step, int stage, int &npesToSend, 
                              int *pelist, int &nextpe) = 0;
    virtual int getNumMessagesExpected(int stage) = 0;
};


class GraphRouter : public Router
{
    PeTable *PeGraph;
    TopologyDescriptor *tp;
    
    int *pesToSend, *gpes;
    int numNeighbors, *neighborPeList, nstages;
    
    int MyPe, NumPes, currentIteration;
    int *recvExpected, *recvCount;
    int curStage, *stageComplete;

    void sendMessages(comID id, int stage);
    void init(int numPes, int myPe, TopologyDescriptor *tp);
    
#if CMK_PERSISTENT_COM
    PersistentHandle *handlerArrayOdd, *handlerArrayEven;
#endif          
    
 public:
    GraphRouter(int numPes, int myPe, Strategy*);
    //GraphRouter(int numPes, int myPe, int topid);
    ~GraphRouter();
  
    //Router enrty methods overridden here.

    void NumDeposits(comID, int);
    void EachToAllMulticast(comID , int , void *, int);
    void EachToManyMulticast(comID , int , void *, int, int *, int);
    void RecvManyMsg(comID, char *);
    void ProcManyMsg(comID, char *);
    void DummyEP(comID id, int);
    
    void SetMap(int *);
};

#endif

---