v2/src/marshal/org/apache/xmlbeans/impl/marshal/util/collections/ObjectAccumulator.java

/*   Copyright 2004 The Apache Software Foundation
 *
 *   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.apache.xmlbeans.impl.marshal.util.collections;


import java.lang.reflect.Array;
import java.util.Collection;


public abstract class ObjectAccumulator
    implements Accumulator
{
    private final Class componentType;
    private final boolean returnCollectionForArray;
    protected final Collection store;
    protected Object[] lastArray;


    public ObjectAccumulator(Class component_type,
                             int initial_capacity,
                             boolean return_collection)
    {
        componentType = component_type;
        returnCollectionForArray = return_collection;

        store = createNewStore(initial_capacity);
        assert store != null;
    }

    public ObjectAccumulator(Class component_type,
                             int initial_capacity)
    {
        this(component_type, initial_capacity, false);
    }

    protected abstract Collection createNewStore(int capacity);

    public ObjectAccumulator(Class component_type)
    {
        this(component_type, Accumulator.DEFAULT_INITIAL_CAPACITY);
    }

    public void append(Object o)
    {
        assert checkInstance(o);

        lastArray = null;
        store.add(o);
    }

    public final void appendDefault()
    {
        append(null);
    }

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

    private boolean checkInstance(Object o)
    {
        if (o == null) return true;
        if (componentType.isPrimitive()) return true; // can't quite deal with this
        if (!componentType.isInstance(o)) {
            String msg = "Invalid type: " + o.getClass().getName() +
                " expecting: " + componentType.getName();
            throw new IllegalArgumentException(msg);
        }
        return true;
    }

    public final Object getFinalArray()
    {
        if (returnCollectionForArray) {
            return store;
        } else {
            if (lastArray != null) {
                return lastArray;
            }

            Object[] out = (Object[])Array.newInstance(componentType, store.size());
            final Object[] retval = store.toArray(out);
            lastArray = retval;
            return retval;
        }
    }

    // use with caution
    public final Collection getStore()
    {
        return store;
    }

}