10d.js (6757B)
1 /** 2 * @license Apache-2.0 3 * 4 * Copyright (c) 2021 The Stdlib Authors. 5 * 6 * Licensed under the Apache License, Version 2.0 (the "License"); 7 * you may not use this file except in compliance with the License. 8 * You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 */ 18 19 /* eslint-disable max-depth */ 20 21 'use strict'; 22 23 // MAIN // 24 25 /** 26 * Applies a unary callback to elements in a ten-dimensional input ndarray and assigns results to elements in an equivalently shaped output ndarray. 27 * 28 * @private 29 * @param {Object} x - object containing input ndarray meta data 30 * @param {string} x.dtype - data type 31 * @param {Collection} x.data - data buffer 32 * @param {NonNegativeIntegerArray} x.shape - dimensions 33 * @param {IntegerArray} x.strides - stride lengths 34 * @param {NonNegativeInteger} x.offset - index offset 35 * @param {string} x.order - specifies whether `x` is row-major (C-style) or column-major (Fortran-style) 36 * @param {Object} y - object containing output ndarray meta data 37 * @param {string} y.dtype - data type 38 * @param {Collection} y.data - data buffer 39 * @param {NonNegativeIntegerArray} y.shape - dimensions 40 * @param {IntegerArray} y.strides - stride lengths 41 * @param {NonNegativeInteger} y.offset - index offset 42 * @param {string} y.order - specifies whether `y` is row-major (C-style) or column-major (Fortran-style) 43 * @param {Callback} fcn - unary callback 44 * @returns {void} 45 * 46 * @example 47 * var Float64Array = require( '@stdlib/array/float64' ); 48 * 49 * function scale( x ) { 50 * return x * 10.0; 51 * } 52 * 53 * // Create data buffers: 54 * var xbuf = new Float64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0 ] ); 55 * var ybuf = new Float64Array( 6 ); 56 * 57 * // Define the shape of the input and output arrays: 58 * var shape = [ 1, 1, 1, 1, 1, 1, 1, 3, 1, 2 ]; 59 * 60 * // Define the array strides: 61 * var sx = [ 12, 12, 12, 12, 12, 12, 12, 4, 4, 1 ]; 62 * var sy = [ 6, 6, 6, 6, 6, 6, 6, 2, 2, 1 ]; 63 * 64 * // Define the index offsets: 65 * var ox = 1; 66 * var oy = 0; 67 * 68 * // Create the input and output ndarray-like objects: 69 * var x = { 70 * 'dtype': 'float64', 71 * 'data': xbuf, 72 * 'shape': shape, 73 * 'strides': sx, 74 * 'offset': ox, 75 * 'order': 'row-major' 76 * }; 77 * var y = { 78 * 'dtype': 'float64', 79 * 'data': ybuf, 80 * 'shape': shape, 81 * 'strides': sy, 82 * 'offset': oy, 83 * 'order': 'row-major' 84 * }; 85 * 86 * // Apply the unary function: 87 * unary10d( x, y, scale ); 88 * 89 * console.log( y.data ); 90 * // => <Float64Array>[ 20.0, 30.0, 60.0, 70.0, 100.0, 110.0 ] 91 */ 92 function unary10d( x, y, fcn ) { // eslint-disable-line max-statements 93 var xbuf; 94 var ybuf; 95 var dx0; 96 var dx1; 97 var dx2; 98 var dx3; 99 var dx4; 100 var dx5; 101 var dx6; 102 var dx7; 103 var dx8; 104 var dx9; 105 var dy0; 106 var dy1; 107 var dy2; 108 var dy3; 109 var dy4; 110 var dy5; 111 var dy6; 112 var dy7; 113 var dy8; 114 var dy9; 115 var sh; 116 var S0; 117 var S1; 118 var S2; 119 var S3; 120 var S4; 121 var S5; 122 var S6; 123 var S7; 124 var S8; 125 var S9; 126 var sx; 127 var sy; 128 var ix; 129 var iy; 130 var i0; 131 var i1; 132 var i2; 133 var i3; 134 var i4; 135 var i5; 136 var i6; 137 var i7; 138 var i8; 139 var i9; 140 141 // Note on variable naming convention: S#, dx#, dy#, i# where # corresponds to the loop number, with `0` being the innermost loop... 142 143 // Extract loop variables for purposes of loop interchange: dimensions and loop offset (pointer) increments... 144 sh = x.shape; 145 sx = x.strides; 146 sy = y.strides; 147 if ( x.order === 'row-major' ) { 148 // For row-major ndarrays, the last dimensions have the fastest changing indices... 149 S0 = sh[ 9 ]; 150 S1 = sh[ 8 ]; 151 S2 = sh[ 7 ]; 152 S3 = sh[ 6 ]; 153 S4 = sh[ 5 ]; 154 S5 = sh[ 4 ]; 155 S6 = sh[ 3 ]; 156 S7 = sh[ 2 ]; 157 S8 = sh[ 1 ]; 158 S9 = sh[ 0 ]; 159 dx0 = sx[ 9 ]; // offset increment for innermost loop 160 dx1 = sx[ 8 ] - ( S0*sx[9] ); 161 dx2 = sx[ 7 ] - ( S1*sx[8] ); 162 dx3 = sx[ 6 ] - ( S2*sx[7] ); 163 dx4 = sx[ 5 ] - ( S3*sx[6] ); 164 dx5 = sx[ 4 ] - ( S4*sx[5] ); 165 dx6 = sx[ 3 ] - ( S5*sx[4] ); 166 dx7 = sx[ 2 ] - ( S6*sx[3] ); 167 dx8 = sx[ 1 ] - ( S7*sx[2] ); 168 dx9 = sx[ 0 ] - ( S8*sx[1] ); // offset increment for outermost loop 169 dy0 = sy[ 9 ]; 170 dy1 = sy[ 8 ] - ( S0*sy[9] ); 171 dy2 = sy[ 7 ] - ( S1*sy[8] ); 172 dy3 = sy[ 6 ] - ( S2*sy[7] ); 173 dy4 = sy[ 5 ] - ( S3*sy[6] ); 174 dy5 = sy[ 4 ] - ( S4*sy[5] ); 175 dy6 = sy[ 3 ] - ( S5*sy[4] ); 176 dy7 = sy[ 2 ] - ( S6*sy[3] ); 177 dy8 = sy[ 1 ] - ( S7*sy[2] ); 178 dy9 = sy[ 0 ] - ( S8*sy[1] ); 179 } else { // order === 'column-major' 180 // For column-major ndarrays, the first dimensions have the fastest changing indices... 181 S0 = sh[ 0 ]; 182 S1 = sh[ 1 ]; 183 S2 = sh[ 2 ]; 184 S3 = sh[ 3 ]; 185 S4 = sh[ 4 ]; 186 S5 = sh[ 5 ]; 187 S6 = sh[ 6 ]; 188 S7 = sh[ 7 ]; 189 S8 = sh[ 8 ]; 190 S9 = sh[ 9 ]; 191 dx0 = sx[ 0 ]; // offset increment for innermost loop 192 dx1 = sx[ 1 ] - ( S0*sx[0] ); 193 dx2 = sx[ 2 ] - ( S1*sx[1] ); 194 dx3 = sx[ 3 ] - ( S2*sx[2] ); 195 dx4 = sx[ 4 ] - ( S3*sx[3] ); 196 dx5 = sx[ 5 ] - ( S4*sx[4] ); 197 dx6 = sx[ 6 ] - ( S5*sx[5] ); 198 dx7 = sx[ 7 ] - ( S6*sx[6] ); 199 dx8 = sx[ 8 ] - ( S7*sx[7] ); 200 dx9 = sx[ 9 ] - ( S8*sx[8] ); // offset increment for outermost loop 201 dy0 = sy[ 0 ]; 202 dy1 = sy[ 1 ] - ( S0*sy[0] ); 203 dy2 = sy[ 2 ] - ( S1*sy[1] ); 204 dy3 = sy[ 3 ] - ( S2*sy[2] ); 205 dy4 = sy[ 4 ] - ( S3*sy[3] ); 206 dy5 = sy[ 5 ] - ( S4*sy[4] ); 207 dy6 = sy[ 6 ] - ( S5*sy[5] ); 208 dy7 = sy[ 7 ] - ( S6*sy[6] ); 209 dy8 = sy[ 8 ] - ( S7*sy[7] ); 210 dy9 = sy[ 9 ] - ( S8*sy[8] ); 211 } 212 // Set the pointers to the first indexed elements in the respective ndarrays... 213 ix = x.offset; 214 iy = y.offset; 215 216 // Cache references to the input and output ndarray buffers... 217 xbuf = x.data; 218 ybuf = y.data; 219 220 // Iterate over the ndarray dimensions... 221 for ( i9 = 0; i9 < S9; i9++ ) { 222 for ( i8 = 0; i8 < S8; i8++ ) { 223 for ( i7 = 0; i7 < S7; i7++ ) { 224 for ( i6 = 0; i6 < S6; i6++ ) { 225 for ( i5 = 0; i5 < S5; i5++ ) { 226 for ( i4 = 0; i4 < S4; i4++ ) { 227 for ( i3 = 0; i3 < S3; i3++ ) { 228 for ( i2 = 0; i2 < S2; i2++ ) { 229 for ( i1 = 0; i1 < S1; i1++ ) { 230 for ( i0 = 0; i0 < S0; i0++ ) { 231 ybuf[ iy ] = fcn( xbuf[ ix ] ); 232 ix += dx0; 233 iy += dy0; 234 } 235 ix += dx1; 236 iy += dy1; 237 } 238 ix += dx2; 239 iy += dy2; 240 } 241 ix += dx3; 242 iy += dy3; 243 } 244 ix += dx4; 245 iy += dy4; 246 } 247 ix += dx5; 248 iy += dy5; 249 } 250 ix += dx6; 251 iy += dy6; 252 } 253 ix += dx7; 254 iy += dy7; 255 } 256 ix += dx8; 257 iy += dy8; 258 } 259 ix += dx9; 260 iy += dy9; 261 } 262 } 263 264 265 // EXPORTS // 266 267 module.exports = unary10d;