org.aksw.commons.collections.multimaps.BiHashMultimap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aksw-commons-collections Show documentation
There is a newer version: 0.9.9
Show newest version
package org.aksw.commons.collections.multimaps;


import java.util.Collection;
import java.util.Map.Entry;
import java.util.Set;

import com.google.common.collect.HashMultimap;
import com.google.common.collect.SetMultimap;

public class BiHashMultimap
	implements IBiSetMultimap
{
	private BiHashMultimap inverse;

	private SetMultimap forward;
	private SetMultimap backward;

	public BiHashMultimap()
	{
		this.forward = HashMultimap.create();
		this.backward = HashMultimap.create();

		inverse = new BiHashMultimap(this);
		inverse.inverse = this;
	}

	public BiHashMultimap(SetMultimap forward, SetMultimap backward)
	{
		this.forward = forward;
		this.backward = backward;

		inverse = new BiHashMultimap(this);
		inverse.inverse = this;
	}

	/**
	 * Constructor for reverse map
	 *
	 * @param original
	 */
	protected BiHashMultimap(BiHashMultimap original)
	{
		this.inverse = original;
		this.forward = original.backward;
		this.backward = original.forward;
	}

	public BiHashMultimap getInverse()
	{
		return inverse;
	}


	// TODO Must be wrapped
//	public Map> asMap()
//	{
//		return forward.asMap();
//	}

	public SetMultimap asMultimap() {
	    return forward;
	}

	public void remove(K key, V value)
	{
		forward.remove(key, value);
		backward.remove(value, key);
	}

	@Override
	public boolean put(K key, V value)
	{
		boolean result = forward.put(key, value);
		backward.put(value, key);

		return result;
	}


	@Override
	public Set removeAll(Object key)
	{
		for(V value : forward.get((K)key)) {
			backward.remove(value, key);
		}

		return forward.removeAll(key);
	}


	@SuppressWarnings("unchecked")
	@Override
	public Set get(Object key)
	{
		return forward.get((K) key);
	}

    public boolean containsEntry(Object key, Object value) {
        return forward.containsEntry(key, value);
    }

    public boolean containsValue(Object value) {
        return forward.containsValue(value);
    }

    public boolean containsKey(Object key) {
        return forward.containsKey(key);
    }

	@Override
	public Set> entries()
	{
		return forward.entries();
	}

	@Override
	public void putAll(K key, Collection values) {
		for(V value : values) {
			put(key, value);
		}
	}

	@Override
	public void putAll(ISetMultimap other)
	{
		for(Entry entry : other.entries()) {
			put(entry.getKey(), entry.getValue());
		}
	}

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

    public int size() {
        return forward.size();
    }

	@Override
	public void clear()
	{
		forward.clear();
		backward.clear();
	}

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        BiHashMultimap that = (BiHashMultimap) o;

        if (forward != null ? !forward.equals(that.forward) : that.forward != null) return false;

        return true;
    }

    @Override
    public int hashCode() {
        return forward != null ? forward.hashCode() : 0;
    }

	@Override
	public Set keySet() {
		return forward.keySet();
	}

	@Override
	public Collection keys() {
		return forward.keys();
	}

    @Override
    public boolean isEmpty() {
        return forward.isEmpty();
    }
}