Skip to content

central_difference_1d(x, D)

Second-order accurate central diffence scheme.

Source code in openterrace/diffusion_schemes/central_difference_1d.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
@nb.njit
def central_difference_1d(x, D):
    """Second-order accurate central diffence scheme.
    """
    _out = np.zeros_like(x)
    for j in range(0, x.shape[0]):
        for i in range(1, x.shape[1]-1):
            _out[j,i] = x[j,i-1]*D[0,j,i] + x[j,i+1]*D[1,j,i]\
                - x[j,i]*(D[0,j,i]+D[1,j,i])             
    return _out