main.js (6121B)
1 /** 2 * @license Apache-2.0 3 * 4 * Copyright (c) 2018 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 'use strict'; 20 21 // MODULES // 22 23 var setReadOnly = require( '@stdlib/utils/define-nonenumerable-read-only-property' ); 24 var setReadOnlyAccessor = require( '@stdlib/utils/define-nonenumerable-read-only-accessor' ); 25 var setReadWriteAccessor = require( '@stdlib/utils/define-nonenumerable-read-write-accessor' ); 26 var constantFunction = require( '@stdlib/utils/constant-function' ); 27 var noop = require( '@stdlib/utils/noop' ); 28 var copy = require( '@stdlib/utils/copy' ); 29 var isObject = require( '@stdlib/assert/is-plain-object' ); 30 var isNonNegativeInteger = require( '@stdlib/assert/is-nonnegative-integer' ).isPrimitive; 31 var hasOwnProp = require( '@stdlib/assert/has-own-property' ); 32 var MAX_VALUE = require( '@stdlib/constants/float64/max' ); 33 var rnorm = require( './../../../base/box-muller' ).factory; 34 var iteratorSymbol = require( '@stdlib/symbol/iterator' ); 35 36 37 // MAIN // 38 39 /** 40 * Returns an iterator for generating pseudorandom numbers drawn from a standard normal distribution using the Box-Muller transform. 41 * 42 * @param {Options} [options] - function options 43 * @param {PRNG} [options.prng] - pseudorandom number generator which generates uniformly distributed pseudorandom numbers 44 * @param {PRNGSeedMT19937} [options.seed] - pseudorandom number generator seed 45 * @param {PRNGStateMT19937} [options.state] - pseudorandom number generator state 46 * @param {boolean} [options.copy=true] - boolean indicating whether to copy a provided pseudorandom number generator state 47 * @param {NonNegativeInteger} [options.iter] - number of iterations 48 * @throws {TypeError} options argument must be an object 49 * @throws {TypeError} must provide valid options 50 * @throws {Error} must provide a valid state 51 * @returns {Iterator} iterator 52 * 53 * @example 54 * var iter = iterator(); 55 * 56 * var r = iter.next().value; 57 * // returns <number> 58 * 59 * r = iter.next().value; 60 * // returns <number> 61 * 62 * r = iter.next().value; 63 * // returns <number> 64 * 65 * // ... 66 */ 67 function iterator( options ) { 68 var opts; 69 var iter; 70 var rand; 71 var FLG; 72 var i; 73 74 if ( arguments.length > 0 ) { 75 if ( !isObject( options ) ) { 76 throw new TypeError( 'invalid argument. Options argument must be an object. Value: `'+options+'`.' ); 77 } 78 opts = copy( options, 1 ); 79 if ( hasOwnProp( opts, 'iter' ) ) { 80 if ( !isNonNegativeInteger( opts.iter ) ) { 81 throw new TypeError( 'invalid option. `iter` option must be a nonnegative integer. Option: `'+opts.iter+'`.' ); 82 } 83 } else { 84 opts.iter = MAX_VALUE; 85 } 86 rand = rnorm( opts ); 87 if ( opts.prng === void 0 && opts.copy !== false ) { 88 opts.state = rand.state; // cache a copy of the PRNG state 89 } 90 } else { 91 rand = rnorm(); 92 opts = { 93 'iter': MAX_VALUE, 94 'state': rand.state // cache a copy of the PRNG state 95 }; 96 } 97 i = 0; 98 99 // Create an iterator protocol-compliant object: 100 iter = {}; 101 setReadOnly( iter, 'next', next ); 102 setReadOnly( iter, 'return', end ); 103 104 if ( opts && opts.prng ) { 105 setReadOnly( iter, 'seed', null ); 106 setReadOnly( iter, 'seedLength', null ); 107 setReadWriteAccessor( iter, 'state', constantFunction( null ), noop ); 108 setReadOnly( iter, 'stateLength', null ); 109 setReadOnly( iter, 'byteLength', null ); 110 } else { 111 setReadOnlyAccessor( iter, 'seed', getSeed ); 112 setReadOnlyAccessor( iter, 'seedLength', getSeedLength ); 113 setReadWriteAccessor( iter, 'state', getState, setState ); 114 setReadOnlyAccessor( iter, 'stateLength', getStateLength ); 115 setReadOnlyAccessor( iter, 'byteLength', getStateSize ); 116 } 117 setReadOnly( iter, 'PRNG', rand.PRNG ); 118 119 // If an environment supports `Symbol.iterator`, make the iterator iterable: 120 if ( iteratorSymbol ) { 121 setReadOnly( iter, iteratorSymbol, factory ); 122 } 123 return iter; 124 125 /** 126 * Returns an iterator protocol-compliant object containing the next iterated value. 127 * 128 * @private 129 * @returns {Object} iterator protocol-compliant object 130 */ 131 function next() { 132 i += 1; 133 if ( FLG || i > opts.iter ) { 134 return { 135 'done': true 136 }; 137 } 138 return { 139 'value': rand(), 140 'done': false 141 }; 142 } 143 144 /** 145 * Finishes an iterator. 146 * 147 * @private 148 * @param {*} [value] - value to return 149 * @returns {Object} iterator protocol-compliant object 150 */ 151 function end( value ) { 152 FLG = true; 153 if ( arguments.length ) { 154 return { 155 'value': value, 156 'done': true 157 }; 158 } 159 return { 160 'done': true 161 }; 162 } 163 164 /** 165 * Returns a new iterator. 166 * 167 * @private 168 * @returns {Iterator} iterator 169 */ 170 function factory() { 171 return iterator( opts ); 172 } 173 174 /** 175 * Returns the PRNG seed. 176 * 177 * @private 178 * @returns {PRNGSeedMT19937} seed 179 */ 180 function getSeed() { 181 return rand.PRNG.seed; 182 } 183 184 /** 185 * Returns the PRNG seed length. 186 * 187 * @private 188 * @returns {PositiveInteger} seed length 189 */ 190 function getSeedLength() { 191 return rand.PRNG.seedLength; 192 } 193 194 /** 195 * Returns the PRNG state length. 196 * 197 * @private 198 * @returns {PositiveInteger} state length 199 */ 200 function getStateLength() { 201 return rand.PRNG.stateLength; 202 } 203 204 /** 205 * Returns the PRNG state size (in bytes). 206 * 207 * @private 208 * @returns {PositiveInteger} state size (in bytes) 209 */ 210 function getStateSize() { 211 return rand.PRNG.byteLength; 212 } 213 214 /** 215 * Returns the current pseudorandom number generator state. 216 * 217 * @private 218 * @returns {PRNGStateMT19937} current state 219 */ 220 function getState() { 221 return rand.PRNG.state; 222 } 223 224 /** 225 * Sets the pseudorandom number generator state. 226 * 227 * @private 228 * @param {PRNGStateMT19937} s - generator state 229 * @throws {Error} must provide a valid state 230 */ 231 function setState( s ) { 232 rand.PRNG.state = s; 233 } 234 } 235 236 237 // EXPORTS // 238 239 module.exports = iterator;