---

TypedSplayTreeMap.java

package ca.mcgill.sable.util;

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 * SableUtil, a clean room implementation of the Collection API.     *
 * Copyright (C) 1997, 1998 Etienne Gagnon (gagnon@sable.mcgill.ca). *
 * All rights reserved.                                              *
 *                                                                   *
 * This work was done as a project of the Sable Research Group,      *
 * School of Computer Science, McGill University, Canada             *
 * (http://www.sable.mcgill.ca/).  It is understood that any         *
 * modification not identified as such is not covered by the         *
 * preceding statement.                                              *
 *                                                                   *
 * This work is free software; you can redistribute it and/or        *
 * modify it under the terms of the GNU Library General Public       *
 * License as published by the Free Software Foundation; either      *
 * version 2 of the License, or (at your option) any later version.  *
 *                                                                   *
 * This work 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 *
 * Library General Public License for more details.                  *
 *                                                                   *
 * You should have received a copy of the GNU Library 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.                                      *
 *                                                                   *
 * To submit a bug report, send a comment, or get the latest news on *
 * this project and other Sable Research Group projects, please      *
 * visit the web site: http://www.sable.mcgill.ca/                   *
 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

/*
 Reference Version
 -----------------
 This is the latest official version on which this file is based.
 The reference version is: $SableUtilVersion: 1.11 $

 Change History
 --------------
 A) Notes:

 Please use the following template.  Most recent changes should
 appear at the top of the list.

 - Modified on [date (March 1, 1900)] by [name]. [(*) if appropriate]
   [description of modification].

 Any Modification flagged with "(*)" was done as a project of the
 Sable Research Group, School of Computer Science,
 McGill University, Canada (http://www.sable.mcgill.ca/).

 You should add your copyright, using the following template, at
 the top of this file, along with other copyrights.

 *                                                                   *
 * Modifications by [name] are                                       *
 * Copyright (C) [year(s)] [your name (or company)].  All rights     *
 * reserved.                                                         *
 *                                                                   *

 B) Changes:

 - Modified on June 7, 1998 by Etienne Gagnon (gagnon@sable.mcgill.ca). (*)
   Changed the license.
*/

import ca.mcgill.sable.util.Map.Entry;

public class TypedSplayTreeMap extends SplayTreeMap
{
    private Cast keyCast;
    private Cast valueCast;
    private Collection entries;

    private class EntryCollection extends AbstractCollection
    {
        private Collection collection;

        EntryCollection(Collection collection)
        {
            this.collection = collection;
        }

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

        public Iterator iterator()
        {
            return new EntryIterator(collection.iterator());
        }
    }

    private class EntryIterator implements Iterator
    {
        private Iterator iterator;

        EntryIterator(Iterator iterator)
        {
            this.iterator = iterator;
        }

        public boolean hasNext()
        {
            return iterator.hasNext();
        }

        public Object next()
        {
            return new TypedEntry((Entry) iterator.next());
        }

        public void remove()
        {
            iterator.remove();
        }
    }

    private class TypedEntry extends AbstractEntry
    {
        private Entry entry;

        TypedEntry(Entry entry)
        {
            this.entry = entry;
        }

        public Object getKey()
        {
            return entry.getKey();
        }

        public Object getValue()
        {
            return entry.getValue();
        }

        public Object setValue(Object value)
        {
            return entry.setValue(valueCast.cast(value));
        }
    }
    public TypedSplayTreeMap()
    {
        super();

        keyCast = NoCast.instance;
        valueCast = NoCast.instance;
    }
    public TypedSplayTreeMap(Cast keyCast, Cast valueCast)
    {
        super();

        this.keyCast = keyCast;
        this.valueCast = valueCast;
    }
    public TypedSplayTreeMap(Comparator comparator)
    {
        super(comparator);

        keyCast = NoCast.instance;
        valueCast = NoCast.instance;
    }
    public TypedSplayTreeMap(Comparator comparator, Cast keyCast, Cast valueCast)
    {
        super(comparator);

        this.keyCast = keyCast;
        this.valueCast = valueCast;
    }
    public TypedSplayTreeMap(Map map)
    {
        super(map);

        keyCast = NoCast.instance;
        valueCast = NoCast.instance;
    }
    public TypedSplayTreeMap(Map map, Cast keyCast, Cast valueCast)
    {
        super(map);

        this.keyCast = keyCast;
        this.valueCast = valueCast;
    }
    public TypedSplayTreeMap(Map map, Comparator comparator)
    {
        super(map, comparator);

        keyCast = NoCast.instance;
        valueCast = NoCast.instance;
    }
    public TypedSplayTreeMap(Map map, Comparator comparator, Cast keyCast, Cast valueCast)
    {
        super(map, comparator);

        this.keyCast = keyCast;
        this.valueCast = valueCast;
    }
    public Object clone()
    {
        return new TypedSplayTreeMap(this, getComparator(), keyCast, valueCast);
    }
    public Collection entries()
    {
        if(entries == null)
        {
            entries = new EntryCollection(super.entries());
        }

        return entries;
    }
    public Cast getKeyCast()
    {
        return keyCast;
    }
    public Cast getValueCast()
    {
        return valueCast;
    }
    public Object put(Object key, Object value)
    {
        return super.put(keyCast.cast(key), valueCast.cast(value));
    }
}

---