indexer.py#
Classes that handle array indexing.
- class openmdao.utils.indexer.ArrayIndexer(arr, flat_src=None)[source]
Bases:
ShapedArrayIndexer
Abstract index array class that may or may not be ‘shaped’.
- Parameters:
- arrndarray
The index array.
- flat_srcbool or None
If True, treat source as flat.
- __init__(arr, flat_src=None)
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- slice
The offset slice.
- as_array(copy=False, flat=True)
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
The index into a flat array.
- property indexed_src_shape
Return the shape of the result of indexing into the source.
- Returns:
- tuple
The shape of the index.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()[source]
Return a ‘shaped’ version of this Indexer type.
- Returns:
- ShapedArrayIndexer or None
Will return a ShapedArrayIndexer if possible, else None.
- to_json()
Return a JSON serializable version of self.
- Returns:
- list of int or int
List or int version of self.
- class openmdao.utils.indexer.EllipsisIndexer(tup, flat_src=None)[source]
Bases:
Indexer
Abstract multi indexer class that is ‘shaped’.
- Parameters:
- tuptuple
Tuple of indices/slices.
- flat_srcbool
If True, treat source array as flat.
- Attributes:
- _tuptuple
The wrapped tuple of indices/slices (it contains an ellipsis).
- __init__(tup, flat_src=None)[source]
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)[source]
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- ndarray
The offset array.
- as_array(copy=False, flat=True)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()[source]
Copy this Indexer.
- Returns:
- EllipsisIndexer
A copy of this Indexer.
- flat(copy=False)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
An index array into a flat array.
- property indexed_src_shape
Return the shape of the result if the indices were applied to a source array.
- Returns:
- tuple
The shape of the result.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()[source]
Return a ‘shaped’ version of this Indexer type.
- Returns:
- A shaped Indexer or None
Will return some kind of shaped Indexer if possible, else None.
- to_json()[source]
Return a JSON serializable version of self.
- Returns:
- list of int or int
A list or int version of self.
- class openmdao.utils.indexer.IndexMaker[source]
Bases:
object
A Factory for Indexer objects.
- __getitem__(idx)[source]
Return an Indexer based on idx.
- Parameters:
- idxint, ndarray, slice or tuple
The passed indices/slices.
- Returns:
- Indexer
The Indexer instance we created based on the args.
- class openmdao.utils.indexer.Indexer(flat_src=None)[source]
Bases:
object
Abstract indexing class.
- Parameters:
- flat_srcbool
True if we’re treating the source as flat.
- Attributes:
- _src_shapetuple or None
Shape of the ‘source’. Used to determine actual index or slice values when indices are negative or slice contains negative start or stop values or ‘:’ or ‘…’.
- _shaped_instIndexer or None
Cached shaped_instance if we’ve computed it before.
- _flat_srcbool
If True, index is into a flat source array.
- _dist_shapetuple
Distributed shape of the source.
- __init__(flat_src=None)[source]
Initialize attributes.
- apply(subidxer)[source]
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- copy(*args)[source]
Copy this Indexer.
- Parameters:
- *argsposition args
Args that are specific to initialization of a derived Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)[source]
Return index array or slice into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- property indexed_src_shape
Return the shape of the result if the indices were applied to a source array.
- Returns:
- tuple
The shape of the result.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- set_src_shape(shape, dist_shape=None)[source]
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)[source]
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()[source]
Return a ‘shaped’ version of this Indexer type.
This should be overridden for all non-shaped derived classes.
- Returns:
- Indexer
The ‘shaped’ Indexer type. ‘shaped’ Indexers know the extent of the array that they are indexing into, or they don’t care what the extent is because they don’t contain negative indices, negative start or stop, ‘:’, or ‘…’.
- to_json()[source]
Return a JSON serializable version of self.
- class openmdao.utils.indexer.IntIndexer(idx, flat_src=None)[source]
Bases:
ShapedIntIndexer
Int indexing class that may or may not be ‘shaped’.
- Parameters:
- idxint
The index.
- flat_srcbool or None
If True, treat source as flat.
- __init__(idx, flat_src=None)
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- int
The offset index.
- as_array(copy=False, flat=True)
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)
Return index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
The index into a flat array.
- property indexed_src_shape
Return the shape of the index ().
- Returns:
- tuple
The shape of the index.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()[source]
Return a ‘shaped’ version of this Indexer type.
- Returns:
- ShapedIntIndexer or None
Will return a ShapedIntIndexer if possible, else None.
- to_json()
Return a JSON serializable version of self.
- Returns:
- int
Int version of self.
- class openmdao.utils.indexer.MultiIndexer(tup, flat_src=False)[source]
Bases:
ShapedMultiIndexer
Abstract multi indexer class that may or may not be ‘shaped’.
- Parameters:
- tuptuple
Tuple of indices/slices.
- flat_srcbool
If True, treat source array as flat.
- __init__(tup, flat_src=False)
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- ndarray
The offset array.
- as_array(copy=False, flat=True)
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array into a flat array.
- copy()
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
An index array into a flat array.
- property indexed_src_shape
Return the shape of the result if the indices were applied to a source array.
- Returns:
- tuple
The shape of the result.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()[source]
Return a ‘shaped’ version of this Indexer type.
- Returns:
- ShapedMultiIndexer or None
Will return a ShapedMultiIndexer if possible, else None.
- to_json()
Return a JSON serializable version of self.
- Returns:
- list of int or int
List or int version of self.
- class openmdao.utils.indexer.ShapedArrayIndexer(arr, flat_src=None)[source]
Bases:
Indexer
Abstract index array class that knows its source shape.
- Parameters:
- arrndarray
The index array.
- flat_srcbool
If True, source is treated as flat.
- Attributes:
- _arrndarray
The wrapped index array object.
- __init__(arr, flat_src=None)[source]
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)[source]
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- slice
The offset slice.
- as_array(copy=False, flat=True)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()[source]
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
The index into a flat array.
- property indexed_src_shape
Return the shape of the result if the indices were applied to a source array.
- Returns:
- tuple
The shape of the result.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()
Return a ‘shaped’ version of this Indexer type.
This should be overridden for all non-shaped derived classes.
- Returns:
- Indexer
The ‘shaped’ Indexer type. ‘shaped’ Indexers know the extent of the array that they are indexing into, or they don’t care what the extent is because they don’t contain negative indices, negative start or stop, ‘:’, or ‘…’.
- to_json()[source]
Return a JSON serializable version of self.
- Returns:
- list of int or int
List or int version of self.
- class openmdao.utils.indexer.ShapedIntIndexer(idx, flat_src=None)[source]
Bases:
Indexer
Int indexing class.
- Parameters:
- idxint
The index.
- flat_srcbool
If True, source is treated as flat.
- Attributes:
- _idxint
The integer index.
- __init__(idx, flat_src=None)[source]
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)[source]
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- int
The offset index.
- as_array(copy=False, flat=True)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()[source]
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)[source]
Return index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
The index into a flat array.
- property indexed_src_shape
Return the shape of the index ().
- Returns:
- tuple
The shape of the index.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()
Return a ‘shaped’ version of this Indexer type.
This should be overridden for all non-shaped derived classes.
- Returns:
- Indexer
The ‘shaped’ Indexer type. ‘shaped’ Indexers know the extent of the array that they are indexing into, or they don’t care what the extent is because they don’t contain negative indices, negative start or stop, ‘:’, or ‘…’.
- to_json()[source]
Return a JSON serializable version of self.
- Returns:
- int
Int version of self.
- class openmdao.utils.indexer.ShapedMultiIndexer(tup, flat_src=False)[source]
Bases:
Indexer
Abstract multi indexer class that is ‘shaped’.
- Parameters:
- tuptuple
Tuple of indices/slices.
- flat_srcbool
If True, treat source array as flat.
- Attributes:
- _tuptuple
The wrapped tuple of indices/slices.
- _idx_listlist
List of Indexers.
- __init__(tup, flat_src=False)[source]
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)[source]
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- ndarray
The offset array.
- as_array(copy=False, flat=True)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array into a flat array.
- copy()[source]
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- ndarray
An index array into a flat array.
- property indexed_src_shape
Return the shape of the result if the indices were applied to a source array.
- Returns:
- tuple
The shape of the result.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)[source]
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()
Return a ‘shaped’ version of this Indexer type.
This should be overridden for all non-shaped derived classes.
- Returns:
- Indexer
The ‘shaped’ Indexer type. ‘shaped’ Indexers know the extent of the array that they are indexing into, or they don’t care what the extent is because they don’t contain negative indices, negative start or stop, ‘:’, or ‘…’.
- to_json()[source]
Return a JSON serializable version of self.
- Returns:
- list of int or int
List or int version of self.
- class openmdao.utils.indexer.ShapedSliceIndexer(slc, flat_src=None)[source]
Bases:
Indexer
Abstract slice class that is ‘shaped’.
- Parameters:
- slcslice
The slice.
- flat_srcbool
If True, source is treated as flat.
- Attributes:
- _sliceslice
The wrapped slice object.
- __init__(slc, flat_src=None)[source]
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)[source]
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- slice
The offset slice.
- as_array(copy=False, flat=True)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()[source]
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)[source]
Return a slice into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- slice
The slice into a flat array.
- property indexed_src_shape
Return the shape of the result if the indices were applied to a source array.
- Returns:
- tuple
The shape of the result.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()
Return a ‘shaped’ version of this Indexer type.
This should be overridden for all non-shaped derived classes.
- Returns:
- Indexer
The ‘shaped’ Indexer type. ‘shaped’ Indexers know the extent of the array that they are indexing into, or they don’t care what the extent is because they don’t contain negative indices, negative start or stop, ‘:’, or ‘…’.
- to_json()[source]
Return a JSON serializable version of self.
- Returns:
- list of int or int
List or int version of self.
- class openmdao.utils.indexer.SliceIndexer(slc, flat_src=None)[source]
Bases:
ShapedSliceIndexer
Abstract slice class that may or may not be ‘shaped’.
- Parameters:
- slcslice
The slice.
- flat_srcbool or None
If True, treat source as flat.
- __init__(slc, flat_src=None)
Initialize attributes.
- apply(subidxer)
Apply a sub-Indexer to this Indexer and return the resulting indices.
- Parameters:
- subidxerIndexer
The Indexer to be applied to this one.
- Returns:
- ndarray
The resulting indices (always flat).
- apply_offset(offset, flat=True)
Apply an offset to this index.
- Parameters:
- offsetint
The offset to apply.
- flatbool
If True, return a flat index.
- Returns:
- slice
The offset slice.
- as_array(copy=False, flat=True)[source]
Return an index array into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
The index array.
- copy()
Copy this Indexer.
- Returns:
- Indexer
A copy of this Indexer.
- flat(copy=False)
Return a slice into a flat array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- Returns:
- slice
The slice into a flat array.
- property indexed_src_shape
Return the shape of the result of indexing into the source.
- Returns:
- tuple
The shape of the index.
- property indexed_src_size
Return the size of the result if the index were applied to the source.
- Returns:
- int
Size of flattened indices.
- property min_src_dim
Return the number of source dimensions.
- Returns:
- int
The number of dimensions expected in the source array.
- set_src_shape(shape, dist_shape=None)
Set the shape of the ‘source’ array .
- Parameters:
- shapetuple or int
The shape of the ‘source’ array.
- dist_shapetuple or None
If not None, the full distributed shape of the source.
- Returns:
- Indexer
Self is returned to allow chaining.
- shaped_array(copy=False, flat=True)
Return an index array version of the indices that index into a flattened array.
- Parameters:
- copybool
If True, make sure the array returned is a copy.
- flatbool
If True, return a flat array.
- Returns:
- ndarray
Version of these indices that index into a flattened array.
- shaped_instance()[source]
Return a ‘shaped’ version of this Indexer type.
- Returns:
- ShapedSliceIndexer or None
Will return a ShapedSliceIndexer if possible, else None.
- to_json()
Return a JSON serializable version of self.
- Returns:
- list of int or int
List or int version of self.
- class openmdao.utils.indexer.Slicer[source]
Bases:
object
Helper class that can be used when a slice is needed for indexing.
- __getitem__(val)[source]
Pass through indices or slice.
- Parameters:
- valint or slice object or tuples of slice objects
Indices or slice to return.
- Returns:
- indicesint or slice object or tuples of slice objects
Indices or slice to return.
- openmdao.utils.indexer.array2slice(arr)[source]
Try to convert an array to slice.
Conversion is only attempted for a 1D array.
- Parameters:
- arrndarray
The index array to be represented as a slice.
- Returns:
- slice or None
If slice conversion is possible, return the slice, else return None.
- openmdao.utils.indexer.combine_ranges(ranges)[source]
Combine a list of (start, end) tuples into the smallest possible list of contiguous ranges.
The ranges are assumed to be non-overlapping and in ascending order.
- Parameters:
- rangeslist
List of (start, end) tuples.
- Returns:
- list of tuples
List of combined ranges.
- openmdao.utils.indexer.ranges2indexer(ranges, src_shape=None)[source]
Convert a list of ranges to an indexer.
- Parameters:
- rangeslist
List of (start, end) tuples.
- src_shapetuple or None
The shape of the source array.
- Returns:
- Indexer
Indexer object.
- class openmdao.utils.indexer.resolve_shape(shape)[source]
Bases:
object
Class that computes the result shape from a source shape and an index.
- Parameters:
- shapetuple
The shape of the source.
- Attributes:
- _shapetuple
The shape of the source.
- __init__(shape)[source]
Initialize attributes.
- Parameters:
- shapetuple or int
Shape of the source.
- get_shape(idx)[source]
Return the shape of the result of indexing into the source with index idx.
- Parameters:
- idxint, slice, tuple, ndarray
The index into the source.
- Returns:
- tuple
The shape after indexing.