consistency check and new test
This commit is contained in:
parent
2aea6df678
commit
892b7eee98
2
PRIVATE
2
PRIVATE
|
@ -1 +1 @@
|
|||
Subproject commit c5683e828c9a761ad046926e26d6cc6cc10c355c
|
||||
Subproject commit 05ac971ce58fc399dd99be9151b7d61d049aec42
|
|
@ -158,7 +158,6 @@ class VTK:
|
|||
return VTK(geom)
|
||||
|
||||
|
||||
# ToDo: If extension is given, check for consistency.
|
||||
def write(self,fname):
|
||||
"""
|
||||
Write to file.
|
||||
|
@ -176,8 +175,11 @@ class VTK:
|
|||
elif(isinstance(self.geom,vtk.vtkPolyData)):
|
||||
writer = vtk.vtkXMLPolyDataWriter()
|
||||
|
||||
writer.SetFileName('{}.{}'.format(os.path.splitext(fname)[0],
|
||||
writer.GetDefaultFileExtension()))
|
||||
default_ext = writer.GetDefaultFileExtension()
|
||||
name, ext = os.path.splitext(fname)
|
||||
if ext and ext != default_ext:
|
||||
raise ValueError('Given extension {} is not {}'.format(ext,default_ext))
|
||||
writer.SetFileName('{}.{}'.format(name,default_ext))
|
||||
writer.SetCompressorTypeToZLib()
|
||||
writer.SetDataModeToBinary()
|
||||
writer.SetInputData(self.geom)
|
||||
|
|
Loading…
Reference in New Issue