UNPKG

@stdlib/math

Version:
156 lines (140 loc) 3.18 kB
/** * @license Apache-2.0 * * Copyright (c) 2020 The Stdlib Authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ 'use strict'; // MODULES // var setReadOnly = require( '@stdlib/utils/define-nonenumerable-read-only-property' ); var iteratorSymbol = require( '@stdlib/symbol/iterator' ); var FLOAT64_MAX_SAFE_INTEGER = require( '@stdlib/constants/float64/max-safe-integer' ); var validate = require( './validate.js' ); // MAIN // /** * Returns an iterator which generates an interleaved sequence of even integers. * * ## Notes * * - If an environment supports `Symbol.iterator`, the returned iterator is iterable. * * @param {Options} [options] - function options * @param {NonNegativeInteger} [options.iter=9007199254740992] - number of iterations * @throws {TypeError} options argument must be an object * @throws {TypeError} must provide valid options * @returns {Iterator} iterator * * @example * var iter = iterEvenIntegersSeq(); * * var v = iter.next().value; * // returns 0 * * v = iter.next().value; * // returns 2 * * v = iter.next().value; * // returns -2 * * // ... */ function iterEvenIntegersSeq( options ) { var opts; var iter; var FLG; var err; var sgn; var i; var j; opts = { 'iter': FLOAT64_MAX_SAFE_INTEGER }; if ( arguments.length ) { err = validate( opts, options ); if ( err ) { throw err; } } sgn = -1; i = 0; j = 0; // Create an iterator protocol-compliant object: iter = {}; setReadOnly( iter, 'next', next ); setReadOnly( iter, 'return', end ); // If an environment supports `Symbol.iterator`, make the iterator iterable: if ( iteratorSymbol ) { setReadOnly( iter, iteratorSymbol, factory ); } return iter; /** * Returns an iterator protocol-compliant object containing the next iterated value. * * @private * @returns {Object} iterator protocol-compliant object */ function next() { i += 1; if ( FLG || i > opts.iter ) { return { 'done': true }; } if ( i === 1 ) { return { 'value': 0, 'done': false }; } // Increment every other iteration... if ( sgn < 0 ) { j += 2; } sgn *= -1; return { 'value': sgn * j, 'done': false }; } /** * Finishes an iterator. * * @private * @param {*} [value] - value to return * @returns {Object} iterator protocol-compliant object */ function end( value ) { FLG = true; if ( arguments.length ) { return { 'value': value, 'done': true }; } return { 'done': true }; } /** * Returns a new iterator. * * @private * @returns {Iterator} iterator */ function factory() { return iterEvenIntegersSeq( opts ); } } // EXPORTS // module.exports = iterEvenIntegersSeq;