You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.2 KiB
64 lines
1.2 KiB
'use strict' |
|
|
|
/** |
|
* Expose `arrayFlatten`. |
|
*/ |
|
module.exports = arrayFlatten |
|
|
|
/** |
|
* Recursive flatten function with depth. |
|
* |
|
* @param {Array} array |
|
* @param {Array} result |
|
* @param {Number} depth |
|
* @return {Array} |
|
*/ |
|
function flattenWithDepth (array, result, depth) { |
|
for (var i = 0; i < array.length; i++) { |
|
var value = array[i] |
|
|
|
if (depth > 0 && Array.isArray(value)) { |
|
flattenWithDepth(value, result, depth - 1) |
|
} else { |
|
result.push(value) |
|
} |
|
} |
|
|
|
return result |
|
} |
|
|
|
/** |
|
* Recursive flatten function. Omitting depth is slightly faster. |
|
* |
|
* @param {Array} array |
|
* @param {Array} result |
|
* @return {Array} |
|
*/ |
|
function flattenForever (array, result) { |
|
for (var i = 0; i < array.length; i++) { |
|
var value = array[i] |
|
|
|
if (Array.isArray(value)) { |
|
flattenForever(value, result) |
|
} else { |
|
result.push(value) |
|
} |
|
} |
|
|
|
return result |
|
} |
|
|
|
/** |
|
* Flatten an array, with the ability to define a depth. |
|
* |
|
* @param {Array} array |
|
* @param {Number} depth |
|
* @return {Array} |
|
*/ |
|
function arrayFlatten (array, depth) { |
|
if (depth == null) { |
|
return flattenForever(array, []) |
|
} |
|
|
|
return flattenWithDepth(array, [], depth) |
|
}
|
|
|