def _runTest(self):
"""
Run test.
"""
#filenameIn = "data/tet4.exo"
#filenameIn = "tri3_200m_gradient.exo"
filenameIn = "tet4_150m.exo"
from pylith.perf.MemoryLogger import MemoryLogger
self.logger = MemoryLogger()
self.logger._configure()
from pylith.utils.petsc import MemoryLogger
sieveLogger = MemoryLogger.singleton()
from pylith.topology.topology import MeshOps_numMaterialCells
from pylith.mpi.Communicator import petsc_comm_world
comm = petsc_comm_world()
self._showStatus("Before reading mesh")
# ------------------------------------------------------------
# Read mesh
from pylith.meshio.MeshIOCubit import MeshIOCubit
io = MeshIOCubit()
io.inventory.filename = filenameIn
io.inventory.useNames = False
io._configure()
from pylith.materials.ElasticIsotropic3D import ElasticIsotropic3D
material = ElasticIsotropic3D()
material.inventory.label = "Elastic material"
material.inventory.id = 1
material._configure()
mesh = io.read(debug=False, interpolate=False)
self.logger.logMesh(mesh.memLoggingStage, mesh)
material.ncells = MeshOps_numMaterialCells(mesh, material.id())
self.logger.logMaterial(mesh.memLoggingStage, material)
self._showStatus("After reading mesh")
# ------------------------------------------------------------
# Reorder mesh
from pylith.topology.ReverseCuthillMcKee import ReverseCuthillMcKee
ordering = ReverseCuthillMcKee()
ordering.reorder(mesh)
# Expect no memory allocation
self.logger.memory["MeshReordering"] = 0
self._showStatus("After reordering mesh")
# ------------------------------------------------------------
# Distribute mesh
if comm.size > 1:
from pylith.topology.Distributor import Distributor
from spatialdata.units.Nondimensional import Nondimensional
distributor = Distributor()
normalizer = Nondimensional()
dmesh = distributor.distribute(mesh, normalizer)
dmesh.memLoggingStage = "DistributedMesh"
self.logger.logMesh(mesh.memLoggingStage, mesh)
material.ncells = MeshOps_numMaterialCells(mesh, material.id())
self.logger.logMaterial(mesh.memLoggingStage, material)
self.logger.logMesh(dmesh.memLoggingStage, dmesh)
material.ncells = MeshOps_numMaterialCells(dmesh, material.id())
self.logger.logMaterial(dmesh.memLoggingStage, material)
self._showStatus("After distributing mesh")
mesh = dmesh
if False: # Refine mesh (if necessary)
from pylith.topology.RefineUniform import RefineUniform
refiner = RefineUniform()
rmesh = refiner.refine(mesh)
rmesh.memLoggingStage = "RefinedMesh"
print "Unrefined mesh logging stage",mesh.memLoggingStage
self.logger.logMesh(mesh.memLoggingStage, mesh)
material.ncells = MeshOps_numMaterialCells(mesh, material.id())
self.logger.logMaterial(mesh.memLoggingStage, material)
self.logger.logMesh(rmesh.memLoggingStage, rmesh)
material.ncells = MeshOps_numMaterialCells(rmesh, material.id())
self.logger.logMaterial(rmesh.memLoggingStage, material)
self._showStatus("After refining mesh")
return
class TestApp(Script):
"""
Test application.
"""
# PUBLIC METHODS /////////////////////////////////////////////////////
def __init__(self, name="testapp"):
"""
Constructor.
"""
Script.__init__(self, name)
return
def main(self, *args, **kwds):
"""
Run the application.
"""
from pylith.utils.PetscManager import PetscManager
manager = PetscManager()
manager.initialize()
self._runTest()
manager.finalize()
return
# PRIVATE METHODS ////////////////////////////////////////////////////
def _runTest(self):
"""
Run test.
"""
#filenameIn = "data/tet4.exo"
#filenameIn = "tri3_200m_gradient.exo"
filenameIn = "tet4_150m.exo"
from pylith.perf.MemoryLogger import MemoryLogger
self.logger = MemoryLogger()
self.logger._configure()
from pylith.utils.petsc import MemoryLogger
sieveLogger = MemoryLogger.singleton()
from pylith.topology.topology import MeshOps_numMaterialCells
from pylith.mpi.Communicator import petsc_comm_world
comm = petsc_comm_world()
self._showStatus("Before reading mesh")
# ------------------------------------------------------------
# Read mesh
from pylith.meshio.MeshIOCubit import MeshIOCubit
io = MeshIOCubit()
io.inventory.filename = filenameIn
io.inventory.useNames = False
io._configure()
from pylith.materials.ElasticIsotropic3D import ElasticIsotropic3D
material = ElasticIsotropic3D()
material.inventory.label = "Elastic material"
material.inventory.id = 1
material._configure()
mesh = io.read(debug=False, interpolate=False)
self.logger.logMesh(mesh.memLoggingStage, mesh)
material.ncells = MeshOps_numMaterialCells(mesh, material.id())
self.logger.logMaterial(mesh.memLoggingStage, material)
self._showStatus("After reading mesh")
# ------------------------------------------------------------
# Reorder mesh
from pylith.topology.ReverseCuthillMcKee import ReverseCuthillMcKee
ordering = ReverseCuthillMcKee()
ordering.reorder(mesh)
# Expect no memory allocation
self.logger.memory["MeshReordering"] = 0
self._showStatus("After reordering mesh")
# ------------------------------------------------------------
# Distribute mesh
if comm.size > 1:
from pylith.topology.Distributor import Distributor
from spatialdata.units.Nondimensional import Nondimensional
distributor = Distributor()
normalizer = Nondimensional()
dmesh = distributor.distribute(mesh, normalizer)
dmesh.memLoggingStage = "DistributedMesh"
self.logger.logMesh(mesh.memLoggingStage, mesh)
material.ncells = MeshOps_numMaterialCells(mesh, material.id())
self.logger.logMaterial(mesh.memLoggingStage, material)
self.logger.logMesh(dmesh.memLoggingStage, dmesh)
material.ncells = MeshOps_numMaterialCells(dmesh, material.id())
self.logger.logMaterial(dmesh.memLoggingStage, material)
self._showStatus("After distributing mesh")
mesh = dmesh
if False: # Refine mesh (if necessary)
from pylith.topology.RefineUniform import RefineUniform
refiner = RefineUniform()
rmesh = refiner.refine(mesh)
rmesh.memLoggingStage = "RefinedMesh"
print "Unrefined mesh logging stage",mesh.memLoggingStage
self.logger.logMesh(mesh.memLoggingStage, mesh)
material.ncells = MeshOps_numMaterialCells(mesh, material.id())
self.logger.logMaterial(mesh.memLoggingStage, material)
self.logger.logMesh(rmesh.memLoggingStage, rmesh)
material.ncells = MeshOps_numMaterialCells(rmesh, material.id())
self.logger.logMaterial(rmesh.memLoggingStage, material)
self._showStatus("After refining mesh")
return
def _showStatus(self, stage):
import sys
from pylith.utils.profiling import resourceUsageString
from pylith.mpi.Communicator import petsc_comm_world
comm = petsc_comm_world()
if comm.rank == 0:
print "\n----------------------------------------------------------------------"
print "STAGE: %s" % stage
print "----------------------------------------------------------------------"
for irank in xrange(comm.size):
comm.barrier()
if comm.rank == irank:
print "\nPROCESSOR %d" % comm.rank
print "\nStatus from ps: %s\n" % resourceUsageString()
self.logger.show()
sys.stdout.flush()
comm.barrier()
return
