Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 107x 107x 107x 107x 107x 107x 107x 107x 107x 107x 107x 107x 416x 416x 416x 416x 88x 88x 328x 328x 256x 256x 103x 103x 103x 153x 153x 153x 153x 153x 256x 72x 72x 70x 70x 53x 53x 17x 17x 9x 9x 9x 9x 9x 8x 8x 8x 8x 70x 2x 2x 2x 2x 2x 2x 416x 99x 107x 3x 3x 3x 3x 3x | /**
* @license Apache-2.0
*
* Copyright (c) 2026 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.
*/
'use strict';
// MODULES //
var isMostlySafeCast = require( '@stdlib/ndarray/base/assert/is-mostly-safe-data-type-cast' );
var isEqualDataType = require( '@stdlib/ndarray/base/assert/is-equal-data-type' );
var maybeBroadcastArray = require( '@stdlib/ndarray/base/maybe-broadcast-array' );
var broadcastScalar = require( '@stdlib/ndarray/base/broadcast-scalar' );
var getOrder = require( '@stdlib/ndarray/base/order' );
var getShape = require( '@stdlib/ndarray/base/shape' );
var baseEmpty = require( '@stdlib/ndarray/base/empty' );
var assign = require( '@stdlib/ndarray/base/assign' );
var format = require( '@stdlib/string/format' );
var ENUMS = require( './type_enums.js' );
// MAIN //
/**
* Normalizes a list of arguments to a list of ndarrays.
*
* @private
* @param {Array} args - list of arguments
* @param {Array} types - argument types
* @param {Array} dtypes - argument data types
* @param {NonNegativeIntegerArray} shape - array shape
* @param {string} order - array order
* @throws {TypeError} only (mostly) safe casts are supported
* @returns {Array<ndarray>} list of ndarrays
*/
function normalizeArguments( args, types, dtypes, shape, order ) {
var odt;
var out;
var N;
var t;
var v;
var i;
N = args.length;
out = [ null, null, null, null ]; // [ base, start, stop, endpoint ]
odt = dtypes[ 4 ];
for ( i = 0; i < N; i++ ) {
t = types[ i ];
// Case: args[ i ] === endpoint
if ( t === ENUMS.BOOLEAN ) {
out[ i ] = broadcastScalar( args[ i ], dtypes[ i ], shape, order );
}
// Case: args[ i ] === <number>
else if ( t === ENUMS.NUMBER ) {
// Case: args[ i ] === base
if ( i === 0 ) {
// As underlying kernels expect a real-valued base, avoid casting the base to a complex number when the output data type is complex:
out[ i ] = broadcastScalar( args[ i ], dtypes[ i ], shape, order ); // eslint-disable-line max-len
}
// Case: args[ i ] === start/stop
else {
// A number primitive should be able to cast to any supported output data type (real or complex):
out[ i ] = broadcastScalar( args[ i ], odt, shape, order );
}
}
// Case: args[ i ] === <ndarray>
else if ( t === ENUMS.NDARRAY ) {
// Case: args[ i ] === base/endpoint || start/stop which have the same dtype as the output dtype
if ( i === 0 || i === 3 || isEqualDataType( dtypes[ i ], odt ) ) {
out[ i ] = maybeBroadcastArray( args[ i ], shape );
}
// Case: args[ i ] === start/stop which have different dtype than output dtype
else if ( isMostlySafeCast( dtypes[ i ], odt ) ) {
// If we have unequal data types, we need to perform a copy...
v = baseEmpty( odt, getShape( args[ i ], false ), getOrder( args[ i ] ) ); // eslint-disable-line max-len
assign( [ args[ i ], v ] );
out[ i ] = maybeBroadcastArray( v, shape );
}
// Case: disallowed complex-to-real data type cast
else {
throw new TypeError( format( 'invalid argument. Argument %d cannot be safely cast to the desired output data type. Output data type: %s. Argument data type: %s.', i+1, String( odt ), String( dtypes[ i ] ) ) );
}
}
// Case: args[ i ] === <complex>
else {
// Complex number scalars should only be able to cast to complex data types...
if ( !isMostlySafeCast( dtypes[ i ], odt ) ) {
throw new TypeError( format( 'invalid argument. Argument %d cannot be safely cast to the desired output data type. Output data type: %s. Argument data type: %s.', i+1, String( odt ), String( dtypes[ i ] ) ) );
}
out[ i ] = broadcastScalar( args[ i ], odt, shape, order );
}
}
return out;
}
// EXPORTS //
module.exports = normalizeArguments;
|