tests/org.eclipse.emf.ecoretools.tests.diagram.ui.core/src/org/eclipse/emf/ecoretools/tests/diagram/ui/core/bypassedclasses/DeleteFromDiagramBypassedAction.java

/***********************************************************************
 * Copyright (c) 2008 Anyware Technologies
 * 
 * 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:
 *    Anyware Technologies - initial API and implementation
 **********************************************************************/

package org.eclipse.emf.ecoretools.tests.diagram.ui.core.bypassedclasses;

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

import org.eclipse.gef.EditPart;
import org.eclipse.gef.commands.Command;
import org.eclipse.gmf.runtime.diagram.ui.actions.internal.DeleteFromDiagramAction;
import org.eclipse.ui.IWorkbenchPart;

/**
 * A DeleteFromDiagramAction bypassed for create the right command
 * 
 * @author Simon Bernard
 */
public class DeleteFromDiagramBypassedAction extends DeleteFromDiagramAction {

	public List<EditPart> partsToDelete;

	public DeleteFromDiagramBypassedAction(IWorkbenchPart part, EditPart objectToDelete) {
		super(part);
		partsToDelete = new ArrayList<EditPart>();
		partsToDelete.add(objectToDelete);
	}

	public DeleteFromDiagramBypassedAction(IWorkbenchPart part, List<EditPart> objectsToDelete) {
		super(part);
		this.partsToDelete = objectsToDelete;
	}

	@Override
	protected List<EditPart> createOperationSet() {
		return partsToDelete;
	}

	@Override
	public Command getCommand() {
		// make it public
		return super.getCommand();
	}
}