org.eclipse.osee.ats/src/org/eclipse/osee/ats/workflow/editor/parts/ShapesEditPartFactory.java

/*******************************************************************************
 * Copyright (c) 2004, 2005 Donald G. Dunne 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:
 *    Donald G. Dunne - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.ats.workflow.editor.parts;

import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPartFactory;
import org.eclipse.osee.ats.workflow.editor.model.CancelledWorkPageShape;
import org.eclipse.osee.ats.workflow.editor.model.CompletedWorkPageShape;
import org.eclipse.osee.ats.workflow.editor.model.Connection;
import org.eclipse.osee.ats.workflow.editor.model.Shape;
import org.eclipse.osee.ats.workflow.editor.model.WorkPageShape;
import org.eclipse.osee.ats.workflow.editor.model.WorkflowDiagram;

/**
 * Factory that maps model elements to edit parts.
 * 
 * @author Donald G. Dunne
 */
public class ShapesEditPartFactory implements EditPartFactory {

   /*
    * (non-Javadoc)
    * @see org.eclipse.gef.EditPartFactory#createEditPart(org.eclipse.gef.EditPart, java.lang.Object)
    */
   public EditPart createEditPart(EditPart context, Object modelElement) {
      // get EditPart for model element
      EditPart part = getPartForElement(modelElement);
      // store model element in EditPart
      part.setModel(modelElement);
      return part;
   }

   /**
    * Maps an object to an EditPart.
    * 
    * @throws RuntimeException if no match was found (programming error)
    */
   private EditPart getPartForElement(Object modelElement) {
      if (modelElement instanceof WorkflowDiagram) {
         return new DiagramEditPart();
      }
      if (modelElement instanceof CompletedWorkPageShape) {
         return new WorkPageEditPart((CompletedWorkPageShape) modelElement);
      }
      if (modelElement instanceof CancelledWorkPageShape) {
         return new WorkPageEditPart((CancelledWorkPageShape) modelElement);
      }
      if (modelElement instanceof WorkPageShape) {
         return new WorkPageEditPart((WorkPageShape) modelElement);
      }
      if (modelElement instanceof Shape) {
         return new ShapeEditPart();
      }
      if (modelElement instanceof Connection) {
         return new ConnectionEditPart();
      }
      throw new RuntimeException(
            "Can't create part for model element: " + ((modelElement != null) ? modelElement.getClass().getName() : "null"));
   }

}