allow uncompressed files

This commit is contained in:
Martin Diehl 2020-09-12 13:46:55 +02:00
parent a6ac875db8
commit a5d5638e4a
1 changed files with 5 additions and 2 deletions

View File

@ -168,7 +168,7 @@ class VTK:
def _write(writer): def _write(writer):
"""Wrapper for parallel writing.""" """Wrapper for parallel writing."""
writer.Write() writer.Write()
def to_file(self,fname,parallel=True): def to_file(self,fname,parallel=True,compress=True):
""" """
Write to file. Write to file.
@ -192,7 +192,10 @@ class VTK:
if ext and ext != '.'+default_ext: if ext and ext != '.'+default_ext:
raise ValueError(f'Given extension {ext} does not match default .{default_ext}') raise ValueError(f'Given extension {ext} does not match default .{default_ext}')
writer.SetFileName(str(Path(fname).with_suffix('.'+default_ext))) writer.SetFileName(str(Path(fname).with_suffix('.'+default_ext)))
writer.SetCompressorTypeToZLib() if compress:
writer.SetCompressorTypeToZLib()
else:
writer.SetCompressorTypeToNone()
writer.SetDataModeToBinary() writer.SetDataModeToBinary()
writer.SetInputData(self.vtk_data) writer.SetInputData(self.vtk_data)