package ca.uhn.fhir.context;
   
   /*
    * #%L
    * HAPI FHIR - Core Library
    * %%
    * Copyright (C) 2014 University Health Network
    * %%
    * 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.
   * #L%
   */
  
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import ca.uhn.fhir.model.api.IElement;
  import ca.uhn.fhir.model.api.IResource;
  import ca.uhn.fhir.model.base.composite.BaseResourceReferenceDt;
  
  public class RuntimeResourceReferenceDefinition extends BaseRuntimeElementDefinition<BaseResourceReferenceDt> {
  
  	private final List<Class<? extends IResource>> myResourceTypes;
  	private HashMap<Class<? extends IResource>, RuntimeResourceDefinition> myResourceTypeToDefinition;
  
  	public RuntimeResourceReferenceDefinition(String theName, List<Class<? extends IResource>> theResourceTypes) {
  		super(theName, BaseResourceReferenceDt.class);
  		if (theResourceTypes == null || theResourceTypes.isEmpty()) {
  			throw new ConfigurationException("Element '" + theName + "' has no resource types noted");
  		}
  		myResourceTypes = theResourceTypes;
  	}
  
  	public List<Class<? extends IResource>> getResourceTypes() {
  		return myResourceTypes;
  	}
  
  	@Override
  	void sealAndInitialize(Map<Class<? extends IElement>, BaseRuntimeElementDefinition<?>> theClassToElementDefinitions) {
  		myResourceTypeToDefinition = new HashMap<Class<? extends IResource>, RuntimeResourceDefinition>();
  		for (Class<? extends IResource> next : myResourceTypes) {
  			if (next.equals(IResource.class)) {
  				continue;
  			}
  			RuntimeResourceDefinition definition = (RuntimeResourceDefinition) theClassToElementDefinitions.get(next);
  			if (definition == null) {
  				throw new ConfigurationException("Couldn't find definition for: " + next.getCanonicalName());
  			}
  			myResourceTypeToDefinition.put(next, definition);
  		}
  	}
  
  	@Override
  	public ca.uhn.fhir.context.BaseRuntimeElementDefinition.ChildTypeEnum getChildType() {
  		return ChildTypeEnum.RESOURCE_REF;
  	}
  
  	public RuntimeResourceDefinition getDefinitionForResourceType(Class<? extends IResource> theType) {
  		RuntimeResourceDefinition retVal = myResourceTypeToDefinition.get(theType);
  		if (retVal == null) {
  			throw new ConfigurationException("Unknown type:  " + theType.getCanonicalName());
  		}
  		return retVal;
  	}
  
  }