Piece ID from vtkXMLPUnstructuredGridReader

When using the parallel VTU file reader (vtkXMLPUnstructuredGridReader), is there a way to have the reader automatically add an array corresponding to the piece of the mesh? (e.g. integers from 0 to NumberOfPeices

I have a workaround by loading each piece and adding a point array of a constant integer to each piece, then appending all of the meshes together, but it would be great if this was available from the vtkXMLPUnstructuredGridReader class

If you’re looking for sample data, this issue has a PVTU dataset you can use: https://github.com/pyvista/pyvista-support/issues/83#issuecomment-565195472