org.eclipse.osee.framework.ui.data.model.editor/src/org/eclipse/osee/framework/ui/data/model/editor/part/ContainerEditPart.java

/*******************************************************************************
 * Copyright (c) 2004, 2007 Boeing.
 * 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:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.framework.ui.data.model.editor.part;

import java.util.List;
import org.eclipse.draw2d.IFigure;
import org.eclipse.gef.editparts.AbstractGraphicalEditPart;
import org.eclipse.osee.framework.ui.data.model.editor.figure.CompartmentFigure;
import org.eclipse.osee.framework.ui.data.model.editor.model.helper.ContainerModel;
import org.eclipse.osee.framework.ui.data.model.editor.model.helper.ContainerModel.ContainerType;

/**
 * @author Roberto E. Escobar
 */
public class ContainerEditPart extends AbstractGraphicalEditPart {

   public ContainerEditPart(Object model) {
      super();
      setModel(model);
   }

   protected void createEditPolicies() {
   }

   protected IFigure createFigure() {
      IFigure figure = new CompartmentFigure();
      return figure;
   }

   @SuppressWarnings("unchecked")
   protected List getModelChildren() {
      if (getModel() instanceof ContainerModel) {
         return ((ContainerModel) getModel()).getChildren();
      } else {
         return (List) getModel();
      }
   }

   public boolean isSelectable() {
      return false;
   }

   public ContainerType getContainerType() {
      if (getModel() instanceof ContainerModel) {
         return ((ContainerModel) getModel()).getContainerType();
      }
      return null;
   }
}