//------------------------------------------------------------------------------
// 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.process.command;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import org.eclipse.emf.common.notify.AdapterFactory;
import org.eclipse.emf.common.util.AbstractTreeIterator;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.emf.edit.provider.ITreeItemContentProvider;
import org.eclipse.emf.edit.provider.ItemProviderAdapter;
import org.eclipse.epf.library.edit.Providers;
import org.eclipse.epf.library.edit.TngAdapterFactory;
import org.eclipse.epf.library.edit.ui.UserInteractionHelper;
import org.eclipse.epf.library.edit.util.DescriptorPropUtil;
import org.eclipse.epf.library.edit.util.LibraryEditUtil;
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.Artifact;
import org.eclipse.epf.uma.MethodConfiguration;
import org.eclipse.epf.uma.TaskDescriptor;
import org.eclipse.epf.uma.UmaPackage;
import org.eclipse.epf.uma.VariabilityElement;
import org.eclipse.epf.uma.WorkProduct;
import org.eclipse.epf.uma.WorkProductDescriptor;
/**
* Command to assign work products to task descriptor. It will set external
* inputs, mandatory inputs, optional inputs and outputs.
*
* @author Shilpa Toraskar
* @author Phong Nguyen Le
* @since 1.0
*/
public class AssignWPToTaskDescriptor extends AddMethodElementCommand {
private List workProducts;
private Activity activity;
private TaskDescriptor taskDesc;
private Collection modifiedResources;
private Collection affectedObjects;
private int action;
private HashMap map = new HashMap();
List existingWPDescList = new ArrayList();
List newWPDescList = new ArrayList();
Collection<WorkProductDescriptor> subartifactDescriptors = new ArrayList<WorkProductDescriptor>();
private MethodConfiguration config;
private boolean calledForExculded = false;
private DescriptorPropUtil propUtil;
public AssignWPToTaskDescriptor(TaskDescriptor taskDesc, List workProducts,
int action, MethodConfiguration config) {
this(taskDesc, workProducts, action, config, false);
}
public AssignWPToTaskDescriptor(TaskDescriptor taskDesc, List workProducts,
int action, MethodConfiguration config, boolean calledForExculded) {
super(TngUtil.getOwningProcess(taskDesc));
this.calledForExculded = calledForExculded;
this.workProducts = workProducts;
this.taskDesc = taskDesc;
this.action = action;
this.config = config;
this.propUtil = DescriptorPropUtil.getDesciptorPropUtil();
AdapterFactory aFactory = TngAdapterFactory.INSTANCE
.getWBS_ComposedAdapterFactory();
ItemProviderAdapter adapter = (ItemProviderAdapter) aFactory.adapt(
taskDesc, ITreeItemContentProvider.class);
Object parent = adapter.getParent(taskDesc);
if (parent instanceof Activity) {
this.activity = (Activity) parent;
}
this.modifiedResources = new HashSet();
this.affectedObjects = 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(workProducts))
return;
for (Iterator it = workProducts.iterator(); it.hasNext();) {
WorkProduct wp = (WorkProduct) it.next();
WorkProductDescriptor newWpDesc = null;
boolean isNewDescriptor = false;
// check for local descriptor
newWpDesc = (WorkProductDescriptor) ProcessCommandUtil
.getDescriptor(wp, activity, config);
if (newWpDesc == null) {
// check for inherited descriptor
newWpDesc = (WorkProductDescriptor) ProcessCommandUtil
.getInheritedDescriptor(wp, activity, config);
if (newWpDesc == null) {
newWpDesc = ProcessUtil.createWorkProductDescriptor(wp);
isNewDescriptor = true;
}
}
if (isNewDescriptor)
newWPDescList.add(newWpDesc);
else
existingWPDescList.add(newWpDesc);
// get deliverable
WorkProductDescriptor deliverable = UserInteractionHelper
.getDeliverable(activity, wp);
if (deliverable != null) {
map.put(newWpDesc, deliverable);
}
}
// add subartifacts to the activity's breakdown element list if there is any
//
for (Iterator<?> iter = workProducts.iterator(); iter
.hasNext();) {
Object element = iter.next();
if (element instanceof Artifact) {
Iterator<?> iterator = new AbstractTreeIterator<Object>(element, false) {
/**
* Comment for <code>serialVersionUID</code>
*/
private static final long serialVersionUID = -4820477887426087262L;
protected Iterator<?> getChildren(Object object) {
Object subArtifacts = Providers.getConfigurationApplicator().getReference(
(VariabilityElement) object,
UmaPackage.eINSTANCE.getArtifact_ContainedArtifacts(),
config);
return ((Collection)subArtifacts).iterator();
}
};
while (iterator.hasNext()) {
Artifact subArtifact = (Artifact) iterator.next();
WorkProductDescriptor newWpDesc = null;
boolean isNewDescriptor = false;
// check for local descriptor
newWpDesc = (WorkProductDescriptor) ProcessCommandUtil
.getDescriptor(subArtifact, activity, config);
if (newWpDesc == null) {
// check for inherited descriptor
newWpDesc = (WorkProductDescriptor) ProcessCommandUtil
.getInheritedDescriptor(subArtifact, activity, config);
if (newWpDesc == null) {
newWpDesc = ProcessUtil.createWorkProductDescriptor(subArtifact);
isNewDescriptor = true;
}
}
if (isNewDescriptor && !subartifactDescriptors.contains(newWpDesc)) {
subartifactDescriptors.add(newWpDesc);
}
}
}
}
redo();
}
/*
* (non-Javadoc)
*
* @see org.eclipse.emf.common.command.Command#redo()
*/
public void redo() {
if (action == IActionTypeConstants.ADD_EXTERNAL_INPUT) {
taskDesc.getExternalInput().addAll(existingWPDescList);
taskDesc.getExternalInput().addAll(newWPDescList);
} else if (action == IActionTypeConstants.ADD_MANDATORY_INPUT) {
taskDesc.getMandatoryInput().addAll(existingWPDescList);
taskDesc.getMandatoryInput().addAll(newWPDescList);
} else if (action == IActionTypeConstants.ADD_OPTIONAL_INPUT) {
taskDesc.getOptionalInput().addAll(existingWPDescList);
taskDesc.getOptionalInput().addAll(newWPDescList);
} else if (action == IActionTypeConstants.ADD_OUTPUT) {
taskDesc.getOutput().addAll(existingWPDescList);
taskDesc.getOutput().addAll(newWPDescList);
}
if (ProcessUtil.isSynFree()) {
if (calledForExculded) {
List excludedList = null;
EReference ref = null;
if (action == IActionTypeConstants.ADD_MANDATORY_INPUT) {
excludedList = taskDesc.getMandatoryInputExclude();
ref = UmaPackage.eINSTANCE.getTaskDescriptor_MandatoryInput();
} else if (action == IActionTypeConstants.ADD_OPTIONAL_INPUT) {
excludedList = taskDesc.getOptionalInputExclude();
ref = UmaPackage.eINSTANCE.getTaskDescriptor_OptionalInput();
} else if (action == IActionTypeConstants.ADD_OUTPUT) {
excludedList = taskDesc.getOutputExclude();
ref = UmaPackage.eINSTANCE.getTaskDescriptor_Output();
}
if (excludedList != null) {
excludedList.removeAll(workProducts);
}
TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(taskDesc);
if (greenParent != null) {
EReference eRef = LibraryEditUtil.getInstance().getExcludeFeature(ref);
List<WorkProduct> parentExecludeList = (List<WorkProduct>) greenParent.eGet(eRef);
for (WorkProduct wp : (List<WorkProduct>) workProducts) {
propUtil.removeGreenRefDelta(taskDesc, wp, eRef, true);
if (parentExecludeList != null && parentExecludeList.contains(wp)) {
propUtil.addGreenRefDelta(taskDesc, wp, eRef, false);
}
}
}
} else {
propUtil.addLocalUsingInfo(existingWPDescList, taskDesc, getFeature(action));
propUtil.addLocalUsingInfo(newWPDescList, taskDesc, getFeature(action));
}
for (WorkProductDescriptor rd : (List<WorkProductDescriptor>) newWPDescList) {
propUtil.setCreatedByReference(rd, true);
}
}
activity.getBreakdownElements().addAll(newWPDescList);
if(!subartifactDescriptors.isEmpty()) {
activity.getBreakdownElements().addAll(subartifactDescriptors);
}
if (map != null) {
Set keyset = map.keySet();
for (Iterator itor = keyset.iterator(); itor.hasNext();) {
Object key = itor.next();
WorkProductDescriptor wpDesc = (WorkProductDescriptor) map
.get(key);
// add to deliverable
wpDesc.getDeliverableParts().add((WorkProductDescriptor) key);
}
}
}
private EReference getFeature(int action) {
UmaPackage up = UmaPackage.eINSTANCE;
if (action == IActionTypeConstants.ADD_MANDATORY_INPUT) {
return up.getTaskDescriptor_MandatoryInput();
}
if (action == IActionTypeConstants.ADD_OPTIONAL_INPUT) {
return up.getTaskDescriptor_OptionalInput();
}
if (action == IActionTypeConstants.ADD_OUTPUT) {
return up.getTaskDescriptor_Output();
}
if (action == IActionTypeConstants.ADD_EXTERNAL_INPUT) {
return up.getTaskDescriptor_ExternalInput();
}
return null;
}
public void undo() {
// remove from configuration if anything was added
super.undo();
if (action == IActionTypeConstants.ADD_EXTERNAL_INPUT) {
taskDesc.getExternalInput().removeAll(existingWPDescList);
taskDesc.getExternalInput().removeAll(newWPDescList);
} else if (action == IActionTypeConstants.ADD_MANDATORY_INPUT) {
taskDesc.getMandatoryInput().removeAll(existingWPDescList);
taskDesc.getMandatoryInput().removeAll(newWPDescList);
} else if (action == IActionTypeConstants.ADD_OPTIONAL_INPUT) {
taskDesc.getOptionalInput().removeAll(existingWPDescList);
taskDesc.getOptionalInput().removeAll(newWPDescList);
} else if (action == IActionTypeConstants.ADD_OUTPUT) {
taskDesc.getOutput().removeAll(existingWPDescList);
taskDesc.getOutput().removeAll(newWPDescList);
}
if (ProcessUtil.isSynFree()) {
if (calledForExculded) {
List excludedList = null;
EReference ref = null;
if (action == IActionTypeConstants.ADD_MANDATORY_INPUT) {
excludedList = taskDesc.getMandatoryInputExclude();
ref = UmaPackage.eINSTANCE.getTaskDescriptor_MandatoryInput();
} else if (action == IActionTypeConstants.ADD_OPTIONAL_INPUT) {
excludedList = taskDesc.getOptionalInputExclude();
ref = UmaPackage.eINSTANCE.getTaskDescriptor_OptionalInput();
} else if (action == IActionTypeConstants.ADD_OUTPUT) {
excludedList = taskDesc.getOutputExclude();
ref = UmaPackage.eINSTANCE.getTaskDescriptor_Output();
}
if (excludedList != null) {
excludedList.addAll(workProducts);
}
TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(taskDesc);
if (greenParent != null) {
EReference eRef = LibraryEditUtil.getInstance().getExcludeFeature(ref);
List<WorkProduct> parentExecludeList = (List<WorkProduct>) greenParent.eGet(eRef);
for (WorkProduct wp : (List<WorkProduct>) workProducts) {
propUtil.addGreenRefDelta(taskDesc, wp, eRef, true);
if (parentExecludeList != null && parentExecludeList.contains(wp)) {
propUtil.removeGreenRefDelta(taskDesc, wp, eRef, false);
}
}
}
} else {
propUtil.removeLocalUsingInfo(existingWPDescList, taskDesc, getFeature(action));
propUtil.removeLocalUsingInfo(newWPDescList, taskDesc, getFeature(action));
}
}
activity.getBreakdownElements().removeAll(newWPDescList);
activity.getBreakdownElements().removeAll(subartifactDescriptors);
if (map != null) {
Set keyset = map.keySet();
for (Iterator itor = keyset.iterator(); itor.hasNext();) {
Object key = itor.next();
WorkProductDescriptor wpDesc = (WorkProductDescriptor) map
.get(key);
// remove it from deliverable
wpDesc.getDeliverableParts()
.remove((WorkProductDescriptor) key);
}
}
}
protected boolean prepare() {
return true;
}
public Collection getModifiedResources() {
if (workProducts != null && !workProducts.isEmpty()) {
if (activity.eResource() != null) {
modifiedResources.add(activity.eResource());
}
if (taskDesc.eResource() != null) {
modifiedResources.add(taskDesc.eResource());
}
}
return modifiedResources;
}
public Collection getAffectedObjects() {
if (workProducts != null && !workProducts.isEmpty()) {
affectedObjects.add(activity);
affectedObjects.add(taskDesc);
return affectedObjects;
}
return super.getAffectedObjects();
}
}