def definition(u, diffusion, *, weight): from __externals__ import fwd_diff, BRANCH with computation(PARALLEL), interval(...): laplacian = lap_op(u=u) laplacian2 = lap_op(u=u) if BRANCH: flux_i, flux_j = fwd_diff(field1=laplacian, field2=laplacian2) else: flux_i = fwd_diff(field1=laplacian) flux_j = fwd_diff_op_y(field=laplacian) diffusion = u[0, 0, 0] - weight * (flux_i[0, 0, 0] - flux_i[-1, 0, 0] + flux_j[0, 0, 0] - flux_j[0, -1, 0])
def definition(u, diffusion, *, weight): from __externals__ import fwd_diff with computation(PARALLEL), interval(...): laplacian = lap_op(u=u) flux_i, flux_j = fwd_diff(field=laplacian) diffusion = u[0, 0, 0] - weight * ( flux_i[0, 0, 0] - flux_i[-1, 0, 0] + flux_j[0, 0, 0] - flux_j[0, -1, 0] )
def definition(u, diffusion, *, weight): from __externals__ import fwd_diff with computation(PARALLEL), interval(...): laplacian = lap_op(u=u) flux_i, flux_j = fwd_diff(field=laplacian) diffusion = u[ # noqa: F841 # Local name is assigned to but never used 0, 0, 0] - weight * (flux_i[0, 0, 0] - flux_i[-1, 0, 0] + flux_j[0, 0, 0] - flux_j[0, -1, 0])
def definition(u, diffusion, *, weight): """ Horizontal diffusion stencil. Parameters ---------- u : 3D float field, input diffusion : 3D float field, output weight : diffusion coefficient """ from __externals__ import fwd_diff, BRANCH with computation(PARALLEL), interval(...): laplacian = lap_op(u=u) laplacian2 = lap_op(u=u) if __INLINED(BRANCH): flux_i, flux_j = fwd_diff(field1=laplacian, field2=laplacian2) else: flux_i = fwd_diff(field1=laplacian) flux_j = fwd_diff_op_y(field=laplacian) diffusion = u[0, 0, 0] - weight * ( flux_i[0, 0, 0] - flux_i[-1, 0, 0] + flux_j[0, 0, 0] - flux_j[0, -1, 0] )