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 176 of file ComputePme.C.

Constructor & Destructor Documentation

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

Definition at line 178 of file ComputePme.C.

179  : ia(i_a), ib(i_b), nb(n_b),
180  size(n), data(newcopyint(n,d)) {
181  }

Member Function Documentation

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

Definition at line 190 of file ComputePme.C.

References NAMD_bug().

190  {
191  int mype = CkMyPe();
192  for ( int i=0; i < size; ++i ) {
193  if ( data[i] == mype ) {
194  CkArrayIndex3D ai(0,0,0);
195  ai.data()[ia] = i / nb;
196  ai.data()[ib] = i % nb;
197  if ( procNum(0,ai) != mype ) NAMD_bug("PmePencilMap is inconsistent");
198  if ( ! msg ) NAMD_bug("PmePencilMap multiple pencils on a pe?");
199  mgr->insertInitial(ai,msg);
200  msg = 0;
201  }
202  }
203  mgr->doneInserting();
204  if ( msg ) CkFreeMsg(msg);
205  }
void NAMD_bug(const char *err_msg)
Definition: common.C:129
virtual int procNum(int, const CkArrayIndex &i)
Definition: ComputePme.C:186
virtual int PmePencilMap::procNum ( int  ,
const CkArrayIndex &  i 
)
inlinevirtual

Definition at line 186 of file ComputePme.C.

186  {
187  //Return the home processor number for this element of this array
188  return data[ i.data()[ia] * nb + i.data()[ib] ];
189  }
virtual int PmePencilMap::registerArray ( CkArrayIndexMax &  ,
CkArrayID   
)
inlinevirtual

Definition at line 182 of file ComputePme.C.

182  {
183  //Return an ``arrayHdl'', given some information about the array
184  return 0;
185  }

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