package uk.co.badgersinfoil.asxsd;

 import java.util.HashMap;

 import java.util.Map;

 import org.eclipse.xsd.XSDConcreteComponent;

 public class MappingFunctionStore {

         private Map store = new HashMap();

         public MappingFunction get(CodegenRole role,

                             XSDConcreteComponent component)

         {

                 Key key = new Key(role, component);

                 return (MappingFunction)store.get(key);

         }

         public void put(CodegenRole role,

                         XSDConcreteComponent component,

                         MappingFunction fn)

         {

                 Key key = new Key(role, component);

                 if (store.containsKey(key)) {

                         throw new IllegalArgumentException("Function already defined for "+role+"+"+component);

                 }

                 store.put(key, fn);

         }

         private static class Key {

                 private CodegenRole role;

                 private XSDConcreteComponent component;

                 public Key(CodegenRole role, XSDConcreteComponent component) {

                         this.role = role;

                         this.component = component;

                 }

                 public int hashCode() {

                         final int PRIME = 31;

                         int result = 1;

                         result = PRIME * result + ((component == null) ? 0 : component.hashCode());

                         result = PRIME * result + ((role == null) ? 0 : role.hashCode());

                         return result;

                 }

                 public boolean equals(Object obj) {

                         if (this == obj)

                                 return true;

                         if (obj == null)

                                 return false;

                         if (getClass() != obj.getClass())

                                 return false;

                         final Key other = (Key) obj;

                         if (component == null) {

                                 if (other.component != null)

                                         return false;

                         } else if (!component.equals(other.component))

                                 return false;

                         if (role == null) {

                                 if (other.role != null)

                                         return false;

                         } else if (!role.equals(other.role))

                                 return false;

                         return true;

                 }

         }

 }