#include <PmeSolver.h>

Inheritance diagram for PmePencilXYMap:

Public Member Functions
	PmePencilXYMap (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 72 of file PmeSolver.h.

Constructor & Destructor Documentation

◆ PmePencilXYMap()

PmePencilXYMap::PmePencilXYMap ( const std::vector< int > & pes )

inline

Definition at line 74 of file PmeSolver.h.

74 : pes(pes) {}

Member Function Documentation

◆ populateInitial()

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

inlinevirtual

Definition at line 84 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("PmePencilXYMap::populateInitial, multiple pencils on a pe?");
         CkArrayIndex3D ai(0,0,0);
         ai.data()[2] = i;
         if ( procNum(0,ai) != CkMyPe() ) NAMD_bug("PmePencilXYMap::populateInitial, map is inconsistent");
         mgr->insertInitial(ai, msg);
         msg = NULL;
       }
     }
     mgr->doneInserting();
     if (msg != NULL) CkFreeMsg(msg);
   }

◆ procNum()

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

inlinevirtual

Definition at line 78 of file PmeSolver.h.

References NAMD_bug().

Referenced by populateInitial().

                                                     {
     int ind = idx.data()[2];
     if (ind < 0 || ind >= pes.size())
       NAMD_bug("PmePencilXYMap::procNum, index out of bounds");
     return pes[ind];
   }

◆ registerArray()

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