DiagramMatchWidthAction.java

/*******************************************************************************
 * Copyright (c) 2007 Exadel, Inc. and Red Hat, Inc.
 * Distributed under license by Red Hat, Inc. All rights reserved.
 * This program is 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:
 *     Exadel, Inc. and Red Hat, Inc. - initial API and implementation
 ******************************************************************************/ 
package org.jboss.tools.common.gef.action;

import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbenchPart;
import org.jboss.tools.common.gef.alignment.xpl.DefaultMatchWidthAction;

/**
 * 
 * @author eskimo(dgolovin@exadel.com)
 *
 */
public class DiagramMatchWidthAction extends DefaultMatchWidthAction {

	public DiagramMatchWidthAction(IWorkbenchPart part) {
		super(part);
	}

	@Override
	public void update() {
		IEditorPart editor = (IEditorPart)getWorkbenchPart();
		if(editor == null) return;
		IDiagramSelectionProvider provider = (IDiagramSelectionProvider)editor.getAdapter(IDiagramSelectionProvider.class);
		if(provider == null) return;
		setSelection(provider.getSelection());
	}

}