def generate_channel1_domain(gpu=True):
#--------------------------------------------------------------------------
# Setup computational domain
#--------------------------------------------------------------------------
#points, vertices, boundary = anuga.rectangular_cross(10, 1,
# len1=10.0, len2=5.0) # Mesh
points, vertices, boundary = anuga.rectangular_cross(1,
4,
len1=0.1,
len2=0.1) # Mesh
if gpu:
domain = GPU_domain(points, vertices, boundary) # Create domain
for i in range(len(sys.argv)):
if sys.argv[i] == '-gpu':
domain.using_gpu = True
print " --> Enable GPU version"
elif sys.argv[i] == '-fs':
finaltime = float(sys.argv[i + 1])
print " --> Finaltime is reset as %f" % finaltime
elif sys.argv[i] == '-test':
domain.cotesting = True
print " --> Enable Cotesting"
elif sys.argv[i] == '-ustore':
domain.store = True
print " --> Disable storing"
else:
domain = anuga.Domain(points, vertices, boundary) # Create domain
domain.set_name('channel1') # Output name
#--------------------------------------------------------------------------
# Setup initial conditions
#--------------------------------------------------------------------------
def topography(x, y):
return -x / 10 # linear bed slope
domain.set_quantity('elevation', topography) # Use function for elevation
domain.set_quantity('friction', 0.01) # Constant friction
domain.set_quantity(
'stage', # Dry bed
expression='elevation')
#--------------------------------------------------------------------------
# Setup boundary conditions
#--------------------------------------------------------------------------
Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
domain.set_boundary({'left': Bi, 'right': Br, 'top': Br, 'bottom': Br})
return domain
def generate_channel1_domain(gpu=True):
# --------------------------------------------------------------------------
# Setup computational domain
# --------------------------------------------------------------------------
# points, vertices, boundary = anuga.rectangular_cross(10, 1,
# len1=10.0, len2=5.0) # Mesh
points, vertices, boundary = anuga.rectangular_cross(1, 4, len1=0.1, len2=0.1) # Mesh
if gpu:
domain = GPU_domain(points, vertices, boundary) # Create domain
for i in range(len(sys.argv)):
if sys.argv[i] == "-gpu":
domain.using_gpu = True
print " --> Enable GPU version"
elif sys.argv[i] == "-fs":
finaltime = float(sys.argv[i + 1])
print " --> Finaltime is reset as %f" % finaltime
elif sys.argv[i] == "-test":
domain.cotesting = True
print " --> Enable Cotesting"
elif sys.argv[i] == "-ustore":
domain.store = True
print " --> Disable storing"
else:
domain = anuga.Domain(points, vertices, boundary) # Create domain
domain.set_name("channel1") # Output name
# --------------------------------------------------------------------------
# Setup initial conditions
# --------------------------------------------------------------------------
def topography(x, y):
return -x / 10 # linear bed slope
domain.set_quantity("elevation", topography) # Use function for elevation
domain.set_quantity("friction", 0.01) # Constant friction
domain.set_quantity("stage", expression="elevation") # Dry bed
# --------------------------------------------------------------------------
# Setup boundary conditions
# --------------------------------------------------------------------------
Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
domain.set_boundary({"left": Bi, "right": Br, "top": Br, "bottom": Br})
return domain
