1 |
3a515b92
|
cagy
|
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;
|