'use strict'; var flatten = require('../../utils/array').flatten; function factory (type, config, load, typed) { var index = load(require('../../type/matrix/MatrixIndex')); var concat = load(require('../matrix/concat')); var size = load(require('../matrix/size')); var subset = load(require('../matrix/subset')); var setIntersect = load(require('../set/setIntersect')); var setSymDifference = load(require('../set/setSymDifference')); /** * Create the union of two (multi)sets. * Multi-dimension arrays will be converted to single-dimension arrays before the operation. * * Syntax: * * math.setUnion(set1, set2) * * Examples: * * math.setUnion([1, 2, 3, 4], [3, 4, 5, 6]); // returns [1, 2, 3, 4, 5, 6] * math.setUnion([[1, 2], [3, 4]], [[3, 4], [5, 6]]); // returns [1, 2, 3, 4, 5, 6] * * See also: * * setIntersect, setDifference * * @param {Array | Matrix} a1 A (multi)set * @param {Array | Matrix} a2 A (multi)set * @return {Array | Matrix} The union of two (multi)sets */ var setUnion = typed('setUnion', { 'Array | Matrix, Array | Matrix': function (a1, a2) { if (subset(size(a1), new index(0)) === 0) { // if any of them is empty, return the other one return flatten(a2); } else if (subset(size(a2), new index(0)) === 0) { return flatten(a1); } var b1 = flatten(a1); var b2 = flatten(a2); return concat(setSymDifference(b1, b2), setIntersect(b1, b2)); } }); return setUnion; } exports.name = 'setUnion'; exports.factory = factory;