MsmBlockKernel< Vtype, Mtype > Class Template Reference

Public Member Functions
	MsmBlockKernel (const msm::BlockIndex &)
	MsmBlockKernel (CkMigrateMessage *m)
void	init ()
void	setupStencils (const msm::Grid< Mtype > *res, const msm::Grid< Mtype > *pro)
void	restrictionKernel ()
void	prolongationKernel ()

Public Attributes
CProxy_ComputeMsmMgr	mgrProxy
ComputeMsmMgr *	mgrLocal
msm::Map *	map
msm::BlockDiagram *	bd
msm::Grid< Vtype >	qh
msm::Grid< Vtype >	eh
const msm::Grid< Mtype > *	resStencil
const msm::Grid< Mtype > *	proStencil
msm::Grid< Vtype >	qhRestricted
msm::Grid< Vtype >	ehProlongated
int	cntRecvsCharge
int	cntRecvsPotential
msm::BlockIndex	blockIndex
msm::Grid< Vtype >	subgrid
int	sequence

Detailed Description

template<class Vtype, class Mtype>
class MsmBlockKernel< Vtype, Mtype >

Definition at line 2733 of file ComputeMsm.C.

Constructor & Destructor Documentation

MsmBlockKernel() [1/2]

template<class Vtype , class Mtype >

MsmBlockKernel< Vtype, Mtype >::MsmBlockKernel

(

const msm::BlockIndex &

bindex

)

Definition at line 2793 of file ComputeMsm.C.

                                                                       {
  blockIndex = bindex;
  mgrProxy = CProxy_ComputeMsmMgr(CkpvAccess(BOCclass_group).computeMsmMgr);
  mgrLocal = CProxy_ComputeMsmMgr::ckLocalBranch(
      CkpvAccess(BOCclass_group).computeMsmMgr);
  map = &(mgrLocal->mapData());
  bd = &(map->blockLevel[blockIndex.level](blockIndex.n));
  qh.init( bd->nrange );
  eh.init( bd->nrange );
#ifndef MSM_GRID_CUTOFF_DECOMP
  ehCutoff.init( bd->nrangeCutoff );
#endif
  qhRestricted.init( bd->nrangeRestricted );
  ehProlongated.init( bd->nrangeProlongated );
#ifdef DEBUG_MSM_GRID
  printf("MsmBlockKernel level=%d, n=%d %d %d:  constructor\n",
      blockIndex.level, blockIndex.n.i, blockIndex.n.j, blockIndex.n.k);
#endif
#ifdef MSM_TIMING
  mgrLocal->addTiming();
#endif
  init();
} // MsmBlockKernel<Vtype,Mtype>::MsmBlockKernel()

MsmBlockKernel() [2/2]

template<class Vtype, class Mtype>

MsmBlockKernel< Vtype, Mtype >::MsmBlockKernel ( CkMigrateMessage *  m )

inline

Definition at line 2758 of file ComputeMsm.C.

{ }

Member Function Documentation

init()

template<class Vtype , class Mtype >

void MsmBlockKernel< Vtype, Mtype >::init

(

)

Definition at line 2819 of file ComputeMsm.C.

                                       {
  qh.reset(0);
  eh.reset(0);
#ifndef MSM_GRID_CUTOFF_DECOMP
  ehCutoff.reset(0);
#endif
  qhRestricted.reset(0);
  ehProlongated.reset(0);
  cntRecvsCharge = 0;
  cntRecvsPotential = 0;
} // MsmBlockKernel<Vtype,Mtype>::init()

prolongationKernel()

template<class Vtype , class Mtype >

void MsmBlockKernel< Vtype, Mtype >::prolongationKernel

(

)

Definition at line 2991 of file ComputeMsm.C.

{
#ifdef DEBUG_MSM_GRID
  printf("MsmBlockKernel level=%d, id=%d %d %d:  prolongation\n",
      blockIndex.level, blockIndex.n.i, blockIndex.n.j, blockIndex.n.k);
#endif

#ifdef MSM_TIMING
  double startTime, stopTime;
  startTime = CkWallTimer();
#endif
#ifndef MSM_COMM_ONLY
  // stencil data for approximating potential on prolongated grid
  const int approx = mgrLocal->approx;
  const int nstencil = ComputeMsmMgr::Nstencil[approx];
  const int *offset = ComputeMsmMgr::IndexOffset[approx];
  const msm::Grid<Mtype>& pro = *proStencil;

  // index range for prolongated h grid potentials
  int ia1 = ehProlongated.ia();
  int ib1 = ehProlongated.ib();
  int ja1 = ehProlongated.ja();
  int jb1 = ehProlongated.jb();
  int ka1 = ehProlongated.ka();
  int kb1 = ehProlongated.kb();

  // index range for 2h grid potentials
  int ia2 = eh.ia();
  int ib2 = eh.ib();
  int ja2 = eh.ja();
  int jb2 = eh.jb();
  int ka2 = eh.ka();
  int kb2 = eh.kb();

  // loop over 2h grid
  for (int k2 = ka2;  k2 <= kb2;  k2++) {
    int k1 = 2 * k2;
    for (int j2 = ja2;  j2 <= jb2;  j2++) {
      int j1 = 2 * j2;
      for (int i2 = ia2;  i2 <= ib2;  i2++) {
        int i1 = 2 * i2;

        // loop over stencils on prolongated h grid
        for (int k = 0;  k < nstencil;  k++) {
          int kn = k1 + offset[k];
          if      (kn < ka1) continue;
          else if (kn > kb1) break;

          for (int j = 0;  j < nstencil;  j++) {
            int jn = j1 + offset[j];
            if      (jn < ja1) continue;
            else if (jn > jb1) break;

            for (int i = 0;  i < nstencil;  i++) {
              int in = i1 + offset[i];
              if      (in < ia1) continue;
              else if (in > ib1) break;

              ehProlongated(in,jn,kn) += pro(i,j,k) * eh(i2,j2,k2);
            }
          }
        } // end loop over stencils on prolongated h grid

      }
    }
  } // end loop over 2h grid
#else
  ehProlongated.reset(0);
#endif // !MSM_COMM_ONLY
#ifdef MSM_TIMING
  stopTime = CkWallTimer();
  mgrLocal->msmTiming[MsmTimer::PROLONGATE] += stopTime - startTime;
#endif
} // MsmBlockKernel<Vtype,Mtype>::prolongationKernel()

restrictionKernel()

template<class Vtype , class Mtype >

void MsmBlockKernel< Vtype, Mtype >::restrictionKernel

(

)

Definition at line 2833 of file ComputeMsm.C.

{
#ifdef DEBUG_MSM_GRID
  printf("MsmBlockKernel level=%d, id=%d %d %d:  restriction\n",
      blockIndex.level, blockIndex.n.i, blockIndex.n.j, blockIndex.n.k);
#endif

#ifdef MSM_TIMING
  double startTime, stopTime;
  startTime = CkWallTimer();
#endif

#ifndef MSM_COMM_ONLY
  // stencil data for approximating charge on restricted grid
  const int approx = mgrLocal->approx;
  const int nstencil = ComputeMsmMgr::Nstencil[approx];
  const int *offset = ComputeMsmMgr::IndexOffset[approx];
  const msm::Grid<Mtype>& res = *resStencil;

  // index range for h grid charges
  int ia1 = qh.ia();
  int ib1 = qh.ib();
  int ja1 = qh.ja();
  int jb1 = qh.jb();
  int ka1 = qh.ka();
  int kb1 = qh.kb();

  // index range for restricted (2h) grid charges
  int ia2 = qhRestricted.ia();
  int ib2 = qhRestricted.ib();
  int ja2 = qhRestricted.ja();
  int jb2 = qhRestricted.jb();
  int ka2 = qhRestricted.ka();
  int kb2 = qhRestricted.kb();

  // reset grid
  qhRestricted.reset(0);

  // loop over restricted (2h) grid
  for (int k2 = ka2;  k2 <= kb2;  k2++) {
    int k1 = 2 * k2;
    for (int j2 = ja2;  j2 <= jb2;  j2++) {
      int j1 = 2 * j2;
      for (int i2 = ia2;  i2 <= ib2;  i2++) {
        int i1 = 2 * i2;

        // loop over stencils on h grid
        Vtype& q2hsum = qhRestricted(i2,j2,k2);

        for (int k = 0;  k < nstencil;  k++) {
          int kn = k1 + offset[k];
          if      (kn < ka1) continue;
          else if (kn > kb1) break;

          for (int j = 0;  j < nstencil;  j++) {
            int jn = j1 + offset[j];
            if      (jn < ja1) continue;
            else if (jn > jb1) break;

            for (int i = 0;  i < nstencil;  i++) {
              int in = i1 + offset[i];
              if      (in < ia1) continue;
              else if (in > ib1) break;

              q2hsum += res(i,j,k) * qh(in,jn,kn);
            }
          }
        } // end loop over stencils on h grid

      }
    }
  } // end loop over restricted (2h) grid
#else
  qhRestricted.reset(0);
#endif // !MSM_COMM_ONLY

#ifdef MSM_TIMING
  stopTime = CkWallTimer();
  mgrLocal->msmTiming[MsmTimer::RESTRICT] += stopTime - startTime;
#endif
} // MsmBlockKernel<Vtype,Mtype>::restrictionKernel()

setupStencils()

template<class Vtype, class Mtype>

void MsmBlockKernel< Vtype, Mtype >::setupStencils ( const msm::Grid< Mtype > *  res, const msm::Grid< Mtype > *  pro  )

inline

Definition at line 2774 of file ComputeMsm.C.

    {
      resStencil = res;
      proStencil = pro;
    }

Member Data Documentation

bd

template<class Vtype, class Mtype>

msm::BlockDiagram* MsmBlockKernel< Vtype, Mtype >::bd

Definition at line 2738 of file ComputeMsm.C.

blockIndex

template<class Vtype, class Mtype>

msm::BlockIndex MsmBlockKernel< Vtype, Mtype >::blockIndex

Definition at line 2751 of file ComputeMsm.C.

cntRecvsCharge

template<class Vtype, class Mtype>

int MsmBlockKernel< Vtype, Mtype >::cntRecvsCharge

Definition at line 2749 of file ComputeMsm.C.

cntRecvsPotential

template<class Vtype, class Mtype>

int MsmBlockKernel< Vtype, Mtype >::cntRecvsPotential

Definition at line 2750 of file ComputeMsm.C.

eh

template<class Vtype, class Mtype>

msm::Grid<Vtype> MsmBlockKernel< Vtype, Mtype >::eh

Definition at line 2740 of file ComputeMsm.C.

ehProlongated

template<class Vtype, class Mtype>

msm::Grid<Vtype> MsmBlockKernel< Vtype, Mtype >::ehProlongated

Definition at line 2748 of file ComputeMsm.C.

map

template<class Vtype, class Mtype>

msm::Map* MsmBlockKernel< Vtype, Mtype >::map

Definition at line 2737 of file ComputeMsm.C.

mgrLocal

template<class Vtype, class Mtype>

ComputeMsmMgr* MsmBlockKernel< Vtype, Mtype >::mgrLocal

Definition at line 2736 of file ComputeMsm.C.

mgrProxy

template<class Vtype, class Mtype>

CProxy_ComputeMsmMgr MsmBlockKernel< Vtype, Mtype >::mgrProxy

Definition at line 2735 of file ComputeMsm.C.

proStencil

template<class Vtype, class Mtype>

const msm::Grid<Mtype>* MsmBlockKernel< Vtype, Mtype >::proStencil

Definition at line 2746 of file ComputeMsm.C.

Referenced by MsmBlockKernel< Float, Float >::setupStencils().

qh

template<class Vtype, class Mtype>

msm::Grid<Vtype> MsmBlockKernel< Vtype, Mtype >::qh

Definition at line 2739 of file ComputeMsm.C.

qhRestricted

template<class Vtype, class Mtype>

msm::Grid<Vtype> MsmBlockKernel< Vtype, Mtype >::qhRestricted

Definition at line 2747 of file ComputeMsm.C.

resStencil

template<class Vtype, class Mtype>

const msm::Grid<Mtype>* MsmBlockKernel< Vtype, Mtype >::resStencil

Definition at line 2745 of file ComputeMsm.C.

Referenced by MsmBlockKernel< Float, Float >::setupStencils().

sequence

template<class Vtype, class Mtype>

int MsmBlockKernel< Vtype, Mtype >::sequence

Definition at line 2755 of file ComputeMsm.C.

subgrid

template<class Vtype, class Mtype>

msm::Grid<Vtype> MsmBlockKernel< Vtype, Mtype >::subgrid

Definition at line 2753 of file ComputeMsm.C.

---

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

• ComputeMsm.C