org.broadinstitute.hellbender.tools.spark.utils.HopscotchMapSpark Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gatk Show documentation

Development on GATK 4

The newest version!

package org.broadinstitute.hellbender.tools.spark.utils;

import com.esotericsoftware.kryo.DefaultSerializer;
import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;

import java.util.Collection;
import java.util.Map;
import java.util.Objects;
import java.util.function.BiPredicate;
import java.util.function.Function;

/**
 * A uniquely keyed map with O(1) operations.
 * Sadly, it's not a java.util.Map, but it does behave like a java.util.Map's entrySet.
 */
@DefaultSerializer(HopscotchMapSpark.Serializer.class)
public final class HopscotchMapSpark>  extends HopscotchCollectionSpark {
    public HopscotchMapSpark() {}
    public HopscotchMapSpark( final int capacity ) { super(capacity); }
    public HopscotchMapSpark( final Collection collection ) { super(collection); }
    private HopscotchMapSpark( final Kryo kryo, final Input input ) { super(kryo, input); }


    /** in a map, uniqueness is on the key value, so we compare keys to see if we already have an entry for that key */
    @Override
    protected BiPredicate entryCollides() { return (t1, t2) -> Objects.equals(t1.getKey(), t2.getKey()); }

    /** getKey returns the key part of a Map.Entry */
    @Override
    protected Function toKey() { return T::getKey; }

    @SuppressWarnings("rawtypes")
    public static final class Serializer extends com.esotericsoftware.kryo.Serializer {
        @Override
        public void write( final Kryo kryo, final Output output, final HopscotchMapSpark hopscotchMapSpark ) {
            hopscotchMapSpark.serialize(kryo, output);
        }

        @Override
        public HopscotchMapSpark read( final Kryo kryo, final Input input, final Class klass ) {
            return new HopscotchMapSpark(kryo, input);
        }
    }
}