Home Research Papers Posters Manuals Talks Download People Help Internal

ck-com/RectMulticastStrategy.h

Go to the documentation of this file.

#ifndef RECT_MULTICAST_STRATEGY
#define RECT_MULTICAST_STRATEGY


// only works on BG/L Native Layer

#include "ComlibManager.h"
#include "ComlibSectionInfo.h"

typedef CkHashtableTslow < unsigned int, void *> comRectHashType;
//typedef CkHashtableT < unsigned int, int > comRectHashType;

CkpvExtern(comRectHashType *, com_rect_ptr);


// need place to record 'src in rectangle' state
class ComlibRectSectionHashObject : public ComlibSectionHashObject
{
 public:
  bool sourceInRectangle;
  int cornerRoot;
  CkArrayID aid;
};

void *DMHandler(void *msg);
#ifdef CMK_RECT_API
#include "bgml.h"
class RectMulticastStrategy: public Strategy, public CharmStrategy {
 protected:
    int handlerId;    
    ComlibSectionInfo sinfo;

    CkHashtableT<ComlibSectionHashKey, ComlibRectSectionHashObject *> sec_ht; 


    inline unsigned int computeKey(int sectionID, int srcPe, CkArrayID arrId) {
      //      fprintf(stderr,"key %d %d %d = %d\n",(((CkGroupID) arrId).idx), (sectionID), srcPe, (((CkGroupID) arrId).idx << 24) | (sectionID<<16)| srcPe);
      return ((unsigned int) (((CkGroupID) arrId).idx << 24) | (sectionID<<16)| srcPe);}
    
    inline unsigned int computeKey(int sectionID, int srcPe, int id) { 
      //      fprintf(stderr,"key %d %d %d = %d", id ,sectionID,  srcPe, (id  << 24) | (sectionID<<16)| srcPe);
      return ((unsigned int) (id  << 24) | (sectionID<<16)| srcPe);}
    
    void insertSectionID(CkSectionID *sid);

    virtual ComlibRectSectionHashObject *createObjectOnSrcPe(int nindices, CkArrayIndex *idx_list, unsigned int sectionid);

    virtual ComlibRectSectionHashObject *createObjectOnIntermediatePe(int nindices, CkArrayIndex *idxlist, int npes, ComlibMulticastIndexCount *counts, int srcpe, int sectionID);
        
    virtual int needSorting() { return 0; }

    void localMulticast(envelope *env, ComlibRectSectionHashObject *obj);
    
    virtual void remoteMulticast(envelope *env, ComlibRectSectionHashObject *obj);

    virtual void forwardMulticast(envelope *env, ComlibRectSectionHashObject *obj);


    void handleNewMulticastMessage(envelope *env);

    void sendRectDest(ComlibRectSectionHashObject *obj, int srcpe, envelope *env);
    BGTsRC_Geometry_t *getRectGeometry(ComlibRectSectionHashObject *obj, int srcpe);
    int assignCornerRoot(BGTsRC_Geometry_t *geom, int srcpe);

 public:



    RectMulticastStrategy(CkMigrateMessage *m): Strategy(m), CharmStrategy(m){}
                
    RectMulticastStrategy(CkArrayID aid);
        
        
    //Destuctor
    ~RectMulticastStrategy();
        
    virtual void insertMessage(CharmMessageHolder *msg);
    virtual void doneInserting();

    virtual void handleMessage(void *msg);    
    virtual void handleMessageForward(void *msg);    

    virtual void pup(PUP::er &p);    
    virtual void beginProcessing(int nelements);
    
    PUPable_decl(RectMulticastStrategy);
};
#else
class RectMulticastStrategy : public Strategy, public CharmStrategy {   
  RectMulticastStrategy(CkMigrateMessage *m): Strategy(m), CharmStrategy(m){}
  //    RectMulticastStrategy(CkArrayID aid){}
  ~RectMulticastStrategy(){}
  void insertMessage(MessageHolder*) {}
  void handleMessage(void*) {}
  void pup(PUP::er &p) {}
  PUPable_decl(RectMulticastStrategy);   
};
#endif


#endif

---