org.javafunk.funk.builders.MultisetBuilder.java Source code

Introduction

Here is the source code for org.javafunk.funk.builders.MultisetBuilder.java

Source

/*
 * Copyright (C) 2011 Funk committers.
 * All rights reserved.
 *
 * The software in this package is published under the terms of the BSD
 * style license a copy of which has been included with this distribution in
 * the LICENSE.txt file.
 */
package org.javafunk.funk.builders;

import com.google.common.collect.HashMultiset;
import com.google.common.collect.Multiset;

public class MultisetBuilder<E> extends AbstractBuilder<E, MultisetBuilder<E>, Multiset<E>>
        implements AbstractBuilder.WithCustomImplementationSupport<Multiset, Multiset<E>> {
    private HashMultiset<E> elements = HashMultiset.create();

    public static <E> MultisetBuilder<E> multisetBuilder() {
        return new MultisetBuilder<E>();
    }

    public static <E> MultisetBuilder<E> multisetBuilder(Class<E> elementClass) {
        return new MultisetBuilder<E>();
    }

    @Override
    public Multiset<E> build() {
        return HashMultiset.create(elements);
    }

    @Override
    public Multiset<E> build(Class<? extends Multiset> implementationClass)
            throws IllegalAccessException, InstantiationException {
        @SuppressWarnings("unchecked")
        Multiset<E> multiset = (Multiset<E>) implementationClass.newInstance();
        multiset.addAll(elements);
        return multiset;
    }

    @Override
    protected void handle(E element) {
        elements.add(element);
    }

    @Override
    protected MultisetBuilder<E> updatedBuilder() {
        return this;
    }
}