Yang Guo | 4fd355c | 2019-09-19 08:59:03 | [diff] [blame] | 1 | var arrayPush = require('./_arrayPush'), |
| 2 | isFlattenable = require('./_isFlattenable'); |
| 3 | |
| 4 | /** |
| 5 | * The base implementation of `_.flatten` with support for restricting flattening. |
| 6 | * |
| 7 | * @private |
| 8 | * @param {Array} array The array to flatten. |
| 9 | * @param {number} depth The maximum recursion depth. |
| 10 | * @param {boolean} [predicate=isFlattenable] The function invoked per iteration. |
| 11 | * @param {boolean} [isStrict] Restrict to values that pass `predicate` checks. |
| 12 | * @param {Array} [result=[]] The initial result value. |
| 13 | * @returns {Array} Returns the new flattened array. |
| 14 | */ |
| 15 | function baseFlatten(array, depth, predicate, isStrict, result) { |
| 16 | var index = -1, |
| 17 | length = array.length; |
| 18 | |
| 19 | predicate || (predicate = isFlattenable); |
| 20 | result || (result = []); |
| 21 | |
| 22 | while (++index < length) { |
| 23 | var value = array[index]; |
| 24 | if (depth > 0 && predicate(value)) { |
| 25 | if (depth > 1) { |
| 26 | // Recursively flatten arrays (susceptible to call stack limits). |
| 27 | baseFlatten(value, depth - 1, predicate, isStrict, result); |
| 28 | } else { |
| 29 | arrayPush(result, value); |
| 30 | } |
| 31 | } else if (!isStrict) { |
| 32 | result[result.length] = value; |
| 33 | } |
| 34 | } |
| 35 | return result; |
| 36 | } |
| 37 | |
| 38 | module.exports = baseFlatten; |