org.javersion.util.AbstractTrieSet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javersion-util Show documentation
The newest version!
/*
 * Copyright 2013 Samppa Saarela
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.javersion.util;

import static com.google.common.base.Objects.equal;
import static com.google.common.collect.Iterators.transform;

import java.util.Collection;
import java.util.Iterator;

import org.javersion.util.AbstractTrieSet.EntryNode;

import com.google.common.base.Function;
import com.google.common.collect.Iterators;


public abstract class AbstractTrieSet> extends AbstractHashTrie, This> implements Iterable {

    @SuppressWarnings("rawtypes")
    private static final Function ELEMENT_TO_ENTRY = new Function() {
        @SuppressWarnings("unchecked")
        @Override
        public EntryNode apply(Object input) {
            return new EntryNode(input);
        }
    };

    @SuppressWarnings("rawtypes")
    private static final Function ENTRY_TO_ELEMENT = input -> input != null ? ((EntryNode) input).element() : null;

    public This conj(E element) {
        final UpdateContext> updateContext = updateContext(1, null);
        try {
            return doAdd(updateContext, new EntryNode(element));
        } finally {
            commit(updateContext);
        }
    }

    public This conjAll(final Collection elements) {
        return conjAll(elements, elements.size());
    }

    public This conjAll(AbstractTrieSet elements) {
        return conjAll(elements, elements.size());
    }

    public This conjAll(final Iterable elements) {
        return conjAll(elements, 32);
    }

    @SuppressWarnings("unchecked")
    private This conjAll(final Iterable elements, int size) {
        final UpdateContext> updateContext = updateContext(size, null);
        try {
            return (This) doAddAll(updateContext, transform(elements.iterator(), ELEMENT_TO_ENTRY));
        } finally {
            commit(updateContext);
        }
    }

    public This disj(Object element) {
        final UpdateContext> updateContext = updateContext(1, null);
        try {
            return doRemove(updateContext, element);
        } finally {
            commit(updateContext);
        }
    }

    public This disjAll(final Iterable elements) {
        final UpdateContext> updateContext = updateContext(1, null);
        try {
            return doRemoveAll(updateContext, elements.iterator());
        } finally {
            commit(updateContext);
        }
    }

    @SuppressWarnings("unchecked")
    public Iterator iterator() {
        return Iterators.transform(doIterator(), ENTRY_TO_ELEMENT);
    }

    protected UpdateContext> updateContext(int expectedSize, Merger> merger) {
        return new UpdateContext<>(expectedSize, merger);
    }

    public boolean contains(Object o) {
        return containsKey(o);
    }

    public static final class EntryNode extends AbstractHashTrie.EntryNode> {

        public EntryNode(E element) {
            super(element);
        }

        public E element() {
            return key;
        }

        @Override
        public Node> assocInternal(final UpdateContext>  currentContext, final int shift, final int hash, final EntryNode newEntry) {
            if (equal(this.key, newEntry.key)) {
                currentContext.merge(this, newEntry);
                return this;
            } else {
                return split(currentContext, shift, hash, newEntry);
            }
        }
    }

    static class ElementSpliterator extends NodeSpliterator> {

        public ElementSpliterator(Node> node, int sizeEstimate, boolean immutable) {
            super(node, sizeEstimate, DISTINCT | (immutable ? IMMUTABLE : 0));
        }

        public ElementSpliterator(Node>[] array, int pos, int limit, int sizeEstimate, boolean immutable) {
            super(array, pos, limit, sizeEstimate, DISTINCT | (immutable ? IMMUTABLE : 0));
        }

        @Override
        protected NodeSpliterator> newSubSpliterator(Node>[] array, int pos, int limit, int sizeEstimate) {
            return new ElementSpliterator<>(array, pos, limit, sizeEstimate, hasCharacteristics(IMMUTABLE));
        }

        @Override
        protected E apply(EntryNode entry) {
            return entry.key;
        }
    }
}