schemes.py 文件源码

def __init__(self, model, jac=False,
                 integrator='vode', **integrator_kwargs):
        def func_scipy_proxy(t, U, fields, pars, hook):
            fields.fill(U)
            fields, pars = hook(t, fields, pars)
            return model.F(fields, pars)

        def jacob_scipy_proxy(t, U, fields, pars, hook):
            fields.fill(U)
            fields, pars = hook(t, fields, pars)
            return model.J(fields, pars, sparse=False)

        self._solv = ode(func_scipy_proxy,
                         jac=jacob_scipy_proxy if jac else None)
        self._solv.set_integrator(integrator, **integrator_kwargs)