191 lines
4.6 KiB
Markdown
191 lines
4.6 KiB
Markdown
<!--
|
|
|
|
@license Apache-2.0
|
|
|
|
Copyright (c) 2018 The Stdlib Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
|
|
-->
|
|
|
|
# sub2ind
|
|
|
|
> Convert subscripts to a linear index.
|
|
|
|
<!-- Section to include introductory text. Make sure to keep an empty line after the intro `section` element and another before the `/section` close. -->
|
|
|
|
<section class="intro">
|
|
|
|
</section>
|
|
|
|
<!-- /.intro -->
|
|
|
|
<!-- Package usage documentation. -->
|
|
|
|
<section class="usage">
|
|
|
|
## Usage
|
|
|
|
```javascript
|
|
var sub2ind = require( '@stdlib/ndarray/sub2ind' );
|
|
```
|
|
|
|
#### sub2ind( shape, ...subscripts\[, options] )
|
|
|
|
Converts subscripts to a linear index.
|
|
|
|
```javascript
|
|
var shape = [ 2, 2 ];
|
|
|
|
var idx = sub2ind( shape, 1, 0 );
|
|
// returns 2
|
|
```
|
|
|
|
The function supports the following `options`:
|
|
|
|
- `mode`: specifies how to handle subscripts which exceed array dimensions. The following modes are supported:
|
|
|
|
- `throw`: specifies that the function should throw an error when a subscript exceeds array dimensions.
|
|
- `wrap`: specifies that the function should wrap around subscripts exceeding array dimensions using modulo arithmetic.
|
|
- `clamp`: specifies that the function should set subscripts exceeding array dimensions to either `0` (minimum index) or the maximum index along a particular dimension.
|
|
|
|
If provided a `mode` array, each array element corresponds to a dimension. If provided fewer modes than dimensions, the function reuses modes using modulo arithmetic. Default: `[ 'throw' ]`.
|
|
|
|
- `order`: specifies whether an array is `row-major` (C-style) or `column-major` (Fortran-style). Default: `'row-major'`.
|
|
|
|
By default, the function assumes a row-major array. To return a linear index for a column-major array, set the `order` option.
|
|
|
|
```javascript
|
|
var shape = [ 2, 2 ];
|
|
var opts = {};
|
|
|
|
opts.order = 'column-major';
|
|
var idx = sub2ind( shape, 1, 0, opts );
|
|
// returns 1
|
|
```
|
|
|
|
By default, the function throws an `error` if provided subscripts which exceed array dimensions. To specify alternative behavior, set the `mode` option.
|
|
|
|
```javascript
|
|
var shape = [ 2, 2 ];
|
|
var opts = {};
|
|
|
|
opts.mode = 'wrap';
|
|
var idx = sub2ind( shape, -2, 0, opts );
|
|
// returns 0
|
|
|
|
opts.mode = 'clamp';
|
|
idx = sub2ind( shape, 10, 10, opts );
|
|
// returns 3
|
|
```
|
|
|
|
To specify separate modes for each dimension, provide a `mode` array.
|
|
|
|
```javascript
|
|
var shape = [ 2, 2, 2 ];
|
|
var opts = {
|
|
'mode': [
|
|
'wrap',
|
|
'clamp'
|
|
]
|
|
};
|
|
|
|
var idx = sub2ind( shape, -2, 10, -1, opts );
|
|
// returns 3
|
|
```
|
|
|
|
</section>
|
|
|
|
<!-- /.usage -->
|
|
|
|
<!-- Package usage notes. Make sure to keep an empty line after the `section` element and another before the `/section` close. -->
|
|
|
|
<section class="notes">
|
|
|
|
</section>
|
|
|
|
<!-- /.notes -->
|
|
|
|
<!-- Package usage examples. -->
|
|
|
|
<section class="examples">
|
|
|
|
## Examples
|
|
|
|
<!-- eslint no-undef: "error" -->
|
|
|
|
```javascript
|
|
var numel = require( '@stdlib/ndarray/base/numel' );
|
|
var sub2ind = require( '@stdlib/ndarray/sub2ind' );
|
|
|
|
var shape = [ 3, 3, 3 ];
|
|
var len = numel( shape );
|
|
|
|
var arr = [];
|
|
var i;
|
|
for ( i = 0; i < len; i++ ) {
|
|
arr.push( i );
|
|
}
|
|
|
|
var opts = {
|
|
'order': 'column-major'
|
|
};
|
|
|
|
console.log( '' );
|
|
console.log( 'Dimensions: %s.', shape.join( 'x' ) );
|
|
console.log( 'View:' );
|
|
console.log( '' );
|
|
|
|
var slice;
|
|
var idx;
|
|
var row;
|
|
var j;
|
|
var k;
|
|
for ( k = 0; k < shape[ 2 ]; k++ ) {
|
|
slice = 'A[:,:,'+k+'] = ';
|
|
console.log( slice );
|
|
for ( i = 0; i < shape[ 0 ]; i++ ) {
|
|
row = ' ';
|
|
for ( j = 0; j < shape[ 1 ]; j++ ) {
|
|
idx = sub2ind( shape, i, j, k, opts );
|
|
row += arr[ idx ];
|
|
if ( j < shape[ 1 ]-1 ) {
|
|
row += ', ';
|
|
}
|
|
}
|
|
console.log( row );
|
|
}
|
|
console.log( '' );
|
|
}
|
|
```
|
|
|
|
</section>
|
|
|
|
<!-- /.examples -->
|
|
|
|
<!-- Section to include cited references. If references are included, add a horizontal rule *before* the section. Make sure to keep an empty line after the `section` element and another before the `/section` close. -->
|
|
|
|
<section class="references">
|
|
|
|
</section>
|
|
|
|
<!-- /.references -->
|
|
|
|
<!-- Section for all links. Make sure to keep an empty line after the `section` element and another before the `/section` close. -->
|
|
|
|
<section class="links">
|
|
|
|
</section>
|
|
|
|
<!-- /.links -->
|