poi/poi/src/main/java/org/apache/poi/util/IntMapper.java

/* ====================================================================
   Licensed to the Apache Software Foundation (ASF) under one or more
   contributor license agreements.  See the NOTICE file distributed with
   this work for additional information regarding copyright ownership.
   The ASF licenses this file to You 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.poi.util;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.apache.poi.common.Duplicatable;

/**
 * A List of objects that are indexed AND keyed by an int; also allows for getting
 * the index of a value in the list
 * <p>
 * I am happy is someone wants to re-implement this without using the
 * internal list and hashmap. If so could you please make sure that
 * you can add elements half way into the list and have the value-key mappings
 * update
 */

public class IntMapper<T> implements Duplicatable {
    private final List<T> elements;
    private final Map<T, Integer> valueKeyMap;

    private static final int _default_size = 10;

    /**
     * create an IntMapper of default size
     */
    public IntMapper() {
        this(_default_size);
    }

    public IntMapper(final int initialCapacity) {
        elements = new ArrayList<>(initialCapacity);
        valueKeyMap = new HashMap<>(initialCapacity);
    }

    public IntMapper(IntMapper<T> other) {
        elements = new ArrayList<>(other.elements);
        valueKeyMap = new HashMap<>(other.valueKeyMap);
    }

    /**
     * Appends the specified element to the end of this list
     *
     * @param value element to be appended to this list.
     * @return true (as per the general contract of the Collection.add
     * method).
     */
    public boolean add(final T value) {
        int index = elements.size();
        elements.add(value);
        valueKeyMap.put(value, index);
        return true;
    }

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

    public T get(int index) {
        return elements.get(index);
    }

    public int getIndex(T o) {
        return valueKeyMap.getOrDefault(o, -1);
    }

    public Iterator<T> iterator() {
        return elements.iterator();
    }

    @Override
    public IntMapper<T> copy() {
        return new IntMapper<>(this);
    }

    public List<T> getElements() {
        return elements;
    }
}