/*
 * 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.diagram.ad.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.EClassifier;
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.ad.expressions.UMLAbstractExpression;
import org.eclipse.epf.diagram.ad.part.ActivityDiagramEditorPlugin;
import org.eclipse.epf.diagram.core.bridge.BridgeHelper;
import org.eclipse.gmf.runtime.common.ui.services.icon.IconService;
import org.eclipse.gmf.runtime.diagram.ui.geoshapes.internal.providers.GeoShapeIconProvider;
import org.eclipse.gmf.runtime.diagram.ui.geoshapes.internal.providers.GeoshapeConstants;
import org.eclipse.gmf.runtime.diagram.ui.util.INotationType;
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.ActivityNode;
import org.eclipse.uml2.uml.ActivityParameterNode;
import org.eclipse.uml2.uml.StructuredActivityNode;
import org.eclipse.uml2.uml.UMLPackage;

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

	/**
	 * @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;
	}
	
	private static GeoShapeIconProvider geoShapeIconProvider = null;

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

	/**
	 * @generated
	 */
	private static ImageDescriptor getProvidedImageDescriptor(
			ENamedElement element) {
		if (element instanceof EStructuralFeature) {
			EStructuralFeature feature = ((EStructuralFeature) element);
			EClass eContainingClass = feature.getEContainingClass();
			EClassifier eType = feature.getEType();
			if (eContainingClass != null && !eContainingClass.isAbstract()) {
				element = eContainingClass;
			} else if (eType instanceof EClass
					&& !((EClass) eType).isAbstract()) {
				element = eType;
			}
		}
		if (element instanceof EClass) {
			EClass eClass = (EClass) element;
			if (!eClass.isAbstract()) {
				return ActivityDiagramEditorPlugin.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);
	}

	/**
	 * @modified
	 */
	public static Image getImage(IAdaptable hint) {
		if(isGeoShape(hint)){
			return getGeoShapeIconProvider().getIcon(((INotationType)hint).getSemanticHint());
		}
		ENamedElement element = getElement(hint);
		if (element == null) {
			return null;
		}
		// modified START
		Object type = hint.getAdapter(IElementType.class);
		if (type != null) {
			Image image = getImage(element, (IElementType) type);
			if (image != null) {
				return image;
			}
		}
		// modified END
		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(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(ActivityPartition_2001, UMLPackage.eINSTANCE
					.getActivityPartition());

			elements.put(ControlFlow_3001, UMLPackage.eINSTANCE
					.getControlFlow());
		}
		return (ENamedElement) elements.get(type);
	}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	/**
	 * @generated
	 */
	public static final IElementType ControlFlow_3001 = getElementType("org.eclipse.epf.diagram.ad.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(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(ActivityPartition_2001);
			KNOWN_ELEMENT_TYPES.add(ControlFlow_3001);
		}
		return KNOWN_ELEMENT_TYPES.contains(elementType);
	}

	/**
	 * @modified
	 */
	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 = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Activity.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1010.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Phase.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1011.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Iteration.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1012.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Milestone.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1009.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/TaskDescriptor.gif"); //$NON-NLS-1$
			}
			if (imageDescriptor != null) {
				getImageRegistry().put(key, imageDescriptor);
				image = getImageRegistry().get(key);
			}
		}
		return image;
	}

	/**
	 * @modified
	 */
	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 = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Activity.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1010.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Phase.gif"); //$NON-NLS-1$
			} else if (key.equals(StructuredActivityNode_1011.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Iteration.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1012.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.getInstance()
						.getImageDescriptor("full/obj16/Milestone.gif"); //$NON-NLS-1$
			} else if (key.equals(ActivityParameterNode_1009.getId())) {
				imageDescriptor = ActivityDiagramEditorPlugin.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;
	}

	private static boolean isGeoShape(IAdaptable hint) {
		if(hint instanceof INotationType){
			String x = ((INotationType)hint).getSemanticHint();
			if(GeoshapeConstants.getSupportedShapes().contains(x))
				return true;
		}
		return false;
	}
	
	private static GeoShapeIconProvider getGeoShapeIconProvider(){
		if(geoShapeIconProvider == null){
			geoShapeIconProvider = new GeoShapeIconProvider();
		}
		return geoShapeIconProvider;
	}
}