class UExpression(Expression):
  def eval(self, values, x):
    values[0] = u(x)
class VExpression(Expression):
  def eval(self, values, x):
    values[0] = v(x)
class WExpression(Expression):
  def eval(self, values, x):
    values[0] = w(x)

# Setup the model
model = model.Model()
model.set_geometry(S, B, deform=True)

# Output the two meshes for comparison
print("Writing meshes")
File(out_dir + "full_mesh.pvd") << mesh_model.mesh
File(out_dir + "part_mesh.pvd") << model.mesh

# Apparently the smb is only used in the transient solver with free-surface  
# so adot isn't projected onto the mesh otherwise by the steady solver.
# In order to view adot in paraview, I'll just output it manually
adot_out = project(adot,model.Q)
File(out_dir + '/adot.pvd') << adot_out

print("Doing some stuff")
u_out = project(UExpression(),model.Q)
v_out = project(VExpression(),model.Q)
Exemplo n.º 2
# Now deform the refined mesh to match the surface and bed velocity
refined_mesh = Mesh(mesh_name + ".xml")

# Expression for the bed elevation
class BedExpression(Expression):
  def eval(self, values, x):
    # Get the z coordinate for this x value
    values[0] = bed(x[0],x[1])

# Expression for the surface elevation
class SurfaceExpression(Expression):
  def eval(self, values, x):
    # Get the surface value at a point
    #values[0] = surface(x[0],x[1])
    values[0] = surface(x[0],x[1])

model = model.Model()
# Set the mesh to the non-deformed anisotropic mesh

# Deform it to match the surface and bed geometry
model.set_geometry(SurfaceExpression(element = model.Q.ufl_element()), BedExpression(element = model.Q.ufl_element()), deform = False)

plot(model.mesh, interactive = True)
# Save the mesh
File(mesh_name + '_deformed.xml') << model.mesh
File(mesh_name + '_deformed.pvd') << model.mesh