Beispiel #1
0
def runclaw(xclawcmd=None,
            outdir=None,
            overwrite=True,
            restart=False,
            rundir=None):
    """
    Run the Fortran version of Clawpack using executable xclawcmd, which is
    typically set to 'xclaw', 'xamr', etc.

    If it is not set by the call, get it from the environment variable
    CLAW_EXE.  Default to 'xclaw' if that's not set.
    
    If rundir is None, all *.data is copied from current directory, if a path 
    is given, data files are copied from there instead.
    """

    import os

    if type(overwrite) is str:
        # convert to boolean
        overwrite = overwrite.lower() in ["true", "t"]

    if type(restart) is str:
        # convert to boolean
        restart = restart.lower() in ["true", "t"]

    # importing these modules requires $CLAW/python in PYTHONPATH:
    from pyclaw.controller import Controller

    if xclawcmd is None:
        # Determine what executable to use from environment variable CLAW_EXE
        # Default to 'xclaw' if it's not set:
        xclawcmd = os.environ.get("CLAW_EXE", "xclaw")

    if outdir is None:
        outdir = "."

    if rundir is None:
        rundir = os.getcwd()
    rundir = os.path.abspath(rundir)
    print("Will take data from ", rundir)

    # directory for fort.* files:
    outdir = os.path.abspath(outdir)
    print("== runclaw: Will write output to ", outdir)

    clawjob = Controller()
    clawjob.xdir = os.getcwd()
    clawjob.rundir = rundir  # use data files from current directory
    clawjob.outdir = outdir  # write fort files to outdir
    clawjob.xclawcmd = xclawcmd  # Clawpack executable
    clawjob.overwrite = overwrite  # Ok to overwrite outdir and plotdir?
    clawjob.restart = restart  # Restarting a previous run?

    returncode = clawjob.runxclaw()
    if returncode != 0:
        print("== runclaw: *** fortran returncode = ", returncode,
              "   aborting")
    print("== runclaw: Done executing %s via pyclaw.runclaw.py" % xclawcmd)
    print("== runclaw: Output is in ", outdir)
Beispiel #2
0
def runclaw(xclawcmd=None, outdir=None, overwrite=True, restart=False, 
            rundir=None):
    """
    Run the Fortran version of Clawpack using executable xclawcmd, which is
    typically set to 'xclaw', 'xamr', etc.

    If it is not set by the call, get it from the environment variable
    CLAW_EXE.  Default to 'xclaw' if that's not set.
    
    If rundir is None, all *.data is copied from current directory, if a path 
    is given, data files are copied from there instead.
    """

    import os
    if type(overwrite) is str:
        # convert to boolean
        overwrite = (overwrite.lower() in ['true','t'])
    
    if type(restart) is str:
        # convert to boolean
        restart = (restart.lower() in ['true','t'])
    
    # importing these modules requires $CLAW/python in PYTHONPATH:
    from pyclaw.controller import Controller

    if xclawcmd is None:
        # Determine what executable to use from environment variable CLAW_EXE
        # Default to 'xclaw' if it's not set:
        xclawcmd = os.environ.get('CLAW_EXE', 'xclaw')
    

    if outdir is None:
        outdir = '.'
        
    if rundir is None:
        rundir = os.getcwd()
    rundir = os.path.abspath(rundir)
    print "Will take data from ", rundir

    # directory for fort.* files:
    outdir = os.path.abspath(outdir)
    print '== runclaw: Will write output to ',outdir
    
    clawjob = Controller()
    clawjob.xdir = os.getcwd()
    clawjob.rundir = rundir           # use data files from current directory
    clawjob.outdir = outdir           # write fort files to outdir
    clawjob.xclawcmd = xclawcmd       # Clawpack executable
    clawjob.overwrite = overwrite     # Ok to overwrite outdir and plotdir?
    clawjob.restart = restart         # Restarting a previous run?
    
    returncode = clawjob.runxclaw()
    if returncode != 0:
        print '== runclaw: *** fortran returncode = ', returncode, '   aborting'
    print '== runclaw: Done executing %s via pyclaw.runclaw.py' % xclawcmd
    print '== runclaw: Output is in ', outdir
Beispiel #3
0
def acoustics(kernel_language='Python',
              petscPlot=False,
              iplot=False,
              htmlplot=False,
              myplot=False,
              outdir='./_output',
              sclaw=False,
              **kwargs):
    import numpy as np
    """
    1D acoustics example.
    """

    from petclaw.patch import Patch
    from petclaw.patch import Dimension
    from pyclaw.solution import Solution
    from pyclaw.controller import Controller
    from petclaw import plot

    petscts = kwargs.get('petscts', False)

    # Initialize patch and solution
    xmin = 0.0
    xmax = 1.0
    ncells = kwargs.get('ncells', 100)
    x = Dimension('x', xmin, xmax, ncells, bc_lower=2, bc_upper=2)
    patch = Patch(x)
    rho = 1.0
    bulk = 1.0
    patch.problem_data['rho'] = rho
    patch.problem_data['bulk'] = bulk
    patch.problem_data['zz'] = np.sqrt(rho * bulk)
    patch.problem_data['cc'] = np.sqrt(rho / bulk)
    from classic1 import cparam
    for key, value in patch.problem_data.iteritems():
        setattr(cparam, key, value)
    patch.num_eqn = 2
    if sclaw:
        patch.num_ghost = 3
    patch.t = 0.0

    # init_q_petsc_structures must be called
    # before patch.x.centers and such can be accessed.
    patch.init_q_petsc_structures()
    xc = patch.x.centers
    q = np.zeros([patch.num_eqn, len(xc)], order='F')
    beta = 100
    gamma = 0
    x0 = 0.75
    q[0, :] = np.exp(-beta * (xc - x0)**2) * np.cos(gamma * (xc - x0))
    q[1, :] = 0.
    patch.q = q

    init_solution = Solution(patch)

    if sclaw:
        from petclaw.sharpclaw import SharpClawSolver1D
        solver = SharpClawSolver1D()
        solver.lim_type = kwargs.get('lim_type', 2)
        solver.time_integrator = 'SSP33'
        solver.num_waves = 2
        solver.char_decomp = 0
    else:
        from petclaw.clawpack import ClawSolver1D
        solver = ClawSolver1D()
        solver.num_waves = 2
        from pyclaw import limiters
        solver.mthlim = limiters.tvd.MC

    if kernel_language == 'Python': solver.set_riemann_solver('acoustics')
    solver.dt = patch.delta[0] / patch.problem_data['cc'] * 0.1
    solver.kernel_language = kernel_language

    claw = Controller()
    #claw.output_style = 3
    claw.keep_copy = True
    claw.num_output_times = 5
    # The output format MUST be set to petsc!
    claw.output_format = 'ascii'
    claw.outdir = outdir
    claw.tfinal = 1.0
    claw.solution = init_solution
    claw.solver = solver

    # Solve
    if petscts:
        # Uses PETSc time integrator. Needs sclaw=1 to get semidiscrete form. Examples:
        # ./acoustics.py sclaw=1 petscts=1 -ts_monitor_solution -ts_type theta -snes_mf_operator -ts_dt 0.02 -ts_max_steps 50 -ts_max_time 10 -draw_pause 0.05
        # ./acoustics.py sclaw=1 petscts=1 -ts_monitor_solution -ts_type ssp -ts_ssp_type rk104
        x = patch.q_da.createGlobalVector()
        f = x.duplicate()

        ts = PETSc.TS().create(PETSc.COMM_WORLD)
        ts.setDM(patch.q_da)
        ts.setProblemType(PETSc.TS.ProblemType.NONLINEAR)
        ts.setType(ts.Type.THETA)

        eqn = AcousticEquation(patch, solver)
        x[:] = patch.q
        ts.setRHSFunction(eqn.rhsfunction, f)
        ts.setIFunction(eqn.ifunction, f)
        ts.setIJacobian(eqn.ijacobian, patch.q_da.createMat())

        ts.setTimeStep(solver.dt)
        ts.setDuration(claw.tfinal, max_steps=1000)
        ts.setFromOptions()
        q0 = x[:].copy()
        ts.solve(x)
        qfinal = x[:].copy()
        dx = patch.delta[0]
    else:
        status = claw.run()
        #This test is set up so that the waves pass through the domain
        #exactly once, and the final solution should be equal to the
        #initial condition.  Here we output the 1-norm of their difference.
        q0 = claw.frames[0].patch.gqVec.getArray().reshape([-1])
        qfinal = claw.frames[
            claw.num_output_times].patch.gqVec.getArray().reshape([-1])
        dx = claw.frames[0].patch.delta[0]

    if htmlplot: plot.html_plot(outdir=outdir, format=output_format)
    if iplot: plot.interactive_plot(outdir=outdir, format=output_format)
    if petscPlot: plot.plotPetsc(output_object)

    print 'Max error:', np.max(qfinal - q0)

    return dx * np.sum(np.abs(qfinal - q0))
Beispiel #4
0
def acoustics(kernel_language='Python',petscPlot=False,iplot=False,htmlplot=False,myplot=False,outdir='./_output',sclaw=False,**kwargs):
    import numpy as np
    """
    1D acoustics example.
    """

    from petclaw.patch import Patch
    from petclaw.patch import Dimension
    from pyclaw.solution import Solution
    from pyclaw.controller import Controller
    from petclaw import plot

    petscts = kwargs.get('petscts', False)

    # Initialize patch and solution
    xmin = 0.0
    xmax = 1.0
    ncells = kwargs.get('ncells',100)
    x = Dimension('x',xmin,xmax,ncells,bc_lower=2,bc_upper=2)
    patch = Patch(x)
    rho = 1.0
    bulk = 1.0
    patch.problem_data['rho']=rho
    patch.problem_data['bulk']=bulk
    patch.problem_data['zz']=np.sqrt(rho*bulk)
    patch.problem_data['cc']=np.sqrt(rho/bulk)
    from classic1 import cparam 
    for key,value in patch.problem_data.iteritems(): setattr(cparam,key,value)
    patch.num_eqn=2
    if sclaw:
        patch.num_ghost=3
    patch.t = 0.0

    # init_q_petsc_structures must be called 
    # before patch.x.centers and such can be accessed.
    patch.init_q_petsc_structures()
    xc=patch.x.centers
    q=np.zeros([patch.num_eqn,len(xc)], order = 'F')
    beta=100; gamma=0; x0=0.75
    q[0,:] = np.exp(-beta * (xc-x0)**2) * np.cos(gamma * (xc - x0))
    q[1,:]=0.
    patch.q=q
    
    init_solution = Solution(patch)

    if sclaw:
        from petclaw.sharpclaw import SharpClawSolver1D
        solver = SharpClawSolver1D()
        solver.lim_type = kwargs.get('lim_type',2)
        solver.time_integrator = 'SSP33'
        solver.num_waves=2
        solver.char_decomp=0
    else:
        from petclaw.clawpack import ClawSolver1D
        solver = ClawSolver1D()
        solver.num_waves=2
        from pyclaw import limiters
        solver.mthlim = limiters.tvd.MC

    if kernel_language=='Python': solver.set_riemann_solver('acoustics')
    solver.dt=patch.delta[0]/patch.problem_data['cc']*0.1
    solver.kernel_language=kernel_language

    claw = Controller()
    #claw.output_style = 3
    claw.keep_copy = True
    claw.num_output_times = 5
    # The output format MUST be set to petsc!
    claw.output_format = 'ascii'
    claw.outdir = outdir
    claw.tfinal = 1.0
    claw.solution = init_solution
    claw.solver = solver

    # Solve
    if petscts:
        # Uses PETSc time integrator. Needs sclaw=1 to get semidiscrete form. Examples:
        # ./acoustics.py sclaw=1 petscts=1 -ts_monitor_solution -ts_type theta -snes_mf_operator -ts_dt 0.02 -ts_max_steps 50 -ts_max_time 10 -draw_pause 0.05
        # ./acoustics.py sclaw=1 petscts=1 -ts_monitor_solution -ts_type ssp -ts_ssp_type rk104
        x = patch.q_da.createGlobalVector()
        f = x.duplicate()

        ts = PETSc.TS().create(PETSc.COMM_WORLD)
        ts.setDM(patch.q_da)
        ts.setProblemType(PETSc.TS.ProblemType.NONLINEAR)
        ts.setType(ts.Type.THETA)

        eqn = AcousticEquation(patch, solver)
        x[:] = patch.q
        ts.setRHSFunction(eqn.rhsfunction, f)
        ts.setIFunction(eqn.ifunction, f)
        ts.setIJacobian(eqn.ijacobian, patch.q_da.createMat())

        ts.setTimeStep(solver.dt)
        ts.setDuration(claw.tfinal, max_steps=1000)
        ts.setFromOptions()
        q0 = x[:].copy()
        ts.solve(x)
        qfinal = x[:].copy()
        dx = patch.delta[0]
    else:
        status = claw.run()
        #This test is set up so that the waves pass through the domain
        #exactly once, and the final solution should be equal to the
        #initial condition.  Here we output the 1-norm of their difference.
        q0=claw.frames[0].patch.gqVec.getArray().reshape([-1])
        qfinal=claw.frames[claw.num_output_times].patch.gqVec.getArray().reshape([-1])
        dx=claw.frames[0].patch.delta[0]

    if htmlplot:  plot.html_plot(outdir=outdir,format=output_format)
    if iplot:     plot.interactive_plot(outdir=outdir,format=output_format)
    if petscPlot: plot.plotPetsc(output_object)

    print 'Max error:', np.max(qfinal - q0)

    return dx*np.sum(np.abs(qfinal-q0))