diffusion_interpolation.py 文件源码

def preprocess_Dr(data, r, normalize=True):
    x = data["x"] #[xx[0] for xx in data["x"]]
    data, x = fields._sorted(data, x)
    Dt = [d[2][2] for d in data["D"]]
    Dn = [d[0][0] for d in data["D"]]

    eps = 1e-2
    x = [-1., -eps, r] + x + [1000.]
    if normalize:
        Dn = [d/Dn[-1] for d in Dn]
        Dt = [d/Dt[-1] for d in Dt]

    Dn = [0., 0., eps] + Dn + [1.]
    Dt = [0., 0., eps] + Dt + [1.]

    fn = interp1d(x, Dn)
    ft = interp1d(x, Dt)
    return fn, ft