PmePencilXMap Class Reference

#include <PmeSolver.h>

Inheritance diagram for PmePencilXMap:

Public Member Functions
	PmePencilXMap (int ia, int ib, int width, const std::vector< int > &pes)
int	registerArray (CkArrayIndex &numElements, CkArrayID aid)
virtual int	procNum (int, const CkArrayIndex &idx)
virtual void	populateInitial (int, CkArrayOptions &, void *msg, CkArrMgr *mgr)

Detailed Description

Definition at line 33 of file PmeSolver.h.

Constructor & Destructor Documentation

PmePencilXMap::PmePencilXMap ( int  ia, int  ib, int  width, const std::vector< int > &  pes  )

inline

Definition at line 35 of file PmeSolver.h.

: ia(ia), ib(ib), width(width), pes(pes) {}

Member Function Documentation

virtual void PmePencilXMap::populateInitial ( int  , CkArrayOptions &  , void *  msg, CkArrMgr *  mgr  )

inlinevirtual

Definition at line 45 of file PmeSolver.h.

References NAMD_bug(), and procNum().

                                                                                {
    for (int i=0;i < pes.size();i++) {
      if (pes[i] == CkMyPe()) {
        if ( msg == NULL ) NAMD_bug("PmePencilXMap::populateInitial, multiple pencils on a pe?");
        CkArrayIndex3D ai(0,0,0);
        ai.data()[ib] = i / width;
        ai.data()[ia] = i % width;
        //fprintf(stderr, "Pe %d i %d at %d %d\n", pes[i], i, ai.data()[ia], ai.data()[ib]);
        if ( procNum(0,ai) != CkMyPe() ) NAMD_bug("PmePencilXMap::populateInitial, map is inconsistent");
        mgr->insertInitial(ai,msg);
        msg = NULL;
      }
    }
    mgr->doneInserting();
    if (msg != NULL) CkFreeMsg(msg);
  }

virtual int PmePencilXMap::procNum ( int  , const CkArrayIndex &  idx  )

inlinevirtual

Definition at line 39 of file PmeSolver.h.

References NAMD_bug().

Referenced by populateInitial().

                                                    {
    int ind = idx.data()[ia] + idx.data()[ib] * width;
    if (ind < 0 || ind >= pes.size())
      NAMD_bug("PmePencilXMap::procNum, index out of bounds");
    return pes[ind];
  }

int PmePencilXMap::registerArray ( CkArrayIndex &  numElements, CkArrayID  aid  )

inline

Definition at line 36 of file PmeSolver.h.

                                                              {
    return 0;
  }

---

The documentation for this class was generated from the following file:

• PmeSolver.h