arrays

add_array

PVGeo._helpers.arrays.add_array(pdo, field, vtkArray)[source]

Adds an array to a vtkDataObject given its field association.

Parameters:
  • pdo (vtkDataObject) – the output data object
  • field (int or str) – the field type id or name
  • vtkArray (vtkDataArray) – the data array to add to the output
Returns:

the output data object with the data array added

Return type:

vtkDataObject

copy_arrays_to_point_data

PVGeo._helpers.arrays.copy_arrays_to_point_data(pdi, pdo, field)[source]

Copys arrays from an input to an ouput’s point data.

Parameters:
  • pdi (vtkDataObject) – The input data object to copy from
  • pdo (vtkDataObject) – The output data object to copy over to
  • field (int or str) – the field type id or name
Returns:

returns the output data object parameter

Return type:

vtkDataObject

get_all_array_names

PVGeo._helpers.arrays.get_all_array_names(dataset, field)[source]

get_numpy_array

PVGeo._helpers.arrays.get_numpy_array(wpdi, field, name)[source]

Grabs an array from vtkDataObject given its name and field association.

Parameters:
  • wpdi (wrapped vtkDataObject) – the input data object wrapped using vtk dataset adapter
  • field (int or str) – the field type id or name
  • name (str) – the name of the input array for the given index
Returns:

a wrapped vtkDataArray for NumPy

Return type:

numpy.array

get_selected_array

PVGeo._helpers.arrays.get_selected_array(algorithm, wpdi, idx)[source]

Gets selectected array at index idx wrapped for NumPy

Parameters:
  • algorithm (vtkAlgorithm) – A vtkAlgorithm class instantiation
  • wpdi (wrapped vtkDataObject) – the input data object wrapped using vtk dataset adapter
  • idx (int) – the input array index
Returns:

a wrapped vtkDataArray for NumPy

Return type:

numpy.array

get_selected_array_field

PVGeo._helpers.arrays.get_selected_array_field(algorithm, idx)[source]

Gets the field of the input array for a given index on a VTK algorithm

Parameters:
  • algorithm (vtkAlgorithm) – A vtkAlgorithm class instantiation
  • idx (int) – the input array index
Returns:

the field type of the input array for the given index

Return type:

int

get_selected_array_name

PVGeo._helpers.arrays.get_selected_array_name(algorithm, idx)[source]

Gets the name of the input array for a given index on a VTK algorithm

Parameters:
  • algorithm (vtkAlgorithm) – A vtkAlgorithm class instantiation
  • idx (int) – the input array index
Returns:

the name of the input array for the given index

Return type:

str

get_vtk_array

PVGeo._helpers.arrays.get_vtk_array(pdi, field, name)[source]

Grabs an array from vtkDataObject given its name and field association.

Parameters:
  • pdi (vtkDataObject) – the input data object
  • field (int or str) – the field type id or name
  • name (str) – the name of the input array for the given index
Returns:

the array from input field and name

Return type:

vtkDataArray

search_for_array

PVGeo._helpers.arrays.search_for_array(pdi, name)[source]