Home Research Papers Posters Manuals Talks Download People Help Internal

conv-com/pipebroadcastconverse.h

Go to the documentation of this file.

#ifndef PIPE_BROADCAST_CONVERSE
#define PIPE_BROADCAST_CONVERSE
#include "ckhashtable.h"
//#include "charm++.h"
//#include "envelope.h"
#include "convcomlibmanager.h"

#define DEFAULT_PIPE   8196

CkpvExtern(int, pipeline_handler);
extern void PipelineHandler(void *msg);
CkpvExtern(int, pipeline_frag_handler);
extern void PipelineFragmentHandler(void *msg);

struct PipeBcastInfo {
  short bcastPe;     
  short seqNumber;   
  int chunkSize;     
  int chunkNumber;   
  int messageSize;   
  short srcPe;       
};

class PipeBcastHashKey {
 public:

    int srcPe;
    int seq;
    PipeBcastHashKey(int _pe, int _seq):srcPe(_pe), seq(_seq){};

    //These routines allow PipeBcastHashKey to be used in
    //  a CkHashtableT
    CkHashCode hash(void) const;
    static CkHashCode staticHash(const void *a,size_t);
    int compare(const PipeBcastHashKey &ind) const;
    static int staticCompare(const void *a,const void *b,size_t);
};

inline CkHashCode PipeBcastHashKey::hash(void) const {
    register int _seq = seq;
    register int _pe = srcPe;
    
    register CkHashCode ret = (_seq << 16) + _pe;
    return ret;
}

inline int PipeBcastHashKey::compare(const PipeBcastHashKey &k2) const {
    if(seq == k2.seq && srcPe == k2.srcPe)
        return 1;
    
    return 0;
}

class PipeBcastHashObj {
 public:
  char *message;
  int dimension; 
  int remaining; 
  PipeBcastHashObj (int dim, int rem, char *msg) :dimension(dim),remaining(rem),message(msg) {};

};

//typedef const UInt  constUInt;
//typedef void (*setFunction)(char*, constUInt);

class PipeBroadcastConverse : public Strategy {
 protected:

  int pipeSize; 
  short topology; 
  //double log_of_2_inv;
  CmiUInt2 seqNumber; 
  //CkQ <MessageHolder*> *messageBuf;

  CkHashtableT<PipeBcastHashKey, PipeBcastHashObj *> fragments;
  //int propagateHandle;
  //int propagateHandle_frag;

  // WARNING: All pure converse messages need to insert a "CmiFragmentHeader"
  // right after the ConverseHeader at the beginning of the user data.

  virtual CmiFragmentHeader *getFragmentHeader(char *msg);

 public:
  PipeBroadcastConverse(short top=USE_HYPERCUBE, int size=DEFAULT_PIPE);
  PipeBroadcastConverse(CkMigrateMessage *m): Strategy(m) {};
  int getPipeSize() { return pipeSize; };

  //void commonInit();
  // the deliver function deletes the message afterwards
  virtual void deliver(char *msg, int dim);

  void handleMessage(void*) {
    CmiAbort("PipeBroadcastConverse::handleMessage, this should never be used!\n");
  }

  void store(char *msg);

  void propagate(char *msg, int isFrag);//, int srcPeNumber, int totalSendingSize, setFunction setPeNumber);

  //void conversePipeBcast(char *env, int size);
  void insertMessage(MessageHolder *msg);
  //void doneInserting();

  virtual void pup(PUP::er &p);
  PUPable_decl(PipeBroadcastConverse);
};

#endif

---