Home Research Papers Posters Manuals Talks Download People Help Internal

ck-core/register.h

Go to the documentation of this file.

#ifndef _REGISTER_H
#define _REGISTER_H


class EntryInfo {
  public:
    const char *name;
    
    CkCallFnPtr call;
    int msgIdx;
    int chareIdx;
    CmiBool traceEnabled; 
    CmiBool noKeep; 
    CmiBool inCharm;
    
#ifdef ADAPT_SCHED_MEM
   CmiBool isMemCritical;
#endif

    CkMarshallUnpackFn marshallUnpack;
    
    CkMessagePupFn messagePup;

    EntryInfo(const char *n, CkCallFnPtr c, int m, int ci) : 
      name(n), call(c), msgIdx(m), chareIdx(ci), 
      marshallUnpack(0), messagePup(0)
    { traceEnabled=CmiTrue; noKeep=CmiFalse; inCharm=CmiFalse;}
};

class MsgInfo {
  public:
    const char *name;
    CkPackFnPtr pack;
    CkUnpackFnPtr unpack;
    CkDeallocFnPtr dealloc;
    size_t size;

    MsgInfo(const char *n,CkPackFnPtr p,CkUnpackFnPtr u,CkDeallocFnPtr d,int s):
      name(n), pack(p), unpack(u), dealloc(d), size(s)
    {}
};

class ChareInfo {
  public:
    const char *name;
    int size;
    
    int defCtor,migCtor; 
    int numbases;
    int bases[16];
    
    int isIrr;
    
    ChareType  chareType;

    CmiBool inCharm;

    int mainChareIdx;      

    ChareInfo(const char *n, int s, ChareType t) : name(n), size(s) {
      defCtor=migCtor=-1;
      isIrr = numbases = 0;
      chareType = t;
      inCharm = CmiFalse;
      mainChareIdx = -1;
    }
    void setDefaultCtor(int idx) { defCtor = idx; }
    int getDefaultCtor(void) { return defCtor; }
    void setMigCtor(int idx) { migCtor = idx; }
    int getMigCtor(void) { return migCtor; }
    void addBase(int idx) { bases[numbases++] = idx; }
    void setInCharm() { inCharm = CmiTrue; }
    CmiBool isInCharm() { return inCharm; }
    void setMainChareType(int idx) { mainChareIdx = idx; }
    int mainChareType() { return mainChareIdx; }
};

class MainInfo {
  public:
    const char *name;
    int chareIdx;
    int entryIdx;
    int entryMigCtor;
    void* obj; // real type is Chare*
    MainInfo(int c, int e) : name("main"), chareIdx(c), entryIdx(e) {}
    inline void* getObj(void) { return obj; }
    inline void setObj(void *_obj) { obj=_obj; }
};

class ReadonlyInfo {
  public:
    const char *name,*type;
    int size;
    void *ptr;
    CkPupReadonlyFnPtr pup;
    
    void pupData(PUP::er &p) {
      if (pup!=NULL)
        (pup)((void *)&p);
      else
        p((char *)ptr,size);
    }
    ReadonlyInfo(const char *n,const char *t,
         int s, void *p,CkPupReadonlyFnPtr pf) 
        : name(n), type(t), size(s), ptr(p), pup(pf) {}
};

class ReadonlyMsgInfo {
  public:
    const char *name, *type;
    void **pMsg;
    ReadonlyMsgInfo(const char *n, const char *t,
        void **p) : name(n), type(t), pMsg(p) {}
};

template <class T>
class CkRegisteredInfo {
        CkVec<T *> vec;
        
        void outOfBounds(int idx) {
                const char *exampleName="";
                if (vec.size()>0) exampleName=vec[0]->name;
                CkPrintf("register.h> CkRegisteredInfo<%d,%s> called with invalid index "
                        "%d (should be less than %d)\n", sizeof(T),exampleName,
                        idx, vec.size());
                CkAbort("Registered idx is out of bounds-- is message or memory corrupted?");
        }
public:
        CkRegisteredInfo() :vec(CkSkipInitialization()) {}
        ~CkRegisteredInfo() {
                for (size_t i=0; i<vec.size(); i++) if (vec[i]) delete vec[i];
        } 

        int add(T *t) {
#if CMK_ERROR_CHECKING
                /* Make sure registrations only happen from rank 0: */
                if (CkMyRank()!=0)
                        CkAbort("Can only do registrations from rank 0 processors");
#endif
                vec.push_back(t);
                return vec.size()-1;
        }
        
        size_t &size(void) {return vec.length();}
        
        T *operator[](size_t idx) {
#if CMK_ERROR_CHECKING
                /* Bounds-check the index: */
                if (idx>=vec.size()) outOfBounds(idx);
#endif
                return vec[idx];
        }
};

extern CkRegisteredInfo<EntryInfo> _entryTable;
extern CkRegisteredInfo<MsgInfo> _msgTable;
extern CkRegisteredInfo<ChareInfo> _chareTable;
extern CkRegisteredInfo<MainInfo> _mainTable;
extern CkRegisteredInfo<ReadonlyInfo> _readonlyTable;
extern CkRegisteredInfo<ReadonlyMsgInfo> _readonlyMsgs;

extern void _registerInit(void);
extern void _registerDone(void);

#endif

---