GenericCopyAction.java example

Explorer
Henshin-Editor-master
/*******************************************************************************
 * Copyright (c) 2010-2015 Henshin developers. 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:
 *     TU Berlin, University of Luxembourg, SES S.A.
 *******************************************************************************/
package de.tub.tfs.muvitor.actions;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.util.EcoreUtil;
import org.eclipse.gef.ConnectionEditPart;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.ui.actions.Clipboard;
import org.eclipse.gef.ui.actions.SelectionAction;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.actions.ActionFactory;

/**
 * This action copies any kind model in form of a serialized String into the
 * system's clipboard. ConnectionEditParts are always copied with their source
 * and targets.
 * 
 * @author Tony Modica
 */
public class GenericCopyAction extends SelectionAction {
	static List<EClass> allowedPasteTargetEClasses = Collections.emptyList();
	private EClass currentContainerEClass;
	final Collection<EObject> selection = new HashSet<EObject>();
	
	private static HashSet<CopyPostSelectStep> postSelect = new HashSet<CopyPostSelectStep>();
	
	
	public interface CopyPostSelectStep {
		public void postProcessSelection(Collection<EObject> selection);		
	}
	
	/**
	 *
	 */
	public GenericCopyAction(final IWorkbenchPart part) {
		super(part);
		setId(ActionFactory.COPY.getId());
		super.setText("Copy");
		super.setDescription("Copy parts to clipboard");
		super.setToolTipText("Copies the selected parts to the clipboard");
	}
	
	@Override
	public boolean calculateEnabled() {
		final List<?> selectedObjects = getSelectedObjects();
		if (selectedObjects == null || selectedObjects.isEmpty()) {
			return false;
		}
		selection.clear();
		currentContainerEClass = null;
		
		List<Object> obj = new LinkedList<Object>(selectedObjects);
		for (Iterator<Object> iterator = obj.iterator(); iterator.hasNext();) {
			Object o = iterator.next();
			if (o instanceof EditPart && ((EditPart)o).getModel() instanceof EObject){
				EObject container = ((EObject) ((EditPart)o).getModel()).eContainer();
				for (Object sel : selectedObjects) {
					if (sel instanceof EditPart && ((EditPart)sel).getModel().equals(container) ){
						iterator.remove();
						break;
					}						
				}					
			}
		}
		
		for (final Object element : obj) {
			if (!(element instanceof EditPart)) {
				return false;
			}
			final EditPart editpart = (EditPart) element;
			if (!(editpart.getModel() instanceof EObject)) {
				return false;
			}
			final EObject model = (EObject) ((EditPart) element).getModel();
			final EObject container = model.eContainer();
			// we do not allow copying of the top element
			if (container == null) {
				return false;
			}
			final EClass containerEClass = container.eClass();
			if (currentContainerEClass != null && currentContainerEClass != containerEClass) {
				// only copy elements that can put into the same parent
					
				return false;
			}
			// store container ID
			currentContainerEClass = containerEClass;
			
			// also copy sources and targets for connections
			if (editpart instanceof ConnectionEditPart) {
				final ConnectionEditPart connection = (ConnectionEditPart) editpart;
				EditPart temp = connection.getSource();
				if (temp != null) {
					selection.add((EObject) temp.getModel());
				}
				temp = connection.getTarget();
				if (temp != null) {
					selection.add((EObject) temp.getModel());
				}
			}
			
			selection.add(model);
		}
		for (CopyPostSelectStep step : postSelect) {
			step.postProcessSelection(selection);
		}
		return !selection.isEmpty();
	}
	
	/**
	 * just serialize the source...
	 */
	@Override
	public void run() {
		final Collection<EObject> copyCollection = EcoreUtil.copyAll(selection);
		Clipboard.getDefault().setContents(copyCollection);
		final List<EClass> newTargetEClasses = new ArrayList<EClass>(
				currentContainerEClass.getEAllSuperTypes());
		newTargetEClasses.add(currentContainerEClass);
		allowedPasteTargetEClasses = Collections.unmodifiableList(newTargetEClasses);
	}
	
	
	public static void registerPostSelectionStep(CopyPostSelectStep step) {
		postSelect.add(step);
	}
	
	
}