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

/*******************************************************************************
 * 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.library.edit.process.command;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import org.eclipse.emf.ecore.EReference;
import org.eclipse.epf.library.edit.LibraryEditPlugin;
import org.eclipse.epf.library.edit.uma.ExtendReferenceMap;
import org.eclipse.epf.library.edit.util.DescriptorPropUtil;
import org.eclipse.epf.library.edit.util.PracticePropUtil;
import org.eclipse.epf.library.edit.util.ProcessUtil;
import org.eclipse.epf.library.edit.util.TngUtil;
import org.eclipse.epf.uma.Activity;
import org.eclipse.epf.uma.BreakdownElement;
import org.eclipse.epf.uma.Checklist;
import org.eclipse.epf.uma.Concept;
import org.eclipse.epf.uma.Descriptor;
import org.eclipse.epf.uma.EstimationConsiderations;
import org.eclipse.epf.uma.Example;
import org.eclipse.epf.uma.Guidance;
import org.eclipse.epf.uma.Guideline;
import org.eclipse.epf.uma.Practice;
import org.eclipse.epf.uma.Report;
import org.eclipse.epf.uma.ReusableAsset;
import org.eclipse.epf.uma.Roadmap;
import org.eclipse.epf.uma.SupportingMaterial;
import org.eclipse.epf.uma.TaskDescriptor;
import org.eclipse.epf.uma.Template;
import org.eclipse.epf.uma.ToolMentor;
import org.eclipse.epf.uma.UmaPackage;

/**
 * Command to add guidance to breakdown element/activity
 *  
 * @author Shilpa Toraskar
 * @since 1.5
 */
public class AddGuidanceToBreakdownElementCommand extends AddMethodElementCommand {

	private List<Guidance> guidances;

	private BreakdownElement brElement;

	private Collection modifiedResources;
	
	private boolean calledForExculded = false;
	
	private Set<Practice> affectedSet = new HashSet<Practice>();
	
	private DescriptorPropUtil propUtil = DescriptorPropUtil.getDesciptorPropUtil();
	
	public AddGuidanceToBreakdownElementCommand(BreakdownElement brElement, List<Guidance> guidances) {
		this(brElement, guidances, false);
	}

	public AddGuidanceToBreakdownElementCommand(BreakdownElement brElement, List<Guidance> guidances, boolean calledForExculded) {

		super(TngUtil.getOwningProcess(brElement));

		this.guidances = guidances;
		this.brElement = brElement;
		this.calledForExculded = calledForExculded;

		this.modifiedResources = new HashSet();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.Command#execute()
	 */
	public void execute() {

		// add to default configuration if not there already
		if (!super.addToDefaultConfiguration(guidances))
			return;

		redo();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.Command#redo()
	 */
	public void redo() {

		List<Practice> utdItems = new ArrayList<Practice>();
		if (!guidances.isEmpty()) {
			for (Iterator it = guidances.iterator(); it.hasNext();) {
				Object obj = it.next();
				if (obj instanceof Guidance) {
					Guidance item = (Guidance) obj;

					// guidances for breakdown element
					if (item instanceof Checklist) {
						brElement.getChecklists().add((Checklist) item);
					} else if (item instanceof Concept) {
						brElement.getConcepts().add((Concept) item);
					} else if (item instanceof Example) {
						brElement.getExamples().add((Example) item);
					} else if (item instanceof SupportingMaterial) {
						brElement.getSupportingMaterials().add((SupportingMaterial) item);
					} else if (item instanceof Guideline) {
						brElement.getGuidelines().add((Guideline) item);
					} else if (item instanceof ReusableAsset) {
						brElement.getReusableAssets().add((ReusableAsset) item);
					} else if (item instanceof Template) {
						brElement.getTemplates().add((Template) item);
					} else if (item instanceof Report) {
						brElement.getReports().add((Report) item);
					} else if (item instanceof EstimationConsiderations) {
						brElement.getEstimationconsiderations().add((EstimationConsiderations) item);
					} else if (item instanceof ToolMentor) {
						brElement.getToolmentor().add((ToolMentor) item);
					} else if (item instanceof Roadmap) {
						if (brElement instanceof Activity) {
							((Activity) brElement).getRoadmaps().add((Roadmap) item);
						}
					} else if (item instanceof Practice) {
						if (PracticePropUtil.getPracticePropUtil().isUdtType((Practice) item)) {
							utdItems.add((Practice) item);
						}
					} else {
						LibraryEditPlugin.getDefault().getLogger()
								.logError("Cant set guidance " + item.getType().getName() + ":" + item.getName()); //$NON-NLS-1$ //$NON-NLS-2$
					}
				}
			}
			
			List<Practice> listValue = propUtil.getUdtList(brElement, true);
			if (listValue != null && ! utdItems.isEmpty()) {
				for (Practice p : utdItems) {
					if (! listValue.contains(p) && listValue.add(p)) {
						affectedSet.add(p);
						propUtil.addOpposite(ExtendReferenceMap.UdtList, brElement, p);
					}
				}
				try {
					propUtil.storeReferences(brElement, false);					
				} catch (Exception e) {
					LibraryEditPlugin.getDefault().getLogger().logError(e);
				}	
			}
			
			if (propUtil.isDescriptor(brElement) && ProcessUtil.isSynFree()) {
				if (calledForExculded) {
					((Descriptor)brElement).getGuidanceExclude().removeAll(guidances);
					
					if (brElement instanceof TaskDescriptor) {
						TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(
								(TaskDescriptor)brElement);						 
						if (greenParent != null) {
							EReference ref = null;
							EReference eRef = UmaPackage.eINSTANCE.getDescriptor_GuidanceExclude();
							List<Guidance> parentExecludeList = (List<Guidance>) greenParent.eGet(eRef);
							for (Guidance guidance : (List<Guidance>) guidances) {
								ref = propUtil.getGuidanceEReference(guidance);								
								propUtil.removeGreenRefDelta((TaskDescriptor)brElement, guidance, eRef, true);
								if (parentExecludeList != null && parentExecludeList.contains(guidance)) {
									propUtil.addGreenRefDelta((TaskDescriptor)brElement, guidance, eRef, false);
								}
							}
						}						
					}					
				} else if (ProcessUtil.isSynFree()) {
					Descriptor des = (Descriptor) brElement;
					des.getGuidanceAdditional().addAll(guidances);
					Descriptor greenParent = propUtil.getGreenParentDescriptor(des);
					if (greenParent != null) {
						EReference aRef = UmaPackage.eINSTANCE.getDescriptor_GuidanceAdditional();
						List<Guidance> parentAdditionalList = (List<Guidance>) greenParent.eGet(aRef);
						for (Guidance guidance : (List<Guidance>) guidances) {
							propUtil.removeGreenRefDelta(des, guidance, aRef, false);
							if (parentAdditionalList == null
									|| ! parentAdditionalList.contains(guidance)) {
								propUtil.addGreenRefDelta(des, guidance, aRef, true);
							}
						}
					}
				}
			}
		}
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.emf.common.command.Command#undo()
	 */
	public void undo() {
		if (!guidances.isEmpty()) {
			// basically remove from configuration if anything was added
			super.undo();

			for (Iterator it = guidances.iterator(); it.hasNext();) {
				Object obj = it.next();
				if (obj instanceof Guidance) {
					Guidance item = (Guidance) obj;

					// guidances for content element
					if (item instanceof Checklist) {
						brElement.getChecklists().remove(item);
					} else if (item instanceof Concept) {
						brElement.getConcepts().remove(item);
					} else if (item instanceof Example) {
						brElement.getExamples().remove(item);
					} else if (item instanceof SupportingMaterial) {
						brElement.getSupportingMaterials().remove(item);
					} else if (item instanceof Guideline) {
						brElement.getGuidelines().remove(item);
					} else if (item instanceof ReusableAsset) {
						brElement.getReusableAssets().remove(item);
					} else if (item instanceof Template) {
						brElement.getTemplates().remove((Template) item);
					} else if (item instanceof Report) {
						brElement.getReports().remove((Report) item);
					} else if (item instanceof EstimationConsiderations) {
						brElement.getEstimationconsiderations().remove((EstimationConsiderations) item);
					} else if (item instanceof ToolMentor) {
						brElement.getToolmentor().remove((ToolMentor) item);
					} else if (item instanceof Roadmap) {
						if (brElement instanceof Activity) {
							((Activity) brElement).getRoadmaps().remove((Roadmap) item);
						}
					} else {
						LibraryEditPlugin.getDefault().getLogger()
								.logError("Cant set guidance " + item.getType().getName() + ":" + item.getName()); //$NON-NLS-1$ //$NON-NLS-2$
					}
				}
			}
			
			if (propUtil.isDescriptor(brElement) && ProcessUtil.isSynFree()) {
				if (calledForExculded) {
					((Descriptor)brElement).getGuidanceExclude().addAll(guidances);
					
					if (brElement instanceof TaskDescriptor) {
						TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(
								(TaskDescriptor)brElement);						 
						if (greenParent != null) {
							EReference ref = null;
							EReference eRef = UmaPackage.eINSTANCE.getDescriptor_GuidanceExclude();
							List<Guidance> parentExecludeList = (List<Guidance>) greenParent.eGet(eRef);
							for (Guidance guidance : (List<Guidance>) guidances) {
								ref = propUtil.getGuidanceEReference(guidance);								
								propUtil.removeGreenRefDelta((TaskDescriptor)brElement, guidance, eRef, true);
								if (parentExecludeList != null && parentExecludeList.contains(guidance)) {
									propUtil.addGreenRefDelta((TaskDescriptor)brElement, guidance, eRef, false);
								}
							}
						}						
					}		
				} else if (ProcessUtil.isSynFree()) {
					Descriptor des = (Descriptor) brElement;
					des.getGuidanceAdditional().removeAll(guidances);
					Descriptor greenParent = propUtil.getGreenParentDescriptor(des);
					if (greenParent != null) {
						EReference aRef = UmaPackage.eINSTANCE.getDescriptor_GuidanceAdditional();
						List<Guidance> parentAdditionalList = (List<Guidance>) greenParent.eGet(aRef);
						for (Guidance guidance : (List<Guidance>) guidances) {
							propUtil.removeGreenRefDelta(des, guidance, aRef, true);
							if (parentAdditionalList != null
									&& parentAdditionalList.contains(guidance)) {
								propUtil.addGreenRefDelta(des, guidance, aRef, false);
							}
						}
					}
				}
			}
		}
		
		try {
			for (Practice p : affectedSet) {
				propUtil.removeOpposite(ExtendReferenceMap.UdtList, brElement, p);	
			}	
			propUtil.storeReferences(brElement, true);
		} catch (Exception e) {
			LibraryEditPlugin.getDefault().getLogger().logError(e);
		}
	}

	protected boolean prepare() {
		return true;
	}

	public Collection getModifiedResources() {
		if (!guidances.isEmpty() ) {
			if (brElement.eResource() != null) {
				modifiedResources.add(brElement.eResource());
			}
		}
		return modifiedResources;
	}

	public Collection getAffectedObjects() {
		if (brElement != null) {
			return Arrays.asList(new Object[] { brElement });
		}
		return super.getAffectedObjects();
	}
}