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)
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
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))
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))
