SourceFunction.c File Reference

Evaluate the source term. More...

#include <stdlib.h>
#include <string.h>
#include <basic.h>
#include <arrayfunctions_gpu.h>
#include <boundaryconditions.h>
#include <mpivars.h>
#include <hypar.h>

Go to the source code of this file.

Functions
int	SourceFunction (double *source, double *u, void *s, void *m, double t)

Detailed Description

Evaluate the source term.

Author

Debojyoti Ghosh, Youngdae Kim

Definition in file SourceFunction.c.

Function Documentation

int SourceFunction	(	double *	source,
		double *	u,
		void *	s,
		void *	m,
		double	t
	)

Evaluate the source term \({\bf S}\left({\bf u}\right)\) in the governing equation, if the physical model specifies one. In addition, if the simulation requires a sponge boundary treatment, the sponge BC function is called.

Parameters

source	the computed source term
u	solution
s	solver object of type HyPar
m	MPI object of type MPIVariables
t	Current simulation time

Definition at line 22 of file SourceFunction.c.

{
  HyPar           *solver   = (HyPar*)        s;
  MPIVariables    *mpi      = (MPIVariables*) m;

  /* initialize to zero */
  int size = solver->ndof_cells_wghosts;
#if defined(HAVE_CUDA)
  if (solver->use_gpu) {
    gpuArraySetValue(source,size, 0.0);
  } else {
#endif
    _ArraySetValue_(source,size,0.0);
#if defined(HAVE_CUDA)
  }
#endif

  /* call the source function of the physics model, if available */
  if (solver->SFunction) {
    solver->SFunction(source,u,solver,mpi,t);
    solver->count_sou++;
  }

  /* Apart from other source terms, implement sponge BC as a source */
  DomainBoundary* boundary = (DomainBoundary*) solver->boundary;
  int n;
  int nb = solver->nBoundaryZones;
#if defined(HAVE_CUDA)
  if (solver->use_gpu) {
    for (n = 0; n < nb; n++) {
      if (!strcmp(boundary[n].bctype,_SPONGE_)) {
        fprintf(stderr,"ERROR: Sponge BC not yet implemented on GPU.\n");
      }
    }
  } else {
#endif
    for (n = 0; n < nb; n++) {
      if (!strcmp(boundary[n].bctype,_SPONGE_)) {
        BCSpongeSource( &boundary[n],
                        solver->ndims,
                        solver->nvars,
                        solver->ghosts,
                        solver->dim_local,
                        solver->x,
                        u,
                        source  ); 
      }
    }
#if defined(HAVE_CUDA)
  }
#endif


  return(0);
}