com.vladsch.flexmark.util.collection.SubClassingBag Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of driver-cql-shaded Show documentation
A Shaded CQL ActivityType driver for http://nosqlbench.io/
There is a newer version: 4.15.102
package com.vladsch.flexmark.util.collection;

import com.vladsch.flexmark.util.collection.iteration.ReversibleIterable;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.BitSet;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;

public class SubClassingBag {
    final private @NotNull ClassificationBag, T> items;
    final private @NotNull HashMap, BitSet> subClassMap;

    public SubClassingBag(@NotNull ClassificationBag, T> items, HashMap, @NotNull List>> subClassMap) {
        this.items = items;
        this.subClassMap = new HashMap<>();

        for (Class clazz : subClassMap.keySet()) {
            List> classList = subClassMap.get(clazz);
            BitSet bitSet = this.items.categoriesBitSet(classList);
            if (!bitSet.isEmpty()) {
                this.subClassMap.put(clazz, bitSet);
            }
        }
    }

    public @NotNull OrderedSet getItems() {
        return items.getItems();
    }

    public boolean contains(@Nullable T item) {
        return items.contains(item);
    }

    public boolean containsType(@Nullable Class type) {
        return items.containsCategory(type);
    }

    public BitSet getTypeSet(@Nullable Class category) {
        return subClassMap.get(category);
    }

    public int getTypeCount(@Nullable Class category) {
        BitSet bitSet = subClassMap.get(category);
        return bitSet == null ? 0 : bitSet.cardinality();
    }

    final public  @NotNull ReversibleIterable itemsOfType(@NotNull Class xClass, @NotNull Class... categories) {
        return items.getCategoryItems(xClass, typeBitSet(xClass, categories));
    }

    final public  @NotNull ReversibleIterable itemsOfType(@NotNull Class xClass, @NotNull Collection> categories) {
        return items.getCategoryItems(xClass, typeBitSet(xClass, categories));
    }

    final public  @NotNull ReversibleIterable reversedItemsOfType(@NotNull Class xClass, @NotNull Class... categories) {
        return items.getCategoryItemsReversed(xClass, typeBitSet(xClass, categories));
    }

    final public  @NotNull ReversibleIterable reversedItemsOfType(@NotNull Class xClass, @NotNull Collection> categories) {
        return items.getCategoryItemsReversed(xClass, typeBitSet(xClass, categories));
    }

    @SuppressWarnings("WeakerAccess")
    final public @NotNull BitSet typeBitSet(@NotNull Class xClass, @NotNull Class... categories) {
        BitSet bitSet = new BitSet();
        for (Class category : categories) {
            if (xClass.isAssignableFrom(category) && subClassMap.containsKey(category)) {
                bitSet.or(subClassMap.get(category));
            }
        }
        return bitSet;
    }

    @SuppressWarnings("WeakerAccess")
    final public @NotNull BitSet typeBitSet(@NotNull Class xClass, @NotNull Collection> categories) {
        BitSet bitSet = new BitSet();
        for (Class category : categories) {
            if (xClass.isAssignableFrom(category) && subClassMap.containsKey(category)) {
                bitSet.or(subClassMap.get(category));
            }
        }
        return bitSet;
    }
}