def to_vtk(self, prefix):
"""
export the result to vtk
the mesh and its triangle data go to prefix_name_mesh.vtk,
the potential/field/pseudopotential go to prefix_name.vtk
all arrays are named
"""
if self.configuration is not None:
self.configuration.to_vtk(prefix)
sp = tvtk.StructuredPoints(
origin=self.grid.get_origin(),
spacing=self.grid.step,
dimensions=self.grid.shape)
# spw = tvtk.StructuredPointsWriter(input=sp)
spw = tvtk.StructuredPointsWriter()
spw.set_input_data(sp)
spw.file_name = "%s_%s.vtk" % (prefix, self.configuration.name)
#xidr = tvtk.XMLImageDataWriter(input=sp)
for data_name in "potential field pseudo_potential".split():
data = getattr(self, data_name)
if data is None:
continue
if len(data.shape) > 3:
data = data.T.reshape(-1, data.shape[0])
else:
data = data.T.flatten()
d = tvtk.DoubleArray(name=data_name)
d.from_array(data)
sp.point_data.add_array(d)
spw.write()
logging.info("wrote %s", spw.file_name)
def test_parent_child_input(self):
"""Case where parent has GetInput and child SetInput."""
vm = tvtk.SmartVolumeMapper()
# In this case if the wrapping is not done right, the input
# trait is made read-only which is a bug. We set the input
# below to test this.
configure_input_data(vm, tvtk.ImageData())
spw = tvtk.StructuredPointsWriter()
spw.input_connection = None
def test_parent_child_input(self):
"""Case where parent has GetInput and child SetInput."""
vm = tvtk.VolumeTextureMapper2D()
# In this case if the wrapping is not done right, the input
# trait is made read-only which is a bug. We set the input
# below to test this.
vm.input = tvtk.ImageData()
spw = tvtk.StructuredPointsWriter()
spw.input = None
def test_parent_child_input(self):
"""Case where parent has GetInput and child SetInput."""
if vtk_major_version == 6 and vtk_minor_version > 1:
vm = tvtk.SmartVolumeMapper()
else:
vm = tvtk.VolumeTextureMapper2D()
# In this case if the wrapping is not done right, the input
# trait is made read-only which is a bug. We set the input
# below to test this.
configure_input_data(vm, tvtk.ImageData())
spw = tvtk.StructuredPointsWriter()
if vtk_major_version < 6:
spw.input = None
else:
spw.input_connection = None
