1.5/plugins/org.eclipse.epf.library.edit/src/org/eclipse/epf/library/edit/process/command/AssignRoleToTaskDescriptor.java - epf/org.eclipse.epf.composer - Git at Google

 //------------------------------------------------------------------------------
 // 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.ecore.EReference;
 import org.eclipse.emf.edit.provider.ITreeItemContentProvider;
 import org.eclipse.emf.edit.provider.ItemProviderAdapter;
 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.MethodConfiguration;
 import org.eclipse.epf.uma.Role;
 import org.eclipse.epf.uma.RoleDescriptor;
 import org.eclipse.epf.uma.TaskDescriptor;
 import org.eclipse.epf.uma.TeamProfile;
 import org.eclipse.epf.uma.UmaPackage;


 /**
  * Command for assign roles to task descriptor. It will assign primary, additional
  * performers and assisted by to a task descriptor
  *
  * @author Shilpa Toraskar
  * @since 1.0
  */
 public class AssignRoleToTaskDescriptor extends AddMethodElementCommand {

 	private List roles;

 	private Activity activity;

 	private TaskDescriptor taskDesc;

 	private Collection modifiedResources;

 	private Collection affectedObjects;

 	private int action;

 	List existingRoleDescList = new ArrayList();

 	List newRoleDescList = new ArrayList();

 	private HashMap map = new HashMap();

 	private MethodConfiguration config;

 	private boolean isNewRoleDescriptor = false;

 	private boolean calledForExculded = false;

 	private DescriptorPropUtil propUtil;

 	/**
 	 * Assign role to task descriptor Used for both additionally performed by
 	 * and assisted by
 	 */
 	public AssignRoleToTaskDescriptor(TaskDescriptor taskDesc, List roles,
 			int action, MethodConfiguration config) {
 		this(taskDesc, roles, action, config, false);
 	}

 	/**
 	 * Assign role to task descriptor Used for both additionally performed by
 	 * and assisted by
 	 */
 	public AssignRoleToTaskDescriptor(TaskDescriptor taskDesc, List roles,
 			int action, MethodConfiguration config, boolean calledForExculded) {
 		super(TngUtil.getOwningProcess(taskDesc));

 		this.calledForExculded = calledForExculded;
 		this.roles = roles;
 		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(roles))
 			return;

 		for (Iterator it = roles.iterator(); it.hasNext();) {
 			// boolean descExists = false;
 			Role role = (Role) it.next();

 			RoleDescriptor roleDesc = null;
 			isNewRoleDescriptor = false;
 			// check for local descriptor
 			roleDesc = (RoleDescriptor) ProcessCommandUtil.getDescriptor(role,
 					activity, config);
 			if (roleDesc == null) {
 				// check for inherited descriptor
 				roleDesc = (RoleDescriptor) ProcessCommandUtil
 						.getInheritedDescriptor(role, activity, config);
 				if (roleDesc == null) {
 					roleDesc = ProcessUtil.createRoleDescriptor(role);
 					isNewRoleDescriptor = true;

 				}
 			}
 			if (isNewRoleDescriptor)
 				newRoleDescList.add(roleDesc);
 			else
 				existingRoleDescList.add(roleDesc);

 			// get team
 			TeamProfile team = UserInteractionHelper.getTeam(activity, role);
 			if (team != null) {
 				map.put(roleDesc, team);
 			}
 		}

 		redo();
 	}

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

 		if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
 			taskDesc.getPerformedPrimarilyBy().addAll(existingRoleDescList);
 			taskDesc.getPerformedPrimarilyBy().addAll(newRoleDescList);
 		} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
 			taskDesc.getAdditionallyPerformedBy().addAll(existingRoleDescList);
 			taskDesc.getAdditionallyPerformedBy().addAll(newRoleDescList);
 		} else if (action == IActionTypeConstants.ADD_ASSISTED_BY) {
 			taskDesc.getAssistedBy().addAll(existingRoleDescList);
 			taskDesc.getAssistedBy().addAll(newRoleDescList);
 		}

 		if (ProcessUtil.isSynFree()) {
 			if (calledForExculded) {
 				List excludedList = null;
 				EReference ref = null;
 				if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
 					excludedList = taskDesc.getPerformedPrimarilyByExcluded();
 					ref = UmaPackage.eINSTANCE.getTaskDescriptor_PerformedPrimarilyBy();
 				} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
 					excludedList = taskDesc.getAdditionallyPerformedByExclude();
 					ref = UmaPackage.eINSTANCE.getTaskDescriptor_AdditionallyPerformedBy();
 				}
 				if (excludedList != null) {
 					excludedList.removeAll(roles);
 				}

 				TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(taskDesc);
 				if (greenParent != null) {
 					EReference eRef = LibraryEditUtil.getInstance().getExcludeFeature(ref);
 					List<Role> parentExecludeList = (List<Role>) greenParent.eGet(eRef);
 					for (Role role : (List<Role>) roles) {
 						propUtil.removeGreenRefDelta(taskDesc, role, eRef, true);
 						if (parentExecludeList != null && parentExecludeList.contains(role)) {
 							propUtil.addGreenRefDelta(taskDesc, role, eRef, false);
 						}
 					}
 				}
 			} else {
 				propUtil.addLocalUsingInfo(existingRoleDescList, taskDesc, getFeature(action));
 				propUtil.addLocalUsingInfo(newRoleDescList, taskDesc, getFeature(action));
 			}

 			for (RoleDescriptor rd : (List<RoleDescriptor>) newRoleDescList) {
 				propUtil.setCreatedByReference(rd, true);
 			}

 		}

 		activity.getBreakdownElements().addAll(newRoleDescList);

 		if (map != null) {
 			Set keyset = map.keySet();
 			for (Iterator itor = keyset.iterator(); itor.hasNext();) {
 				Object key = itor.next();
 				TeamProfile team = (TeamProfile) map.get(key);

 				// add to team
 				team.getTeamRoles().add((RoleDescriptor) key);
 			}
 		}


 	}

 	private EReference getFeature(int action) {
 		UmaPackage up = UmaPackage.eINSTANCE;

 		if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
 			return up.getTaskDescriptor_PerformedPrimarilyBy();
 		}

 		if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
 			return up.getTaskDescriptor_AdditionallyPerformedBy();
 		}

 		if (action == IActionTypeConstants.ADD_ASSISTED_BY) {
 			return up.getTaskDescriptor_AssistedBy();
 		}

 		return null;
 	}

 	public void undo() {

 		// basically remove from configuration if anything was added
 		super.undo();

 		if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
 			taskDesc.getPerformedPrimarilyBy().removeAll(
 					existingRoleDescList);
 			taskDesc.getPerformedPrimarilyBy().removeAll(newRoleDescList);
 		} else	if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
 			taskDesc.getAdditionallyPerformedBy().removeAll(
 					existingRoleDescList);
 			taskDesc.getAdditionallyPerformedBy().removeAll(newRoleDescList);
 		} else if (action == IActionTypeConstants.ADD_ASSISTED_BY) {
 			taskDesc.getAssistedBy().removeAll(existingRoleDescList);
 			taskDesc.getAssistedBy().removeAll(newRoleDescList);
 		}

 		if (ProcessUtil.isSynFree()) {
 			if (calledForExculded) {
 				List excludedList = null;
 				EReference ref = null;
 				if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
 					excludedList = taskDesc.getPerformedPrimarilyByExcluded();
 					ref = UmaPackage.eINSTANCE.getTaskDescriptor_PerformedPrimarilyBy();
 				} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
 					excludedList = taskDesc.getAdditionallyPerformedByExclude();
 					ref = UmaPackage.eINSTANCE.getTaskDescriptor_AdditionallyPerformedBy();
 				}
 				if (excludedList != null) {
 					excludedList.addAll(roles);
 				}

 				TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(taskDesc);
 				if (greenParent != null) {
 					EReference eRef = LibraryEditUtil.getInstance().getExcludeFeature(ref);
 					List<Role> parentExecludeList = (List<Role>) greenParent.eGet(eRef);
 					for (Role role : (List<Role>) roles) {
 						propUtil.addGreenRefDelta(taskDesc, role, eRef, true);
 						if (parentExecludeList != null && parentExecludeList.contains(role)) {
 							propUtil.removeGreenRefDelta(taskDesc, role, eRef, false);
 						}
 					}
 				}
 			} else {
 				propUtil.removeLocalUsingInfo(existingRoleDescList, taskDesc, getFeature(action));
 				propUtil.removeLocalUsingInfo(newRoleDescList, taskDesc, getFeature(action));
 			}
 		}

 		activity.getBreakdownElements().removeAll(newRoleDescList);

 		if (map != null) {
 			Set keyset = map.keySet();
 			for (Iterator itor = keyset.iterator(); itor.hasNext();) {
 				Object key = itor.next();
 				TeamProfile team = (TeamProfile) map.get(key);

 				// remove it from team
 				team.getTeamRoles().remove((RoleDescriptor) key);
 			}
 		}
 	}

 	protected boolean prepare() {
 		return true;
 	}

 	public Collection getModifiedResources() {

 		if (roles != null && !roles.isEmpty()) {
 			if (activity.eResource() != null) {
 				modifiedResources.add(activity.eResource());
 			}
 			if (taskDesc.eResource() != null) {
 				modifiedResources.add(taskDesc.eResource());
 			}
 		}
 		return modifiedResources;
 	}

 	public Collection getAffectedObjects() {
 		if (roles != null && !roles.isEmpty()) {
 			affectedObjects.add(activity);
 			affectedObjects.add(taskDesc);
 			return affectedObjects;
 		}
 		return super.getAffectedObjects();
 	}

 }
	//------------------------------------------------------------------------------
	// 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.ecore.EReference;
	import org.eclipse.emf.edit.provider.ITreeItemContentProvider;
	import org.eclipse.emf.edit.provider.ItemProviderAdapter;
	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.MethodConfiguration;
	import org.eclipse.epf.uma.Role;
	import org.eclipse.epf.uma.RoleDescriptor;
	import org.eclipse.epf.uma.TaskDescriptor;
	import org.eclipse.epf.uma.TeamProfile;
	import org.eclipse.epf.uma.UmaPackage;


	/**
	* Command for assign roles to task descriptor. It will assign primary, additional
	* performers and assisted by to a task descriptor
	*
	* @author Shilpa Toraskar
	* @since 1.0
	*/
	public class AssignRoleToTaskDescriptor extends AddMethodElementCommand {

	private List roles;

	private Activity activity;

	private TaskDescriptor taskDesc;

	private Collection modifiedResources;

	private Collection affectedObjects;

	private int action;

	List existingRoleDescList = new ArrayList();

	List newRoleDescList = new ArrayList();

	private HashMap map = new HashMap();

	private MethodConfiguration config;

	private boolean isNewRoleDescriptor = false;

	private boolean calledForExculded = false;

	private DescriptorPropUtil propUtil;

	/**
	* Assign role to task descriptor Used for both additionally performed by
	* and assisted by
	*/
	public AssignRoleToTaskDescriptor(TaskDescriptor taskDesc, List roles,
	int action, MethodConfiguration config) {
	this(taskDesc, roles, action, config, false);
	}

	/**
	* Assign role to task descriptor Used for both additionally performed by
	* and assisted by
	*/
	public AssignRoleToTaskDescriptor(TaskDescriptor taskDesc, List roles,
	int action, MethodConfiguration config, boolean calledForExculded) {
	super(TngUtil.getOwningProcess(taskDesc));

	this.calledForExculded = calledForExculded;
	this.roles = roles;
	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(roles))
	return;

	for (Iterator it = roles.iterator(); it.hasNext();) {
	// boolean descExists = false;
	Role role = (Role) it.next();

	RoleDescriptor roleDesc = null;
	isNewRoleDescriptor = false;
	// check for local descriptor
	roleDesc = (RoleDescriptor) ProcessCommandUtil.getDescriptor(role,
	activity, config);
	if (roleDesc == null) {
	// check for inherited descriptor
	roleDesc = (RoleDescriptor) ProcessCommandUtil
	.getInheritedDescriptor(role, activity, config);
	if (roleDesc == null) {
	roleDesc = ProcessUtil.createRoleDescriptor(role);
	isNewRoleDescriptor = true;

	}
	}
	if (isNewRoleDescriptor)
	newRoleDescList.add(roleDesc);
	else
	existingRoleDescList.add(roleDesc);

	// get team
	TeamProfile team = UserInteractionHelper.getTeam(activity, role);
	if (team != null) {
	map.put(roleDesc, team);
	}
	}

	redo();
	}

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

	if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
	taskDesc.getPerformedPrimarilyBy().addAll(existingRoleDescList);
	taskDesc.getPerformedPrimarilyBy().addAll(newRoleDescList);
	} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
	taskDesc.getAdditionallyPerformedBy().addAll(existingRoleDescList);
	taskDesc.getAdditionallyPerformedBy().addAll(newRoleDescList);
	} else if (action == IActionTypeConstants.ADD_ASSISTED_BY) {
	taskDesc.getAssistedBy().addAll(existingRoleDescList);
	taskDesc.getAssistedBy().addAll(newRoleDescList);
	}

	if (ProcessUtil.isSynFree()) {
	if (calledForExculded) {
	List excludedList = null;
	EReference ref = null;
	if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
	excludedList = taskDesc.getPerformedPrimarilyByExcluded();
	ref = UmaPackage.eINSTANCE.getTaskDescriptor_PerformedPrimarilyBy();
	} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
	excludedList = taskDesc.getAdditionallyPerformedByExclude();
	ref = UmaPackage.eINSTANCE.getTaskDescriptor_AdditionallyPerformedBy();
	}
	if (excludedList != null) {
	excludedList.removeAll(roles);
	}

	TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(taskDesc);
	if (greenParent != null) {
	EReference eRef = LibraryEditUtil.getInstance().getExcludeFeature(ref);
	List<Role> parentExecludeList = (List<Role>) greenParent.eGet(eRef);
	for (Role role : (List<Role>) roles) {
	propUtil.removeGreenRefDelta(taskDesc, role, eRef, true);
	if (parentExecludeList != null && parentExecludeList.contains(role)) {
	propUtil.addGreenRefDelta(taskDesc, role, eRef, false);
	}
	}
	}
	} else {
	propUtil.addLocalUsingInfo(existingRoleDescList, taskDesc, getFeature(action));
	propUtil.addLocalUsingInfo(newRoleDescList, taskDesc, getFeature(action));
	}

	for (RoleDescriptor rd : (List<RoleDescriptor>) newRoleDescList) {
	propUtil.setCreatedByReference(rd, true);
	}

	}

	activity.getBreakdownElements().addAll(newRoleDescList);

	if (map != null) {
	Set keyset = map.keySet();
	for (Iterator itor = keyset.iterator(); itor.hasNext();) {
	Object key = itor.next();
	TeamProfile team = (TeamProfile) map.get(key);

	// add to team
	team.getTeamRoles().add((RoleDescriptor) key);
	}
	}


	}

	private EReference getFeature(int action) {
	UmaPackage up = UmaPackage.eINSTANCE;

	if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
	return up.getTaskDescriptor_PerformedPrimarilyBy();
	}

	if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
	return up.getTaskDescriptor_AdditionallyPerformedBy();
	}

	if (action == IActionTypeConstants.ADD_ASSISTED_BY) {
	return up.getTaskDescriptor_AssistedBy();
	}

	return null;
	}

	public void undo() {

	// basically remove from configuration if anything was added
	super.undo();

	if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
	taskDesc.getPerformedPrimarilyBy().removeAll(
	existingRoleDescList);
	taskDesc.getPerformedPrimarilyBy().removeAll(newRoleDescList);
	} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
	taskDesc.getAdditionallyPerformedBy().removeAll(
	existingRoleDescList);
	taskDesc.getAdditionallyPerformedBy().removeAll(newRoleDescList);
	} else if (action == IActionTypeConstants.ADD_ASSISTED_BY) {
	taskDesc.getAssistedBy().removeAll(existingRoleDescList);
	taskDesc.getAssistedBy().removeAll(newRoleDescList);
	}

	if (ProcessUtil.isSynFree()) {
	if (calledForExculded) {
	List excludedList = null;
	EReference ref = null;
	if (action == IActionTypeConstants.ADD_PRIMARY_PERFORMER) {
	excludedList = taskDesc.getPerformedPrimarilyByExcluded();
	ref = UmaPackage.eINSTANCE.getTaskDescriptor_PerformedPrimarilyBy();
	} else if (action == IActionTypeConstants.ADD_ADDITIONAL_PERFORMER) {
	excludedList = taskDesc.getAdditionallyPerformedByExclude();
	ref = UmaPackage.eINSTANCE.getTaskDescriptor_AdditionallyPerformedBy();
	}
	if (excludedList != null) {
	excludedList.addAll(roles);
	}

	TaskDescriptor greenParent = (TaskDescriptor) propUtil.getGreenParentDescriptor(taskDesc);
	if (greenParent != null) {
	EReference eRef = LibraryEditUtil.getInstance().getExcludeFeature(ref);
	List<Role> parentExecludeList = (List<Role>) greenParent.eGet(eRef);
	for (Role role : (List<Role>) roles) {
	propUtil.addGreenRefDelta(taskDesc, role, eRef, true);
	if (parentExecludeList != null && parentExecludeList.contains(role)) {
	propUtil.removeGreenRefDelta(taskDesc, role, eRef, false);
	}
	}
	}
	} else {
	propUtil.removeLocalUsingInfo(existingRoleDescList, taskDesc, getFeature(action));
	propUtil.removeLocalUsingInfo(newRoleDescList, taskDesc, getFeature(action));
	}
	}

	activity.getBreakdownElements().removeAll(newRoleDescList);

	if (map != null) {
	Set keyset = map.keySet();
	for (Iterator itor = keyset.iterator(); itor.hasNext();) {
	Object key = itor.next();
	TeamProfile team = (TeamProfile) map.get(key);

	// remove it from team
	team.getTeamRoles().remove((RoleDescriptor) key);
	}
	}
	}

	protected boolean prepare() {
	return true;
	}

	public Collection getModifiedResources() {

	if (roles != null && !roles.isEmpty()) {
	if (activity.eResource() != null) {
	modifiedResources.add(activity.eResource());
	}
	if (taskDesc.eResource() != null) {
	modifiedResources.add(taskDesc.eResource());
	}
	}
	return modifiedResources;
	}

	public Collection getAffectedObjects() {
	if (roles != null && !roles.isEmpty()) {
	affectedObjects.add(activity);
	affectedObjects.add(taskDesc);
	return affectedObjects;
	}
	return super.getAffectedObjects();
	}

	}