bundles/org.eclipse.wst.wsdl.ui/src-asd/org/eclipse/wst/wsdl/ui/internal/asd/design/editparts/MessageReferenceEditPart.java - webservices/webtools.webservices - Git at Google

 /*******************************************************************************
  * Copyright (c) 2001, 2008 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.wsdl.ui.internal.asd.design.editparts;

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

 import org.eclipse.draw2d.ColorConstants;
 import org.eclipse.draw2d.Figure;
 import org.eclipse.draw2d.Graphics;
 import org.eclipse.draw2d.IFigure;
 import org.eclipse.draw2d.Label;
 import org.eclipse.draw2d.MarginBorder;
 import org.eclipse.draw2d.PositionConstants;
 import org.eclipse.draw2d.ToolbarLayout;
 import org.eclipse.draw2d.geometry.Point;
 import org.eclipse.gef.DragTracker;
 import org.eclipse.gef.EditPart;
 import org.eclipse.gef.EditPolicy;
 import org.eclipse.gef.GraphicalEditPart;
 import org.eclipse.gef.Request;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.DesignViewGraphicsConstants;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.editpolicies.ASDDragAndDropEditPolicy;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.editpolicies.ASDGraphNodeDragTracker;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.editpolicies.ASDSelectionEditPolicy;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.figures.ListFigure;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.figures.ModelDiagnosticInfo;
 import org.eclipse.wst.wsdl.ui.internal.asd.design.layouts.RowLayout;
 import org.eclipse.wst.wsdl.ui.internal.asd.facade.IMessageReference;
 import org.eclipse.wst.wsdl.ui.internal.asd.outline.ITreeElement;

 public class MessageReferenceEditPart extends BaseEditPart implements IFeedbackHandler
 {
   protected Figure contentPane;
   protected Label label;
   protected RowLayout rowLayout;

   private Figure messageLabelWrapper;
   private Label messageLabel;

   protected ASDSelectionEditPolicy selectionHandlesEditPolicy = new ASDSelectionEditPolicy();

   protected IFigure createFigure()
   {
     ListFigure figure = new ListFigure(true);
     figure.setOpaque(true);
     figure.setPaintFirstLine(false);
     figure.setBackgroundColor(ColorConstants.tooltipBackground);

     rowLayout = new RowLayout();
     figure.setLayoutManager(rowLayout);
     label = new Label()
     {
       public void paint(Graphics graphics)
       {
         super.paint(graphics);
         // bug146932
         paintFocusCursor(getBounds(), graphics);
       }
     };
     label.setLabelAlignment(Label.LEFT);
     //label.setFont(DesignViewGraphicsConstants.mediumFont);
     label.setBorder(new MarginBorder(2, 16, 2 ,10));
     figure.add(label);

     contentPane = new ListFigure();
     ((ListFigure)contentPane).setOpaque(true);
     contentPane.setBackgroundColor(ColorConstants.listBackground);
     ToolbarLayout toolbarLayout = new ToolbarLayout(false);

     /*
     ToolbarLayout toolbarLayout = new ToolbarLayout(false)
     {
        // TODO (cs) consider minor tweak here to ensure that the row fills up the available space
        // vertically... without this a bit of trim is visible at the bottom of the list
        // when the param labels are given a margin width top and bottom <  '4'
        public void layout(IFigure parent)
        {
          super.layout(parent);
          if (children.size() == 1)
          {
             Figure child = (Figure)children.get(0);
             //child.getBounds().height = parent.getClientArea().height;
          }
        }
     };
      */
     toolbarLayout.setStretchMinorAxis(true);
     contentPane.setLayoutManager(toolbarLayout);
     figure.add(contentPane);

     rowLayout.setConstraint(label, "MessageLabel"); //$NON-NLS-1$
     rowLayout.setConstraint(contentPane, "MessageContentPane"); //$NON-NLS-1$

     // rmah: The block of code below has been moved from refreshVisuals().  We're
     // assuming the read-only state of the EditPart will never change once the
     // EditPart has been created.
     if (isReadOnly())
     {
       label.setForegroundColor(DesignViewGraphicsConstants.readOnlyLabelColor);
       label.getParent().setBackgroundColor(DesignViewGraphicsConstants.readOnlyMessageRefHeadingColor);
     }
     else
     {
       label.setForegroundColor(DesignViewGraphicsConstants.defaultForegroundColor);
       label.getParent().setBackgroundColor(ColorConstants.tooltipBackground);
     }

     return figure;
   }

   protected void createEditPolicies()
   {
       super.createEditPolicies();
 	  if (!isReadOnly() && !isFileReadOnly()) {
 		  installEditPolicy(EditPolicy.PRIMARY_DRAG_ROLE, new ASDDragAndDropEditPolicy(getViewer(), selectionHandlesEditPolicy));
 	  }
 	  installEditPolicy(EditPolicy.SELECTION_FEEDBACK_ROLE, selectionHandlesEditPolicy);
   }

   public DragTracker getDragTracker(Request request)
   {
     return new ASDGraphNodeDragTracker((EditPart)this);
   }

   public IFigure getContentPane()
   {
     return contentPane;
   }

   protected void refreshVisuals()
   {
     super.refreshVisuals();
     IMessageReference message = (IMessageReference)getModel();

     if (message instanceof ITreeElement) {
     	label.setText(((ITreeElement) message).getText());
     	label.setIcon(((ITreeElement) message).getImage());
     }

     // Resize column widths.  Sizes may have shrunk.
     rowLayout.getColumnData().clearColumnWidths();
     for (EditPart parent = getParent(); parent != null; parent = parent.getParent())
     {
       if (parent instanceof InterfaceEditPart)
       {
         ((GraphicalEditPart)parent).getFigure().invalidateTree();
         break;
       }
     }
   }

   protected void refreshChildren() {
 	  super.refreshChildren();
       refreshMessage();
   }

   protected void refreshMessage() {
 	  if (getModelChildren().size() > 0) {
 		  if (messageLabelWrapper != null) {
 			  contentPane.remove(messageLabelWrapper);
 			  messageLabelWrapper = null;
 		  }
 	  }
 	  else {
 		  if (messageLabelWrapper == null) {
 			  messageLabelWrapper = new Figure();
 			  messageLabel = new Label();

 			  ToolbarLayout toolbarLayout = new ToolbarLayout(false)
 			    {
 				  // We want to center the text
 			       public void layout(IFigure parent)
 			       {
 			         super.layout(parent);

 			         if (parent.getChildren().size() == 1 && parent.getChildren().get(0) instanceof Label)
 			         {
 			            parent.setSize(parent.getSize().width, parent.getSize().height + 3);
 			            Label child = (Label) parent.getChildren().get(0);
 			            int newXAmount = (parent.getSize().width - child.getSize().width) / 2;
 			            Point p = child.getLocation();
 			            child.setLocation(new Point(p.x + newXAmount, p.y + 3));
 			         }
 			       }
 			    };

 			  messageLabelWrapper.setLayoutManager(toolbarLayout);
 			  messageLabelWrapper.add(messageLabel);
 			  contentPane.add(messageLabelWrapper);
 		  }

 		  List errorList = getErrors();
 		  List warnList = getWarnings();

 		  if (errorList.size() > 0) {
 			  ModelDiagnosticInfo info = (ModelDiagnosticInfo) errorList.get(0);
 			  messageLabel.setText("  " + info.getDescriptionText() + "  ");  //$NON-NLS-1$ //$NON-NLS-2$
 			  messageLabel.setForegroundColor(info.getDescriptionTextColor());
 		  }
 		  else if (warnList.size() > 0){
 			  ModelDiagnosticInfo info = (ModelDiagnosticInfo) warnList.get(0);
 			  messageLabel.setText("  " + info.getDescriptionText() + "  ");  //$NON-NLS-1$ //$NON-NLS-2$
 			  messageLabel.setForegroundColor(info.getDescriptionTextColor());
 		  }
 		  else {
 			  messageLabel.setText(""); //$NON-NLS-1$
 			  messageLabel.setForegroundColor(ColorConstants.black);
 		  }
 	  }
   }

   protected List getErrors() {
 	  IMessageReference messageRef = (IMessageReference) getModel();
 	  return getDiagnosticMessageType(messageRef.getDiagnosticMessages(), ModelDiagnosticInfo.ERROR_TYPE);
   }

   protected List getWarnings() {
 	  IMessageReference messageRef = (IMessageReference) getModel();
 	  return getDiagnosticMessageType(messageRef.getDiagnosticMessages(), ModelDiagnosticInfo.WARNING_TYPE);
   }

   protected List getDiagnosticMessageType(List diagnosticInfo, int type) {
 	  List listType = new ArrayList();
 	  for (int index = 0; index < diagnosticInfo.size(); index++) {
 		  ModelDiagnosticInfo info = (ModelDiagnosticInfo) diagnosticInfo.get(index);
 		  if (info.getType() == type) {
 			  listType.add(info);
 		  }
 	  }

 	  return listType;
   }

   protected List getModelChildren()
   {
     IMessageReference theMessage = (IMessageReference)getModel();
     return theMessage.getParameters();
   }

   public void addNotify()
   {
     InterfaceEditPart.attachToInterfaceEditPart(this, rowLayout);
     super.addNotify();
   }

   public void addFeedback() {
 	  label.getParent().setBackgroundColor(DesignViewGraphicsConstants.tableCellSelectionColor);
   }

   public void removeFeedback() {
 	  if (isReadOnly()) {
 	    label.getParent().setBackgroundColor(DesignViewGraphicsConstants.readOnlyMessageRefHeadingColor);
 	  }
 	  else {
 		  label.getParent().setBackgroundColor(ColorConstants.tooltipBackground);
 	  }
   }

   public EditPart getRelativeEditPart(int direction)
   {
     EditPart editPart = super.getRelativeEditPart(direction);
     if (direction == PositionConstants.SOUTH && editPart == null)
     {
       editPart = EditPartNavigationHandlerUtil.getNextInterface(this);
     }
     return editPart;
   }
 }
	/*******************************************************************************
	* Copyright (c) 2001, 2008 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.wsdl.ui.internal.asd.design.editparts;

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

	import org.eclipse.draw2d.ColorConstants;
	import org.eclipse.draw2d.Figure;
	import org.eclipse.draw2d.Graphics;
	import org.eclipse.draw2d.IFigure;
	import org.eclipse.draw2d.Label;
	import org.eclipse.draw2d.MarginBorder;
	import org.eclipse.draw2d.PositionConstants;
	import org.eclipse.draw2d.ToolbarLayout;
	import org.eclipse.draw2d.geometry.Point;
	import org.eclipse.gef.DragTracker;
	import org.eclipse.gef.EditPart;
	import org.eclipse.gef.EditPolicy;
	import org.eclipse.gef.GraphicalEditPart;
	import org.eclipse.gef.Request;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.DesignViewGraphicsConstants;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.editpolicies.ASDDragAndDropEditPolicy;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.editpolicies.ASDGraphNodeDragTracker;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.editpolicies.ASDSelectionEditPolicy;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.figures.ListFigure;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.figures.ModelDiagnosticInfo;
	import org.eclipse.wst.wsdl.ui.internal.asd.design.layouts.RowLayout;
	import org.eclipse.wst.wsdl.ui.internal.asd.facade.IMessageReference;
	import org.eclipse.wst.wsdl.ui.internal.asd.outline.ITreeElement;

	public class MessageReferenceEditPart extends BaseEditPart implements IFeedbackHandler
	{
	protected Figure contentPane;
	protected Label label;
	protected RowLayout rowLayout;

	private Figure messageLabelWrapper;
	private Label messageLabel;

	protected ASDSelectionEditPolicy selectionHandlesEditPolicy = new ASDSelectionEditPolicy();

	protected IFigure createFigure()
	{
	ListFigure figure = new ListFigure(true);
	figure.setOpaque(true);
	figure.setPaintFirstLine(false);
	figure.setBackgroundColor(ColorConstants.tooltipBackground);

	rowLayout = new RowLayout();
	figure.setLayoutManager(rowLayout);
	label = new Label()
	{
	public void paint(Graphics graphics)
	{
	super.paint(graphics);
	// bug146932
	paintFocusCursor(getBounds(), graphics);
	}
	};
	label.setLabelAlignment(Label.LEFT);
	//label.setFont(DesignViewGraphicsConstants.mediumFont);
	label.setBorder(new MarginBorder(2, 16, 2 ,10));
	figure.add(label);

	contentPane = new ListFigure();
	((ListFigure)contentPane).setOpaque(true);
	contentPane.setBackgroundColor(ColorConstants.listBackground);
	ToolbarLayout toolbarLayout = new ToolbarLayout(false);

	/*
	ToolbarLayout toolbarLayout = new ToolbarLayout(false)
	{
	// TODO (cs) consider minor tweak here to ensure that the row fills up the available space
	// vertically... without this a bit of trim is visible at the bottom of the list
	// when the param labels are given a margin width top and bottom < '4'
	public void layout(IFigure parent)
	{
	super.layout(parent);
	if (children.size() == 1)
	{
	Figure child = (Figure)children.get(0);
	//child.getBounds().height = parent.getClientArea().height;
	}
	}
	};
	*/
	toolbarLayout.setStretchMinorAxis(true);
	contentPane.setLayoutManager(toolbarLayout);
	figure.add(contentPane);

	rowLayout.setConstraint(label, "MessageLabel"); //$NON-NLS-1$
	rowLayout.setConstraint(contentPane, "MessageContentPane"); //$NON-NLS-1$

	// rmah: The block of code below has been moved from refreshVisuals(). We're
	// assuming the read-only state of the EditPart will never change once the
	// EditPart has been created.
	if (isReadOnly())
	{
	label.setForegroundColor(DesignViewGraphicsConstants.readOnlyLabelColor);
	label.getParent().setBackgroundColor(DesignViewGraphicsConstants.readOnlyMessageRefHeadingColor);
	}
	else
	{
	label.setForegroundColor(DesignViewGraphicsConstants.defaultForegroundColor);
	label.getParent().setBackgroundColor(ColorConstants.tooltipBackground);
	}

	return figure;
	}

	protected void createEditPolicies()
	{
	super.createEditPolicies();
	if (!isReadOnly() && !isFileReadOnly()) {
	installEditPolicy(EditPolicy.PRIMARY_DRAG_ROLE, new ASDDragAndDropEditPolicy(getViewer(), selectionHandlesEditPolicy));
	}
	installEditPolicy(EditPolicy.SELECTION_FEEDBACK_ROLE, selectionHandlesEditPolicy);
	}

	public DragTracker getDragTracker(Request request)
	{
	return new ASDGraphNodeDragTracker((EditPart)this);
	}

	public IFigure getContentPane()
	{
	return contentPane;
	}

	protected void refreshVisuals()
	{
	super.refreshVisuals();
	IMessageReference message = (IMessageReference)getModel();

	if (message instanceof ITreeElement) {
	label.setText(((ITreeElement) message).getText());
	label.setIcon(((ITreeElement) message).getImage());
	}

	// Resize column widths. Sizes may have shrunk.
	rowLayout.getColumnData().clearColumnWidths();
	for (EditPart parent = getParent(); parent != null; parent = parent.getParent())
	{
	if (parent instanceof InterfaceEditPart)
	{
	((GraphicalEditPart)parent).getFigure().invalidateTree();
	break;
	}
	}
	}

	protected void refreshChildren() {
	super.refreshChildren();
	refreshMessage();
	}

	protected void refreshMessage() {
	if (getModelChildren().size() > 0) {
	if (messageLabelWrapper != null) {
	contentPane.remove(messageLabelWrapper);
	messageLabelWrapper = null;
	}
	}
	else {
	if (messageLabelWrapper == null) {
	messageLabelWrapper = new Figure();
	messageLabel = new Label();

	ToolbarLayout toolbarLayout = new ToolbarLayout(false)
	{
	// We want to center the text
	public void layout(IFigure parent)
	{
	super.layout(parent);

	if (parent.getChildren().size() == 1 && parent.getChildren().get(0) instanceof Label)
	{
	parent.setSize(parent.getSize().width, parent.getSize().height + 3);
	Label child = (Label) parent.getChildren().get(0);
	int newXAmount = (parent.getSize().width - child.getSize().width) / 2;
	Point p = child.getLocation();
	child.setLocation(new Point(p.x + newXAmount, p.y + 3));
	}
	}
	};

	messageLabelWrapper.setLayoutManager(toolbarLayout);
	messageLabelWrapper.add(messageLabel);
	contentPane.add(messageLabelWrapper);
	}

	List errorList = getErrors();
	List warnList = getWarnings();

	if (errorList.size() > 0) {
	ModelDiagnosticInfo info = (ModelDiagnosticInfo) errorList.get(0);
	messageLabel.setText(" " + info.getDescriptionText() + " "); //$NON-NLS-1$ //$NON-NLS-2$
	messageLabel.setForegroundColor(info.getDescriptionTextColor());
	}
	else if (warnList.size() > 0){
	ModelDiagnosticInfo info = (ModelDiagnosticInfo) warnList.get(0);
	messageLabel.setText(" " + info.getDescriptionText() + " "); //$NON-NLS-1$ //$NON-NLS-2$
	messageLabel.setForegroundColor(info.getDescriptionTextColor());
	}
	else {
	messageLabel.setText(""); //$NON-NLS-1$
	messageLabel.setForegroundColor(ColorConstants.black);
	}
	}
	}

	protected List getErrors() {
	IMessageReference messageRef = (IMessageReference) getModel();
	return getDiagnosticMessageType(messageRef.getDiagnosticMessages(), ModelDiagnosticInfo.ERROR_TYPE);
	}

	protected List getWarnings() {
	IMessageReference messageRef = (IMessageReference) getModel();
	return getDiagnosticMessageType(messageRef.getDiagnosticMessages(), ModelDiagnosticInfo.WARNING_TYPE);
	}

	protected List getDiagnosticMessageType(List diagnosticInfo, int type) {
	List listType = new ArrayList();
	for (int index = 0; index < diagnosticInfo.size(); index++) {
	ModelDiagnosticInfo info = (ModelDiagnosticInfo) diagnosticInfo.get(index);
	if (info.getType() == type) {
	listType.add(info);
	}
	}

	return listType;
	}

	protected List getModelChildren()
	{
	IMessageReference theMessage = (IMessageReference)getModel();
	return theMessage.getParameters();
	}

	public void addNotify()
	{
	InterfaceEditPart.attachToInterfaceEditPart(this, rowLayout);
	super.addNotify();
	}

	public void addFeedback() {
	label.getParent().setBackgroundColor(DesignViewGraphicsConstants.tableCellSelectionColor);
	}

	public void removeFeedback() {
	if (isReadOnly()) {
	label.getParent().setBackgroundColor(DesignViewGraphicsConstants.readOnlyMessageRefHeadingColor);
	}
	else {
	label.getParent().setBackgroundColor(ColorConstants.tooltipBackground);
	}
	}

	public EditPart getRelativeEditPart(int direction)
	{
	EditPart editPart = super.getRelativeEditPart(direction);
	if (direction == PositionConstants.SOUTH && editPart == null)
	{
	editPart = EditPartNavigationHandlerUtil.getNextInterface(this);
	}
	return editPart;
	}
	}