com.landawn.abacus.util.IdentityHashSet Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2019 HaiYang Li
 *
 * 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 com.landawn.abacus.util;

import java.util.AbstractSet;
import java.util.Collection;
import java.util.IdentityHashMap;
import java.util.Iterator;

/**
 *
 * @param 
 */
public final class IdentityHashSet extends AbstractSet {

    private static final Object VAL = Boolean.TRUE;

    private final IdentityHashMap map;

    public IdentityHashSet() {
        map = new IdentityHashMap<>();
    }

    /**
     *
     * @param initialCapacity
     */
    public IdentityHashSet(final int initialCapacity) {
        map = N.newIdentityHashMap(initialCapacity);
    }

    /**
     *
     * @param c
     */
    public IdentityHashSet(final Collection c) {
        map = N.newIdentityHashMap(N.size(c));

        addAll(c); // NOSONAR
    }

    /**
     *
     * @param e
     * @return
     */
    @Override
    public boolean add(final T e) {
        return map.put(e, VAL) == null;
    }

    /**
     *
     * @param o
     * @return
     */
    @Override
    public boolean remove(final Object o) {
        return map.remove(o) != null;
    }

    /**
     *
     * @param c
     * @return
     */
    @Override
    public boolean containsAll(final Collection c) {
        if (N.isEmpty(c)) {
            return true;
        }

        return map.keySet().containsAll(c);
    }

    /**
     * Adds the all.
     *
     * @param c
     * @return
     */
    @Override
    public boolean addAll(final Collection c) {
        boolean modified = false;

        if (N.notEmpty(c)) {
            for (final T e : c) {
                if (add(e)) {
                    modified = true;
                }
            }
        }

        return modified;
    }

    /**
     * Removes the all.
     *
     * @param c
     * @return
     */
    @Override
    public boolean removeAll(final Collection c) {
        boolean modified = false;

        if (N.notEmpty(c)) {
            for (final Object e : c) {
                if (remove(e)) {
                    modified = true;
                }
            }
        }

        return modified;
    }

    /**
     *
     * @param c
     * @return
     */
    @Override
    public boolean retainAll(final Collection c) {
        if (N.isEmpty(c)) {
            if (!map.isEmpty()) {
                map.clear();
                return true;
            }
        } else {
            final IdentityHashSet kept = new IdentityHashSet<>(N.min(c.size(), size()));

            for (final Object e : c) {
                if (this.contains(e)) {
                    kept.add((T) e);
                }
            }

            if (kept.size() < this.size()) {
                clear();
                addAll(kept);
                return true;
            }
        }

        return false;
    }

    /**
     *
     * @param valueToFind
     * @return
     */
    @Override
    public boolean contains(final Object valueToFind) {
        //noinspection SuspiciousMethodCalls
        return map.containsKey(valueToFind);
    }

    @Override
    public Iterator iterator() {
        return map.keySet().iterator();
    }

    @Override
    public Object[] toArray() {
        return map.keySet().toArray();
    }

    /**
     *
     * @param 
     * @param a
     * @return
     */
    @Override
    public  A[] toArray(final A[] a) {
        return map.keySet().toArray(a);
    }

    @Override
    public int size() {
        return map.size();
    }

    /**
     * Checks if is empty.
     *
     * @return {@code true}, if is empty
     */
    @Override
    public boolean isEmpty() {
        return map.isEmpty();
    }

    /**
     * Clear.
     */
    @Override
    public void clear() {
        map.clear();
    }

    /**
     *
     * @param o
     * @return
     */
    @SuppressWarnings("rawtypes")
    @Override
    public boolean equals(final Object o) {
        if (o == this) {
            return true;
        }

        if (o instanceof IdentityHashSet) {
            return ((IdentityHashSet) o).map.equals(map);
        }

        return false;
    }

    @Override
    public int hashCode() {
        return map.keySet().hashCode();
    }

    @Override
    public String toString() {
        return map.keySet().toString();
    }
}