support comments also in vtr

This commit is contained in:
Martin Diehl 2020-08-25 00:40:14 +02:00
parent 9a2ac31545
commit 2751cdb6af
2 changed files with 6 additions and 4 deletions

View File

@ -359,11 +359,12 @@ class Geom:
"""
v = VTK.from_file(fname if str(fname).endswith('.vtr') else str(fname)+'.vtr')
comments = v.get_comments()
grid = np.array(v.geom.GetDimensions())-1
bbox = np.array(v.geom.GetBounds()).reshape(3,2).T
size = bbox[1] - bbox[0]
return Geom(v.get('materialpoint').reshape(grid,order='F'),size,bbox[0])
return Geom(v.get('materialpoint').reshape(grid,order='F'),size,bbox[0],comments=comments)
@staticmethod
@ -527,6 +528,7 @@ class Geom:
"""
v = VTK.from_rectilinearGrid(self.grid,self.size,self.origin)
v.add(self.microstructure.flatten(order='F'),'materialpoint')
v.add_comments(self.comments)
if fname:
v.write(fname if str(fname).endswith('.vtr') else str(fname)+'.vtr')

View File

@ -284,7 +284,7 @@ class VTK:
def set_comments(self,comments):
"""
Set Comments.
Set comments.
Parameters
----------
@ -301,7 +301,7 @@ class VTK:
def add_comments(self,comments):
"""
Add Comments.
Add comments.
Parameters
----------
@ -309,7 +309,7 @@ class VTK:
Comments to add.
"""
self.set_comments(self.get_comments + ([comments] if isinstance(comments,str) else comments))
self.set_comments(self.get_comments() + ([comments] if isinstance(comments,str) else comments))
def __repr__(self):