def setUp(self):
"""
Setup for test.
"""
TestQuad4.setUp(self)
run_pylith(AxialApp, GenerateDB)
self.outputRoot = "axialdisp"
self.soln = AnalyticalSoln()
return
def run(self):
"""
Generate the database.
"""
from axialdisp_soln import AnalyticalSoln
soln = AnalyticalSoln()
from spatialdata.geocoords.CSCart import CSCart
cs = CSCart()
cs.inventory.spaceDim = 3
cs._configure()
from spatialdata.spatialdb.SimpleIOAscii import SimpleIOAscii
io = SimpleIOAscii()
for component in ["x", "y", "z"]:
if component == "x":
xyz = numpy.array([[-40.0e+3, 0.0, 0.0], [+40.0e+3, 0.0, 0.0]],
dtype=numpy.float64)
ic = 0
elif component == "y":
xyz = numpy.array([[0.0, -40.0e+3, 0.0], [0.0, +40.0e+3, 0.0]],
dtype=numpy.float64)
ic = 1
elif component == "z":
xyz = numpy.array([[0.0, 0.0, -40.0e+3], [0.0, 0.0, 0.0]],
dtype=numpy.float64)
ic = 2
disp = soln.displacement(xyz)
io.inventory.filename = "axial_disp%s.spatialdb" % component
io._configure()
data = {
'points':
xyz,
'coordsys':
cs,
'data_dim':
1,
'values': [
{
'name': "displacement-%s" % component,
'units': "m",
'data': numpy.ravel(disp[0, :, ic])
},
]
}
io.write(data)
return
def run(self):
"""Generate the database.
"""
# Domain
x = numpy.arange(-4000.0, 4000.1, 1000.0)
y = numpy.arange(-4000.0, 4000.1, 1000.0)
npts = x.shape[0]
xx = x * numpy.ones((npts, 1), dtype=numpy.float64)
yy = y * numpy.ones((npts, 1), dtype=numpy.float64)
xy = numpy.zeros((npts**2, 2), dtype=numpy.float64)
xy[:, 0] = numpy.ravel(xx)
xy[:, 1] = numpy.ravel(numpy.transpose(yy))
from axialdisp_soln import AnalyticalSoln
soln = AnalyticalSoln()
disp = soln.displacement(xy)
from spatialdata.geocoords.CSCart import CSCart
cs = CSCart()
cs.inventory.spaceDim = 2
cs._configure()
data = {
'points':
xy,
'coordsys':
cs,
'data_dim':
2,
'values': [{
'name': "initial_amplitude_x",
'units': "m",
'data': numpy.ravel(disp[0, :, 0])
}, {
'name': "initial_amplitude_y",
'units': "m",
'data': numpy.ravel(disp[0, :, 1])
}]
}
from spatialdata.spatialdb.SimpleIOAscii import createWriter
io = createWriter("axialdisp_bc.spatialdb")
io.write(data)
data["values"][0]["name"] = "displacement_x"
data["values"][1]["name"] = "displacement_y"
io = createWriter("axialdisp_ic.spatialdb")
io.write(data)
return
def run(self):
"""Generate the database.
"""
# Domain
x = numpy.arange(-4.0e+3, 4.01e+3, 1.0e+3)
y = numpy.arange(-4.0e+3, 4.01e+3, 1.0e+3)
z = numpy.arange(-8.0e+3, 0.01e+3, 1.0e+3)
x3, y3, z3 = numpy.meshgrid(x, y, z)
nptsX = x.shape[0]
nptsY = y.shape[0]
nptsZ = z.shape[0]
xyz = numpy.zeros((nptsX * nptsY * nptsZ, 3), dtype=numpy.float64)
xyz[:, 0] = x3.ravel()
xyz[:, 1] = y3.ravel()
xyz[:, 2] = z3.ravel()
from axialdisp_soln import AnalyticalSoln
soln = AnalyticalSoln()
disp = soln.displacement(xyz)
from spatialdata.geocoords.CSCart import CSCart
cs = CSCart()
cs.inventory.spaceDim = 3
cs._configure()
data = {
"x":
x,
"y":
y,
"z":
z,
"points":
xyz,
"coordsys":
cs,
"data_dim":
3,
"values": [
{
"name": "initial_amplitude_x",
"units": "m",
"data": numpy.ravel(disp[0, :, 0])
},
{
"name": "initial_amplitude_y",
"units": "m",
"data": numpy.ravel(disp[0, :, 1])
},
{
"name": "initial_amplitude_z",
"units": "m",
"data": numpy.ravel(disp[0, :, 2])
},
]
}
from spatialdata.spatialdb.SimpleGridAscii import SimpleGridAscii
io = SimpleGridAscii()
io.inventory.filename = "axialdisp_bc.spatialdb"
io._configure()
io.write(data)
return
def setUp(self):
"""Setup for test.
"""
FullTestCase.setUp(self)
self.exactsoln = AnalyticalSoln()
return
