지니가 GEF를 만났을때4 (Connection)


package net.jlab.get.example.editor.model;





public class SimpleConnection extends SimpleModel {

  public static final int SOLID_CONNECTION = 1;

  public static final int DASHED_CONNECTION = 2;



  private SimpleModel startPoint;

  private SimpleModel endPoint;

  private boolean connected;



  public SimpleModel getStartPoint() {

    return startPoint;

  }



  public void setStartPoint(SimpleModel startPoint) {

    this.startPoint = startPoint;

  }



  public SimpleModel getEndPoint() {

    return endPoint;

  }



  public void setEndPoint(SimpleModel endPoint) {

    this.endPoint = endPoint;

  }



  public boolean isConnected() {

    return connected;

  }



  public void setConnected(boolean connected) {

    this.connected = connected;

  }

  

  public void disconnect() {

    if (connected) {

      startPoint.removeConnection(this);

      endPoint.removeConnection(this);

      setConnected(false);

    }

  }

  public void reconnect() {

    if (!connected) {

      startPoint.addConnection(this);

      endPoint.addConnection(this);

      setConnected(true);

    }

  }



  public void reconnect(SimpleModel start, SimpleModel end) {

    if (start == null || end == null || start == end) {

      throw new IllegalArgumentException();

    }

    disconnect();

    this.startPoint = start;

    this.endPoint = end;

    reconnect();

  }



}


package net.jlab.get.example.editor.model;



import java.beans.PropertyChangeListener;

import java.beans.PropertyChangeSupport;

import java.util.ArrayList;

import java.util.List;



import org.eclipse.draw2d.geometry.Dimension;

import org.eclipse.draw2d.geometry.Point;



public class SimpleModel {

  /** Property ID to use when the location of this shape is modified. */

  public static final String LOCATION_PROP = "Simple.Location";

  /** Property ID to use then the size of this shape is modified. */

  public static final String SIZE_PROP = "Simple.Size";

  /** Property ID to use when the list of outgoing connections is modified. */

  public static final String SOURCE_CONNECTIONS_PROP = "Simple.SourceConn";

  /** Property ID to use when the list of incoming connections is modified. */

  public static final String TARGET_CONNECTIONS_PROP = "Simple.TargetConn";



  private int type;



  public void setType(int type) {

    this.type = type;

  }



  public int getType() {

    return type;

  }



  private String content;



  public String getContent() {

    return content;

  }



  public void setContent(String content) {

    this.content = content;

  }



  /** Location of this shape. */

  private Point location = new Point(0, 0);

  /** Size of this shape. */

  private Dimension size = new Dimension(50, 50);



  public Dimension getSize() {

    return size;

  }



  public void setSize(Dimension size) {

    this.size = size;

    firePropertyChange(SIZE_PROP, null, size);



  }



  public Point getLocation() {

    return location;

  }



  public void setLocation(Point location) {

    this.location = location;

    firePropertyChange(LOCATION_PROP, null, location);



  }



  private transient PropertyChangeSupport pcsDelegate = new PropertyChangeSupport(

      this);



  /**

   * Attach a non-null PropertyChangeListener to this object.

   * 

   * @param l

   *            a non-null PropertyChangeListener instance

   * @throws IllegalArgumentException

   *             if the parameter is null

   */

  public synchronized void addPropertyChangeListener(PropertyChangeListener l) {

    if (l == null) {

      throw new IllegalArgumentException();

    }

    pcsDelegate.addPropertyChangeListener(l);

  }



  /**

   * Remove a PropertyChangeListener from this component.

   * 

   * @param l

   *            a PropertyChangeListener instance

   */

  public synchronized void removePropertyChangeListener(

      PropertyChangeListener l) {

    if (l != null) {

      pcsDelegate.removePropertyChangeListener(l);

    }

  }



  /**

   * Report a property change to registered listeners (for example edit

   * parts).

   * 

   * @param property

   *            the programmatic name of the property that changed

   * @param oldValue

   *            the old value of this property

   * @param newValue

   *            the new value of this property

   */

  protected void firePropertyChange(String property, Object oldValue,

      Object newValue) {

    if (pcsDelegate.hasListeners(property)) {

      pcsDelegate.firePropertyChange(property, oldValue, newValue);

    }

  }



  /** List of outgoing Connections. */

  private List<SimpleConnection> sourceConnections = new ArrayList<SimpleConnection>();



  public List<SimpleConnection> getSourceConnections() {

    return sourceConnections;

  }



  public List<SimpleConnection> getTargetConnections() {

    return targetConnections;

  }



  /** List of incoming Connections. */

  private List<SimpleConnection> targetConnections = new ArrayList<SimpleConnection>();



  void addConnection(SimpleConnection simpleConnection) {

    if (simpleConnection == null) {

      throw new IllegalArgumentException();

    }

    if (simpleConnection.getStartPoint() == this) {

      sourceConnections.add(simpleConnection);

      firePropertyChange(SOURCE_CONNECTIONS_PROP, null, simpleConnection);

    } else if (simpleConnection.getEndPoint() == this) {

      targetConnections.add(simpleConnection);

      firePropertyChange(TARGET_CONNECTIONS_PROP, null, simpleConnection);

    }

  }



  public void removeConnection(SimpleConnection simpleConnection) {

    if (simpleConnection == null) {

      throw new IllegalArgumentException();

    }

    if (simpleConnection.getStartPoint() == this) {

      sourceConnections.remove(simpleConnection);

      firePropertyChange(SOURCE_CONNECTIONS_PROP, null, simpleConnection);

    } else if (simpleConnection.getEndPoint() == this) {

      targetConnections.remove(simpleConnection);

      firePropertyChange(TARGET_CONNECTIONS_PROP, null, simpleConnection);

    }



  }

}

SimpleModel.java

실제 추가한 내용은 이곳 부터 이다. sourceConnection은 startPoint가 자신인 Connecion을 뜻 한다. 반대로 targetConnection은 자신이 endPoint인 Connecion을 뜻 한다. 왜 이 필드가 필요할까 생각했었는데 GEF의 Connection refresh 메카니즘 상 필요하다. 이 부분의 코드가 없으면 나중에 SimpleModelEditPart에서 별도의 분류 작업을 하는 코드를 삽입해야한다. 다음 부분들은 자세히 볼 부분은 없고 SOURCE_CONNECTIONS_PROP와 TARGET_CONNECTIONS_PROP가 있는데 이것은 어떤 Component를 refresh할것인가 문제 때문에 존재 한다.


package net.jlab.get.example.editor;



import net.jlab.get.example.editor.model.SimpleConnection;

import net.jlab.get.example.editor.model.SimpleModel;



import org.eclipse.gef.palette.CombinedTemplateCreationEntry;

import org.eclipse.gef.palette.ConnectionCreationToolEntry;

import org.eclipse.gef.palette.MarqueeToolEntry;

import org.eclipse.gef.palette.PaletteContainer;

import org.eclipse.gef.palette.PaletteDrawer;

import org.eclipse.gef.palette.PaletteEntry;

import org.eclipse.gef.palette.PaletteRoot;

import org.eclipse.gef.palette.PaletteToolbar;

import org.eclipse.gef.palette.PanningSelectionToolEntry;

import org.eclipse.gef.palette.ToolEntry;

import org.eclipse.gef.requests.CreationFactory;

import org.eclipse.gef.requests.SimpleFactory;

import org.eclipse.jface.resource.ImageDescriptor;



final class SimplePaletteFactory {

  static PaletteRoot createPalette() {

    PaletteRoot palette = new PaletteRoot();

    palette.add(createToolsGroup(palette));

    palette.add(createShapesDrawer());

    return palette;

  }



  private static PaletteEntry createToolsGroup(PaletteRoot palette) {

    PaletteToolbar toolbar = new PaletteToolbar("Tools");



    // Add a selection tool to the group

    ToolEntry tool = new PanningSelectionToolEntry();

    toolbar.add(tool);

    palette.setDefaultEntry(tool);

    

    // Add a marquee tool to the group

    toolbar.add(new MarqueeToolEntry());

    

    // Add (solid-line) connection tool 

    tool = new ConnectionCreationToolEntry(

        "Solid connection",

        "Create a solid-line connection",

        new CreationFactory() {

          public Object getNewObject() { return null; }

          // see ShapeEditPart#createEditPolicies() 

          // this is abused to transmit the desired line style 

          public Object getObjectType() { return SimpleConnection.SOLID_CONNECTION; }

        },

        ImageDescriptor.createFromFile(SimpleEditor.class, "icons/connection_s16.gif"),

        ImageDescriptor.createFromFile(SimpleEditor.class, "icons/connection_s24.gif"));

    toolbar.add(tool);

    

    // Add (dashed-line) connection tool

    tool = new ConnectionCreationToolEntry(

        "Dashed connection",

        "Create a dashed-line connection",

        new CreationFactory() {

          public Object getNewObject() { return null; }

          // see ShapeEditPart#createEditPolicies()

          // this is abused to transmit the desired line style 

          public Object getObjectType() { return SimpleConnection.DASHED_CONNECTION; }

        },

        ImageDescriptor.createFromFile(SimpleEditor.class, "icons/connection_d16.gif"),

        ImageDescriptor.createFromFile(SimpleEditor.class, "icons/connection_d24.gif"));

    toolbar.add(tool);

    

    return toolbar;

  }

  

  private static PaletteContainer createShapesDrawer() {

    PaletteDrawer componentsDrawer = new PaletteDrawer("Simple");



    CombinedTemplateCreationEntry component = new CombinedTemplateCreationEntry(

        "Ellipse", 

        "Create an elliptical shape", 

        SimpleModel.class,

        new SimpleFactory(SimpleModel.class), 

        ImageDescriptor.createFromFile(SimpleEditor.class, "icons/ellipse16.gif"), 

        ImageDescriptor.createFromFile(SimpleEditor.class, "icons/ellipse24.gif"));

    componentsDrawer.add(component);



    return componentsDrawer;

  }

}

SimplePaletteFactory.java

이 부분도 딱히 설명할 부분은 없다. 코드가 추가된 부분은 부분이다. 그냥 ConnecionToolEntry는 이렇게 추가하는구나 라고 받아들이자. getObjectType() 부분만 일단 봐두자 나중에 연결된 코드가 있다.


public class SimpleEditPartFactory implements EditPartFactory {



  @Override

  public EditPart createEditPart(EditPart context, Object model) {

    EditPart part = null;



    if (model instanceof SimpleConnection) {

      part = new SimpleConnectionEditPart();

      part.setModel(model);

      return part;

    }

    if (model instanceof SimpleDiagram) {

      part = new SimpleDiagramEditPart();

      part.setModel(model);

      return part;

    }

    if (model instanceof SimpleModel) {

      part = new SimpleModelEditPart();

      part.setModel(model);

      return part;

    }

    part.setModel(model);

    return part;

  }



}

SimpleEditPartFactory.java


package net.jlab.get.example.editor.edit;



import net.jlab.get.example.editor.command.SimpleConnectionDeleteCommand;

import net.jlab.get.example.editor.model.SimpleConnection;



import org.eclipse.draw2d.Graphics;

import org.eclipse.draw2d.IFigure;

import org.eclipse.draw2d.PolygonDecoration;

import org.eclipse.draw2d.PolylineConnection;

import org.eclipse.gef.EditPolicy;

import org.eclipse.gef.commands.Command;

import org.eclipse.gef.editparts.AbstractConnectionEditPart;

import org.eclipse.gef.editpolicies.ConnectionEditPolicy;

import org.eclipse.gef.editpolicies.ConnectionEndpointEditPolicy;

import org.eclipse.gef.requests.GroupRequest;



public class SimpleConnectionEditPart extends AbstractConnectionEditPart {



  /*

   * (non-Javadoc)

   * 

   * @see org.eclipse.gef.editparts.AbstractEditPart#createEditPolicies()

   */

  protected void createEditPolicies() {

    installEditPolicy(EditPolicy.CONNECTION_ENDPOINTS_ROLE,

        new ConnectionEndpointEditPolicy());

    // Allows the removal of the connection model element

    installEditPolicy(EditPolicy.CONNECTION_ROLE,

        new ConnectionEditPolicy() {

          protected Command getDeleteCommand(GroupRequest request) {

            return new SimpleConnectionDeleteCommand(

                getCastedModel());

          }

        });

  }



  /*

   * (non-Javadoc)

   * 

   * @see org.eclipse.gef.editparts.AbstractGraphicalEditPart#createFigure()

   */

  protected IFigure createFigure() {

    PolylineConnection connection = (PolylineConnection) super

        .createFigure();

    connection.setTargetDecoration(new PolygonDecoration());

    connection.setLineStyle(getCastedModel().getType());

    if (getCastedModel().getType() == SimpleConnection.SOLID_CONNECTION)

      connection.setLineStyle(Graphics.LINE_SOLID);

    if (getCastedModel().getType() == SimpleConnection.DASHED_CONNECTION)

      connection.setLineStyle(Graphics.LINE_DASH);

    return connection;

  }



  private SimpleConnection getCastedModel() {

    return (SimpleConnection) getModel();

  }



}


package net.jlab.get.example.editor.edit;



import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.util.List;



import net.jlab.get.example.editor.command.SimpleConnectionCreateCommand;

import net.jlab.get.example.editor.command.SimpleConnectionReconnectCommand;

import net.jlab.get.example.editor.model.SimpleConnection;

import net.jlab.get.example.editor.model.SimpleModel;



import org.eclipse.draw2d.ChopboxAnchor;

import org.eclipse.draw2d.ColorConstants;

import org.eclipse.draw2d.ConnectionAnchor;

import org.eclipse.draw2d.Ellipse;

import org.eclipse.draw2d.IFigure;

import org.eclipse.draw2d.RectangleFigure;

import org.eclipse.draw2d.geometry.Rectangle;

import org.eclipse.gef.ConnectionEditPart;

import org.eclipse.gef.EditPolicy;

import org.eclipse.gef.GraphicalEditPart;

import org.eclipse.gef.NodeEditPart;

import org.eclipse.gef.Request;

import org.eclipse.gef.commands.Command;

import org.eclipse.gef.editparts.AbstractGraphicalEditPart;

import org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy;

import org.eclipse.gef.requests.CreateConnectionRequest;

import org.eclipse.gef.requests.ReconnectRequest;



public class SimpleModelEditPart extends AbstractGraphicalEditPart implements

    NodeEditPart, PropertyChangeListener {



  private ConnectionAnchor anchor;



  @Override

  /*

   * (non-Javadoc)

   * 

   * @see org.eclipse.gef.editparts.AbstractGraphicalEditPart#createFigure()

   */

  protected IFigure createFigure() {

    SimpleModel model = (SimpleModel) this.getModel();

    IFigure f = null;

    if (model.getType() == 1) {

      f = new Ellipse();

      f.setOpaque(true); // non-transparent figure

      f.setBackgroundColor(ColorConstants.green);

    }



    if (model.getType() != 2) {

      f = new RectangleFigure();

      f.setOpaque(true); // non-transparent figure

      f.setBackgroundColor(ColorConstants.yellow);

      f.setLocation(model.getLocation());

      f.setSize(50, 50);

    }

    return f;

  }



  /*

   * (non-Javadoc)

   * 

   * @seejava.beans.PropertyChangeListener#propertyChange(java.beans.

   * PropertyChangeEvent)

   */

  public void propertyChange(PropertyChangeEvent evt) {

    String prop = evt.getPropertyName();

    if (SimpleModel.SIZE_PROP.equals(prop)

        || SimpleModel.LOCATION_PROP.equals(prop)) {

      refreshVisuals();

    } else if (SimpleModel.SOURCE_CONNECTIONS_PROP.equals(prop)) {

      refreshSourceConnections();

    } else if (SimpleModel.TARGET_CONNECTIONS_PROP.equals(prop)) {

      refreshTargetConnections();

    }

  }



  /**

   * Upon activation, attach to the model element as a property change

   * listener.

   */

  public void activate() {

    if (!isActive()) {

      super.activate();

      ((SimpleModel) getModel()).addPropertyChangeListener(this);

    }

  }



  /**

   * Upon deactivation, detach from the model element as a property change

   * listener.

   */

  public void deactivate() {

    if (isActive()) {

      super.deactivate();

      ((SimpleModel) getModel()).removePropertyChangeListener(this);

    }

  }



  private SimpleModel getCastedModel() {

    return (SimpleModel) getModel();

  }



  protected void refreshVisuals() {

    // notify parent container of changed position & location

    // if this line is removed, the XYLayoutManager used by the parent

    // container

    // (the Figure of the ShapesDiagramEditPart), will not know the bounds

    // of this figure

    // and will not draw it correctly.

    Rectangle bounds = new Rectangle(getCastedModel().getLocation(),

        getCastedModel().getSize());

    ((GraphicalEditPart) getParent()).setLayoutConstraint(this,

        getFigure(), bounds);

  }



  @Override

  protected void createEditPolicies() {

    installEditPolicy(EditPolicy.COMPONENT_ROLE,

        new SimpleModelEditPolicy());



    installEditPolicy(EditPolicy.GRAPHICAL_NODE_ROLE,

        new GraphicalNodeEditPolicy() {



          @Override

          protected Command getConnectionCompleteCommand(

              CreateConnectionRequest request) {

            SimpleConnectionCreateCommand cmd = (SimpleConnectionCreateCommand) request

                .getStartCommand();

            cmd.setEndPoint((SimpleModel) getHost().getModel());

            return cmd;

          }



          @Override

          protected Command getConnectionCreateCommand(

              CreateConnectionRequest request) {

            SimpleModel source = (SimpleModel) getHost().getModel();

            int style = ((Integer) request.getNewObjectType())

                .intValue();

            SimpleConnectionCreateCommand cmd = new SimpleConnectionCreateCommand(

                source, style);

            request.setStartCommand(cmd);

            return cmd;

          }



          @Override

          protected Command getReconnectSourceCommand(

              ReconnectRequest request) {

            SimpleConnection conn = (SimpleConnection) request

                .getConnectionEditPart().getModel();

            SimpleModel newSource = (SimpleModel) getHost()

                .getModel();

            SimpleConnectionReconnectCommand cmd = new SimpleConnectionReconnectCommand(

                conn);

            cmd.setNewSource(newSource);

            return cmd;

          }



          @Override

          protected Command getReconnectTargetCommand(

              ReconnectRequest request) {

            SimpleConnection conn = (SimpleConnection) request

                .getConnectionEditPart().getModel();

            SimpleModel newTarget = (SimpleModel) getHost()

                .getModel();

            SimpleConnectionReconnectCommand cmd = new SimpleConnectionReconnectCommand(

                conn);

            cmd.setNewTarget(newTarget);

            return cmd;

          }

        });



  }



  protected ConnectionAnchor getConnectionAnchor() {

    if (anchor == null) {

      anchor = new ChopboxAnchor(getFigure());

    }

    return anchor;

  }



  @Override

  public ConnectionAnchor getSourceConnectionAnchor(

      ConnectionEditPart connection) {

    if (anchor == null) {

      anchor = new ChopboxAnchor(getFigure());

    }

    return anchor;

  }



  @Override

  public ConnectionAnchor getSourceConnectionAnchor(Request request) {

    return getConnectionAnchor();

  }



  @Override

  public ConnectionAnchor getTargetConnectionAnchor(

      ConnectionEditPart connection) {

    return getConnectionAnchor();

  }



  @Override

  public ConnectionAnchor getTargetConnectionAnchor(Request request) {

    return getConnectionAnchor();

  }



  /*

   * (non-Javadoc)

   * 

   * @see

   * org.eclipse.gef.editparts.AbstractGraphicalEditPart#getModelSourceConnections

   * ()

   */

  protected List getModelSourceConnections() {

    return getCastedModel().getSourceConnections();

  }



  /*

   * (non-Javadoc)

   * 

   * @see

   * org.eclipse.gef.editparts.AbstractGraphicalEditPart#getModelTargetConnections

   * ()

   */

  protected List getModelTargetConnections() {

    return getCastedModel().getTargetConnections();

  }



}

Model Section에서 설명한 Refresh 메카니즘과 연결해서 생각하면 된다.

Connection에서 delete를 제외한 필요한 명령어 집합이다.


package net.jlab.draw2d.figure.sequence;



import org.eclipse.draw2d.AbstractConnectionAnchor;

import org.eclipse.draw2d.IFigure;

import org.eclipse.draw2d.geometry.Point;

import org.eclipse.draw2d.geometry.Rectangle;



public class MessageOutgoingConnectionAnchor extends AbstractConnectionAnchor {



  public MessageOutgoingConnectionAnchor(IFigure owner) {

    super(owner);

  }



  Point result = new Point();



  @Override

  public Point getLocation(Point reference) {

     System.out.println("Out "+reference.x + " " + reference.y);

    Rectangle rectangle = this.getOwner().getClientArea();

    if(reference.x<rectangle.x){

      result.x = rectangle.x;

      result.y = rectangle.y + rectangle.height;

    }else{

      result.x = rectangle.x + rectangle.width;

      result.y = rectangle.y + rectangle.height;

    }

      

    

    return result;