examples/org.eclipse.gmf.examples.eclipsecon.diagram.custom/src/org/eclipse/gmf/examples/eclipsecon/diagram/custom/providers/EditPartProvider.java

package org.eclipse.gmf.examples.eclipsecon.diagram.custom.providers;

import org.eclipse.gmf.examples.eclipsecon.Presenter;
import org.eclipse.gmf.examples.eclipsecon.diagram.custom.editparts.BetterLookingPresenterEditPart;
import org.eclipse.gmf.runtime.diagram.ui.services.editpart.AbstractEditPartProvider;
import org.eclipse.gmf.runtime.notation.View;


public class EditPartProvider extends AbstractEditPartProvider {

	/* 
	 * (non-Javadoc)
	 * @see org.eclipse.gmf.runtime.diagram.ui.services.editpart.AbstractEditPartProvider#getNodeEditPartClass(org.eclipse.gmf.runtime.notation.View)
	 */
	protected Class getNodeEditPartClass(View view) {
        
		if (view.getElement() instanceof Presenter &&
            view.getType() == "" ) { //$NON-NLS-1$
            return BetterLookingPresenterEditPart.class;
        }
        
		return super.getNodeEditPartClass(view);
	}

}