/*
 * Copyright (C) 2008 Herve Quiroz
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 * 
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 
 *
 * $Id$
 */
package org.trancecode.collection;

import com.google.common.base.Function;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Maps;

import java.util.Map;
import java.util.Map.Entry;

import org.trancecode.lang.TcObjects;

/**
 * Utility methods related to {@link Map}.
 * 
 * @author Herve Quiroz
 */
public final class TcMaps
{
    private TcMaps()
    {
        // No instantiation
    }

    public static <K, V> Map<K, V> newSmallWriteOnceMap()
    {
        // TODO
        return Maps.newLinkedHashMap();
    }

    public static <K, V> Map<K, V> merge(final Map<K, V> map1, final Map<? extends K, ? extends V> map2)
    {
        final Map<K, V> map = Maps.newHashMapWithExpectedSize(map1.size() + map2.size());
        map.putAll(map1);
        map.putAll(map2);

        return ImmutableMap.copyOf(map);
    }

    public static <K, V> Map<K, V> copyAndPut(final Map<K, V> map, final K key, final V value)
    {
        Preconditions.checkNotNull(map);
        Preconditions.checkNotNull(key);
        if (map instanceof ImmutableMap && TcObjects.equals(map.get(key), value))
        {
            return map;
        }

        final Map<K, V> newMap = Maps.newLinkedHashMap(map);
        newMap.put(key, value);

        return ImmutableMap.copyOf(newMap);
    }

    public static <K, V> V get(final Map<K, V> map, final K key, final V defaultValue)
    {
        if (map.containsKey(key))
        {
            return map.get(key);
        }

        return defaultValue;
    }

    public static <K, V> Map<K, V> fromEntries(final Iterable<Entry<K, V>> entries)
    {
        final Function<Entry<K, V>, K> keyFunction = MapFunctions.getKey();
        final Function<Entry<K, V>, V> valueFunction = MapFunctions.getValue();
        final Map<K, Entry<K, V>> intermediate = Maps.uniqueIndex(entries, keyFunction);
        return Maps.transformValues(intermediate, valueFunction);
    }
}

@@ -949,24 +949,79 @@
 mutableMap;%0A
+import com.google.common.collect.ImmutableMap.Builder;%0A
 import com.g
@@ -1047,16 +1047,16 @@
 t.Maps;%0A
-
 %0Aimport 
@@ -1563,175 +1563,111 @@
 nal 
-Map%3CK, V%3E map = Maps.newHashMapWithExpectedSize(map1.size() + map2.size());%0A        map.putAll(map1);%0A        map.putAll(map2);%0A%0A        return ImmutableMap.copyOf(map
+Builder%3CK, V%3E builder = ImmutableMap.builder();%0A        return builder.putAll(map1).putAll(map2).build(
 );%0A 
@@ -2000,69 +2000,90 @@
 nal 
-Map%3CK, V%3E newMap = Maps.newLinkedHashMap(map);%0A        newM
+Builder%3CK, V%3E builder = ImmutableMap.builder();%0A        return builder.putAll(m
 ap
+)
 .put
@@ -2098,52 +2098,15 @@
 lue)
-;%0A%0A        return ImmutableMap.copyOf(newMap
+.build(
 );%0A 

Improve TcMaps utility methods