//------------------------------------------------------------------------------
// Copyright (c) 2005, 2007 IBM Corporation and others.
// All rights reserved. This program and the accompanying materials
// are made available under the terms of the Eclipse Public License v1.0
// which accompanies this distribution, and is available at
// http://www.eclipse.org/legal/epl-v10.html
//
// Contributors:
// IBM Corporation - initial implementation
//------------------------------------------------------------------------------

package org.eclipse.epf.diagramming.providers;

import java.util.HashSet;
import java.util.IdentityHashMap;
import java.util.Map;
import java.util.Set;

import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.ENamedElement;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.EcorePackage;
import org.eclipse.epf.diagram.core.bridge.BridgeHelper;
import org.eclipse.epf.diagramming.expressions.UMLAbstractExpression;
import org.eclipse.epf.diagramming.part.EPFDiagramEditorPlugin;
import org.eclipse.gmf.runtime.emf.type.core.ElementTypeRegistry;
import org.eclipse.gmf.runtime.emf.type.core.IElementType;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.jface.resource.ImageRegistry;
import org.eclipse.swt.graphics.Image;
import org.eclipse.uml2.uml.ActivityParameterNode;
import org.eclipse.uml2.uml.StructuredActivityNode;
import org.eclipse.uml2.uml.UMLPackage;

/**
 * @generated
 */
public class UMLElementTypes {

	/**
	 * @generated
	 */
	private UMLElementTypes() {
	}

	/**
	 * @generated
	 */
	private static Map elements;

	/**
	 * @generated
	 */
	private static ImageRegistry imageRegistry;

	/**
	 * @generated
	 */
	private static ImageRegistry getImageRegistry() {
		if (imageRegistry == null) {
			imageRegistry = new ImageRegistry();
		}
		return imageRegistry;
	}

	/**
	 * @generated
	 */
	private static String getImageRegistryKey(ENamedElement element) {
		return element.getName();
	}

	/**
	 * @generated
	 */
	private static ImageDescriptor getProvidedImageDescriptor(
			ENamedElement element) {
		if (element instanceof EStructuralFeature) {
			element = ((EStructuralFeature) element).getEContainingClass();
		}
		if (element instanceof EClass) {
			EClass eClass = (EClass) element;
			if (!eClass.isAbstract()) {
				return EPFDiagramEditorPlugin.getInstance()
						.getItemImageDescriptor(
								eClass.getEPackage().getEFactoryInstance()
										.create(eClass));
			}
		}
		// TODO : support structural features
		return null;
	}

	/**
	 * @generated
	 */
	public static ImageDescriptor getImageDescriptor(ENamedElement element) {
		String key = getImageRegistryKey(element);
		ImageDescriptor imageDescriptor = getImageRegistry().getDescriptor(key);
		if (imageDescriptor == null) {
			imageDescriptor = getProvidedImageDescriptor(element);
			if (imageDescriptor == null) {
				imageDescriptor = ImageDescriptor.getMissingImageDescriptor();
			}
			getImageRegistry().put(key, imageDescriptor);
		}
		return imageDescriptor;
	}

	/**
	 * @generated
	 */
	public static Image getImage(ENamedElement element) {
		String key = getImageRegistryKey(element);
		Image image = getImageRegistry().get(key);
		if (image == null) {
			ImageDescriptor imageDescriptor = getProvidedImageDescriptor(element);
			if (imageDescriptor == null) {
				imageDescriptor = ImageDescriptor.getMissingImageDescriptor();
			}
			getImageRegistry().put(key, imageDescriptor);
			image = getImageRegistry().get(key);
		}
		return image;
	}

	/**
	 * @generated
	 */
	public static ImageDescriptor getImageDescriptor(IAdaptable hint) {
		ENamedElement element = getElement(hint);
		if (element == null) {
			return null;
		}
		return getImageDescriptor(element);
	}

	/**
	 * @custom
	 */
	public static Image getImage(ENamedElement element, IElementType type) {
		String key = type.getId();
		Image image = getImageRegistry().get(key);

		if (image == null) {
			ImageDescriptor imageDescriptor = null;
			if (key.equals(StructuredActivityNode_1007.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Activity.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1010.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Phase.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1011.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Iteration.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1012.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Milestone.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1009.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/TaskDescriptor.gif"); //$NON-NLS-1$
			}
			if (imageDescriptor != null) {
				getImageRegistry().put(key, imageDescriptor);
				image = getImageRegistry().get(key);
			}
		}
		return image;
	}

	/**
	 * @custom
	 */
	public static ImageDescriptor getImageDescriptor(ENamedElement element,
			IElementType type) {
		String key = type.getId();
		ImageDescriptor imageDescriptor = getImageRegistry().getDescriptor(key);

		if (imageDescriptor == null) {
			imageDescriptor = null;
			if (key.equals(StructuredActivityNode_1007.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Activity.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1010.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Phase.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1011.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Iteration.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1012.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Milestone.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1009.getId())) {
				imageDescriptor = EPFDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/TaskDescriptor.gif"); //$NON-NLS-1$
			}
			if (imageDescriptor != null) {
				getImageRegistry().put(key, imageDescriptor);
				imageDescriptor = getImageRegistry().getDescriptor(key);
			} else {
				imageDescriptor = getImageDescriptor(element);
			}
		}
		return imageDescriptor;
	}

	/**
	 * @modified
	 */
	public static Image getImage(IAdaptable hint) {
		ENamedElement element = getElement(hint);
		if (element == null) {
			return null;
		}
		Object type = hint.getAdapter(IElementType.class);
		if (type != null) {
			Image image = getImage(element, (IElementType) type);
			if (image != null) {
				return image;
			}
		}
		return getImage(element);
	}

	/**
	 * Returns 'type' of the ecore object associated with the hint.
	 * 
	 * @generated
	 */
	public static ENamedElement getElement(IAdaptable hint) {
		Object type = hint.getAdapter(IElementType.class);
		if (elements == null) {
			elements = new IdentityHashMap();
			elements.put(Activity_79, UMLPackage.eINSTANCE.getActivity());
			elements.put(ActivityPartition_2001, UMLPackage.eINSTANCE
					.getActivityPartition());
			elements.put(ActivityFinalNode_1001, UMLPackage.eINSTANCE
					.getActivityFinalNode());
			elements.put(MergeNode_1002, UMLPackage.eINSTANCE.getMergeNode());
			elements.put(ForkNode_1003, UMLPackage.eINSTANCE.getForkNode());
			elements.put(InitialNode_1004, UMLPackage.eINSTANCE
					.getInitialNode());
			elements.put(DecisionNode_1005, UMLPackage.eINSTANCE
					.getDecisionNode());
			elements.put(JoinNode_1006, UMLPackage.eINSTANCE.getJoinNode());
			elements.put(StructuredActivityNode_1007, UMLPackage.eINSTANCE
					.getStructuredActivityNode());
			elements.put(ActivityPartition_1008, UMLPackage.eINSTANCE
					.getActivityPartition());
			elements.put(ActivityParameterNode_1009, UMLPackage.eINSTANCE
					.getActivityParameterNode());
			elements.put(StructuredActivityNode_1010, UMLPackage.eINSTANCE
					.getStructuredActivityNode());
			elements.put(StructuredActivityNode_1011, UMLPackage.eINSTANCE
					.getStructuredActivityNode());
			elements.put(ActivityParameterNode_1012, UMLPackage.eINSTANCE
					.getActivityParameterNode());
			elements.put(ControlFlow_3001, UMLPackage.eINSTANCE
					.getControlFlow());
		}
		return (ENamedElement) elements.get(type);
	}

	/**
	 * @generated
	 */
	public static final IElementType Activity_79 = getElementType("org.eclipse.epf.diagramming.Activity_79"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ActivityPartition_2001 = getElementType("org.eclipse.epf.diagramming.ActivityPartition_2001"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ActivityFinalNode_1001 = getElementType("org.eclipse.epf.diagramming.ActivityFinalNode_1001"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType MergeNode_1002 = getElementType("org.eclipse.epf.diagramming.MergeNode_1002"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ForkNode_1003 = getElementType("org.eclipse.epf.diagramming.ForkNode_1003"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType InitialNode_1004 = getElementType("org.eclipse.epf.diagramming.InitialNode_1004"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType DecisionNode_1005 = getElementType("org.eclipse.epf.diagramming.DecisionNode_1005"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType JoinNode_1006 = getElementType("org.eclipse.epf.diagramming.JoinNode_1006"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType StructuredActivityNode_1007 = getElementType("org.eclipse.epf.diagramming.StructuredActivityNode_1007"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ActivityPartition_1008 = getElementType("org.eclipse.epf.diagramming.ActivityPartition_1008"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ActivityParameterNode_1009 = getElementType("org.eclipse.epf.diagramming.ActivityParameterNode_1009"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType StructuredActivityNode_1010 = getElementType("org.eclipse.epf.diagramming.StructuredActivityNode_1010"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType StructuredActivityNode_1011 = getElementType("org.eclipse.epf.diagramming.StructuredActivityNode_1011"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ActivityParameterNode_1012 = getElementType("org.eclipse.epf.diagramming.ActivityParameterNode_1012"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final IElementType ControlFlow_3001 = getElementType("org.eclipse.epf.diagramming.ControlFlow_3001"); //$NON-NLS-1$

	/**
	 * @generated
	 */
	private static IElementType getElementType(String id) {
		return ElementTypeRegistry.getInstance().getType(id);
	}

	/**
	 * @generated
	 */
	private static Set KNOWN_ELEMENT_TYPES;

	/**
	 * @generated
	 */
	public static boolean isKnownElementType(IElementType elementType) {
		if (KNOWN_ELEMENT_TYPES == null) {
			KNOWN_ELEMENT_TYPES = new HashSet();
			KNOWN_ELEMENT_TYPES.add(Activity_79);
			KNOWN_ELEMENT_TYPES.add(ActivityPartition_2001);
			KNOWN_ELEMENT_TYPES.add(ActivityFinalNode_1001);
			KNOWN_ELEMENT_TYPES.add(MergeNode_1002);
			KNOWN_ELEMENT_TYPES.add(ForkNode_1003);
			KNOWN_ELEMENT_TYPES.add(InitialNode_1004);
			KNOWN_ELEMENT_TYPES.add(DecisionNode_1005);
			KNOWN_ELEMENT_TYPES.add(JoinNode_1006);
			KNOWN_ELEMENT_TYPES.add(StructuredActivityNode_1007);
			KNOWN_ELEMENT_TYPES.add(ActivityPartition_1008);
			KNOWN_ELEMENT_TYPES.add(ActivityParameterNode_1009);
			KNOWN_ELEMENT_TYPES.add(StructuredActivityNode_1010);
			KNOWN_ELEMENT_TYPES.add(StructuredActivityNode_1011);
			KNOWN_ELEMENT_TYPES.add(ActivityParameterNode_1012);
			KNOWN_ELEMENT_TYPES.add(ControlFlow_3001);
		}
		return KNOWN_ELEMENT_TYPES.contains(elementType);
	}

	/**
	 * @generated
	 */
	public static class Initializers {

		/**
		 * @generated
		 */
		public static final ObjectInitializer StructuredActivityNode_1007 = new ObjectInitializer(
				new FeatureInitializer[] { new FeatureInitializer(
						new UMLAbstractExpression(UMLPackage.eINSTANCE
								.getStructuredActivityNode()) {
							protected Object doEvaluate(Object context, Map env) {
								StructuredActivityNode self = (StructuredActivityNode) context;
								return Java.createUmaType_Activity(self);
							}
						}, EcorePackage.eINSTANCE
								.getEModelElement_EAnnotations()) });

		/**
		 * @generated
		 */
		public static final ObjectInitializer ActivityParameterNode_1009 = new ObjectInitializer(
				new FeatureInitializer[] { new FeatureInitializer(
						new UMLAbstractExpression(UMLPackage.eINSTANCE
								.getActivityParameterNode()) {
							protected Object doEvaluate(Object context, Map env) {
								ActivityParameterNode self = (ActivityParameterNode) context;
								return Java.createUmaType_TaskDescriptor(self);
							}
						}, EcorePackage.eINSTANCE
								.getEModelElement_EAnnotations()) });

		/**
		 * @generated
		 */
		public static final ObjectInitializer StructuredActivityNode_1010 = new ObjectInitializer(
				new FeatureInitializer[] { new FeatureInitializer(
						new UMLAbstractExpression(UMLPackage.eINSTANCE
								.getStructuredActivityNode()) {
							protected Object doEvaluate(Object context, Map env) {
								StructuredActivityNode self = (StructuredActivityNode) context;
								return Java.createUmaType_Phase(self);
							}
						}, EcorePackage.eINSTANCE
								.getEModelElement_EAnnotations()) });

		/**
		 * @generated
		 */
		public static final ObjectInitializer StructuredActivityNode_1011 = new ObjectInitializer(
				new FeatureInitializer[] { new FeatureInitializer(
						new UMLAbstractExpression(UMLPackage.eINSTANCE
								.getStructuredActivityNode()) {
							protected Object doEvaluate(Object context, Map env) {
								StructuredActivityNode self = (StructuredActivityNode) context;
								return Java.createUmaType_Iteration(self);
							}
						}, EcorePackage.eINSTANCE
								.getEModelElement_EAnnotations()) });

		/**
		 * @generated
		 */
		public static final ObjectInitializer ActivityParameterNode_1012 = new ObjectInitializer(
				new FeatureInitializer[] { new FeatureInitializer(
						new UMLAbstractExpression(UMLPackage.eINSTANCE
								.getActivityParameterNode()) {
							protected Object doEvaluate(Object context, Map env) {
								ActivityParameterNode self = (ActivityParameterNode) context;
								return Java.createUmaType_Milestone(self);
							}
						}, EcorePackage.eINSTANCE
								.getEModelElement_EAnnotations()) });

		/** 
		 * @generated
		 */
		private Initializers() {
		}

		/** 
		 * @generated
		 */
		public static class ObjectInitializer {
			/** 
			 * @generated
			 */
			private FeatureInitializer[] initExpressions;

			/** 
			 * @generated
			 */
			ObjectInitializer(FeatureInitializer[] initExpressions) {
				this.initExpressions = initExpressions;
			}

			/** 
			 * @generated
			 */
			public void init(EObject instance) {
				for (int i = 0; i < initExpressions.length; i++) {
					FeatureInitializer nextExpr = initExpressions[i];
					try {
						nextExpr.init(instance);
					} catch (RuntimeException e) {
						EPFDiagramEditorPlugin.getInstance().logError(
								"Feature initialization failed", e); //$NON-NLS-1$						
					}
				}
			}
		} // end of ObjectInitializer

		/** 
		 * @generated
		 */
		static class FeatureInitializer {

			/** 
			 * @generated
			 */
			private EStructuralFeature sFeature;

			/** 
			 * @generated
			 */
			private UMLAbstractExpression expression;

			/**
			 * @generated
			 */
			FeatureInitializer(UMLAbstractExpression expression,
					EStructuralFeature sFeature) {
				this.sFeature = sFeature;
				this.expression = expression;
			}

			/** 
			 * @generated
			 */
			void init(EObject contextInstance) {
				expression.assignTo(sFeature, contextInstance);
			}
		} // end of FeatureInitializer

		/** 
		 * @generated
		 */
		static class Java {

			/**
			 * @modified
			 */
			private static EList createUmaType_Phase(StructuredActivityNode self) {
				return BridgeHelper.addEAnnotationType(self,
						BridgeHelper.UMA_PHASE);
			}

			/**
			 * @modified
			 */
			private static EList createUmaType_Milestone(
					ActivityParameterNode self) {
				return BridgeHelper.addEAnnotationType(self,
						BridgeHelper.UMA_MILESTONE);
			}

			/**
			 * @modified
			 */
			private static EList createUmaType_Activity(
					StructuredActivityNode self) {
				return BridgeHelper.addEAnnotationType(self,
						BridgeHelper.UMA_ACTIVITY);
			}

			/**
			 * @modified
			 */
			private static EList createUmaType_TaskDescriptor(
					ActivityParameterNode self) {
				return BridgeHelper.addEAnnotationType(self,
						BridgeHelper.UMA_TASK_DESCRIPTOR);
			}

			/**
			 * @modified
			 */
			private static EList createUmaType_Iteration(
					StructuredActivityNode self) {
				return BridgeHelper.addEAnnotationType(self,
						BridgeHelper.UMA_ITERATION);
			}
		} //Java
	} // end of Initializers
}