1.5/plugins/org.eclipse.epf.library.edit/src/org/eclipse/epf/library/edit/command/RemoveReferencesCommand.java

//------------------------------------------------------------------------------
// Copyright (c) 2005, 2006 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.library.edit.command;

import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;

import org.eclipse.emf.common.command.AbstractCommand;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.util.EContentsEList;
import org.eclipse.epf.uma.MethodPlugin;
import org.eclipse.epf.uma.ProcessElement;
import org.eclipse.epf.uma.UmaPackage;
import org.eclipse.epf.uma.util.UmaUtil;

/**
 * This command is used to remove method element references.
 * 
 * @author Phong Nguyen Le
 * @since 1.0
 */
public class RemoveReferencesCommand extends AbstractCommand implements
		IResourceAwareCommand {

	private MethodPlugin plugin;

	private MethodPlugin base;

	private Map<FeatureValuePair, Object> featureValuePairToObjectMap;

	private Collection<Resource> modifiedResources;

	/**
	 * Constructs a command to remove all references in <code>plugin</code>
	 * that are elements of <code>base</code>
	 * 
	 * @param plugin
	 * @param base
	 */
	public RemoveReferencesCommand(MethodPlugin plugin, MethodPlugin base) {
		super();
		this.plugin = plugin;
		this.base = base;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.AbstractCommand#prepare()
	 */
	protected boolean prepare() {
		return true;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see com.ibm.library.edit.command.IResourceAwareCommand#getModifiedResources()
	 */
	public Collection<Resource> getModifiedResources() {
		return modifiedResources;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.Command#execute()
	 */
	public void execute() {
		featureValuePairToObjectMap = new HashMap<FeatureValuePair, Object>();
		modifiedResources = new HashSet<Resource>();

		for (EContentsEList.FeatureIterator featureIterator = (EContentsEList.FeatureIterator) plugin
				.eCrossReferences().iterator(); featureIterator.hasNext();) {
			EObject ref = (EObject) featureIterator.next();
			EStructuralFeature f = featureIterator.feature();
			if (f != UmaPackage.eINSTANCE.getMethodPlugin_Bases()
					&& UmaUtil.getMethodPlugin(ref) == base) {
				featureValuePairToObjectMap.put(new FeatureValuePair(f, ref),
						plugin);
			}
		}

		for (Iterator iter = plugin.eAllContents(); iter.hasNext();) {
			EObject element = (EObject) iter.next();

			// ignore ProcessElement b/c it can references anything
			//
			if (element instanceof ProcessElement) {
				continue;
			}

			for (EContentsEList.FeatureIterator featureIterator = (EContentsEList.FeatureIterator) element
					.eCrossReferences().iterator(); featureIterator.hasNext();) {
				EObject ref = (EObject) featureIterator.next();

				if (UmaUtil.getMethodPlugin(ref) == base) {
					EStructuralFeature f = featureIterator.feature();
					featureValuePairToObjectMap.put(
							new FeatureValuePair(f, ref), element);
					if(f == UmaPackage.Literals.VARIABILITY_ELEMENT__VARIABILITY_BASED_ON_ELEMENT) {
						EStructuralFeature feature = UmaPackage.Literals.VARIABILITY_ELEMENT__VARIABILITY_TYPE;
						featureValuePairToObjectMap.put(
								new FeatureValuePair(feature, element.eGet(feature)), element);
					}
				}
			}
		}

		redo();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.Command#redo()
	 */
	public void redo() {
		for (Iterator iter = featureValuePairToObjectMap.entrySet().iterator(); iter
				.hasNext();) {
			Map.Entry entry = (Map.Entry) iter.next();
			FeatureValuePair fvPair = (FeatureValuePair) entry.getKey();
			EObject element = (EObject) entry.getValue();
			if (fvPair.feature.isMany()) {
				((Collection) element.eGet(fvPair.feature))
						.remove(fvPair.value);
			} else {
				element.eSet(fvPair.feature, null);
			}
			if (element.eResource() != null) {
				modifiedResources.add(element.eResource());
			}
		}
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.AbstractCommand#undo()
	 */
	public void undo() {
		for (Iterator iter = featureValuePairToObjectMap.entrySet().iterator(); iter
				.hasNext();) {
			Map.Entry entry = (Map.Entry) iter.next();
			FeatureValuePair fvPair = (FeatureValuePair) entry.getKey();
			EObject element = (EObject) entry.getValue();
			if (fvPair.feature.isMany()) {
				((Collection) element.eGet(fvPair.feature)).add(fvPair.value);
			} else {
				element.eSet(fvPair.feature, fvPair.value);
			}
		}
	}
	
	@Override
	public void dispose() {
		base = plugin = null;
		if(featureValuePairToObjectMap != null) {
			featureValuePairToObjectMap.clear();
		}
		if(modifiedResources != null) {
			modifiedResources.clear();
		}
		super.dispose();
	}

	private static class FeatureValuePair {
		EStructuralFeature feature;

		Object value;

		FeatureValuePair(EStructuralFeature f, Object v) {
			feature = f;
			value = v;
		}
	}
}