10d_accessors.js (7443B)
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 {Function} x.getter - callback for accessing `x` data buffer elements 37 * @param {Object} y - object containing output ndarray meta data 38 * @param {string} y.dtype - data type 39 * @param {Collection} y.data - data buffer 40 * @param {NonNegativeIntegerArray} y.shape - dimensions 41 * @param {IntegerArray} y.strides - stride lengths 42 * @param {NonNegativeInteger} y.offset - index offset 43 * @param {string} y.order - specifies whether `y` is row-major (C-style) or column-major (Fortran-style) 44 * @param {Function} y.setter - callback for setting `y` data buffer elements 45 * @param {Callback} fcn - unary callback 46 * @returns {void} 47 * 48 * @example 49 * var Complex64Array = require( '@stdlib/array/complex64' ); 50 * var Complex64 = require( '@stdlib/complex/float32' ); 51 * var real = require( '@stdlib/complex/real' ); 52 * var imag = require( '@stdlib/complex/imag' ); 53 * 54 * function scale( z ) { 55 * return new Complex64( real(z)*10.0, imag(z)*10.0 ); 56 * } 57 * 58 * // Create data buffers: 59 * var xbuf = new Complex64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0 ] ); 60 * var ybuf = new Complex64Array( 4 ); 61 * 62 * // Define the shape of the input and output arrays: 63 * var shape = [ 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 ]; 64 * 65 * // Define the array strides: 66 * var sx = [ 2, 2, 2, 2, 2, 2, 2, 2, 2, 1 ]; 67 * var sy = [ 2, 2, 2, 2, 2, 2, 2, 2, 2, 1 ]; 68 * 69 * // Define the index offsets: 70 * var ox = 0; 71 * var oy = 0; 72 * 73 * // Define getters and setters: 74 * function getter( buf, idx ) { 75 * return buf.get( idx ); 76 * } 77 * 78 * function setter( buf, idx, value ) { 79 * buf.set( value, idx ); 80 * } 81 * 82 * // Create the input and output ndarray-like objects: 83 * var x = { 84 * 'dtype': 'complex64', 85 * 'data': xbuf, 86 * 'shape': shape, 87 * 'strides': sx, 88 * 'offset': ox, 89 * 'order': 'row-major', 90 * 'getter': getter 91 * }; 92 * var y = { 93 * 'dtype': 'complex64', 94 * 'data': ybuf, 95 * 'shape': shape, 96 * 'strides': sy, 97 * 'offset': oy, 98 * 'order': 'row-major', 99 * 'setter': setter 100 * }; 101 * 102 * // Apply the unary function: 103 * unary10d( x, y, scale ); 104 * 105 * var v = y.data.get( 0 ); 106 * 107 * var re = real( v ); 108 * // returns 10.0 109 * 110 * var im = imag( v ); 111 * // returns 20.0 112 */ 113 function unary10d( x, y, fcn ) { // eslint-disable-line max-statements 114 var xbuf; 115 var ybuf; 116 var get; 117 var set; 118 var dx0; 119 var dx1; 120 var dx2; 121 var dx3; 122 var dx4; 123 var dx5; 124 var dx6; 125 var dx7; 126 var dx8; 127 var dx9; 128 var dy0; 129 var dy1; 130 var dy2; 131 var dy3; 132 var dy4; 133 var dy5; 134 var dy6; 135 var dy7; 136 var dy8; 137 var dy9; 138 var sh; 139 var S0; 140 var S1; 141 var S2; 142 var S3; 143 var S4; 144 var S5; 145 var S6; 146 var S7; 147 var S8; 148 var S9; 149 var sx; 150 var sy; 151 var ix; 152 var iy; 153 var i0; 154 var i1; 155 var i2; 156 var i3; 157 var i4; 158 var i5; 159 var i6; 160 var i7; 161 var i8; 162 var i9; 163 164 // Note on variable naming convention: S#, dx#, dy#, i# where # corresponds to the loop number, with `0` being the innermost loop... 165 166 // Extract loop variables for purposes of loop interchange: dimensions and loop offset (pointer) increments... 167 sh = x.shape; 168 sx = x.strides; 169 sy = y.strides; 170 if ( x.order === 'row-major' ) { 171 // For row-major ndarrays, the last dimensions have the fastest changing indices... 172 S0 = sh[ 9 ]; 173 S1 = sh[ 8 ]; 174 S2 = sh[ 7 ]; 175 S3 = sh[ 6 ]; 176 S4 = sh[ 5 ]; 177 S5 = sh[ 4 ]; 178 S6 = sh[ 3 ]; 179 S7 = sh[ 2 ]; 180 S8 = sh[ 1 ]; 181 S9 = sh[ 0 ]; 182 dx0 = sx[ 9 ]; // offset increment for innermost loop 183 dx1 = sx[ 8 ] - ( S0*sx[9] ); 184 dx2 = sx[ 7 ] - ( S1*sx[8] ); 185 dx3 = sx[ 6 ] - ( S2*sx[7] ); 186 dx4 = sx[ 5 ] - ( S3*sx[6] ); 187 dx5 = sx[ 4 ] - ( S4*sx[5] ); 188 dx6 = sx[ 3 ] - ( S5*sx[4] ); 189 dx7 = sx[ 2 ] - ( S6*sx[3] ); 190 dx8 = sx[ 1 ] - ( S7*sx[2] ); 191 dx9 = sx[ 0 ] - ( S8*sx[1] ); // offset increment for outermost loop 192 dy0 = sy[ 9 ]; 193 dy1 = sy[ 8 ] - ( S0*sy[9] ); 194 dy2 = sy[ 7 ] - ( S1*sy[8] ); 195 dy3 = sy[ 6 ] - ( S2*sy[7] ); 196 dy4 = sy[ 5 ] - ( S3*sy[6] ); 197 dy5 = sy[ 4 ] - ( S4*sy[5] ); 198 dy6 = sy[ 3 ] - ( S5*sy[4] ); 199 dy7 = sy[ 2 ] - ( S6*sy[3] ); 200 dy8 = sy[ 1 ] - ( S7*sy[2] ); 201 dy9 = sy[ 0 ] - ( S8*sy[1] ); 202 } else { // order === 'column-major' 203 // For column-major ndarrays, the first dimensions have the fastest changing indices... 204 S0 = sh[ 0 ]; 205 S1 = sh[ 1 ]; 206 S2 = sh[ 2 ]; 207 S3 = sh[ 3 ]; 208 S4 = sh[ 4 ]; 209 S5 = sh[ 5 ]; 210 S6 = sh[ 6 ]; 211 S7 = sh[ 7 ]; 212 S8 = sh[ 8 ]; 213 S9 = sh[ 9 ]; 214 dx0 = sx[ 0 ]; // offset increment for innermost loop 215 dx1 = sx[ 1 ] - ( S0*sx[0] ); 216 dx2 = sx[ 2 ] - ( S1*sx[1] ); 217 dx3 = sx[ 3 ] - ( S2*sx[2] ); 218 dx4 = sx[ 4 ] - ( S3*sx[3] ); 219 dx5 = sx[ 5 ] - ( S4*sx[4] ); 220 dx6 = sx[ 6 ] - ( S5*sx[5] ); 221 dx7 = sx[ 7 ] - ( S6*sx[6] ); 222 dx8 = sx[ 8 ] - ( S7*sx[7] ); 223 dx9 = sx[ 9 ] - ( S8*sx[8] ); // offset increment for outermost loop 224 dy0 = sy[ 0 ]; 225 dy1 = sy[ 1 ] - ( S0*sy[0] ); 226 dy2 = sy[ 2 ] - ( S1*sy[1] ); 227 dy3 = sy[ 3 ] - ( S2*sy[2] ); 228 dy4 = sy[ 4 ] - ( S3*sy[3] ); 229 dy5 = sy[ 5 ] - ( S4*sy[4] ); 230 dy6 = sy[ 6 ] - ( S5*sy[5] ); 231 dy7 = sy[ 7 ] - ( S6*sy[6] ); 232 dy8 = sy[ 8 ] - ( S7*sy[7] ); 233 dy9 = sy[ 9 ] - ( S8*sy[8] ); 234 } 235 // Set the pointers to the first indexed elements in the respective ndarrays... 236 ix = x.offset; 237 iy = y.offset; 238 239 // Cache references to the input and output ndarray buffers... 240 xbuf = x.data; 241 ybuf = y.data; 242 243 // Cache accessors: 244 get = x.getter; 245 set = y.setter; 246 247 // Iterate over the ndarray dimensions... 248 for ( i9 = 0; i9 < S9; i9++ ) { 249 for ( i8 = 0; i8 < S8; i8++ ) { 250 for ( i7 = 0; i7 < S7; i7++ ) { 251 for ( i6 = 0; i6 < S6; i6++ ) { 252 for ( i5 = 0; i5 < S5; i5++ ) { 253 for ( i4 = 0; i4 < S4; i4++ ) { 254 for ( i3 = 0; i3 < S3; i3++ ) { 255 for ( i2 = 0; i2 < S2; i2++ ) { 256 for ( i1 = 0; i1 < S1; i1++ ) { 257 for ( i0 = 0; i0 < S0; i0++ ) { 258 set( ybuf, iy, fcn( get( xbuf, ix ) ) ); // eslint-disable-line max-len 259 ix += dx0; 260 iy += dy0; 261 } 262 ix += dx1; 263 iy += dy1; 264 } 265 ix += dx2; 266 iy += dy2; 267 } 268 ix += dx3; 269 iy += dy3; 270 } 271 ix += dx4; 272 iy += dy4; 273 } 274 ix += dx5; 275 iy += dy5; 276 } 277 ix += dx6; 278 iy += dy6; 279 } 280 ix += dx7; 281 iy += dy7; 282 } 283 ix += dx8; 284 iy += dy8; 285 } 286 ix += dx9; 287 iy += dy9; 288 } 289 } 290 291 292 // EXPORTS // 293 294 module.exports = unary10d;