from varglas.utilities import DataInput
from fenics import *
import varglas.model as model
# Output directory
out_dir = 'results_dir3/'
#set_log_active(True)
parameters["allow_extrapolation"] = True
mesh = Mesh('data/meshes/ant_mesh.xml')
# Get a bunch of data to use in the simulation
thklim = 100.0
measures = DataFactory.get_ant_measures(res=900)
bedmap1 = DataFactory.get_bedmap1(thklim=thklim)
bedmap2 = DataFactory.get_bedmap2(thklim=thklim)
dm = DataInput(measures, mesh=mesh)
db1 = DataInput(bedmap1, mesh=mesh)
db2 = DataInput(bedmap2, mesh=mesh)
# Fix some stuff?
db2.data['B'] = db2.data['S'] - db2.data['H']
db2.set_data_val('H', 32767, thklim)
db2.data['S'] = db2.data['B'] + db2.data['H']
S = db2.get_spline_expression("S")
B = db2.get_spline_expression("B")
T_s = db1.get_spline_expression("srfTemp")
q_geo = db1.get_spline_expression("q_geo")
from fenics import *
# get the input args :
i = int(sys.argv[2]) # assimilation number
dir_b = sys.argv[1] + '/0' # directory to save
# set the output directory :
out_dir = dir_b + str(i) + '/'
in_dir = 'vars/'
set_log_active(True)
thklim = 1.0
measures = DataFactory.get_ant_measures(res=450)
bedmap1 = DataFactory.get_bedmap1(thklim=thklim)
bedmap2 = DataFactory.get_bedmap2(thklim=thklim)
mesh = MeshFactory.get_ronne_3D_50H()
dm = DataInput(measures, mesh=mesh)
db1 = DataInput(bedmap1, mesh=mesh)
db2 = DataInput(bedmap2, mesh=mesh)
db2.data['B'] = db2.data['S'] - db2.data['H']
db2.set_data_val('H', 32767, thklim)
db2.data['S'] = db2.data['B'] + db2.data['H']
H = db2.get_nearest_expression("H")
S = db2.get_nearest_expression("S")
B = db2.get_nearest_expression("B")
S = Function(Q) B = Function(Q) File(in_dir + 'S_s.xml') >> S File(in_dir + 'B_s.xml') >> B config = default_config() config['output_path'] = out_dir config['balance_velocity']['kappa'] = 5.0 config['model_order'] = 'SSA' model = model.Model(config) model.set_mesh(mesh) model.set_surface_and_bed(S, B) model.initialize_variables() model.init_adot(in_dir + 'adot_s.xml') F = solvers.BalanceVelocitySolver(model, config) F.solve() model.save_xml(model.Ubar, 'Ubar_5') bedmap = DataFactory.get_bedmap1() bm = DataInput(bedmap, gen_space=False) do = DataOutput(out_dir) do.write_matlab(bm, model.Ubar, 'Ubar_5', val=0.0)
