#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 35 of file PmeSolver.h.

Constructor & Destructor Documentation

◆ PmePencilXMap()

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

inline

Definition at line 37 of file PmeSolver.h.

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

Member Function Documentation

◆ populateInitial()

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

inlinevirtual

Definition at line 47 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);
   }

◆ procNum()

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

inlinevirtual

Definition at line 41 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];
   }

◆ registerArray()

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