TmMolStar/src/mol-util/set.ts

/**
 * Copyright (c) 2018 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 */

// TODO remove Array.from workaround when targeting ES6

/** Test if set a contains all elements of set b. */
export function isSuperset<T>(setA: Set<T>, setB: Set<T>) {
    for (const elm of Array.from(setB)) {
        if (!setA.has(elm)) return false;
    }
    return true;
}

/** Create set containing elements of both set a and set b. */
export function union<T>(setA: Set<T>, setB: Set<T>): Set<T> {
    const union = new Set(setA);
    for (const elem of Array.from(setB)) union.add(elem);
    return union;
}

export function unionMany<T>(sets: Set<T>[]) {
    if (sets.length === 0) return new Set<T>();
    if (sets.length === 1) return sets[0];
    const union = new Set(sets[0]);
    for (let i = 1; i < sets.length; i++) {
        for (const elem of Array.from(sets[i])) union.add(elem);
    }
    return union;
}

export function unionManyArrays<T>(arrays: T[][]) {
    if (arrays.length === 0) return new Set<T>();
    const union = new Set(arrays[0]);
    for (let i = 1; i < arrays.length; i++) {
        for (const elem of arrays[i]) union.add(elem);
    }
    return union;
}

/** Create set containing elements of set a that are also in set b. */
export function intersection<T>(setA: Set<T>, setB: Set<T>): Set<T> {
    const intersection = new Set();
    for (const elem of Array.from(setB)) {
        if (setA.has(elem)) intersection.add(elem);
    }
    return intersection;
}

/** Create set containing elements of set a that are not in set b. */
export function difference<T>(setA: Set<T>, setB: Set<T>): Set<T> {
    const difference = new Set(setA);
    for (const elem of Array.from(setB)) difference.delete(elem);
    return difference;
}