Home Research Papers Posters Manuals Talks Download People Help Internal

ck-com/ComlibSectionInfo.h

Go to the documentation of this file.

#ifndef COMLIB_SECTION_INFO
#define COMLIB_SECTION_INFO

/***********
  Helper classes that help strategies manage array sections 
***********/

class ComlibSectionHashKey{
 public:
    int srcPe;
    int id;
    ComlibSectionHashKey(int _pe, int _id):srcPe(_pe), id(_id){};

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

/*********** CkHashTable functions ******************/
inline CkHashCode ComlibSectionHashKey::hash(void) const
{
    register int _id = id;
    register int _pe = srcPe;
    
    register CkHashCode ret = (_id << 16) + _pe;
    return ret;
}

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

inline int intCompare(const void *a, const void *b){
    int a1 = *(int *) a;
    int b1 = *(int *) b;

    if(a1 < b1)
        return -1;
    
    if(a1 == b1)
        return 0;

    if(a1 > b1)
        return 1;

    return 0;
}    

//ComlibSectionHashKey CODE
inline int ComlibSectionHashKey::staticCompare(const void *k1, const void *k2, 
                                                size_t ){
    return ((const ComlibSectionHashKey *)k1)->
                compare(*(const ComlibSectionHashKey *)k2);
}

inline CkHashCode ComlibSectionHashKey::staticHash(const void *v,size_t){
    return ((const ComlibSectionHashKey *)v)->hash();
}

/*************** End CkHashtable Functions *****************/


class ComlibSectionHashObject {
 public:
    //My local indices
    CkVec<CkArrayIndex> indices;
    
    //Other processors to send this message to
    int npes;
    int *pelist;

    void *msg;

    //Flags associated with the section
    int isOld; // 1 if the section is indeed old
    
    ComlibSectionHashObject(): indices(0), isOld(0) {
        npes = 0;
        pelist = NULL;
        msg = NULL;
    }

    ~ComlibSectionHashObject() {
        delete [] pelist;
        //delete msg;
    }
};


class ComlibSectionInfo {
 
  int MaxSectionID;

 public:

    ComlibSectionInfo() { 
      MaxSectionID = 1;
    }
   
    inline void initSectionID(CkSectionID *sid) {
      if (MaxSectionID > 65000) {
        CkAbort("Too many sections allocated, wrapping of ints should be done!\n");
      }
      ComlibPrintf("[%d] ComlibSectionInfo::initSectionID: creating section number %d for proc %d\n",
          CkMyPe(), MaxSectionID, sid->_cookie.get_pe());
      sid->_cookie.info.sInfo.cInfo.id = MaxSectionID ++;
      sid->_cookie.get_pe() = CkMyPe();
    }
    
    void processOldSectionMessage(CharmMessageHolder *cmsg);

    CkMcastBaseMsg *getNewDeliveryErrorMsg(CkMcastBaseMsg *base);

    ComlibMulticastMsg *getNewMulticastMessage(CharmMessageHolder *cmsg, int needSort, int instanceID);

    void unpack(envelope *cb_env, int &nLocalElems, CkArrayIndex *&destIndices, 
                envelope *&env);

    void localMulticast(envelope *env);

    void getPeCount(int nindices, CkArrayIndex *idxlist, const CkArrayID &destArrayID,
                    int &npes, int &nidx,
                    ComlibMulticastIndexCount *&counts, int *&belongs);

    void getPeList(envelope *cb_env, int npes, int *&pelist);

    void getRemotePelist(int nindices, CkArrayIndex *idxlist, CkArrayID &destArrayID,
                         int &npes, int *&pelist);

    void getPeList(int nindices, CkArrayIndex *idxlist, CkArrayID &destArrayID,
                   int &npes, int *&pelist);

    void getLocalIndices(int nindices, CkArrayIndex *idxlist, CkArrayID &destArrayID,
                         CkVec<CkArrayIndex> &idx_vec);   
        
    void getNodeLocalIndices(int nindices, CkArrayIndex *idxlist, CkArrayID &destArrayID,
                         CkVec<CkArrayIndex> &idx_vec);   
        
    void pup(PUP::er &p) {
        p | MaxSectionID;
    }
};

#endif

---