コード例 #1
0
ファイル: poisson.py プロジェクト: TheBB/aroma
def rsolve(xcenter, ycenter, num, nred):
    hcase = get_case(num)
    rcase = get_reduced(num, nred)
    mu = rcase.parameter(xcenter=xcenter, ycenter=ycenter)

    with util.time():
        lhs = solvers.poisson(rcase, mu)
コード例 #2
0
@click.option('--fast/--no-fast', default=False)
@util.common_args
def disp(fast):
    case = get_case(fast)
    print(case)


@main.command()
@click.option('--velocity', default=1.0)
@click.option('--height', default=1.0)
@click.option('--fast/--no-fast', default=False)
@util.common_args
def solve(velocity, height, fast):
    case = get_case(fast)
    mu = case.parameter(velocity=velocity, height=height)
    with util.time():
        lhs = solvers.stokes(case, mu)
    visualization.velocity(case,
                           mu,
                           lhs,
                           figsize=(10, 10),
                           name='full',
                           colorbar=True)
    visualization.pressure(case,
                           mu,
                           lhs,
                           figsize=(10, 10),
                           name='full',
                           colorbar=True)