/*

  *  Copyright 2003-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.commons.convert1.util;

 import java.util.HashMap;

 import java.util.Iterator;

 /**

  * A map which stores objects by a key Class.

  * When obtaining the object, it will check inheritence and 

  * interface trees to see if the Class matches.

  */

 public class ClassMap extends ProxyMap {

     private Inheritor inh;

     public ClassMap() {

         this(new BasicInheritor());

     }

     public ClassMap(Inheritor inh) {

         super(new HashMap());

         this.inh = inh;

     }

     public Inheritor getInheritor() {

         return inh;

     }

     /**

      * Get the object from the map. If the key is not 

      * a Class object, then it uses the Class of the object.

      */

     public Object get(Object key) {

         if(key == null) {

             return null;

         }

         Class clss = null;

         if(key instanceof Class) {

             clss = (Class) key;

         } else {

             clss = key.getClass();

         }

         Object obj = super.get(clss);

         if(obj == null) {

             inh.setTarget(clss);

             Iterator itr = inh.iterator();

             while(itr.hasNext() && obj == null) {

                 obj = super.get(itr.next());

             }

         }

         return obj;

     }

 }