diagramgraphicalnodeeditpolicy.java

GEF初学者可以用来对照的小例子

package com.example.policies;

import org.eclipse.gef.commands.Command;
import org.eclipse.gef.editpolicies.GraphicalNodeEditPolicy;
import org.eclipse.gef.requests.CreateConnectionRequest;
import org.eclipse.gef.requests.ReconnectRequest;

import com.example.commands.CreateConnectionCommand;
import com.example.commands.ReconnectSourceCommand;
import com.example.commands.ReconnectTargetCommand;
import com.example.model.Connection;
import com.example.model.Diagram;

public class DiagramGraphicalNodeEditPolicy extends GraphicalNodeEditPolicy {

	protected Command getConnectionCompleteCommand(CreateConnectionRequest request) {
        CreateConnectionCommand command = (CreateConnectionCommand) request.getStartCommand();
        command.setTarget((Diagram) getHost().getModel());
        return command;
    }

    protected Command getConnectionCreateCommand(CreateConnectionRequest request) {
        CreateConnectionCommand command = new CreateConnectionCommand();
        command.setSource((Diagram) getHost().getModel());
        request.setStartCommand(command);
        return command;
    }

    protected Command getReconnectSourceCommand(ReconnectRequest request) {
		ReconnectSourceCommand cmd = new ReconnectSourceCommand();
		cmd.setConnection((Connection)request.getConnectionEditPart().getModel());
		cmd.setSource((Diagram)getHost().getModel());
		return cmd;
    }

    protected Command getReconnectTargetCommand(ReconnectRequest request) {
    	ReconnectTargetCommand cmd = new ReconnectTargetCommand();
    	cmd.setConnection((Connection)request.getConnectionEditPart().getModel());
    	cmd.setTarget((Diagram)getHost().getModel());
        return cmd;
    }

}