main.js (6134B)
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/improved-ziggurat' ).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 Improved Ziggurat algorithm. 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 if ( arguments.length > 0 ) { 74 if ( !isObject( options ) ) { 75 throw new TypeError( 'invalid argument. Options argument must be an object. Value: `'+options+'`.' ); 76 } 77 opts = copy( options, 1 ); 78 if ( hasOwnProp( opts, 'iter' ) ) { 79 if ( !isNonNegativeInteger( opts.iter ) ) { 80 throw new TypeError( 'invalid option. `iter` option must be a nonnegative integer. Option: `'+opts.iter+'`.' ); 81 } 82 } else { 83 opts.iter = MAX_VALUE; 84 } 85 rand = rnorm( opts ); 86 if ( opts.prng === void 0 && opts.copy !== false ) { 87 opts.state = rand.state; // cache a copy of the PRNG state 88 } 89 } else { 90 rand = rnorm(); 91 opts = { 92 'iter': MAX_VALUE, 93 'state': rand.state // cache a copy of the PRNG state 94 }; 95 } 96 i = 0; 97 98 // Create an iterator protocol-compliant object: 99 iter = {}; 100 setReadOnly( iter, 'next', next ); 101 setReadOnly( iter, 'return', end ); 102 103 if ( opts && opts.prng ) { 104 setReadOnly( iter, 'seed', null ); 105 setReadOnly( iter, 'seedLength', null ); 106 setReadWriteAccessor( iter, 'state', constantFunction( null ), noop ); 107 setReadOnly( iter, 'stateLength', null ); 108 setReadOnly( iter, 'byteLength', null ); 109 } else { 110 setReadOnlyAccessor( iter, 'seed', getSeed ); 111 setReadOnlyAccessor( iter, 'seedLength', getSeedLength ); 112 setReadWriteAccessor( iter, 'state', getState, setState ); 113 setReadOnlyAccessor( iter, 'stateLength', getStateLength ); 114 setReadOnlyAccessor( iter, 'byteLength', getStateSize ); 115 } 116 setReadOnly( iter, 'PRNG', rand.PRNG ); 117 118 // If an environment supports `Symbol.iterator`, make the iterator iterable: 119 if ( iteratorSymbol ) { 120 setReadOnly( iter, iteratorSymbol, factory ); 121 } 122 return iter; 123 124 /** 125 * Returns an iterator protocol-compliant object containing the next iterated value. 126 * 127 * @private 128 * @returns {Object} iterator protocol-compliant object 129 */ 130 function next() { 131 i += 1; 132 if ( FLG || i > opts.iter ) { 133 return { 134 'done': true 135 }; 136 } 137 return { 138 'value': rand(), 139 'done': false 140 }; 141 } 142 143 /** 144 * Finishes an iterator. 145 * 146 * @private 147 * @param {*} [value] - value to return 148 * @returns {Object} iterator protocol-compliant object 149 */ 150 function end( value ) { 151 FLG = true; 152 if ( arguments.length ) { 153 return { 154 'value': value, 155 'done': true 156 }; 157 } 158 return { 159 'done': true 160 }; 161 } 162 163 /** 164 * Returns a new iterator. 165 * 166 * @private 167 * @returns {Iterator} iterator 168 */ 169 function factory() { 170 return iterator( opts ); 171 } 172 173 /** 174 * Returns the PRNG seed. 175 * 176 * @private 177 * @returns {PRNGSeedMT19937} seed 178 */ 179 function getSeed() { 180 return rand.PRNG.seed; 181 } 182 183 /** 184 * Returns the PRNG seed length. 185 * 186 * @private 187 * @returns {PositiveInteger} seed length 188 */ 189 function getSeedLength() { 190 return rand.PRNG.seedLength; 191 } 192 193 /** 194 * Returns the PRNG state length. 195 * 196 * @private 197 * @returns {PositiveInteger} state length 198 */ 199 function getStateLength() { 200 return rand.PRNG.stateLength; 201 } 202 203 /** 204 * Returns the PRNG state size (in bytes). 205 * 206 * @private 207 * @returns {PositiveInteger} state size (in bytes) 208 */ 209 function getStateSize() { 210 return rand.PRNG.byteLength; 211 } 212 213 /** 214 * Returns the current pseudorandom number generator state. 215 * 216 * @private 217 * @returns {PRNGStateMT19937} current state 218 */ 219 function getState() { 220 return rand.PRNG.state; 221 } 222 223 /** 224 * Sets the pseudorandom number generator state. 225 * 226 * @private 227 * @param {PRNGStateMT19937} s - generator state 228 * @throws {Error} must provide a valid state 229 */ 230 function setState( s ) { 231 rand.PRNG.state = s; 232 } 233 } 234 235 236 // EXPORTS // 237 238 module.exports = iterator;