DAMASK.GeomGrid to pyvista.ImageData

[eisenlohr]

A function to create a pyvista ImageData object from an existing damask.GeomGrid:

import pyvista as pv

def GeomGrid_to_pyvistaImageData(grid):
    """
    Transform a damask.GeomGrid into a corresponding pyvista ImageData object.

    Parameters
    ----------
    grid : damask.GeomGrid
        The GeomGrid to transform.

    Returns
    -------
    img : pyvista.ImageData
        Corresponding pyvista object with a data array 'material'.

    """
    img = pv.ImageData(dimensions=grid.cells+1,
                       spacing=grid.size/grid.cells,
                       origin=grid.origin,
                      )
    img['material'] = grid.material.flatten('F')
    return img