plugins/org.eclipse.epf.authoring.ui/src/org/eclipse/epf/authoring/gef/viewer/NewWPDependencyDiagramViewer.java

/**
 * 
 */
package org.eclipse.epf.authoring.gef.viewer;

import org.eclipse.draw2d.DelegatingLayout;
import org.eclipse.draw2d.FreeformLayer;
import org.eclipse.draw2d.LayeredPane;
import org.eclipse.epf.diagram.model.Diagram;
import org.eclipse.epf.diagram.wpdd.edit.parts.DiagramEditPartFactory;
import org.eclipse.epf.library.edit.IFilter;
import org.eclipse.epf.library.edit.util.Suppression;
import org.eclipse.epf.library.util.ResourceHelper;
import org.eclipse.epf.uma.Activity;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.LayerConstants;
import org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramRootEditPart;
import org.eclipse.gmf.runtime.diagram.ui.parts.DiagramEditDomain;
import org.eclipse.gmf.runtime.diagram.ui.parts.DiagramGraphicalViewer;
import org.eclipse.gmf.runtime.diagram.ui.parts.IDiagramGraphicalViewer;
import org.eclipse.gmf.runtime.diagram.ui.services.editpart.EditPartService;
import org.eclipse.swt.widgets.Composite;

/**
 * 
 * @author Shashidhar Kannoori
 */
public class NewWPDependencyDiagramViewer extends
		NewAbstractDiagramGraphicalViewer {

	/**
	 * @param parent
	 * @param wrapper
	 */
	public NewWPDependencyDiagramViewer(Composite parent, Object wrapper) {
		super(parent, wrapper);
		// TODO Auto-generated constructor stub
	}

	/* (non-Javadoc)
	 * @see org.eclipse.epf.authoring.gef.viewer.AbstractDiagramGraphicalViewer#createEditPart(java.lang.Object, org.eclipse.epf.library.edit.IFilter, org.eclipse.epf.library.edit.util.Suppression)
	 */
	@Override
	protected EditPart createEditPart(Object e, IFilter filter, Suppression sup) {
		if(e instanceof Activity){
			Diagram model =(Diagram)diagram.getElement();
			model.setSuppression(sup);
//			model.setObject(e);
			DiagramEditDomain domain = new DiagramEditDomain(null);
			((IDiagramGraphicalViewer)graphicalViewer).setEditDomain(domain);
			part = EditPartService.getInstance().createGraphicEditPart(diagram);
			return part;
		}
		if(part == null){
			System.err.println("Editpart for "+ e +" is not created");
		}
		return part;
	}

	/* (non-Javadoc)
	 * @see org.eclipse.epf.authoring.gef.viewer.AbstractDiagramGraphicalViewer#getDiagramType()
	 */
	@Override
	protected String getDiagramType() {
		return ResourceHelper.DIAGRAM_TYPE_WP_DEPENDENCY;
	}
	@Override
	protected void configureGraphicalViewer() {
		DiagramGraphicalViewer viewer = (DiagramGraphicalViewer) getGraphicalViewer();
		DiagramRootEditPart root = (DiagramRootEditPart) EditPartService
		.getInstance().createRootEditPart(getDiagram());
		LayeredPane printableLayers = (LayeredPane) root
				.getLayer(LayerConstants.PRINTABLE_LAYERS);
		FreeformLayer extLabelsLayer = new FreeformLayer();
		extLabelsLayer.setLayoutManager(new DelegatingLayout());
		printableLayers.addLayerAfter(extLabelsLayer,
				DiagramEditPartFactory.EXTERNAL_NODE_LABELS_LAYER,
				LayerConstants.PRIMARY_LAYER);
		LayeredPane scalableLayers = (LayeredPane) root
				.getLayer(LayerConstants.SCALABLE_LAYERS);
		FreeformLayer scaledFeedbackLayer = new FreeformLayer();
		scaledFeedbackLayer.setEnabled(false);
		scalableLayers.addLayerAfter(scaledFeedbackLayer,
				LayerConstants.SCALED_FEEDBACK_LAYER,
				DiagramRootEditPart.DECORATION_UNPRINTABLE_LAYER);
		viewer.setRootEditPart(root);
		viewer.setEditPartFactory(createEditPartFactory());
	}
}