EDIT

package eu.etaxonomy.taxeditor.editor.view.uses;

import java.util.HashMap;

import java.util.Map;

import org.eclipse.jface.action.Action;

import org.eclipse.jface.action.GroupMarker;

import org.eclipse.jface.action.IToolBarManager;

import org.eclipse.jface.action.MenuManager;

import org.eclipse.jface.resource.ImageDescriptor;

import org.eclipse.jface.viewers.ISelection;

import org.eclipse.jface.viewers.StructuredSelection;

import org.eclipse.jface.viewers.TreeViewer;

import org.eclipse.jface.viewers.Viewer;

import org.eclipse.swt.SWT;

import org.eclipse.swt.dnd.DND;

import org.eclipse.swt.dnd.Transfer;

import org.eclipse.swt.graphics.ImageData;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Control;

import org.eclipse.swt.widgets.Menu;

import org.eclipse.swt.widgets.Tree;

import org.eclipse.ui.IEditorPart;

import org.eclipse.ui.IWorkbenchActionConstants;

import org.eclipse.ui.IWorkbenchPart;

import eu.etaxonomy.cdm.model.common.CdmBase;

import eu.etaxonomy.cdm.model.description.DescriptionBase;

import eu.etaxonomy.cdm.model.description.DescriptionElementBase;

import eu.etaxonomy.cdm.model.description.TaxonDescription;

import eu.etaxonomy.cdm.api.service.IDescriptionService;

import eu.etaxonomy.taxeditor.bulkeditor.BulkEditor;

import eu.etaxonomy.taxeditor.editor.EditorUtil;

import eu.etaxonomy.taxeditor.editor.MultiPageTaxonEditor;

import eu.etaxonomy.taxeditor.editor.TaxonEditorInput;

import eu.etaxonomy.taxeditor.editor.key.AbstractGraphKeyEditor;

import eu.etaxonomy.taxeditor.editor.view.descriptive.DescriptionElementDragListener;

import eu.etaxonomy.taxeditor.editor.view.descriptive.DescriptionElementDropAdapter;

import eu.etaxonomy.taxeditor.editor.view.descriptive.DescriptionElementTransfer;

import eu.etaxonomy.taxeditor.editor.view.descriptive.DescriptiveViewerSorter;

import eu.etaxonomy.taxeditor.model.FeatureNodeContainer;

import eu.etaxonomy.taxeditor.model.FeatureNodeContainerTree;

import eu.etaxonomy.taxeditor.model.IPartContentHasDetails;

import eu.etaxonomy.taxeditor.model.IPartContentHasSupplementalData;

import eu.etaxonomy.taxeditor.model.ImageResources;

import eu.etaxonomy.taxeditor.view.AbstractCdmEditorViewPart;

public class UsesViewPart extends AbstractCdmEditorViewPart implements IPartContentHasDetails, IPartContentHasSupplementalData {

	public static final String ID = "eu.etaxonomy.taxeditor.editor.view.uses";

	private TreeViewer viewer;

	private Map<TaxonDescription, FeatureNodeContainerTree> featureNodeContainerCache = new HashMap<TaxonDescription, FeatureNodeContainerTree>();

	private ToggleDescriptionAction showAllElementsAction;

	private ToggleDescriptionAction hideAllElementsAction;

	private int dndOperations = DND.DROP_COPY | DND.DROP_MOVE;

	/** {@inheritDoc} */

	@Override

	public void createViewer(Composite parent) {

		viewer = new TreeViewer(new Tree(parent, SWT.MULTI | SWT.H_SCROLL

				| SWT.V_SCROLL | SWT.FULL_SELECTION));

		viewer.setContentProvider(new UsesContentProvider(featureNodeContainerCache));		

		//viewer.setInput(descService.listTaxonDescriptions(Taxon, null, null, null, null, null, null));

		viewer.setLabelProvider(new UsesLabelProvider());

		//viewer.setSorter(new DescriptiveViewerSorter());

		viewer.setAutoExpandLevel(2);

		Transfer[] transfers = new Transfer[] { DescriptionElementTransfer.getInstance() };

		/*viewer.addDragSupport(dndOperations, transfers, new DescriptionElementDragListener(

				this));

		viewer.addDropSupport(dndOperations, transfers,

				new DescriptionElementDropAdapter(viewer));*/

		// Propagate selection from viewer

		getSite().setSelectionProvider(viewer);

		showAllElementsAction = new ToggleDescriptionAction(false);		

		hideAllElementsAction = new ToggleDescriptionAction(true);

		// Add context menu to tree

		createMenu();

		createToolbar();

		// set initial input

//		if(getEditor() != null){

//			viewer.setInput(getEditor().getEditorInput());

//		}

	}

	private void createToolbar() {

		IToolBarManager toolBarManager = getViewSite().getActionBars().getToolBarManager();

		toolBarManager.add(showAllElementsAction);

		toolBarManager.add(hideAllElementsAction);

	}

	private void createMenu(){

		MenuManager menuManager = new MenuManager();

		menuManager.add(new GroupMarker(IWorkbenchActionConstants.MB_ADDITIONS));

		getSite().registerContextMenu(menuManager, viewer);

		// FIXME for some reason this is not working

		menuManager.add(showAllElementsAction);

		menuManager.add(hideAllElementsAction);

		Control control = viewer.getControl();

		Menu menu = menuManager.createContextMenu(control);

		control.setMenu(menu);	

	}

	public void changed(Object object) {

		for(FeatureNodeContainerTree containerTree : featureNodeContainerCache.values()){

			containerTree.buildTree();

		}

		viewer.refresh();

		/*if(object instanceof DescriptionElementBase){

			DescriptionElementBase descriptionElement = (DescriptionElementBase) object;

			DescriptionBase description = descriptionElement.getInDescription();

			FeatureNodeContainerTree containerTree = featureNodeContainerCache.get(description);

			FeatureNodeContainer featureNodeContainer = containerTree.getFeatureNodeContainerForDescriptionElement(descriptionElement);

			viewer.expandToLevel(featureNodeContainer, 1);

		}*/

		if(object != null){

			StructuredSelection selection = new StructuredSelection(object);

			viewer.setSelection(selection, true);

		}

		super.changed(object);

	}

	/* (non-Javadoc)

	 * @see eu.etaxonomy.taxeditor.model.AbstractCdmViewPart#getInitialSelection()

	 */

	/** {@inheritDoc} */

	@Override

	protected ISelection getInitialSelection() {

		if(getEditor() != null){

			return new StructuredSelection(getEditor().getEditorInput());

		}	

		return super.getInitialSelection();

	}

	/** {@inheritDoc} */

	@Override

	public boolean postOperation(CdmBase objectAffectedByOperation) {

		return super.postOperation(objectAffectedByOperation);

	}

	@Override

	public void selectionChanged(IWorkbenchPart part, ISelection selection) {

		if(EditorUtil.getActiveEditor() == null){

			showEmptyPage();

			return;

		}

		if(part instanceof BulkEditor){

			showEmptyPage();

			return;

		}

		if(part instanceof AbstractGraphKeyEditor){

			showEmptyPage();

			return;

		}

		if(part instanceof MultiPageTaxonEditor){

			if(! part.equals(this.part)){

				TaxonEditorInput input = (TaxonEditorInput)((IEditorPart) part).getEditorInput();

				featureNodeContainerCache.clear();

				//showViewer(part, new StructuredSelection(CdmStore.getService(IDescriptionService.class).listTaxonDescriptions(input.getTaxonNode().getTaxon(), null, null, null, null, null, null)));

				showViewer(part, new StructuredSelection(input));

			}

			showViewer();

		}

	}

	@Override

	public boolean onComplete() {

		return false;

	}

	/*@Override

	public void showViewer(IWorkbenchPart part, IStructuredSelection selection) {

		this.part = part;		

		Object element = selection.getFirstElement();

		getViewer().setInput(descService.listTaxonDescriptions((Taxon) element, null, null, null, null, null, null));

		showViewer();	

	}*/

	@Override

	public Viewer getViewer() {

		return viewer;

	}

	private class ToggleDescriptionAction extends Action{

		private boolean expanded;

		public ToggleDescriptionAction(boolean expand){

			super(null, Action.AS_PUSH_BUTTON);

			expanded = expand;

			setImageAndTooltip();

		}

		private void setImageAndTooltip(){

			setImageDescriptor(new ImageDescriptor() {

				@Override

				public ImageData getImageData() {

					setText(expanded ? "Collapse All" : "Expand All");

					String resource = expanded ? ImageResources.COLLAPSE_ALL : ImageResources.EXPAND_ALL;

					return ImageResources.getImage(resource).getImageData();

				}

			});

			String toolTipText = expanded ? "Collapse all" : "Show all descriptive data";

			setToolTipText(toolTipText);

		}

		@Override

		public void run() {

			if(expanded){

				viewer.collapseAll();

			}else{

				viewer.expandAll();

			}

			setImageAndTooltip();

		}

	}

}