plugins/org.eclipse.epf.diagram/src/org/eclipse/epf/diagram/model/impl/TaskNodeImpl.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.diagram.model.impl;

import java.util.Collection;
import java.util.Iterator;

import org.eclipse.draw2d.geometry.Point;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.common.notify.NotificationChain;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.InternalEObject;
import org.eclipse.emf.ecore.util.InternalEList;
import org.eclipse.epf.diagram.model.Link;
import org.eclipse.epf.diagram.model.ModelPackage;
import org.eclipse.epf.diagram.model.Node;
import org.eclipse.epf.diagram.model.TaskNode;
import org.eclipse.epf.uma.MethodElement;
import org.eclipse.epf.uma.TaskDescriptor;
import org.eclipse.epf.uma.UmaPackage;


/**
 * <!-- begin-user-doc --> An implementation of the model object '<em><b>Task Node</b></em>'.
 * <!-- end-user-doc -->
 * <p>
 * </p>
 *
 * @generated
 */
public class TaskNodeImpl extends NamedNodeImpl implements TaskNode {
	/**
	 * <!-- begin-user-doc --> <!-- end-user-doc -->
	 */
	protected TaskNodeImpl() {
		super();

		methodElementAdapter = new TaskDescriptorAdapter();
	}

	/**
	 * <!-- begin-user-doc --> <!-- end-user-doc -->
	 * @generated
	 */
	protected EClass eStaticClass() {
		return ModelPackage.Literals.TASK_NODE;
	}

	private class TaskDescriptorAdapter extends MethodElementAdapter {
		/*
		 * (non-Javadoc)
		 * 
		 * @see org.eclipse.emf.common.notify.impl.AdapterImpl#notifyChanged(org.eclipse.emf.common.notify.Notification)
		 */
		public void notifyChanged(Notification msg) {
			if (!notificationEnabled)
				return;
			notificationEnabled = false;
			try {
				MethodElement obj;
				switch (msg.getFeatureID(TaskDescriptor.class)) {
				case UmaPackage.TASK_DESCRIPTOR__MANDATORY_INPUT:
				case UmaPackage.TASK_DESCRIPTOR__OPTIONAL_INPUT:
				case UmaPackage.TASK_DESCRIPTOR__EXTERNAL_INPUT:
					switch (msg.getEventType()) {
					case Notification.ADD:
						obj = (MethodElement) msg.getNewValue();
						addIncomingConnection(obj);
						return;
					case Notification.REMOVE:
						obj = (MethodElement) msg.getOldValue();
						removeIncomingConnection(obj);
						return;
					case Notification.ADD_MANY:
						Collection collection = (Collection) msg.getNewValue();
						for (Iterator iter = collection.iterator(); iter
								.hasNext();) {
							obj = (MethodElement) iter.next();
							addIncomingConnection(obj);
						}
						return;
					case Notification.REMOVE_MANY:
						collection = (Collection) msg.getOldValue();
						for (Iterator iter = collection.iterator(); iter
								.hasNext();) {
							obj = (MethodElement) iter.next();
							removeIncomingConnection(obj);
						}
						return;
					}
					break;
				case UmaPackage.TASK_DESCRIPTOR__OUTPUT:
					switch (msg.getEventType()) {
					case Notification.ADD:
						obj = (MethodElement) msg.getNewValue();
						addOutgoingConnection(obj);
						return;
					case Notification.REMOVE:
						obj = (MethodElement) msg.getOldValue();
						removeOutgoingConnection(obj);
						return;
					case Notification.ADD_MANY:
						Collection collection = (Collection) msg.getNewValue();
						for (Iterator iter = collection.iterator(); iter
								.hasNext();) {
							obj = (MethodElement) iter.next();
							addOutgoingConnection(obj);
						}
						return;
					case Notification.REMOVE_MANY:
						collection = (Collection) msg.getOldValue();
						for (Iterator iter = collection.iterator(); iter
								.hasNext();) {
							obj = (MethodElement) iter.next();
							removeOutgoingConnection(obj);
						}
						return;
					}
					break;
				}
			} finally {
				notificationEnabled = true;
			}

			super.notifyChanged(msg);
		}

		/*
		 * (non-Javadoc)
		 * 
		 * @see org.eclipse.epf.diagram.model.util.INodeChangeListener#getNode()
		 */
		public Node getNode() {
			return TaskNodeImpl.this;
		}
	}

	// /* (non-Javadoc)
	// * @see
	// org.eclipse.epf.diagram.model.impl.NamedNodeImpl#setObject(java.lang.Object)
	// */
	// public void setObject(Object newObject) {
	// Object oldObject = object;
	// object = newObject;
	// if (eNotificationRequired())
	// eNotify(new ENotificationImpl(this, Notification.SET,
	// ModelPackage.LINKED_OBJECT__OBJECT, oldObject, object));
	//
	// if(newObject == null) return;
	//		
	// MethodElement element = (MethodElement) newObject;
	// if(element != null && getMethodElementAdapterType() != null) {
	// // remove old methodElementAdapter
	// //
	// Object oldAdapter = UmaUtil.getAdapter(element,
	// getMethodElementAdapterType());
	// if(oldAdapter != null) {
	// element.eAdapters().remove(oldAdapter);
	// }
	// }
	// graphNode =
	// GraphicalDataManager.getInstance().getGraphNode(getUMADiagram(),
	// element);
	//        
	// // System.out.println("NodeImpl.setObject(): element: '" +
	// element.getName() + "', graphNode: " + graphNode);
	//        
	// // populate data
	// //
	// setHeight(graphNode.getSize().getHeight().intValue());
	// setLocation(GraphicalDataHelper.toPoint(graphNode.getPosition()));
	// setWidth(graphNode.getSize().getWidth().intValue());
	//        
	// // add methodElementAdapter
	// //
	// if(methodElementAdapter != null) {
	// element.eAdapters().add(methodElementAdapter);
	// }
	//        
	// this.eAdapters().add(createNodeAdapter());
	//
	// setName(element.getName());
	// }

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.epf.diagram.model.impl.NodeImpl#addToUMA(org.eclipse.epf.diagram.model.Link)
	 */
	protected boolean addToUMA(Link link) {
		if (!super.addToUMA(link))
			return false;

		//TaskDescriptor taskDesc = (TaskDescriptor) getObject();
		NodeImpl nodeImpl = (NodeImpl) link.getSource();
		boolean notify = notificationEnabled;
		boolean notifySource = nodeImpl.notificationEnabled;
		try {
			notificationEnabled = false;
			nodeImpl.notificationEnabled = false;
			// taskDesc.getMandatoryInput().add(link.getSource().getObject());
		} finally {
			notificationEnabled = notify;
			nodeImpl.notificationEnabled = notifySource;
		}

		return true;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.epf.diagram.model.impl.NodeImpl#removeFromUMA(org.eclipse.epf.diagram.model.Link,
	 *      org.eclipse.epf.diagram.model.Node, org.eclipse.epf.diagram.model.Node)
	 */
	protected void removeFromUMA(Link link, Node oldSource, Node oldTarget) {
		TaskDescriptor taskDesc = (TaskDescriptor) getObject();
		taskDesc.getOutput().remove(link.getTarget().getObject());

		super.removeFromUMA(link, oldSource, oldTarget);
	}

} // TaskNodeImpl