Home Research Papers Posters Manuals Talks Download People Help Internal

libs/ck-libs/ParFUM/ParFUM_SA.h

Go to the documentation of this file.

#ifndef __PARFUM_SA_H
#define __PARFUM_SA_H

#include "tcharm.h"
#include "charm++.h"
#include "ParFUM.h"
#include "idxl.h"
#include "ParFUM_internals.h"


class RegionID{
public:
    int chunkID;
    int localID;
    double prio;
    operator CkHashCode() const {
        return (CkHashCode )(localID + 1<< chunkID); 
    };
    void pup(PUP::er &p) {
      p|chunkID; p|localID; p|prio;
    } 
    inline bool operator==(const RegionID &rhs) const {
      return (rhs.chunkID == chunkID) && (rhs.localID == localID);
    }
    inline RegionID& operator=(const RegionID& rhs) {
      chunkID = rhs.chunkID;
      localID = rhs.localID;
      prio = rhs.prio;
      return *this;
    }
};
//PUPbytes(RegionID);

class LockRegion {
public:
    RegionID myID;
    CkVec<int> localNodes;
    CkVec<int> sharedIdxls;
    CkHashtableT<CkHashtableAdaptorT<int>, CkVec<adaptAdj> *> remoteElements;
    CthThread tid;
    int numReplies;
    int success;
    ~LockRegion(){
        CkHashtableIterator *iter = remoteElements.iterator();
        while(iter->hasNext()){
            CkVec<adaptAdj> *list = *((CkVec<adaptAdj> **)iter->next());
            //printf("[%d] LockRegion destructor deleting list %p\n",myID.chunkID,list);
            delete list;
        }
    }
};

#include "ParFUM_SA.decl.h"


class BulkAdapt;
class ParFUMShadowArray : public CBase_ParFUMShadowArray {
 private:
  int numChunks;
  int idx;
  TCharm *tc;
  CProxy_TCharm tproxy;
  FEM_Mesh *fmMesh;
  CkHashtableT<CkHashtableAdaptorT<RegionID>,LockRegion *> regionTable;
  int regionCount;
  
 public:
  BulkAdapt *bulkAdapt;
  RegionID holdingLock;
  RegionID pendingLock;
  
 public:
  ParFUMShadowArray(int s, int i);
  ParFUMShadowArray(CkMigrateMessage *m);
  ~ParFUMShadowArray();

  void pup(PUP::er &p);
  void ckJustMigrated(void);

  int getNumChunks(){return numChunks;}
  int getIdx(){return idx;}
  FEM_Mesh *getfmMesh(){return fmMesh;}

  void setFemMesh(FEMMeshMsg *m);

  void setRunningTCharm(){CtvAccess(_curTCharm)= tc;};

  void sort(int *chkList, int chkListSize);
  
  //Lock all the nodes belonging to these elements
  //Some of the elements might be remote
  //You also have to lock idxls with all those chunks with which
  //any of the nodes are shared
  int lockRegion(int numElements,adaptAdj *elements,RegionID *regionID, double prio);
  void unlockRegion(RegionID regionID);
  void unpendRegion(RegionID regionID);

  void collectLocalNodes(int numElements,adaptAdj *elements,CkVec<int> &localNodes);
  bool lockLocalNodes(LockRegion *region);
  bool lockSharedIdxls(LockRegion *region);
  void lockRegionForRemote(RegionID regionID,int *sharedIdxls,int numSharedIdxls,adaptAdj *elements,int numElements);
  void lockReply(int remoteChunk,RegionID regionID,int success,int tag, int otherSuccess);
  void unlockRegion(LockRegion *region);
  void unlockLocalNodes(LockRegion *region);
  void unlockSharedIdxls(LockRegion *region);
  void unlockForRemote(RegionID regionID);
  void unpendForRemote(RegionID regionID);
  void unlockReply(int remoteChunk,RegionID regionID);
  void freeRegion(LockRegion *region);
  
  FEM_Comm *FindIdxlSide(int idxlType);


  int IdxlAddPrimary(int localId, int sharedChk, int idxlType);
  bool IdxlAddSecondary(int localId, int sharedChk, int sharedIdx, int idxlType);
  int IdxlRemovePrimary(int localId, int sharedChk, int idxlType);
  bool IdxlRemoveSecondary(int sharedChk, int sharedIdx, int idxlType);

  int IdxlLookUpPrimary(int localId, int sharedChk, int idxlType);
  int IdxlLookUpSecondary(int sharedChk, int sharedIdx, int idxlType);

  adaptAdjMsg *remote_bulk_edge_bisect_2D(adaptAdj &nbrElem, adaptAdj &splitElem, int new_idxl, int n1_idxl, int n2_idxl, int partitionID);

  void remote_adaptAdj_replace(adaptAdj &elem, adaptAdj &oldElem, adaptAdj &newElem);
  void remote_edgeAdj_replace(int remotePartID, adaptAdj &adj, adaptAdj &elem, 
                  adaptAdj &splitElem, int n1_idxl, int n2_idxl);
  void remote_edgeAdj_add(int remotePartID, adaptAdj &adj, adaptAdj &splitElem,
              int n1_idxl, int n2_idxl);
  void recv_split_3D(int pos, int tableID, adaptAdj &elem, adaptAdj &splitElem);
  void handle_split_3D(int remotePartID, int pos, int tableID, adaptAdj &elem, RegionID lockRegionID,
               int n1_idxl, int n2_idxl, int n5_idxl);
  void recv_splits(int tableID, int expectedSplits);

  longestMsg *isLongest(int elem, int elemType, double len);

  void update_asterisk_3D(int remotePartID, int i, adaptAdj &elem, 
              int numElemPairs, adaptAdj *elemPairs, RegionID lockRegionID,
              int n1_idxl, int n2_idxl, int n5_idxl);
};

class lockChunksMsg : public CMessage_lockChunksMsg {
 public:
  int *chkList;
  int chkListSize;
  int idxlType;

 public:
  lockChunksMsg(int *c, int s, int type) {
    chkListSize = s;
    idxlType = type;
  }

  ~lockChunksMsg() {
  }

  int *getChks() {
    return chkList;
  }

  int getSize() {
    return chkListSize;
  }

  int getType() {
    return idxlType;
  }
};

class adaptAdjMsg : public CMessage_adaptAdjMsg {
 public:
  adaptAdj elem;
};

class longestMsg : public CMessage_longestMsg {
 public:
  bool longest;
};
  
#endif

---