223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

support pathlib.Path

This commit is contained in:
Martin Diehl 2020-06-28 07:17:51 +02:00
parent 8d5a72e936
commit 8a9a768a77
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/damask/_vtk.py
View File

@ -121,7 +121,7 @@ class VTK:
Parameters Parameters
---------- ----------
fname : str fname : str or pathlib.Path
Filename for reading. Valid extensions are .vtr, .vtu, .vtp, and .vtk. Filename for reading. Valid extensions are .vtr, .vtu, .vtp, and .vtk.
dataset_type : str, optional dataset_type : str, optional
Name of the vtk.vtkDataSet subclass when opening an .vtk file. Valid types are vtkRectilinearGrid, Name of the vtk.vtkDataSet subclass when opening an .vtk file. Valid types are vtkRectilinearGrid,
@ -131,7 +131,7 @@ class VTK:
ext = Path(fname).suffix ext = Path(fname).suffix
if ext == '.vtk' or dataset_type: if ext == '.vtk' or dataset_type:
reader = vtk.vtkGenericDataObjectReader() reader = vtk.vtkGenericDataObjectReader()
reader.SetFileName(fname) reader.SetFileName(str(fname))
reader.Update() reader.Update()
if dataset_type is None: if dataset_type is None:
raise TypeError('Dataset type for *.vtk file not given.') raise TypeError('Dataset type for *.vtk file not given.')
@ -153,7 +153,7 @@ class VTK:
else: else:
raise TypeError(f'Unknown file extension {ext}') raise TypeError(f'Unknown file extension {ext}')
reader.SetFileName(fname) reader.SetFileName(str(fname))
reader.Update() reader.Update()
geom = reader.GetOutput() geom = reader.GetOutput()
@ -166,7 +166,7 @@ class VTK:
Parameters Parameters
---------- ----------
fname : str fname : str or pathlib.Path
Filename for writing. Filename for writing.
""" """