bundles/org.eclipse.wst.xsd.ui/src-adt-xsd/org/eclipse/wst/xsd/ui/internal/design/editparts/ConnectableEditPart.java

/*******************************************************************************
 * Copyright (c) 2001, 2006 IBM Corporation 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:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.wst.xsd.ui.internal.design.editparts;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.core.runtime.Assert;
import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.IFigure;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPartFactory;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.LayerConstants;
import org.eclipse.wst.xsd.ui.internal.adapters.XSDBaseAdapter;
import org.eclipse.wst.xsd.ui.internal.adt.design.editparts.BaseEditPart;
import org.eclipse.wst.xsd.ui.internal.adt.design.editpolicies.ADTSelectionFeedbackEditPolicy;
import org.eclipse.wst.xsd.ui.internal.design.figures.CenteredIconFigure;
import org.eclipse.wst.xsd.ui.internal.design.figures.GenericGroupFigure;
import org.eclipse.wst.xsd.ui.internal.design.figures.IExtendedFigureFactory;
import org.eclipse.xsd.XSDConcreteComponent;

public abstract class ConnectableEditPart extends BaseEditPart
{
  protected ArrayList connectionFigures = new ArrayList();
  
  public IExtendedFigureFactory getExtendedFigureFactory()
  {
    EditPartFactory factory = getViewer().getEditPartFactory();
    Assert.isTrue(factory instanceof IExtendedFigureFactory, "EditPartFactory must be an instanceof of IExtendedFigureFactory");     //$NON-NLS-1$
    return (IExtendedFigureFactory)factory; 
  }
  
  public ConnectableEditPart()
  {
    super();
  }
  
  protected IFigure createFigure()
  {
    GenericGroupFigure figure = new GenericGroupFigure();
    return figure;
  }

  public XSDConcreteComponent getXSDConcreteComponent()
  {
    return (XSDConcreteComponent)((XSDBaseAdapter)getModel()).getTarget();
  }
  
  public List getConnectionFigures()
  {
    return connectionFigures;
  }
  
  public abstract ReferenceConnection createConnectionFigure(BaseEditPart child);
  
  public void activate()
  {
    super.activate();
    activateConnection();
  }

  protected void activateConnection()
  {
    if (connectionFigures == null)
    {
      connectionFigures = new ArrayList();
    }
    for (Iterator i = getChildren().iterator(); i.hasNext();)
    {
      Object o = i.next();
      if (o instanceof BaseEditPart)
      {
        BaseEditPart g = (BaseEditPart) o;
        ReferenceConnection figure = createConnectionFigure(g);
        connectionFigures.add(figure);
        figure.setPoints(figure.getPoints());

        getLayer(LayerConstants.CONNECTION_LAYER).add(figure);
      }
    }
  }
  
  public void deactivate()
  {
    super.deactivate();
    deactivateConnection();
  }

  protected void deactivateConnection()
  {
    // if we have a connection, remove it
    ReferenceConnection connectionFigure;
    if (connectionFigures != null && !connectionFigures.isEmpty())
    {
      for (Iterator i = connectionFigures.iterator(); i.hasNext();)
      {
        connectionFigure = (ReferenceConnection) i.next();

        if (getLayer(LayerConstants.CONNECTION_LAYER).getChildren().contains(connectionFigure))
        {
          getLayer(LayerConstants.CONNECTION_LAYER).remove(connectionFigure);
        }
      }
      connectionFigures = null;
    }
  }

  public void refresh()
  {
    super.refresh();
    refreshConnection();
  }

  protected void refreshConnection()
  {
    if (!isActive())
      return;

    if (connectionFigures == null || connectionFigures.isEmpty())
    {
      activateConnection();
    }
    else
    {
      deactivateConnection();
      activateConnection();
    }
  }
  
  public void addFeedback()
  {
    ReferenceConnection connectionFigure;
    if (connectionFigures != null && !connectionFigures.isEmpty())
    {
      for (Iterator i = connectionFigures.iterator(); i.hasNext();)
      {
        connectionFigure = (ReferenceConnection) i.next();
        connectionFigure.setHighlight(true);
      }
    }
    GenericGroupFigure figure = (GenericGroupFigure)getFigure();
    figure.getIconFigure().setMode(CenteredIconFigure.SELECTED);
    figure.getIconFigure().refresh();
  }
  
  public void removeFeedback()
  {
    ReferenceConnection connectionFigure;
    if (connectionFigures != null && !connectionFigures.isEmpty())
    {
      for (Iterator i = connectionFigures.iterator(); i.hasNext();)
      {
        connectionFigure = (ReferenceConnection) i.next();
        connectionFigure.setHighlight(false);
      }
    }
    GenericGroupFigure figure = (GenericGroupFigure)getFigure();
    figure.getIconFigure().setMode(CenteredIconFigure.NORMAL);
    figure.getIconFigure().refresh();
  }
  
  protected void refreshVisuals()
  {
    super.refreshVisuals();
    GenericGroupFigure figure = (GenericGroupFigure)getFigure();
    figure.getIconFigure().refresh();
  }

  protected void createEditPolicies()
  {
    installEditPolicy(EditPolicy.SELECTION_FEEDBACK_ROLE, new ADTSelectionFeedbackEditPolicy());
  }

  protected void addChildVisual(EditPart childEditPart, int index)
  {
    IFigure child = ((GraphicalEditPart) childEditPart).getFigure();
    getContentPane().add(child, index);
  }

  protected void removeChildVisual(EditPart childEditPart)
  {
    IFigure child = ((GraphicalEditPart) childEditPart).getFigure();
    getContentPane().remove(child);
  }
  
  public IFigure getContentPane()
  {
    return ((GenericGroupFigure)getFigure()).getContentFigure();
  }

  public Figure getTargetFigure()
  {
    return ((GenericGroupFigure)getFigure()).getTargetFigure();
  }

}