| /******************************************************************************* |
| * Copyright (c) 2005, 2009 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 implementation |
| *******************************************************************************/ |
| package org.eclipse.epf.diagram.ui.actions; |
| |
| import java.util.List; |
| |
| import org.eclipse.gef.GraphicalViewer; |
| import org.eclipse.gmf.runtime.diagram.ui.actions.internal.ArrangeAction; |
| import org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart; |
| import org.eclipse.gmf.runtime.diagram.ui.parts.DiagramCommandStack; |
| import org.eclipse.gmf.runtime.diagram.ui.parts.IDiagramEditDomain; |
| import org.eclipse.ui.IWorkbenchPage; |
| |
| /** |
| * @author skannoor |
| * |
| */ |
| public class CustomArrangeAction extends ArrangeAction { |
| |
| |
| private DiagramEditPart diagramEditPart; |
| private GraphicalViewer viewer; |
| |
| /** |
| * @param workbenchPage |
| * @param selectionOnly |
| */ |
| public CustomArrangeAction(IWorkbenchPage workbenchPage, |
| boolean selectionOnly, DiagramEditPart diagramEditPart, GraphicalViewer viewer) { |
| super(workbenchPage, selectionOnly); |
| this.diagramEditPart = diagramEditPart; |
| this.viewer = viewer; |
| // TODO Auto-generated constructor stub |
| } |
| |
| @Override |
| protected List getSelectedObjects() { |
| return diagramEditPart.getChildren(); |
| } |
| |
| @Override |
| protected DiagramEditPart getDiagramEditPart() { |
| return diagramEditPart; |
| } |
| @Override |
| protected DiagramCommandStack getDiagramCommandStack() { |
| return new DiagramCommandStack((IDiagramEditDomain)viewer.getEditDomain()); |
| } |
| } |