Exemplos de PeriodInfo em Python

Exemplo n.º 1

Arquivo: bp2_kuramoto.py Projeto: mjanusz/sdepy

local_vars = {
        'ns0': lambda sdei: sympy.sqrt(sdei.S.d0 * sdei.S.dt * 2.0 / sdei.S.gam1),
        'ns1': lambda sdei: sympy.sqrt(sdei.S.d0 * sdei.S.dt * 2.0 / sdei.S.gam2),
        }

const_pars = {
        'phi0': lambda sdei: numpy.random.uniform(0.0, 2.0 * numpy.pi, sdei.num_threads)
    }

code = """
    dx0 = (is * sinf(x0 - x1) + f1 + amp * cosf(t + phi0)) / gam1;
    dx1 = (-is * sinf(x0 - x1) + f2) / gam2;
"""

ns_map = {0: ['ns0', 0], 1: [0, 'ns1']}
period_map = {0: sde.PeriodInfo(period=2.0 * math.pi, freq=1.0)}

sdei = sde.SDE(code, sim_params, num_vars=2, num_noises=2, noise_map=ns_map, period_map=period_map,
               local_vars=local_vars, const_pars=const_pars)

output = {'path': {
            'main': [sde.OutputDecl(func=sde.avg_moments, vars=[0, 1])],
            },
          'summary': {
            'main': [sde.OutputDecl(func=sde.drift_velocity, vars=[0, 1])],
            }
        }

sdei.prepare(sde.SRK2, init_vector)
sdei.simulate(output, block_size=128)

Exemplo n.º 2

Arquivo: jj2_identical.py Projeto: mjanusz/sdepy

    'a2': 'AC drive amplitude for the 2nd particle',
    'omega': 'AC drive frequency'
}

local_vars = {
    'ns0': lambda sdei: sympy.sqrt(sdei.S.d0 * sdei.S.dt),
}

code = """
    dx0 = i1 - sinf(x0) + alpha * (i2 - sinf(x1)) + (a1 + alpha * a2) * cosf(omega * t + phi0);
    dx1 = i2 - sinf(x1) + alpha * (i1 - sinf(x0)) + (alpha * a1 + a2) * cosf(omega * t + phi0);
"""

ns_map = {0: ['ns0', 0], 1: [0, 'ns0']}
period_map = {
    0: sde.PeriodInfo(period=2.0 * math.pi, freq=1),
    1: sde.PeriodInfo(period=2.0 * math.pi, freq=1)
}

sdei = sde.SDE(code,
               sim_params,
               num_vars=2,
               num_noises=2,
               const_pars=const_pars,
               noise_map=ns_map,
               period_map=period_map,
               local_vars=local_vars)

output = {
    'path': {
        'main': [sde.OutputDecl(func=sde.avg_moments, vars=[0, 1])],

Exemplo n.º 3

Arquivo: bistable.py Projeto: mjanusz/sdepy

    'gamma_': 'damping constant',
    'psd': 'potential strenght',
    'd0': 'noise strength'
}

local_vars = {
    'ns': lambda sdei: sympy.sqrt(sdei.S.d0 * sdei.S.dt * 2.0 * sdei.S.gamma_)
}

code = """
    dx0 = x1;
    dx1 = -psd * sinf(x0) - gamma_ * x1 + force;
"""

noise_map = {1: ['ns']}
period_map = {0: sde.PeriodInfo(period=2.0 * numpy.pi, freq=1)}

sde_ = sde.SDE(code,
               params,
               num_vars=2,
               num_noises=1,
               noise_map=noise_map,
               period_map=period_map,
               local_vars=local_vars)

output = {
    'summary': {
        'main': [sde.OutputDecl(func=sde.diffusion_coefficient, vars=[0])],
    },
    'path': {
        'main': [