def __init__(self,
coordinates,
vertices,
boundary = None,
tagged_elements = None,
geo_reference = None,
use_inscribed_circle=False,
velocity = None,
full_send_dict=None,
ghost_recv_dict=None,
processor=0,
numproc=1
):
conserved_quantities = ['stage']
other_quantities = []
Generic_Domain.__init__(self,
source=coordinates,
triangles=vertices,
boundary=boundary,
conserved_quantities=conserved_quantities,
other_quantities=other_quantities,
tagged_elements=tagged_elements,
geo_reference=geo_reference,
use_inscribed_circle=use_inscribed_circle,
full_send_dict=full_send_dict,
ghost_recv_dict=ghost_recv_dict,
processor=processor,
numproc=numproc)
if velocity is None:
self.velocity = num.array([1,0],'d')
else:
self.velocity = num.array(velocity,'d')
#Only first is implemented for advection
self.set_default_order(1)
self.set_beta(1.0)
self.smooth = True
self.max_flux_update_frequency=1
def __init__(self,
coordinates,
vertices,
boundary=None,
tagged_elements=None,
geo_reference=None,
use_inscribed_circle=False,
velocity=None,
full_send_dict=None,
ghost_recv_dict=None,
processor=0,
numproc=1):
conserved_quantities = ['stage']
other_quantities = []
Generic_Domain.__init__(self,
source=coordinates,
triangles=vertices,
boundary=boundary,
conserved_quantities=conserved_quantities,
other_quantities=other_quantities,
tagged_elements=tagged_elements,
geo_reference=geo_reference,
use_inscribed_circle=use_inscribed_circle,
full_send_dict=full_send_dict,
ghost_recv_dict=ghost_recv_dict,
processor=processor,
numproc=numproc)
if velocity is None:
self.velocity = num.array([1, 0], 'd')
else:
self.velocity = num.array(velocity, 'd')
#Only first is implemented for advection
self.set_default_order(1)
self.set_beta(1.0)
self.smooth = True
self.max_flux_update_frequency = 1
def test_simple(self):
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe, daf, dae
vertices = [[1, 0, 2], [1, 2, 4], [4, 2, 5], [3, 1, 4]]
conserved_quantities = ['stage', 'xmomentum', 'ymomentum']
other_quantities = ['elevation', 'friction']
domain = Generic_Domain(points, vertices, None, conserved_quantities,
None, other_quantities)
domain.check_integrity()
for name in conserved_quantities + other_quantities:
assert domain.quantities.has_key(name)
assert num.alltrue(domain.get_conserved_quantities(0, edge=1) == 0.)
def test_simple(self):
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0,0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0,0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe, daf, dae
vertices = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4]]
conserved_quantities = ['stage', 'xmomentum', 'ymomentum']
other_quantities = ['elevation', 'friction']
domain = Generic_Domain(points, vertices, None,
conserved_quantities, None, other_quantities)
domain.check_integrity()
for name in conserved_quantities + other_quantities:
assert domain.quantities.has_key(name)
assert num.alltrue(domain.get_conserved_quantities(0, edge=1) == 0.)
def check_integrity(self):
Generic_Domain.check_integrity(self)
msg = 'Conserved quantity must be "stage"'
assert self.conserved_quantities[0] == 'stage', msg
def test_time(self):
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0,0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0,0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4] ]
domain = Generic_Domain(points, elements)
domain.check_integrity()
domain.conserved_quantities = ['stage', 'ymomentum']
domain.evolved_quantities = ['stage', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Test time bdry, you need to provide a domain and function
try:
T = Time_boundary(domain)
except:
pass
else:
raise Exception('Should have raised exception')
#Test time bdry, you need to provide a function
try:
T = Time_boundary()
except:
pass
else:
raise Exception('Should have raised exception')
def function(t):
return [1.0, 0.0]
T = Time_boundary(domain, function)
from anuga.config import default_boundary_tag
domain.set_boundary( {default_boundary_tag: T} )
#FIXME: should not necessarily be true always.
#E.g. with None as a boundary object.
assert len(domain.boundary) == len(domain.boundary_objects)
q = T.evaluate(0, 2) #Vol=0, edge=2
assert num.allclose(q, [1.0, 0.0])
def test_fileboundary_exception(self):
"""Test that boundary object complains if number of
conserved quantities are wrong
"""
import time, os
from math import sin, pi
from anuga.config import time_format
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0,0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0,0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4] ]
domain = Generic_Domain(points, elements)
domain.conserved_quantities = ['stage', 'xmomentum', 'ymomentum']
domain.evolved_quantities = ['stage', 'xmomentum', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['xmomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Write file (with only two values)
filename = 'boundarytest' + str(time.time())
fid = open(filename + '.txt', 'w')
start = time.mktime(time.strptime('2000', '%Y'))
dt = 5*60 #Five minute intervals
for i in range(10):
t = start + i*dt
t_string = time.strftime(time_format, time.gmtime(t))
fid.write('%s,%f %f\n' %(t_string, 1.0*i, sin(i*2*pi/10)))
fid.close()
#Convert ASCII file to NetCDF (Which is what we really like!)
from anuga.file_conversion.file_conversion import timefile2netcdf
timefile2netcdf(filename+'.txt', quantity_names = ['stage', 'xmomentum'])
try:
F = File_boundary(filename + '.tms',
domain)
except:
pass
else:
raise Exception('Should have raised an exception')
os.remove(filename + '.txt')
os.remove(filename + '.tms')
def NOtest_fileboundary_time_only(self):
"""Test that boundary values can be read from file and interpolated
This is using the .tms file format
See also test_util for comprenhensive testing of the underlying
file_function and also tests in test_datamanager which tests
file_function using the sts format
"""
#FIXME (Ole): This test was disabled 18 August 2008 as no
# need for this was found. Rather I implemented an Exception
# to catch possible errors in the model setup
import time, os
from math import sin, pi
from anuga.config import time_format
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4] ]
domain = Generic_Domain(points, elements)
domain.conserved_quantities = ['stage', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Write file
filename = 'boundarytest' + str(time.time())
fid = open(filename + '.txt', 'w')
start = time.mktime(time.strptime('2000', '%Y'))
dt = 5*60 #Five minute intervals
for i in range(10):
t = start + i*dt
t_string = time.strftime(time_format, time.gmtime(t))
fid.write('%s,%f %f\n' %(t_string, 1.0*i, sin(i*2*pi/10)))
fid.close()
#Convert ASCII file to NetCDF (Which is what we really like!)
from anuga.shallow_water.data_manager import timefile2netcdf
timefile2netcdf(filename, quantity_names = ['stage', 'ymomentum'])
F = File_boundary(filename + '.tms', domain)
os.remove(filename + '.txt')
os.remove(filename + '.tms')
#Check that midpoint coordinates at boundary are correctly computed
assert num.allclose( F.midpoint_coordinates,
[[1.0, 0.0], [0.0, 1.0], [3.0, 0.0],
[3.0, 1.0], [1.0, 3.0], [0.0, 3.0]])
#assert allclose(F.midpoint_coordinates[(3,2)], [0.0, 3.0])
#assert allclose(F.midpoint_coordinates[(3,1)], [1.0, 3.0])
#assert allclose(F.midpoint_coordinates[(0,2)], [0.0, 1.0])
#assert allclose(F.midpoint_coordinates[(0,0)], [1.0, 0.0])
#assert allclose(F.midpoint_coordinates[(2,0)], [3.0, 0.0])
#assert allclose(F.midpoint_coordinates[(2,1)], [3.0, 1.0])
#Check time interpolation
from anuga.config import default_boundary_tag
domain.set_boundary( {default_boundary_tag: F} )
domain.time = 5*30/2 #A quarter way through first step
q = F.evaluate()
assert num.allclose(q, [1.0/4, sin(2*pi/10)/4])
domain.time = 2.5*5*60 #Half way between steps 2 and 3
q = F.evaluate()
assert num.allclose(q, [2.5, (sin(2*2*pi/10) + sin(3*2*pi/10))/2])
def test_transmissive(self):
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0,0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0,0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4] ]
domain = Generic_Domain(points, elements)
domain.check_integrity()
domain.conserved_quantities = ['stage', 'ymomentum']
domain.evolved_quantities = ['stage', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Test transmissve bdry
try:
T = Transmissive_boundary()
except:
pass
else:
raise Exception('Should have raised exception')
T = Transmissive_boundary(domain)
from anuga.config import default_boundary_tag
domain.set_boundary( {default_boundary_tag: T} )
#FIXME: should not necessarily be true always.
#E.g. with None as a boundary object.
assert len(domain.boundary) == len(domain.boundary_objects)
q = T.evaluate(0, 2) #Vol=0, edge=2
assert num.allclose(q, [1.5, 2.5])
# Now set the centroid_transmissive_bc flag to true
domain.set_centroid_transmissive_bc(True)
q = T.evaluate(0, 2) #Vol=0, edge=2
assert num.allclose(q, [2.0 ,3.0]) # centroid value
def test_time(self):
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [[1, 0, 2], [1, 2, 4], [4, 2, 5], [3, 1, 4]]
domain = Generic_Domain(points, elements)
domain.check_integrity()
domain.conserved_quantities = ['stage', 'ymomentum']
domain.evolved_quantities = ['stage', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Test time bdry, you need to provide a domain and function
try:
T = Time_boundary(domain)
except:
pass
else:
raise Exception('Should have raised exception')
#Test time bdry, you need to provide a function
try:
T = Time_boundary()
except:
pass
else:
raise Exception('Should have raised exception')
def function(t):
return [1.0, 0.0]
T = Time_boundary(domain, function)
from anuga.config import default_boundary_tag
domain.set_boundary({default_boundary_tag: T})
#FIXME: should not necessarily be true always.
#E.g. with None as a boundary object.
assert len(domain.boundary) == len(domain.boundary_objects)
q = T.evaluate(0, 2) #Vol=0, edge=2
assert num.allclose(q, [1.0, 0.0])
def test_fileboundary_exception(self):
"""Test that boundary object complains if number of
conserved quantities are wrong
"""
import time, os
from math import sin, pi
from anuga.config import time_format
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [[1, 0, 2], [1, 2, 4], [4, 2, 5], [3, 1, 4]]
domain = Generic_Domain(points, elements)
domain.conserved_quantities = ['stage', 'xmomentum', 'ymomentum']
domain.evolved_quantities = ['stage', 'xmomentum', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['xmomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Write file (with only two values)
filename = 'boundarytest' + str(time.time())
fid = open(filename + '.txt', 'w')
start = time.mktime(time.strptime('2000', '%Y'))
dt = 5 * 60 #Five minute intervals
for i in range(10):
t = start + i * dt
t_string = time.strftime(time_format, time.gmtime(t))
fid.write('%s,%f %f\n' % (t_string, 1.0 * i, sin(i * 2 * pi / 10)))
fid.close()
#Convert ASCII file to NetCDF (Which is what we really like!)
from anuga.file_conversion.file_conversion import timefile2netcdf
timefile2netcdf(filename + '.txt',
quantity_names=['stage', 'xmomentum'])
try:
F = File_boundary(filename + '.tms', domain)
except:
pass
else:
raise Exception('Should have raised an exception')
os.remove(filename + '.txt')
os.remove(filename + '.tms')
def NOtest_fileboundary_time_only(self):
"""Test that boundary values can be read from file and interpolated
This is using the .tms file format
See also test_util for comprenhensive testing of the underlying
file_function and also tests in test_datamanager which tests
file_function using the sts format
"""
#FIXME (Ole): This test was disabled 18 August 2008 as no
# need for this was found. Rather I implemented an Exception
# to catch possible errors in the model setup
import time, os
from math import sin, pi
from anuga.config import time_format
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [[1, 0, 2], [1, 2, 4], [4, 2, 5], [3, 1, 4]]
domain = Generic_Domain(points, elements)
domain.conserved_quantities = ['stage', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Write file
filename = 'boundarytest' + str(time.time())
fid = open(filename + '.txt', 'w')
start = time.mktime(time.strptime('2000', '%Y'))
dt = 5 * 60 #Five minute intervals
for i in range(10):
t = start + i * dt
t_string = time.strftime(time_format, time.gmtime(t))
fid.write('%s,%f %f\n' % (t_string, 1.0 * i, sin(i * 2 * pi / 10)))
fid.close()
#Convert ASCII file to NetCDF (Which is what we really like!)
from anuga.shallow_water.data_manager import timefile2netcdf
timefile2netcdf(filename, quantity_names=['stage', 'ymomentum'])
F = File_boundary(filename + '.tms', domain)
os.remove(filename + '.txt')
os.remove(filename + '.tms')
#Check that midpoint coordinates at boundary are correctly computed
assert num.allclose(F.midpoint_coordinates,
[[1.0, 0.0], [0.0, 1.0], [3.0, 0.0], [3.0, 1.0],
[1.0, 3.0], [0.0, 3.0]])
#assert allclose(F.midpoint_coordinates[(3,2)], [0.0, 3.0])
#assert allclose(F.midpoint_coordinates[(3,1)], [1.0, 3.0])
#assert allclose(F.midpoint_coordinates[(0,2)], [0.0, 1.0])
#assert allclose(F.midpoint_coordinates[(0,0)], [1.0, 0.0])
#assert allclose(F.midpoint_coordinates[(2,0)], [3.0, 0.0])
#assert allclose(F.midpoint_coordinates[(2,1)], [3.0, 1.0])
#Check time interpolation
from anuga.config import default_boundary_tag
domain.set_boundary({default_boundary_tag: F})
domain.time = 5 * 30 / 2 #A quarter way through first step
q = F.evaluate()
assert num.allclose(q, [1.0 / 4, sin(2 * pi / 10) / 4])
domain.time = 2.5 * 5 * 60 #Half way between steps 2 and 3
q = F.evaluate()
assert num.allclose(
q, [2.5, (sin(2 * 2 * pi / 10) + sin(3 * 2 * pi / 10)) / 2])
def test_transmissive(self):
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
points = [a, b, c, d, e, f]
#bac, bce, ecf, dbe
elements = [[1, 0, 2], [1, 2, 4], [4, 2, 5], [3, 1, 4]]
domain = Generic_Domain(points, elements)
domain.check_integrity()
domain.conserved_quantities = ['stage', 'ymomentum']
domain.evolved_quantities = ['stage', 'ymomentum']
domain.quantities['stage'] =\
Quantity(domain, [[1,2,3], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.quantities['ymomentum'] =\
Quantity(domain, [[2,3,4], [5,5,5],
[0,0,9], [-6, 3, 3]])
domain.check_integrity()
#Test transmissve bdry
try:
T = Transmissive_boundary()
except:
pass
else:
raise Exception('Should have raised exception')
T = Transmissive_boundary(domain)
from anuga.config import default_boundary_tag
domain.set_boundary({default_boundary_tag: T})
#FIXME: should not necessarily be true always.
#E.g. with None as a boundary object.
assert len(domain.boundary) == len(domain.boundary_objects)
q = T.evaluate(0, 2) #Vol=0, edge=2
assert num.allclose(q, [1.5, 2.5])
# Now set the centroid_transmissive_bc flag to true
domain.set_centroid_transmissive_bc(True)
q = T.evaluate(0, 2) #Vol=0, edge=2
assert num.allclose(q, [2.0, 3.0]) # centroid value
def check_integrity(self):
Generic_Domain.check_integrity(self)
msg = 'Conserved quantity must be "stage"'
assert self.conserved_quantities[0] == 'stage', msg
