示例#1
0
def solver_scaled(I, dt, C, T):
    """
    Solve 1D wave equation in dimensionless form.
    """
    # Make a hash of the arguments
    import inspect, hashlib
    data = inspect.getsource(I) + '_' + str(dt) + '_' + \
           str(C) + '_' + str(T)
    # Not fool proof: if x0 changes value, the source code of I
    # is still the same, and no recomputation takes place...
    hashed_input = hashlib.sha1(data).hexdigest()

    # Use joblib-based tool (class Storage) to store already
    # computed solutions in files
    cachedir = 'tmp_%s' % hashed_input
    is_computed = os.path.isdir(cachedir)
    print 'cachedir:', cachedir, is_computed
    storage = Storage(cachedir, verbose=0)

    def action(u, x, t, n):
        if n == 0:
            storage.save('x', x)
            storage.save('t', t)
        storage.save('u%d' % n, u)

    if is_computed:
        print 'No need to compute the numerical solution'
        return storage
    else:
        print 'Computing the numerical solution'
        solver_unscaled(
            I=I, V=0, f=0, c=1, L=1, dt=dt, C=C, T=T,
            user_action=action)
        return storage
def solver_scaled(I, dt, C, T):
    """
    Solve 1D wave equation in dimensionless form.
    """
    # Make a hash of the arguments
    import inspect, hashlib
    data = inspect.getsource(I) + '_' + str(dt) + '_' + \
           str(C) + '_' + str(T)
    # Not fool proof: if x0 changes value, I source is the same...
    hashed_input = hashlib.sha1(data).hexdigest()

    cachedir = 'tmp_%s' % hashed_input
    is_computed = os.path.isdir(cachedir)

    import joblib
    memory = joblib.Memory(cachedir=cachedir, verbose=1)

    def retrieve(name, data=None):
        print 'joblib save of', name
        return data

    retrieve = memory.cache(retrieve, ignore=['data'])
    save = retrieve

    def action(u, x, t, n):
        if n == 0:
            save('x', x)
            save('t', t)
        save('u%d' % n, u)

    if is_computed:
        print 'No need to compute the numerical solution'
        return retrieve
    else:
        print 'Computing the numerical solution'
        solver_unscaled(I=I,
                        V=0,
                        f=0,
                        c=1,
                        L=1,
                        dt=dt,
                        C=C,
                        T=T,
                        user_action=action)
        return retrieve
def solver_scaled_prev(I, dt, C, T):
    """
    Solve 1D wave equation in dimensionless form.
    """

    # store solution in files? joblib? or just list in memory?
    # joblib doesn't handle I... Store u in joblib? retrieve/store Yes!
    def action(u, x, t, n):
        if n == 0:
            save('x', x)
            save('t', t)
        save('u%d' % n, u)

    # Make a hash of the arguments
    import inspect, hashlib
    data = inspect.getsource(I) + '_' + str(dt) + '_' + \
           str(C) + '_' + str(T)
    hashed_input = hashlib.sha1(data).hexdigest()

    print 'hash:', hashed_input
    cachedir = 'tmp_%s' % hashed_input
    import joblib
    memory = joblib.Memory(cachedir=cachedir, verbose=1)

    @memory.cache(ignore=['data'])
    def retrieve(name, data=None):
        print 'joblib save of', name
        return data

    save = retrieve

    if os.path.isdir(cachedir):
        return retrieve
    else:
        print 'Computing the numerical solution'
        solver_unscaled(I=I,
                        V=0,
                        f=0,
                        c=1,
                        L=1,
                        dt=dt,
                        C=C,
                        T=T,
                        user_action=action)
        return retrieve
def solver_scaled(I, dt, C, T):
    """
    Solve 1D wave equation in dimensionless form.
    """
    # Make a hash of the arguments
    import inspect, hashlib
    data = inspect.getsource(I) + '_' + str(dt) + '_' + \
           str(C) + '_' + str(T)
    # Not fool proof: if x0 changes value, I source is the same...
    hashed_input = hashlib.sha1(data).hexdigest()

    cachedir = 'tmp_%s' % hashed_input
    is_computed = os.path.isdir(cachedir)

    import joblib
    memory = joblib.Memory(cachedir=cachedir, verbose=1)

    def retrieve(name, data=None):
        print 'joblib save of', name
        return data

    retrieve = memory.cache(retrieve, ignore=['data'])
    save = retrieve

    def action(u, x, t, n):
        if n == 0:
            save('x', x)
            save('t', t)
        save('u%d' % n, u)

    if is_computed:
        print 'No need to compute the numerical solution'
        return retrieve
    else:
        print 'Computing the numerical solution'
        solver_unscaled(
            I=I, V=0, f=0, c=1, L=1, dt=dt, C=C, T=T,
            user_action=action)
        return retrieve
def solver_scaled_prev(I, dt, C, T):
    """
    Solve 1D wave equation in dimensionless form.
    """
    # store solution in files? joblib? or just list in memory?
    # joblib doesn't handle I... Store u in joblib? retrieve/store Yes!
    def action(u, x, t, n):
        if n == 0:
            save('x', x)
            save('t', t)
        save('u%d' % n, u)

    # Make a hash of the arguments
    import inspect, hashlib
    data = inspect.getsource(I) + '_' + str(dt) + '_' + \
           str(C) + '_' + str(T)
    hashed_input = hashlib.sha1(data).hexdigest()

    print 'hash:', hashed_input
    cachedir = 'tmp_%s' % hashed_input
    import joblib
    memory = joblib.Memory(cachedir=cachedir, verbose=1)

    @memory.cache(ignore=['data'])
    def retrieve(name, data=None):
        print 'joblib save of', name
        return data

    save = retrieve

    if os.path.isdir(cachedir):
        return retrieve
    else:
        print 'Computing the numerical solution'
        solver_unscaled(
            I=I, V=0, f=0, c=1, L=1, dt=dt, C=C, T=T,
            user_action=action)
        return retrieve