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

fast and for arbitrary type

This commit is contained in:
Martin Diehl 2020-08-23 09:02:17 +02:00
parent b731b1e768
commit 5fb2d30ee4
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/damask/_geom.py
View File

@ -6,6 +6,7 @@ from functools import partial
import numpy as np import numpy as np
from scipy import ndimage,spatial from scipy import ndimage,spatial
from vtk.util.numpy_support import vtk_to_numpy as vtk_to_np
from . import environment from . import environment
from . import Rotation from . import Rotation
@ -357,8 +358,6 @@ class Geom:
""" """
g = VTK.from_file(fname).geom g = VTK.from_file(fname).geom
N_cells = g.GetNumberOfCells()
microstructure = np.zeros(N_cells,'i')
grid = np.array(g.GetDimensions())-1 grid = np.array(g.GetDimensions())-1
bbox = np.array(g.GetBounds()).reshape(3,2).T bbox = np.array(g.GetBounds()).reshape(3,2).T
size = bbox[1] - bbox[0] size = bbox[1] - bbox[0]
@ -366,9 +365,7 @@ class Geom:
celldata = g.GetCellData() celldata = g.GetCellData()
for a in range(celldata.GetNumberOfArrays()): for a in range(celldata.GetNumberOfArrays()):
if celldata.GetArrayName(a) == 'microstructure': if celldata.GetArrayName(a) == 'microstructure':
array = celldata.GetArray(a) microstructure = vtk_to_np(celldata.GetArray(a))
for c in range(N_cells):
microstructure[c] = array.GetValue(c)
return Geom(microstructure.reshape(grid,order='F'),size,bbox[0]) return Geom(microstructure.reshape(grid,order='F'),size,bbox[0])