Home Research Papers Posters Manuals Talks Download People Help Internal

ck-com/ChunkMulticastStrategy.h

Go to the documentation of this file.

#ifndef CHUNK_MULTICAST_STRATEGY
#define CHUNK_MULTICAST_STRATEGY

#include "ComlibManager.h"
#include "ComlibSectionInfo.h"
#include <list>

/*class blkMsg1 {
public:
  double *data;
};*/

struct ChunkInfo {
  //int bcastPe;     ///< pe who is doing the broadcast, used for the hash key
  int chunkNumber;   
  int chunkSize;     
  int numChunks;   
  //int messageSize;   ///< the entire message size, all included 
  int idx;           
  int srcPe;       
};



struct recvBuffer {
  int numChunks;
  envelope** recvChunks;
  int nrecv;
  int srcPe;
  int idx;
};

class ChunkMulticastStrategy: public Strategy, public CharmStrategy {
 private:
  
  ComlibSectionInfo sinfo; // This is used to create the multicast messages themselves

  void remoteMulticast(ComlibMulticastMsg * multMsg, bool rootPE, int chunkNumber, int numChunks);
  void localMulticast(CharmMessageHolder *cmsg);
  
 public:

  //int numChunks;
  //int nrecv;
  std::list< recvBuffer* > recvList;
  int sentCount;

  virtual void determineNextHopPEs(const int totalDestPEs, const ComlibMulticastIndexCount* destPEs, const int myIndex, int * &pelist, int &npes, int chunkNumber, int numChunks);

    ChunkMulticastStrategy(CkMigrateMessage *m): Strategy(m), CharmStrategy(m){}
  
  ChunkMulticastStrategy();
  ~ChunkMulticastStrategy();
  
  void insertMessage(MessageHolder *msg) {insertMessage((CharmMessageHolder*)msg);}
  void insertMessage(CharmMessageHolder *msg);
  
  void handleMessage(void *msg);    

  void pup(PUP::er &p);

  PUPable_decl(ChunkMulticastStrategy);

};





class ChunkRingMulticastStrategy: public ChunkMulticastStrategy {
  
 public:
  void determineNextHopPEs(const int totalDestPEs, const ComlibMulticastIndexCount* destPEs, const int myIndex, int * &pelist, int &npes, int chunkNumber, int numChunks );

 ChunkRingMulticastStrategy(CkMigrateMessage *m): ChunkMulticastStrategy(m) {}
 ChunkRingMulticastStrategy(): ChunkMulticastStrategy() {}
  ~ChunkRingMulticastStrategy() {}
  
  void pup(PUP::er &p){ ChunkMulticastStrategy::pup(p); }
  
  PUPable_decl(ChunkRingMulticastStrategy);

};



class ChunkTreeMulticastStrategy: public ChunkMulticastStrategy {
 private:
  int degree;
  
 public:
  
  void determineNextHopPEs(const int totalDestPEs, const ComlibMulticastIndexCount* destPEs, const int myIndex, int * &pelist, int &npes, int chunkNumber, int numChunks );
  
 ChunkTreeMulticastStrategy(CkMigrateMessage *m): ChunkMulticastStrategy(m) {}

 ChunkTreeMulticastStrategy(int treeDegree=4): ChunkMulticastStrategy(), degree(treeDegree) {}

  ~ChunkTreeMulticastStrategy() {}
  
  void pup(PUP::er &p){ 
    ChunkMulticastStrategy::pup(p); 
    p | degree;
  }
  
  PUPable_decl(ChunkTreeMulticastStrategy);
};


class ChunkPipeTreeMulticastStrategy: public ChunkMulticastStrategy {
 private:
  int degree;
  
 public:
  
  void determineNextHopPEs(const int totalDestPEs, const ComlibMulticastIndexCount* destPEs, const int myIndex, int * &pelist, int &npes, int chunkNumber, int numChunks );
  
 ChunkPipeTreeMulticastStrategy(CkMigrateMessage *m): ChunkMulticastStrategy(m) {}

 ChunkPipeTreeMulticastStrategy(int treeDegree=4): ChunkMulticastStrategy(), degree(treeDegree) {}

  ~ChunkPipeTreeMulticastStrategy() {}
  
  void pup(PUP::er &p){ 
    ChunkMulticastStrategy::pup(p); 
    p | degree;
  }
  
  PUPable_decl(ChunkPipeTreeMulticastStrategy);
};







/*class ChunkNodeTreeMulticastStrategy: public ChunkMulticastStrategy {
 private:
  int degree;
  
 public:
  
  void determineNextHopPEs(const int totalDestPEs, const ComlibMulticastIndexCount* destPEs, const int myIndex, int * &pelist, int &npes );
  
 ChunkNodeTreeMulticastStrategy(CkMigrateMessage *m): ChunkMulticastStrategy(m) {}
  
 ChunkNodeTreeMulticastStrategy(int treeDegree=4): ChunkMulticastStrategy(), degree(treeDegree) {}
  
  ~ChunkNodeTreeMulticastStrategy() {}
  
  void pup(PUP::er &p){ 
    ChunkMulticastStrategy::pup(p); 
    p | degree;
  }
  
  PUPable_decl(ChunkNodeTreeMulticastStrategy);
};*/



/*class ChunkNodeTreeRingMulticastStrategy: public ChunkMulticastStrategy {
 private:
  int degree;
  
 public:
  
  void determineNextHopPEs(const int totalDestPEs, const ComlibMulticastIndexCount* destPEs, const int myIndex, int * &pelist, int &npes );
  
 ChunkNodeTreeRingMulticastStrategy(CkMigrateMessage *m): ChunkMulticastStrategy(m) {}
  
 ChunkNodeTreeRingMulticastStrategy(int treeDegree=4): ChunkMulticastStrategy(), degree(treeDegree) {}
  
  ~ChunkNodeTreeRingMulticastStrategy() {}
  
  void pup(PUP::er &p){ 
    ChunkMulticastStrategy::pup(p); 
    p | degree;
  }
  
  PUPable_decl(ChunkNodeTreeRingMulticastStrategy);
};*/





#endif

---