Diffusion

\[\dtempdif{1}\]

src/fluid/predict/t.c

static int diffusion_x (
    const domain_t * domain,
    const double diffusivity,
    const double * restrict t,
    double * restrict src
) {
  const int isize = domain->mysizes[0];
  const int jsize = domain->mysizes[1];
#if NDIMS == 3
  const int ksize = domain->mysizes[2];
#endif
  const laplacian_t * lapx = laplacians.lapx;
  BEGIN
    src[cnt] += diffusivity * (
#if NDIMS == 2
        + LAPX(i).l * T(i-1, j  )
        + LAPX(i).c * T(i  , j  )
        + LAPX(i).u * T(i+1, j  )
#else
        + LAPX(i).l * T(i-1, j  , k  )
        + LAPX(i).c * T(i  , j  , k  )
        + LAPX(i).u * T(i+1, j  , k  )
#endif
    );
  END
  return 0;
}

src/fluid/predict/t.c

const int isize = domain->glsizes[0];
const double * hxxf = domain->hxxf;
const double * jdxf = domain->jdxf;
const double * jdxc = domain->jdxc;
laplacians.lapx = memory_calloc(isize, sizeof(laplacian_t));
for (int i = 1; i <= isize; i++) {
  const double l = 1. / JDXC(i  ) * JDXF(i  ) / HXXF(i  ) / HXXF(i  );
  const double u = 1. / JDXC(i  ) * JDXF(i+1) / HXXF(i+1) / HXXF(i+1);
  const double c = - l - u;
  laplacians.LAPX(i).l = l;
  laplacians.LAPX(i).c = c;
  laplacians.LAPX(i).u = u;
}

\[\dtempdif{2}\]

src/fluid/predict/t.c

static int diffusion_y (
    const domain_t * domain,
    const double diffusivity,
    const double * restrict t,
    double * restrict src
) {
  const int isize = domain->mysizes[0];
  const int jsize = domain->mysizes[1];
#if NDIMS == 3
  const int ksize = domain->mysizes[2];
#endif
  const laplacian_t * lapy = &laplacians.lapy;
  BEGIN
    src[cnt] += diffusivity * (
#if NDIMS == 2
        + (*lapy).l * T(i  , j-1)
        + (*lapy).c * T(i  , j  )
        + (*lapy).u * T(i  , j+1)
#else
        + (*lapy).l * T(i  , j-1, k  )
        + (*lapy).c * T(i  , j  , k  )
        + (*lapy).u * T(i  , j+1, k  )
#endif
    );
  END
  return 0;
}

src/fluid/predict/t.c

const double hy = domain->hy;
const double l = 1. / hy / hy;
const double u = 1. / hy / hy;
const double c = - l - u;
laplacians.lapy.l = l;
laplacians.lapy.c = c;
laplacians.lapy.u = u;

\[\dtempdif{3}\]

src/fluid/predict/t.c

static int diffusion_z (
    const domain_t * domain,
    const double diffusivity,
    const double * restrict t,
    double * restrict src
) {
  const int isize = domain->mysizes[0];
  const int jsize = domain->mysizes[1];
  const int ksize = domain->mysizes[2];
  const laplacian_t * lapz = &laplacians.lapz;
  BEGIN
    src[cnt] += diffusivity * (
        + (*lapz).l * T(i  , j  , k-1)
        + (*lapz).c * T(i  , j  , k  )
        + (*lapz).u * T(i  , j  , k+1)
    );
  END
  return 0;
}

src/fluid/predict/t.c

const double hz = domain->hz;
const double l = 1. / hz / hz;
const double u = 1. / hz / hz;
const double c = - l - u;
laplacians.lapz.l = l;
laplacians.lapz.c = c;
laplacians.lapz.u = u;