NAMD
Public Member Functions | List of all members
PmePencilMap Class Reference
Inheritance diagram for PmePencilMap:

Public Member Functions

 PmePencilMap (int i_a, int i_b, int n_b, int n, int *d)
 
virtual int registerArray (CkArrayIndexMax &, CkArrayID)
 
virtual int procNum (int, const CkArrayIndex &i)
 
virtual void populateInitial (int, CkArrayIndexMax &, void *msg, CkArrMgr *mgr)
 

Detailed Description

Definition at line 196 of file ComputePme.C.

Constructor & Destructor Documentation

◆ PmePencilMap()

PmePencilMap::PmePencilMap ( int  i_a,
int  i_b,
int  n_b,
int  n,
int *  d 
)
inline

Definition at line 198 of file ComputePme.C.

199  : ia(i_a), ib(i_b), nb(n_b),
200  size(n), data(newcopyint(n,d)) {
201  }

Member Function Documentation

◆ populateInitial()

virtual void PmePencilMap::populateInitial ( int  ,
CkArrayIndexMax &  ,
void *  msg,
CkArrMgr *  mgr 
)
inlinevirtual

Definition at line 210 of file ComputePme.C.

References NAMD_bug().

210  {
211  int mype = CkMyPe();
212  for ( int i=0; i < size; ++i ) {
213  if ( data[i] == mype ) {
214  CkArrayIndex3D ai(0,0,0);
215  ai.data()[ia] = i / nb;
216  ai.data()[ib] = i % nb;
217  if ( procNum(0,ai) != mype ) NAMD_bug("PmePencilMap is inconsistent");
218  if ( ! msg ) NAMD_bug("PmePencilMap multiple pencils on a pe?");
219  mgr->insertInitial(ai,msg);
220  msg = 0;
221  }
222  }
223  mgr->doneInserting();
224  if ( msg ) CkFreeMsg(msg);
225  }
void NAMD_bug(const char *err_msg)
Definition: common.C:195
virtual int procNum(int, const CkArrayIndex &i)
Definition: ComputePme.C:206

◆ procNum()

virtual int PmePencilMap::procNum ( int  ,
const CkArrayIndex &  i 
)
inlinevirtual

Definition at line 206 of file ComputePme.C.

206  {
207  //Return the home processor number for this element of this array
208  return data[ i.data()[ia] * nb + i.data()[ib] ];
209  }

◆ registerArray()

virtual int PmePencilMap::registerArray ( CkArrayIndexMax &  ,
CkArrayID   
)
inlinevirtual

Definition at line 202 of file ComputePme.C.

202  {
203  //Return an ``arrayHdl'', given some information about the array
204  return 0;
205  }

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