time-to-botec

Benchmark sampling in different programming languages
Log | Files | Refs | README

factory.js (7719B)


      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 isObject = require( '@stdlib/assert/is-plain-object' );
     27 var isFunction = require( '@stdlib/assert/is-function' );
     28 var hasOwnProp = require( '@stdlib/assert/has-own-property' );
     29 var constantFunction = require( '@stdlib/utils/constant-function' );
     30 var noop = require( '@stdlib/utils/noop' );
     31 var randu = require( './../../../base/mt19937' ).factory;
     32 var isnan = require( '@stdlib/math/base/assert/is-nan' );
     33 var typedarray2json = require( '@stdlib/array/to-json' );
     34 var validate = require( './validate.js' );
     35 var frechet0 = require( './frechet.js' );
     36 
     37 
     38 // MAIN //
     39 
     40 /**
     41 * Returns a pseudorandom number generator for generating random numbers drawn from a Fréchet distribution.
     42 *
     43 * @param {PositiveNumber} [alpha] - shape parameter
     44 * @param {PositiveNumber} [s] - scale parameter
     45 * @param {number} [m] - location parameter
     46 * @param {Options} [options] - function options
     47 * @param {PRNG} [options.prng] - pseudorandom number generator which generates uniformly distributed pseudorandom numbers
     48 * @param {PRNGSeedMT19937} [options.seed] - pseudorandom number generator seed
     49 * @param {PRNGStateMT19937} [options.state] - pseudorandom number generator state
     50 * @param {boolean} [options.copy=true] - boolean indicating whether to copy a provided pseudorandom number generator state
     51 * @throws {TypeError} `alpha` must be a positive number
     52 * @throws {TypeError} `s` must be a positive number
     53 * @throws {TypeError} `m` must be a number
     54 * @throws {TypeError} options argument must be an object
     55 * @throws {TypeError} must provide valid options
     56 * @throws {Error} must provide a valid state
     57 * @returns {PRNG} pseudorandom number generator
     58 *
     59 * @example
     60 * var frechet = factory( 1.0, 1.0, 0.8 );
     61 * var v = frechet();
     62 * // returns <number>
     63 *
     64 * @example
     65 * var frechet = factory( 1.0, 1.0, -2.0, {
     66 *     'seed': 297
     67 * });
     68 * var v = frechet();
     69 * // returns <number>
     70 */
     71 function factory() {
     72 	var alpha;
     73 	var opts;
     74 	var rand;
     75 	var prng;
     76 	var err;
     77 	var s;
     78 	var m;
     79 
     80 	if ( arguments.length === 0 ) {
     81 		rand = randu();
     82 	} else if ( arguments.length === 1 ) {
     83 		opts = arguments[ 0 ];
     84 		if ( !isObject( opts ) ) {
     85 			throw new TypeError( 'invalid argument. Options argument must be an object. Value: `'+opts+'`.' );
     86 		}
     87 		if ( hasOwnProp( opts, 'prng' ) ) {
     88 			if ( !isFunction( opts.prng ) ) {
     89 				throw new TypeError( 'invalid option. `prng` option must be a pseudorandom number generator function. Option: `' + opts.prng + '`.' );
     90 			}
     91 			rand = opts.prng;
     92 		} else {
     93 			rand = randu( opts );
     94 		}
     95 	} else {
     96 		alpha = arguments[ 0 ];
     97 		s = arguments[ 1 ];
     98 		m = arguments[ 2 ];
     99 		err = validate( alpha, s, m );
    100 		if ( err ) {
    101 			throw err;
    102 		}
    103 		if ( arguments.length > 3 ) {
    104 			opts = arguments[ 3 ];
    105 			if ( !isObject( opts ) ) {
    106 				throw new TypeError( 'invalid argument. Options argument must be an object. Value: `' + opts + '`.' );
    107 			}
    108 			if ( hasOwnProp( opts, 'prng' ) ) {
    109 				if ( !isFunction( opts.prng ) ) {
    110 					throw new TypeError( 'invalid option. `prng` option must be a pseudorandom number generator function. Option: `' + opts.prng + '`.' );
    111 				}
    112 				rand = opts.prng;
    113 			} else {
    114 				rand = randu( opts );
    115 			}
    116 		} else {
    117 			rand = randu();
    118 		}
    119 	}
    120 	if ( alpha === void 0 ) {
    121 		prng = frechet2;
    122 	} else {
    123 		prng = frechet1;
    124 	}
    125 	setReadOnly( prng, 'NAME', 'frechet' );
    126 
    127 	// If we are provided an "external" PRNG, we don't support getting or setting PRNG state, as we'd need to check for compatible state value types, etc, entailing considerable complexity.
    128 	if ( opts && opts.prng ) {
    129 		setReadOnly( prng, 'seed', null );
    130 		setReadOnly( prng, 'seedLength', null );
    131 		setReadWriteAccessor( prng, 'state', constantFunction( null ), noop );
    132 		setReadOnly( prng, 'stateLength', null );
    133 		setReadOnly( prng, 'byteLength', null );
    134 		setReadOnly( prng, 'toJSON', constantFunction( null ) );
    135 		setReadOnly( prng, 'PRNG', rand );
    136 	} else {
    137 		setReadOnlyAccessor( prng, 'seed', getSeed );
    138 		setReadOnlyAccessor( prng, 'seedLength', getSeedLength );
    139 		setReadWriteAccessor( prng, 'state', getState, setState );
    140 		setReadOnlyAccessor( prng, 'stateLength', getStateLength );
    141 		setReadOnlyAccessor( prng, 'byteLength', getStateSize );
    142 		setReadOnly( prng, 'toJSON', toJSON );
    143 		setReadOnly( prng, 'PRNG', rand );
    144 		rand = rand.normalized;
    145 	}
    146 	return prng;
    147 
    148 	/**
    149 	* Returns the PRNG seed.
    150 	*
    151 	* @private
    152 	* @returns {PRNGSeedMT19937} seed
    153 	*/
    154 	function getSeed() {
    155 		return rand.seed;
    156 	}
    157 
    158 	/**
    159 	* Returns the PRNG seed length.
    160 	*
    161 	* @private
    162 	* @returns {PositiveInteger} seed length
    163 	*/
    164 	function getSeedLength() {
    165 		return rand.seedLength;
    166 	}
    167 
    168 	/**
    169 	* Returns the PRNG state length.
    170 	*
    171 	* @private
    172 	* @returns {PositiveInteger} state length
    173 	*/
    174 	function getStateLength() {
    175 		return rand.stateLength;
    176 	}
    177 
    178 	/**
    179 	* Returns the PRNG state size (in bytes).
    180 	*
    181 	* @private
    182 	* @returns {PositiveInteger} state size (in bytes)
    183 	*/
    184 	function getStateSize() {
    185 		return rand.byteLength;
    186 	}
    187 
    188 	/**
    189 	* Returns the current pseudorandom number generator state.
    190 	*
    191 	* @private
    192 	* @returns {PRNGStateMT19937} current state
    193 	*/
    194 	function getState() {
    195 		return rand.state;
    196 	}
    197 
    198 	/**
    199 	* Sets the pseudorandom number generator state.
    200 	*
    201 	* @private
    202 	* @param {PRNGStateMT19937} s - generator state
    203 	* @throws {Error} must provide a valid state
    204 	*/
    205 	function setState( s ) {
    206 		rand.state = s;
    207 	}
    208 
    209 	/**
    210 	* Serializes the pseudorandom number generator as a JSON object.
    211 	*
    212 	* ## Notes
    213 	*
    214 	* -   `JSON.stringify()` implicitly calls this method when stringifying a PRNG.
    215 	*
    216 	* @private
    217 	* @returns {Object} JSON representation
    218 	*/
    219 	function toJSON() {
    220 		var out = {};
    221 		out.type = 'PRNG';
    222 		out.name = prng.NAME;
    223 		out.state = typedarray2json( rand.state );
    224 		if ( alpha === void 0 ) {
    225 			out.params = [];
    226 		} else {
    227 			out.params = [ alpha, s, m ];
    228 		}
    229 		return out;
    230 	}
    231 
    232 	/**
    233 	* Returns a pseudorandom number drawn from a Fréchet distribution with bound parameters.
    234 	*
    235 	* @private
    236 	* @returns {number} pseudorandom number
    237 	*
    238 	* @example
    239 	* var v = frechet1();
    240 	* // returns <number>
    241 	*/
    242 	function frechet1() {
    243 		return frechet0( rand, alpha, s, m );
    244 	}
    245 
    246 	/**
    247 	* Returns a pseudorandom number drawn from a Fréchet distribution with shape `alpha`, scale `s`, and location `m`.
    248 	*
    249 	* @private
    250 	* @param {PositiveNumber} alpha - shape parameter
    251 	* @param {PositiveNumber} s - scale parameter
    252 	* @param {number} m - location parameter
    253 	* @returns {number} pseudorandom number
    254 	*
    255 	* @example
    256 	* var v = frechet2( 2.0, 1.0, 0.5 );
    257 	* // returns <number>
    258 	*
    259 	* @example
    260 	* var v = frechet2( 1.0, 1.0, 0.5 );
    261 	* // returns NaN
    262 	*
    263 	* @example
    264 	* var v = frechet2( 1.0, 2.0, NaN );
    265 	* // returns NaN
    266 	*/
    267 	function frechet2( alpha, s, m ) {
    268 		if (
    269 			isnan( alpha ) ||
    270 			isnan( s ) ||
    271 			isnan( m ) ||
    272 			alpha <= 0.0 ||
    273 			s <= 0.0
    274 		) {
    275 			return NaN;
    276 		}
    277 		return frechet0( rand, alpha, s, m );
    278 	}
    279 }
    280 
    281 
    282 // EXPORTS //
    283 
    284 module.exports = factory;