import org.eclipse.gmf.runtime.diagram.ui.internal.editparts.TreeNodeEditPart;

import org.eclipse.gmf.runtime.notation.Node;

import org.eclipse.jface.viewers.StructuredSelection;

    class GMFTreeViewer extends TreeViewer

    {

		/** 

		 * A work-around to convert selected tree edit parts into graphical edit

		 * parts right before firing selection changed event to the listeners. This

		 * is done to ensure that Properties View stays in sync with the selection

		 * made from the tree viewer

		 *  

		 * @see org.eclipse.gef.ui.parts.AbstractEditPartViewer#getSelection()

		 */

		public ISelection getSelection() {

			ISelection selected = super.getSelection();

			if (getDiagramGraphicalViewer() == null || !(selected instanceof IStructuredSelection))

				return selected;

			ArrayList selectedItems = new ArrayList();

			Iterator iter = ((IStructuredSelection) selected).iterator();

			while (iter.hasNext()) {

				Object o = iter.next();

				EditPart ep = null;

				if (o instanceof EditPart) {

					try {

						ep = (EditPart) getDiagramGraphicalViewer()

								.getEditPartRegistry().get(

										((EditPart) o).getModel());

					} catch (Exception e) {

					}

				}

				if (ep != null)

					selectedItems.add(ep);

				else

					selectedItems.add(o);

			}

			return new StructuredSelection(selectedItems);

		}	

    }

/******************************************************************************

 * Copyright (c) 2002, 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 API and implementation 

 ****************************************************************************/

package org.eclipse.gmf.runtime.diagram.ui.internal.editparts;

import java.util.ArrayList;

import java.util.List;

import java.util.ListIterator;

import org.eclipse.emf.common.notify.Notification;

import org.eclipse.gmf.runtime.notation.DrawerStyle;

import org.eclipse.gmf.runtime.notation.Node;

import org.eclipse.gmf.runtime.notation.NotationPackage;

import org.eclipse.gmf.runtime.notation.View;

/**

 * Implementation of a tree edit part for node views

 * @author aboyko

 * @canBeSeenBy org.eclipse.gmf.runtime.diagram.ui.*

 */

public class TreeNodeEditPart extends TreeContainerEditPart {

	public TreeNodeEditPart(Object model) {

		super(model);

	}

	/* (non-Javadoc)

	 * @see org.eclipse.gmf.runtime.diagram.ui.internal.editparts.TreeContainerEditPart#getModelChildren()

	 */

	protected List getModelChildren() {

		if (!(getModel() instanceof Node))

			return super.getModelChildren();

		return getNodeChildren((Node)getModel());

	}

    /**

     * Checks whether the node passed in as an argument is compartment

     * @param node - The node to check 

     * @return <Code>true</Code> if node is a compartment, <Code>false</Code> otherwise

     */

    private boolean isCompartment(Node node)

    {

    	return node.getStyle(NotationPackage.eINSTANCE.getDrawerStyle())!=null;

    }

	/* (non-Javadoc)

	 * @see org.eclipse.gmf.runtime.diagram.ui.internal.editparts.TreeContainerEditPart#handleNotificationEvent(org.eclipse.emf.common.notify.Notification)

	 */

	protected void handleNotificationEvent(Notification event) {

		Object feature = event.getFeature();

		if (NotationPackage.eINSTANCE.getView_Visible()==feature 

				|| NotationPackage.eINSTANCE.getDrawerStyle_Collapsed()==feature)

			refreshChildren();

		else

			super.handleNotificationEvent(event);

	}

	/**

	 * Collect non-compartment children of the node. If one of the children is a compartment then children of

	 * this non-compartment children of this compartment child are appended to the list. 

	 * @param node - the node from which children are collected

	 * @return <Code>List</Code> of non-compartment children of the node passed in as an argument 

	 */

	private List getNodeChildren(Node node)

	{

		List children = new ArrayList(node.getVisibleChildren().size());

		ListIterator iter = node.getVisibleChildren().listIterator();

		while (iter.hasNext())

		{

			Object element = iter.next();

			if (element instanceof Node)

			{

				Node child = (Node) element;

				if (isCompartment(child))

				{

					children.addAll(getNodeChildren(child));

				}

				else if (isCompartment(node) && 

							!((DrawerStyle) node.getStyle(NotationPackage.eINSTANCE.getDrawerStyle())).isCollapsed()

						)

				{

					children.add(child);

				}

			}

		}

		return children;

	}

	/* (non-Javadoc)

	 * @see org.eclipse.gmf.runtime.diagram.ui.internal.editparts.TreeEditPart#activate()

	 */

	public void activate() {

		super.activate();

		// add listeners to catch compartments notification events		

		ListIterator iter = getCompartmentChildren().listIterator();

		while(iter.hasNext())

		{

			View view = (View)iter.next();

	        getDiagramEventBroker().addNotificationListener(view, this);

		}

	}

	/* (non-Javadoc)

	 * @see org.eclipse.gmf.runtime.diagram.ui.internal.editparts.TreeEditPart#deactivate()

	 */

	public void deactivate() {

		super.deactivate();

		// remove listeners catching compartments notification events

		ListIterator iter = getCompartmentChildren().listIterator();

		while(iter.hasNext())

		{

			View view = (View)iter.next();

	        getDiagramEventBroker().removeNotificationListener(view, this);

		}		

	}

	/**

	 * Collects all compartment children of the node including comaprtments within compartments

	 * @return <Code>List</Code> of all compartments contained within this node

	 */

	private List getCompartmentChildren()

	{

		ArrayList compartments = new ArrayList();

		ListIterator iter = ((Node)getModel()).getChildren().listIterator();

		while(iter.hasNext())

		{

			Object obj = iter.next();

			if (obj instanceof Node && isCompartment((Node)obj))

				compartments.add(obj);

		}

		return compartments;

	}

}