ShallowWater2DModifiedSolution.c

Go to the documentation of this file.

#include <basic.h>
#include <arrayfunctions.h>
#include <physicalmodels/shallowwater2d.h>
#include <mpivars.h>
#include <hypar.h>

int ShallowWater2DModifiedSolution(
                            double  *uC, 
                            double  *u,  
                            int     d,   
                            void    *s,  
                            void    *m,  
                            double  waqt 
                           )
{
  HyPar           *solver = (HyPar*) s;
  ShallowWater2D  *param  = (ShallowWater2D*) solver->physics;

  int     ghosts  = solver->ghosts;
  int     *dim    = solver->dim_local;
  int     ndims   = solver->ndims;
  int     index[ndims], bounds[ndims], offset[ndims];

  /* set bounds for array index to include ghost points */
  _ArrayCopy1D_(dim,bounds,ndims);
  int i; for (i=0; i<ndims; i++) bounds[i] += 2*ghosts;

  /* set offset such that index is compatible with ghost point arrangement */
  _ArraySetValue_(offset,ndims,-ghosts);

  int done = 0; _ArraySetValue_(index,ndims,0);
  while (!done) {
    int p; _ArrayIndex1DWO_(ndims,dim,index,offset,ghosts,p);
    double h,uvel,vvel;
    _ShallowWater2DGetFlowVar_((u+_MODEL_NVARS_*p),h,uvel,vvel);
    uC[_MODEL_NVARS_*p+0] = h + param->b[p];
    uC[_MODEL_NVARS_*p+1] = h * uvel;
    uC[_MODEL_NVARS_*p+2] = h * vvel;
    _ArrayIncrementIndex_(ndims,bounds,index,done);
  }
  return(0);
}