def parse_args(argument_adder, from_commandline=False, **kwargs):
"""
Return Namespace of arguments.
If from_commandline, the arguments are parsed from sys.argv;
otherwise, defaults are returned.
argument_adder is a function which modifies an argparse.ArgumentParser
to give it simulation-specific arguments.
Optional kwargs allow overriding default argument values.
"""
# set up default arguments common to all simulations
parser = anuga.create_standard_parser()
# add any benchmark-specific arguments
argument_adder(parser)
# override default argument values
parser.set_defaults(**kwargs)
if from_commandline:
return parser.parse_args()
else:
return parser.parse_args([])
def parse_args_and_parameters(argument_adder=None, from_commandline=False, **kwargs):
"""
Return Namespace of arguments.
If from_commandline, the arguments are parsed from sys.argv;
otherwise, defaults are returned.
argument_adder is a function which modifies an argparse.ArgumentParser
to give it simulation-specific arguments.
Optional kwargs allow overriding default argument values.
Combine with local variables from project.py
"""
parser = anuga.create_standard_parser()
try:
import project
# Make variables in project.py into a dictionary
project_dict = vars(project)
del project_dict['__builtins__']
del project_dict['__doc__']
del project_dict['__file__']
del project_dict['__name__']
del project_dict['__package__']
del project_dict['join']
except:
project_dict = {}
# add any benchmark-specific arguments
if argument_adder:
argument_adder(parser)
# override default argument values from project.py
parser.set_defaults(**project_dict)
# override default argument values
parser.set_defaults(**kwargs)
if from_commandline:
return parser.parse_args()
else:
return parser.parse_args([])
"""
Quick plot of the subcritical-flow outputs
"""
import anuga.utilities.plot_utils as util
import matplotlib
matplotlib.use("Agg")
from matplotlib import pyplot as pyplot
from analytical_supercritical import analytic_sol
from numpy import ones, arange
import anuga
parser = anuga.create_standard_parser()
parser.add_argument("-tid", type=int, default=-1, help="timestep id")
args = parser.parse_args()
tid = args.tid
verbose = args.verbose
if verbose:
print "Read in swwfile"
p_st = util.get_output("supercritical.sww")
p2_st = util.get_centroids(p_st)
# v = p2_st.y[10]
# v2=(p2_st.y==v)
# v2=(p2_st.y>-1.0)
"""
Quick plot of the subcritical-flow outputs
"""
import anuga.utilities.plot_utils as util
import matplotlib
matplotlib.use('Agg')
from matplotlib import pyplot as pyplot
from analytical_supercritical import analytic_sol
from numpy import ones, arange
import anuga
parser = anuga.create_standard_parser()
parser.add_argument('-tid', type=int, default=-1, help='timestep id')
args = parser.parse_args()
tid = args.tid
verbose = args.verbose
if verbose: print('Read in swwfile')
p_st = util.get_output('supercritical.sww')
p2_st = util.get_centroids(p_st)
#v = p2_st.y[10]
#v2=(p2_st.y==v)
#v2=(p2_st.y>-1.0)
v2 = arange(len(p2_st.y))
if verbose: print('Calculate analytical solution')
h, z = analytic_sol(p2_st.x[v2])
