/*
* 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;
}
}