def __init__(self, discr, pcontext=None, basename=None, compressor=None):
hedge.tools.Closable.__init__(self)
from pytools import assert_not_a_file
if basename is not None:
self.pvd_name = basename+".pvd"
assert_not_a_file(self.pvd_name)
else:
self.pvd_name = None
self.pcontext = pcontext
self.compressor = compressor
if self.pcontext is None or self.pcontext.is_head_rank:
self.timestep_to_pathnames = {}
else:
self.timestep_to_pathnames = None
from pyvisfile.vtk import UnstructuredGrid, DataArray, \
VTK_LINE, VTK_TRIANGLE, VTK_TETRA, VF_LIST_OF_VECTORS
from hedge.mesh.element import Interval, Triangle, Tetrahedron
# For now, we use IntVector here because the Python allocator
# is somewhat reluctant to return allocated chunks of memory
# to the OS.
from hedge._internal import IntVector
cells = IntVector()
cell_types = IntVector()
for eg in discr.element_groups:
ldis = eg.local_discretization
smi = ldis.get_submesh_indices()
cells.reserve(len(cells)+len(smi)*len(eg.members))
for el, el_slice in zip(eg.members, eg.ranges):
for element in smi:
for j in element:
cells.append(el_slice.start+j)
if ldis.geometry is Interval:
vtk_eltype = VTK_LINE
elif ldis.geometry is Triangle:
vtk_eltype = VTK_TRIANGLE
elif ldis.geometry is Tetrahedron:
vtk_eltype = VTK_TETRA
else:
raise RuntimeError("unsupported element type: %s"
% ldis.geometry)
cell_types.extend([vtk_eltype] * len(smi) * len(eg.members))
self.grid = UnstructuredGrid(
(len(discr),
DataArray("points", discr.nodes, vector_format=VF_LIST_OF_VECTORS)),
numpy.asarray(cells),
cell_types=numpy.asarray(cell_types, dtype=numpy.uint8))
def _cp(src, dest):
from pytools import assert_not_a_file
assert_not_a_file(dest)
inf = open(src, "rb")
try:
outf = open(dest, "wb")
try:
outf.write(inf.read())
finally:
outf.close()
finally:
inf.close()
def do_close(self):
from pytools import assert_not_a_file
assert_not_a_file(self.pathname)
outf = file(self.pathname, "w")
from pyvisfile.vtk import AppendedDataXMLGenerator
AppendedDataXMLGenerator(self.compressor)(self.grid).write(outf)
#from pyvisfile.vtk import InlineXMLGenerator
#InlineXMLGenerator(self.compressor)(self.grid).write(outf)
outf.close()
def do_close(self):
from pytools import assert_not_a_file
assert_not_a_file(self.pathname)
outf = file(self.pathname, "w")
from pyvisfile.vtk import AppendedDataXMLGenerator
AppendedDataXMLGenerator(self.compressor)(self.grid).write(outf)
#from pyvisfile.vtk import InlineXMLGenerator
#InlineXMLGenerator(self.compressor)(self.grid).write(outf)
outf.close()
def __init__(self, discr, pcontext=None, basename=None, compressor=None):
logger.info("init vtk visualizer: start")
hedge.tools.Closable.__init__(self)
from pytools import assert_not_a_file
if basename is not None:
self.pvd_name = basename + ".pvd"
assert_not_a_file(self.pvd_name)
else:
self.pvd_name = None
self.pcontext = pcontext
self.compressor = compressor
if self.pcontext is None or self.pcontext.is_head_rank:
self.timestep_to_pathnames = {}
else:
self.timestep_to_pathnames = None
from pyvisfile.vtk import UnstructuredGrid, DataArray, \
VTK_LINE, VTK_TRIANGLE, VTK_TETRA, VF_LIST_OF_VECTORS
from hedge.mesh.element import Interval, Triangle, Tetrahedron
# For now, we use IntVector here because the Python allocator
# is somewhat reluctant to return allocated chunks of memory
# to the OS.
from hedge._internal import IntVector
cells = IntVector()
cell_types = IntVector()
for eg in discr.element_groups:
ldis = eg.local_discretization
smi = ldis.get_submesh_indices()
cells.reserve(len(cells) + len(smi) * len(eg.members))
for el, el_slice in zip(eg.members, eg.ranges):
for element in smi:
for j in element:
cells.append(el_slice.start + j)
if ldis.geometry is Interval:
vtk_eltype = VTK_LINE
elif ldis.geometry is Triangle:
vtk_eltype = VTK_TRIANGLE
elif ldis.geometry is Tetrahedron:
vtk_eltype = VTK_TETRA
else:
raise RuntimeError("unsupported element type: %s" %
ldis.geometry)
cell_types.extend([vtk_eltype] * len(smi) * len(eg.members))
self.grid = UnstructuredGrid(
(len(discr),
DataArray("points", discr.nodes,
vector_format=VF_LIST_OF_VECTORS)),
np.asarray(cells),
cell_types=np.asarray(cell_types, dtype=np.uint8))
logger.info("init vtk visualizer: done")
