class TestShearDisp(TestTri3):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestTri3.setUp(self)
run_pylith(ShearApp, GenerateDB)
self.outputRoot = "sheardisp"
self.soln = AnalyticalSoln()
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros((ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress" or name == "cauchy_stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
class TestShearDisp(TestTet4):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestTet4.setUp(self)
run_pylith()
self.outputRoot = "sheardisp"
self.soln = AnalyticalSoln()
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
class TestFaultsIntersectNoSlip(TestTet4):
"""
Test suite for testing pylith with shear slip on two faults.
"""
def setUp(self):
"""
Setup for test.
"""
TestTet4.setUp(self)
self.nverticesO = self.mesh['nvertices']
# Fault x
self.mesh['nvertices'] += 8
self.faultMeshX = {'nvertices': 19,
'spaceDim': 3,
'ncells': 20,
'ncorners': 3}
# Fault y
self.mesh['nvertices'] += 2
self.faultMeshY = {'nvertices': 9,
'spaceDim': 3,
'ncells': 8,
'ncorners': 3}
run_pylith()
self.outputRoot = "faultsintersectnoslip"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
from pylith.tests.Fault import check_vertex_fields
fields = ["normal_dir", "final_slip", "slip_time"]
self.fault = "x"
filename = "%s-faultx_info.h5" % self.outputRoot
check_vertex_fields(self, filename, self.faultMeshX, fields)
self.fault = "y"
filename = "%s-faulty_info.h5" % self.outputRoot
check_vertex_fields(self, filename, self.faultMeshY, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
from pylith.tests.Fault import check_vertex_fields
fields = ["slip"]
filename = "%s-faultx.h5" % self.outputRoot
self.fault = "x"
check_vertex_fields(self, filename, self.faultMeshX, fields)
filename = "%s-faulty.h5" % self.outputRoot
self.fault = "y"
check_vertex_fields(self, filename, self.faultMeshY, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
if self.fault == "x":
normalDir = (+1.0, 0.0, 0.0)
finalSlip = 0.0
faultMesh = self.faultMeshX
elif self.fault == "y":
normalDir = (0.0,-1.0, 0.0)
finalSlip = 0.0
faultMesh = self.faultMeshY
slipTime = 0.0
dim = 3
nvertices = faultMesh['nvertices']
if name == "normal_dir":
field = numpy.zeros( (1, nvertices, dim), dtype=numpy.float64)
field[0,:,0] = normalDir[0]
field[0,:,1] = normalDir[1]
field[0,:,2] = normalDir[2]
elif name == "final_slip":
field = numpy.zeros( (1, nvertices, dim), dtype=numpy.float64)
field[0,:,0] = finalSlip
elif name == "slip_time":
field = slipTime*numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros( (1, nvertices, dim), dtype=numpy.float64)
field[0,:,0] = finalSlip
elif name == "traction_change":
field = numpy.zeros( (1, nvertices, dim), dtype=numpy.float64)
field[0,:,0] = 0.0
else:
raise ValueError("Unknown fault field '%s'." % name)
return field
class TestShearDispNoSlip(TestTet4):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestTet4.setUp(self)
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 8
self.faultMesh = {'nvertices': 19,
'spaceDim': 3,
'ncells': 20,
'ncorners': 3}
run_pylith()
self.outputRoot = "sheardispnoslip"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["normal_dir", "final_slip", "slip_time"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress" or name == "cauchy_stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (+1.0, 0.0, 0.0)
finalSlip = 0.0
slipTime = 0.0
nvertices = self.faultMesh['nvertices']
if name == "normal_dir":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = normalDir[0]
field[0,:,1] = normalDir[1]
field[0,:,2] = normalDir[2]
elif name == "final_slip":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = finalSlip
elif name == "slip_time":
field = slipTime*numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = finalSlip
elif name == "traction_change":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = 0.0
else:
raise ValueError("Unknown fault field '%s'." % name)
# Mask clamped edges
maskX = numpy.fabs(vertices[:,0]) <= 1.0
maskY = numpy.fabs(vertices[:,1]) <= 25.001e+3
maskZ = vertices[:,2] >= -20.001e+3
mask = numpy.bitwise_and(numpy.bitwise_and(maskX,maskY),maskZ)
field[:,~mask,:] = 0.0
return field
class TestShearDispNoSlipRefine(TestHex8):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestHex8.setUp(self)
self.mesh = {'ncells-elastic': 180*8,
'ncells-viscoelastic': 180*8,
'ncorners': 8,
'nvertices': 3591,
'spaceDim': 3,
'tensorSize': 6}
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 44
self.faultMesh = {'nvertices': 65,
'spaceDim': 3,
'ncells': 48,
'ncorners': 4}
run_pylith()
self.outputRoot = "sheardispnosliprefine"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["normal_dir", "final_slip", "slip_time"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (-1.0, 0.0, 0.0)
finalSlip = 0.0
slipTime = 0.0
nvertices = self.faultMesh['nvertices']
if name == "normal_dir":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = normalDir[0]
field[0,:,1] = normalDir[1]
field[0,:,2] = normalDir[2]
elif name == "final_slip":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = finalSlip
elif name == "slip_time":
field = slipTime*numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = finalSlip
elif name == "traction_change":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = 0.0
else:
raise ValueError("Unknown fault field '%s'." % name)
return field
class TestShearDispFriction(TestTri3):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestTri3.setUp(self)
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 1
self.faultMesh = {'nvertices': 3,
'spaceDim': 2,
'ncells': 2,
'ncorners': 2}
run_pylith()
self.outputRoot = "sheardispfriction"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["strike_dir", "normal_dir", "traction_initial","friction_coefficient","cohesion"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
strikeDir = (-1.0, 0.0)
normalDir = (0.0, 1.0)
initialTraction = (0.0, -100.0e+6)
frictionCoefficient = 1.0
nvertices = self.faultMesh['nvertices']
if name == "strike_dir":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = strikeDir[0]
field[0,:,1] = strikeDir[1]
elif name == "normal_dir":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = normalDir[0]
field[0,:,1] = normalDir[1]
elif name == "traction_initial":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = initialTraction[0]
field[0,:,1] = initialTraction[1]
elif name == "friction_coefficient":
field = numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
field[0,:,0] = frictionCoefficient
elif name == "cohesion":
field = numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
elif name == "traction":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = 1.0e+6
field[0,:,1] = -1.0e+7
else:
raise ValueError("Unknown fault field '%s'." % name)
# Mask clamped vertices
maskX = numpy.bitwise_and(vertices[:,0] >= -2.0001e+3, vertices[:,0] <= 1.0)
maskY = numpy.fabs(vertices[:,1]) <= 0.01e+3
mask = numpy.bitwise_and(maskX,maskY)
field[:,~mask,:] = 0
return field
class TestShearDispNoSlipRefine(TestTet4):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestTet4.setUp(self)
self.mesh = {
'ncells-elastic': 1266 * 8,
'ncells-viscoelastic': 1276 * 8,
'ncorners': 4,
'nvertices': 4016,
'spaceDim': 3,
'tensorSize': 6
}
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 36
self.faultMesh = {
'nvertices': 57,
'spaceDim': 3,
'ncells': 80,
'ncorners': 3
}
run_pylith(ShearApp, GenerateDB, nprocs=4)
self.outputRoot = "sheardispnosliprefine"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["normal_dir", "final_slip", "slip_time"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros((ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress" or name == "cauchy_stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (+1.0, 0.0, 0.0)
finalSlip = 0.0
slipTime = 0.0
nvertices = self.faultMesh['nvertices']
if name == "normal_dir":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = normalDir[0]
field[0, :, 1] = normalDir[1]
field[0, :, 2] = normalDir[2]
elif name == "final_slip":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = finalSlip
elif name == "slip_time":
field = slipTime * numpy.zeros(
(1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = finalSlip
elif name == "traction_change":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = 0.0
else:
raise ValueError("Unknown fault field '%s'." % name)
# Mask clamped edges
maskX = numpy.fabs(vertices[:, 0]) <= 1.0
maskY = numpy.fabs(vertices[:, 1]) <= 25.001e+3
maskZ = vertices[:, 2] >= -20.001e+3
mask = numpy.bitwise_and(numpy.bitwise_and(maskX, maskY), maskZ)
field[:, ~mask, :] = 0.0
return field
class TestFaultsIntersectNoSlip(TestTet4):
"""
Test suite for testing pylith with shear slip on two faults.
"""
def setUp(self):
"""
Setup for test.
"""
TestTet4.setUp(self)
self.nverticesO = self.mesh['nvertices']
# Fault x
self.mesh['nvertices'] += 8
self.faultMeshX = {
'nvertices': 19,
'spaceDim': 3,
'ncells': 20,
'ncorners': 3
}
# Fault y
self.mesh['nvertices'] += 2
self.faultMeshY = {
'nvertices': 9,
'spaceDim': 3,
'ncells': 8,
'ncorners': 3
}
run_pylith(FaultsIntersectNoSlipApp, GenerateDB)
self.outputRoot = "faultsintersectnoslip"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
from pylith.tests.Fault import check_vertex_fields
fields = ["normal_dir", "final_slip", "slip_time"]
self.fault = "x"
filename = "%s-faultx_info.h5" % self.outputRoot
check_vertex_fields(self, filename, self.faultMeshX, fields)
self.fault = "y"
filename = "%s-faulty_info.h5" % self.outputRoot
check_vertex_fields(self, filename, self.faultMeshY, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
from pylith.tests.Fault import check_vertex_fields
fields = ["slip"]
filename = "%s-faultx.h5" % self.outputRoot
self.fault = "x"
check_vertex_fields(self, filename, self.faultMeshX, fields)
filename = "%s-faulty.h5" % self.outputRoot
self.fault = "y"
check_vertex_fields(self, filename, self.faultMeshY, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros((ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress" or name == "cauchy_stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
if self.fault == "x":
normalDir = (+1.0, 0.0, 0.0)
finalSlip = 0.0
faultMesh = self.faultMeshX
elif self.fault == "y":
normalDir = (0.0, -1.0, 0.0)
finalSlip = 0.0
faultMesh = self.faultMeshY
slipTime = 0.0
dim = 3
nvertices = faultMesh['nvertices']
if name == "normal_dir":
field = numpy.zeros((1, nvertices, dim), dtype=numpy.float64)
field[0, :, 0] = normalDir[0]
field[0, :, 1] = normalDir[1]
field[0, :, 2] = normalDir[2]
elif name == "final_slip":
field = numpy.zeros((1, nvertices, dim), dtype=numpy.float64)
field[0, :, 0] = finalSlip
elif name == "slip_time":
field = slipTime * numpy.zeros(
(1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros((1, nvertices, dim), dtype=numpy.float64)
field[0, :, 0] = finalSlip
elif name == "traction_change":
field = numpy.zeros((1, nvertices, dim), dtype=numpy.float64)
field[0, :, 0] = 0.0
else:
raise ValueError("Unknown fault field '%s'." % name)
return field
class TestFrictionNoSlip(TestHex8):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestHex8.setUp(self)
self.nverticesO = self.mesh["nvertices"]
self.mesh["nvertices"] += 10
self.faultMesh = {"nvertices": 21, "spaceDim": 3, "ncells": 12, "ncorners": 4}
run_pylith()
self.outputRoot = "frictionnoslip"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["normal_dir", "strike_dir", "static_coefficient", "traction_initial"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh["ncells"]
pts = numpy.zeros((ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (-1.0, 0.0, 0.0)
strikeDir = (0.0, -1.0, 0.0)
staticCoef = 0.6
initialTraction = (0.0, 0.0, -100.0e6)
nvertices = self.faultMesh["nvertices"]
if name == "normal_dir":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = normalDir[0]
field[0, :, 1] = normalDir[1]
field[0, :, 2] = normalDir[2]
elif name == "strike_dir":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = strikeDir[0]
field[0, :, 1] = strikeDir[1]
field[0, :, 2] = strikeDir[2]
elif name == "static_coefficient":
field = staticCoef * numpy.ones((1, nvertices, 1), dtype=numpy.float64)
elif name == "traction_initial":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
field[0, :, 0] = initialTraction[0]
field[0, :, 1] = initialTraction[1]
field[0, :, 2] = initialTraction[2]
elif name == "slip":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
else:
raise ValueError("Unknown fault field '%s'." % name)
# Mask clamped vertices
maskX = numpy.fabs(vertices[:, 0]) <= 1.0
maskY = numpy.fabs(vertices[:, 1]) <= 25.001e3
maskZ = vertices[:, 2] >= -20.001e3
mask = numpy.bitwise_and(numpy.bitwise_and(maskX, maskY), maskZ)
field[:, ~mask, :] = 0.0
return field
class TestFrictionNoSlipHalo(TestHex8):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestHex8.setUp(self)
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 21
self.faultMesh = {'nvertices': 34,
'spaceDim': 3,
'ncells': 23,
'ncorners': 4}
run_pylith()
self.outputRoot = "frictionnoslip_halo"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["static_coefficient"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (-1.0, 0.0, 0.0)
strikeDir = (0.0, -1.0, 0.0)
staticCoef = 0.6
initialTraction = (0.0,0.0,-100.0e+6)
nvertices = self.faultMesh['nvertices']
if name == "static_coefficient":
field = staticCoef*numpy.ones( (1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
else:
raise ValueError("Unknown fault field '%s'." % name)
return field
class TestFrictionNoSlip(TestHex8):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestHex8.setUp(self)
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 10
self.faultMesh = {'nvertices': 21,
'spaceDim': 3,
'ncells': 12,
'ncorners': 4}
run_pylith(ShearApp, GenerateDB, nprocs=3)
self.outputRoot = "frictionnoslip"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["normal_dir","strike_dir","static_coefficient", "traction_initial"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros( (ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (-1.0, 0.0, 0.0)
strikeDir = (0.0, -1.0, 0.0)
staticCoef = 0.6
initialTraction = (0.0,0.0,-100.0e+6)
nvertices = self.faultMesh['nvertices']
if name == "normal_dir":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = normalDir[0]
field[0,:,1] = normalDir[1]
field[0,:,2] = normalDir[2]
elif name == "strike_dir":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = strikeDir[0]
field[0,:,1] = strikeDir[1]
field[0,:,2] = strikeDir[2]
elif name == "static_coefficient":
field = staticCoef*numpy.ones( (1, nvertices, 1), dtype=numpy.float64)
elif name == "traction_initial":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
field[0,:,0] = initialTraction[0]
field[0,:,1] = initialTraction[1]
field[0,:,2] = initialTraction[2]
elif name == "slip":
field = numpy.zeros( (1, nvertices, 3), dtype=numpy.float64)
else:
raise ValueError("Unknown fault field '%s'." % name)
# Mask clamped vertices
maskX = numpy.fabs(vertices[:,0]) <= 1.0
maskY = numpy.fabs(vertices[:,1]) <= 25.001e+3
maskZ = vertices[:,2] >= -20.001e+3
mask = numpy.bitwise_and(numpy.bitwise_and(maskX,maskY),maskZ)
field[:,~mask,:] = 0.0
return field
class TestFrictionNoSlipHalo(TestHex8):
"""
Test suite for testing pylith with 2-D shear extension.
"""
def setUp(self):
"""
Setup for test.
"""
TestHex8.setUp(self)
self.nverticesO = self.mesh['nvertices']
self.mesh['nvertices'] += 21
self.faultMesh = {
'nvertices': 34,
'spaceDim': 3,
'ncells': 23,
'ncorners': 4
}
run_pylith(ShearApp, GenerateDB, nprocs=1)
self.outputRoot = "frictionnoslip_halo"
self.soln = AnalyticalSoln()
return
def test_fault_info(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault_info.h5" % self.outputRoot
fields = ["static_coefficient"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def test_fault_data(self):
"""
Check fault information.
"""
if not self.checkResults:
return
filename = "%s-fault.h5" % self.outputRoot
fields = ["slip"]
from pylith.tests.Fault import check_vertex_fields
check_vertex_fields(self, filename, self.faultMesh, fields)
return
def calcDisplacements(self, vertices):
"""
Calculate displacement field given coordinates of vertices.
"""
return self.soln.displacement(vertices)
def calcStateVar(self, name, vertices, cells):
"""
Calculate state variable.
"""
ncells = self.mesh['ncells']
pts = numpy.zeros((ncells, 3), dtype=numpy.float64)
if name == "total_strain":
stateVar = self.soln.strain(pts)
elif name == "stress":
stateVar = self.soln.stress(pts)
else:
raise ValueError("Unknown state variable '%s'." % name)
return stateVar
def calcFaultField(self, name, vertices):
"""
Calculate fault info.
"""
normalDir = (-1.0, 0.0, 0.0)
strikeDir = (0.0, -1.0, 0.0)
staticCoef = 0.6
initialTraction = (0.0, 0.0, -100.0e+6)
nvertices = self.faultMesh['nvertices']
if name == "static_coefficient":
field = staticCoef * numpy.ones(
(1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros((1, nvertices, 3), dtype=numpy.float64)
else:
raise ValueError("Unknown fault field '%s'." % name)
return field
