//------------------------------------------------------------------------------
// Copyright (c) 2005, 2007 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.add.edit.parts;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.FreeformLayer;
import org.eclipse.draw2d.FreeformLayout;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.MarginBorder;
import org.eclipse.draw2d.geometry.Dimension;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.transaction.Transaction;
import org.eclipse.epf.common.CommonPlugin;
import org.eclipse.epf.diagram.add.edit.policies.ActivityDetailDiagramCanonicalEditPolicy;
import org.eclipse.epf.diagram.add.edit.policies.ActivityDetailDiagramItemSemanticEditPolicy;
import org.eclipse.epf.diagram.core.part.AbstractDiagramEditor;
import org.eclipse.epf.diagram.core.part.DiagramEditorInputProxy;
import org.eclipse.epf.diagram.core.part.IDiagramEditorInputProvider;
import org.eclipse.epf.diagram.core.part.util.DiagramEditorUtil;
import org.eclipse.epf.diagram.core.services.DiagramHelper;
import org.eclipse.epf.diagram.model.TaskNode;
import org.eclipse.epf.diagram.model.WorkProductComposite;
import org.eclipse.epf.diagram.model.WorkProductDescriptorNode;
import org.eclipse.epf.diagram.model.util.TxUtil;
import org.eclipse.epf.uma.TaskDescriptor;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPartViewer;
import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.NodeEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editpolicies.EditPolicyRoles;
import org.eclipse.gmf.runtime.diagram.ui.util.MeasurementUnitHelper;
import org.eclipse.gmf.runtime.draw2d.ui.mapmode.IMapMode;
import org.eclipse.gmf.runtime.notation.Location;
import org.eclipse.gmf.runtime.notation.Node;
import org.eclipse.gmf.runtime.notation.NotationPackage;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.ui.IEditorInput;
import org.eclipse.ui.IEditorPart;

/**
 * @modified
 */
public class ActivityDetailDiagramEditPart extends DiagramEditPart {

	Point cachedPoint = new Point(-1, -1);

	protected boolean debug = false;

	static int cachedWidth = 10;

	static int cachedHeight = 10;

	int y = 60;

	boolean xaxisbegin = false;

	boolean loading = true;

	final static int spacing = 40;

	final static int yspacing = 20;

	boolean miscellaneous = false;

	int x = 10;

	public static final String OUTPUTS = "outputs"; //$NON-NLS-1$

	public static final String INPUTS = "inputs"; //$NON-NLS-1$

	private List<EditPart> recentlyAddedParts = new ArrayList<EditPart>();

	private IMapMode mapMode;
	/**
	 * @generated
	 */
	public static final String MODEL_ID = "ADD"; //$NON-NLS-1$

	/**
	 * @generated
	 */
	public static final int VISUAL_ID = 79;

	/**
	 * @generated
	 */
	public ActivityDetailDiagramEditPart(View view) {
		super(view);
		
		mapMode = MeasurementUnitHelper.getMapMode(view.getDiagram().getMeasurementUnit());
	}
	
	private View getView() {
		return (View) getModel();
	}
	
	private void doLayout() {
		EditPartViewer viewer = getViewer();
		if(viewer instanceof IDiagramEditorInputProvider) {
			IEditorPart editor = ((IDiagramEditorInputProvider)viewer).getEditor();
			if(editor instanceof AbstractDiagramEditor) {
				((AbstractDiagramEditor)editor).resetLayout();
			}
		}
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart#createFigure()
	 */
	protected IFigure createFigure() {
		Figure f = new FreeformLayer() {
			protected void layout() {
				if (hasMisplacedChildren()) {
					resetChildrenLocations();
				}
				else {
					super.layout();
				}
			}
		};

		f.setBorder(new MarginBorder(20));
		f.setLayoutManager(new FreeformLayout());

		return f;
	}

	/**
	 * Checks whether there are any misplaced children
	 * @return
	 */
	protected boolean hasMisplacedChildren() {
		List locationList = new ArrayList();

		for (Iterator iter = getChildren().iterator(); iter.hasNext();) {
			Object child = iter.next();
			if (child instanceof EditPart) {
				Object model = ((EditPart) child).getModel();
				if (model instanceof Node) {
					Node node = (Node) model;
					Location p = (Location) node.getLayoutConstraint();
					locationList.add(p);
					if (debug) {
						System.out.println("Node-" + node); //$NON-NLS-1$
						System.out.println("loc-" + p); //$NON-NLS-1$
					}
				}
			}
		}
		if (locationList.size() == 1)
			return false;
		if (locationList.size() > 0) {
			boolean needToReset = true;

			for (int i = 1; i < locationList.size(); i++) {
				Location currentLoc = (Location) locationList.get(i);
				Location previousLoc = (Location) locationList.get(i - 1);
				if (!((currentLoc.getX() == previousLoc.getX() + 10) && (currentLoc
						.getY() == previousLoc.getY() + 10))) {
					needToReset = false;
					break;
				}
			}
			return needToReset;
		}
		return false;
	}

	/**
	 * Reset all children locations
	 * In this, All the Task's MandatoryInput workproducts placed on top and RoleTask 
	 * composite in the middle. and Task's output workproducts bottom.
	 * 
	 * TODO - Have to revisit this method and do some cleanup
	 * 
	 */
	protected void resetChildrenLocations() {

		final int yspacing = 40;
		List rolecompositeList = new ArrayList();
		List inputworkproductCompositeList = new ArrayList();
		List outputWorkProductCompositeList = new ArrayList();
		List miscList = new ArrayList();
		List taskList = new ArrayList();
		this.recentlyAddedParts = new ArrayList();

		//		HashMap map = new HashMap();

		for (Iterator iter = getChildren().iterator(); iter.hasNext();) {
			Object child = iter.next();
			Node n = (Node) ((EditPart) child).getModel();
			//			if (n.getLocation() == null || n.getLocation().x == -1) {
			this.recentlyAddedParts.add((EditPart) child);
			//			}
			if (child instanceof RoleTaskCompositeEditPart) {
				rolecompositeList.add(child);

				List list = ((RoleTaskCompositeEditPart) child).getChildren();
				for (Iterator it = list.iterator(); it.hasNext();) {
					Object obj = it.next();
					if (obj instanceof TaskNodeEditPart) {
						View viewObj = (View) ((NodeEditPart) obj).getModel();
						Object o = viewObj.getElement();
						if (o instanceof TaskNode) {
							taskList.add(o);
						}
					}

				}
			} else if (child instanceof WorkProductCompositeEditPart) {
				WorkProductCompositeEditPart part = (WorkProductCompositeEditPart) child;
				View viewObj = (View) ((NodeEditPart) part).getModel();
				Object o = viewObj.getElement();

				if (o instanceof WorkProductComposite) {
					WorkProductComposite node = (WorkProductComposite) o;
					if (node.getType() == WorkProductComposite.INPUTS) {
						inputworkproductCompositeList.add(child);
					} else if (node.getType() == WorkProductComposite.OUTPUTS) {
						outputWorkProductCompositeList.add(child);
					}
					//					calcSize(child);
				}
			} else {
				miscList.add(child);
			}
		}

		// Set size for the rolecomposite.
		if (!rolecompositeList.isEmpty()) {
			for (Iterator itor = rolecompositeList.iterator(); itor.hasNext();) {
				Object object = itor.next();
				List list = ((RoleTaskCompositeEditPart) object).getChildren();
				for (Iterator itor1 = list.iterator(); itor1.hasNext();) {
					Object obj = itor1.next();
					if (obj instanceof NodeEditPart) {
						View viewObj = (View) ((NodeEditPart) obj).getModel();
						Object objNode = viewObj.getElement();
						if (objNode instanceof TaskNode) {
							TaskNode taskNode = (TaskNode) objNode;

							TaskDescriptor task = (TaskDescriptor) taskNode
									.getObject();
							for (Iterator wpitor = inputworkproductCompositeList
									.iterator(); wpitor.hasNext();) {
								WorkProductCompositeEditPart wpEditPart = (WorkProductCompositeEditPart) wpitor
										.next();
								WorkProductComposite wpCompositeNode = (WorkProductComposite) ((View) wpEditPart
										.getModel()).getElement();
								Object wpObject = wpCompositeNode.getObject();
								if (wpObject != null && wpObject.equals(task)) {
									int width = wpEditPart.getFigure()
											.getPreferredSize().width;
									int height = ((NodeEditPart) obj)
											.getFigure().getPreferredSize().height;
									((NodeEditPart) obj)
											.getFigure()
											.setPreferredSize(
													new Dimension(width, height));
									if (((WorkProductComposite) ((View) wpEditPart
											.getModel()).getElement())
											.getType() == WorkProductComposite.INPUTS) {
										//										List inputList = (List) ((HashMap) map
										//												.get(taskNode)).get(INPUTS);
										//										inputList.add(wpEditPart);
									} else if (((WorkProductComposite) ((View) wpEditPart
											.getModel()).getElement())
											.getType() == WorkProductComposite.OUTPUTS) {
										//										List outputList = (List) ((HashMap) map
										//												.get(taskNode)).get(OUTPUTS);
										//										outputList.add(wpEditPart);
									}
								}
							}
						}
					}
				}
			}
		}

		if (!rolecompositeList.isEmpty()) {
			for (int i = 0; i < rolecompositeList.size(); i++) {
				List confirmedSeqInputList = new ArrayList();
				List confirmedSeqOutputList = new ArrayList();
				List list = ((RoleTaskCompositeEditPart) rolecompositeList
						.get(i)).getChildren();
				for (int j = 0; j < list.size(); j++) {
					Object o = list.get(j);
					if (o instanceof NodeEditPart) {
						NodeEditPart task = (NodeEditPart) list.get(j);
						if (((View) task.getModel()).getElement() instanceof TaskNode) {
							TaskNode tasknode = (TaskNode) ((View) task
									.getModel()).getElement();
							for (int k = 0; k < inputworkproductCompositeList
									.size(); k++) {
								WorkProductComposite wpcomp = (WorkProductComposite) ((View) ((WorkProductCompositeEditPart) inputworkproductCompositeList
										.get(k)).getModel()).getElement();

								if (wpcomp != null && wpcomp.getObject().equals(
										tasknode.getObject())) {
									if (wpcomp.getType() == WorkProductComposite.INPUTS) {
										confirmedSeqInputList
												.add(inputworkproductCompositeList
														.get(k));
									}
								}
							}
							for (int k = 0; k < outputWorkProductCompositeList
									.size(); k++) {
								WorkProductComposite wpcomp = (WorkProductComposite) ((View) ((WorkProductCompositeEditPart) outputWorkProductCompositeList
										.get(k)).getModel()).getElement();

								if (wpcomp.getObject().equals(
										tasknode.getObject())) {
									if (wpcomp.getType() == WorkProductComposite.OUTPUTS) {
										confirmedSeqOutputList
												.add(outputWorkProductCompositeList
														.get(k));
									}
								}
							}
						}
					}
				}
				if (!confirmedSeqInputList.isEmpty()) {
					setObjectsLocations(confirmedSeqInputList);
					y = cachedPoint.y + cachedHeight + yspacing;
				}
				List roleC = new ArrayList();
				roleC.add(rolecompositeList.get(i));
				setObjectsLocations(roleC);

				y = cachedPoint.y + cachedHeight + yspacing;
				if (!confirmedSeqOutputList.isEmpty()) {
					setObjectsLocations(confirmedSeqOutputList);
					y = cachedPoint.y + cachedHeight + yspacing;
				}
			}

		}

		//		if (!miscList.isEmpty()) {
		//			if (debug) {
		//				System.out.println("Print Size of misc: " + miscList.size()); //$NON-NLS-1$
		//			}
		//			if (!loading) {
		//				miscellaneous = true;
		//			}
		//			setObjectsLocations(miscList);
		//			miscellaneous = false;
		//		}

		final View view = getView();
		try {
			TxUtil.runInTransaction(view.getDiagram(), new Runnable() {

				public void run() {
					view.persist();
					view.persistChildren();
				}
				
			});
		} catch (Exception e) {
			CommonPlugin.getDefault().getLogger().logError(e);
		}
		loading = false;

	}

	/**
	 * @generated
	 */
	protected void createDefaultEditPolicies() {
		super.createDefaultEditPolicies();
		installEditPolicy(EditPolicyRoles.SEMANTIC_ROLE,
				new ActivityDetailDiagramItemSemanticEditPolicy());
		installEditPolicy(EditPolicyRoles.CANONICAL_ROLE,
				new ActivityDetailDiagramCanonicalEditPolicy());
	}

	/**
	 * It sets object location on cavas
	 * @param list
	 * 
	 */
	private void setObjectsLocations(List list) {
		xaxisbegin = true;
		x = 50;
		for (Iterator iter = list.iterator(); iter.hasNext();) {
			Object child = iter.next();
			if (child instanceof GraphicalEditPart) {
				final GraphicalEditPart part = (GraphicalEditPart) child;
				final Node node = (Node) part.getModel();

				if (node instanceof Node) {
					final Location location = (Location) node.getLayoutConstraint();

					if (debug) {
						System.out.println("Node - " + part); //$NON-NLS-1$
						System.out.println("Location - " + location); //$NON-NLS-1$
					}

					if (!xaxisbegin) {
						x = cachedPoint.x + cachedWidth + spacing;
					}

					if (!loading && miscellaneous) {
						location.setX(mapMode.DPtoLP(x));
						location.setY(mapMode.DPtoLP(10));

					} else {
						try {
							TxUtil.runInTransaction(getView().getDiagram(), new Runnable() {

								public void run() {
									location.setX(mapMode.DPtoLP(x));
									location.setY(mapMode.DPtoLP(y));

									if (part instanceof RoleTaskCompositeEditPart) {
										((RoleTaskCompositeEditPart) part)
												.updatebounds();
									}
									if (part instanceof WorkProductCompositeEditPart) {
										((WorkProductCompositeEditPart) part)
												.updatebounds();
									}

								}
								
							});
						} catch (Exception e) {
							CommonPlugin.getDefault().getLogger().logError(e);
						}

						if (debug)
							System.out.println("NewLocation - " + location); //$NON-NLS-1$

						if (!miscellaneous)
							cachedPoint = new Point(location.getX(), location
									.getY());
						cachedWidth = part.getFigure().getPreferredSize().width;
						if (part.getFigure().getPreferredSize().height > cachedHeight)
							cachedHeight = part.getFigure().getPreferredSize().height;
						xaxisbegin = false;
						if (debug) {
							//							System.out
							//									.println("Point : " + new Point(x, y).toString() + "And width=" //$NON-NLS-1$ //$NON-NLS-2$
							//											+ cachedWidth
							//											+ ", Height=" + cachedHeight + ", Width of Viewer: " + viewerWidth); //$NON-NLS-1$ //$NON-NLS-2$
						}
					}
				}
			}
		}
	}

	/**
	 * Calculates and sets size of the given node
	 * @param child
	 * 
	 * TODO - Not used currently, we need to see whether we need this method
	 */
	private void calcSize(Object child) {
		List list = ((GraphicalEditPart) child).getChildren();
		int largestWidth = 0;
		int largestHeight = 0;
		for (Iterator it = list.iterator(); it.hasNext();) {
			Object obj = it.next();

			if (obj instanceof NodeEditPart) {
				View viewObj = (View) ((NodeEditPart) obj).getModel();
				Object objNode = viewObj.getElement();
				if (objNode instanceof WorkProductDescriptorNode) {
					int nodeWidth = ((NodeEditPart) obj).getFigure()
							.getPreferredSize().width;
					int nodeHeight = ((NodeEditPart) obj).getFigure()
							.getPreferredSize().height;
					if (nodeWidth > largestWidth)
						largestWidth = nodeWidth;
					if (nodeHeight > largestHeight)
						largestHeight = nodeHeight;
				}
			}

		}
		((WorkProductCompositeEditPart) child).getFigure().setPreferredSize(
				new Dimension(largestWidth * 2 + 10, 20 + largestHeight
						* (list.size() / 2 + 1)));
	}

	/**
	 * Returns recently addded editparts. Usually called 
	 * by diagram services to do autolayout adjustment.
	 */
	public List<EditPart> getRecentlyAddedParts() {
		return recentlyAddedParts;
	}

	/**
	 * Clear the recently added editparts from the list.
	 */
	public void clearRecentlyAddedParts() {
		recentlyAddedParts.clear();
	}

	@Override
	protected void handleNotificationEvent(Notification event) {
		if (NotationPackage.Literals.VIEW__VISIBLE.equals(event.getFeature())) {
			Object notifier = event.getNotifier();
			if (notifier == getModel())
				refreshVisibility();
			else if (event.getNotifier() instanceof Node) {
				refreshChildren();
				DiagramEditorUtil.refreshConnectionEditParts(this);
			}
		} else
			super.handleNotificationEvent(event);
	}
	
	@Override
	public void activate() {
		super.activate();
		Object model = getModel();
		if (model instanceof View) {
			EObject e = ((View) model).getElement();
			if (e instanceof org.eclipse.epf.diagram.model.Node) {
				((org.eclipse.epf.diagram.model.Node) e).addConsumer(this);
			}
		}
	}
	
	@Override
	public void deactivate() {
		super.deactivate();
		Object model = getModel();
		if (model instanceof View) {
			EObject e = ((View) model).getElement();
			if (e instanceof org.eclipse.epf.diagram.model.Node) {
				((org.eclipse.epf.diagram.model.Node) e).removeConsumer(this);
			}
		}
	}
	
}