deleteconnectioncommand.java

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

/*
 * Created on 2005-1-27
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package com.example.commands;

import org.eclipse.gef.commands.Command;

import com.example.model.Connection;
import com.example.model.Diagram;
import com.example.model.Node;

/**
 * @author zhanghao
 *
 * TODO To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
public class DeleteConnectionCommand extends Command {

	Diagram source;

	Diagram target;

    Connection connection;

    //Setters
    public void setConnection(Connection connection) {
        this.connection = connection;
    }

    public void setSource(Diagram source) {
        this.source = source;
    }

    public void setTarget(Diagram target) {
        this.target = target;
    }

    public void execute() {
        source.removeOutput(connection);
        target.removeInput(connection);
        connection.setSource(null);
        connection.setTarget(null);
    }

    public String getLabel() {
        return "Delete Connection";
    }

    public void redo() {
        execute();
    }

    public void undo() {
        connection.setSource(source);
        connection.setTarget(target);
        source.addOutput(connection);
        target.addInput(connection);
    }
}