import org.eclipse.core.runtime.IAdapterFactory;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.graphiti.mm.pictograms.PictogramElement;
import org.eclipse.graphiti.ui.editor.DiagramEditor;
import org.eclipse.graphiti.ui.platform.GraphitiShapeEditPart;
import org.eclipse.ui.views.properties.IPropertySource;


public class EditPartToIPropertySourceAdapterFactory implements IAdapterFactory {
	
	public EditPartToIPropertySourceAdapterFactory(){
		super();
	}
	
	
	public Object getAdapter(Object adaptableObject, @SuppressWarnings("rawtypes") Class adapterType) {
		if (IPropertySource.class.equals(adapterType)) {
			if (adaptableObject instanceof GraphitiShapeEditPart) {
				GraphitiShapeEditPart editPart = (GraphitiShapeEditPart) adaptableObject;
				PictogramElement pe = editPart.getPictogramElement();
			Object object = editPart.getFeatureProvider().getBusinessObjectForPictogramElement(pe);
				if (object instanceof EClass) {
					return new EClassPropertySource((EClass)object);
				}
			}
		}
		return null;
	}
	
	@SuppressWarnings("rawtypes")
	public Class[] getAdapterList() {
		return new Class[] { IPropertySource.class };
	}