geometry0 itself is useful

can be combined with 'place' if in-memory VTK is required for spatial
operations
This commit is contained in:
Martin Diehl 2021-04-03 08:58:22 +02:00
parent 39c0bc9de6
commit 885aeb62e5
1 changed files with 12 additions and 9 deletions

View File

@ -530,6 +530,17 @@ class Result:
with h5py.File(self.fname,'r') as f: with h5py.File(self.fname,'r') as f:
return f['geometry/x_n'][()] return f['geometry/x_n'][()]
@property
def geometry0(self):
if self.structured:
return VTK.from_rectilinear_grid(self.cells,self.size,self.origin)
else:
with h5py.File(self.fname,'r') as f:
return VTK.from_unstructured_grid(f['/geometry/x_n'][()],
f['/geometry/T_c'][()]-1,
f['/geometry/T_c'].attrs['VTK_TYPE'] if h5py3 else \
f['/geometry/T_c'].attrs['VTK_TYPE'].decode())
@staticmethod @staticmethod
def _add_absolute(x): def _add_absolute(x):
@ -1267,15 +1278,7 @@ class Result:
""" """
if mode.lower()=='cell': if mode.lower()=='cell':
v = self.geometry0
if self.structured:
v = VTK.from_rectilinear_grid(self.cells,self.size,self.origin)
else:
with h5py.File(self.fname,'r') as f:
v = VTK.from_unstructured_grid(f['/geometry/x_n'][()],
f['/geometry/T_c'][()]-1,
f['/geometry/T_c'].attrs['VTK_TYPE'] if h5py3 else \
f['/geometry/T_c'].attrs['VTK_TYPE'].decode())
elif mode.lower()=='point': elif mode.lower()=='point':
v = VTK.from_poly_data(self.coordinates0_point) v = VTK.from_poly_data(self.coordinates0_point)