EDIT

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

import java.util.Collection;

import java.util.HashSet;

import java.util.Map.Entry;

import java.util.Set;

import org.eclipse.core.runtime.IProgressMonitor;

import org.eclipse.jface.action.MenuManager;

import org.eclipse.jface.util.LocalSelectionTransfer;

import org.eclipse.jface.viewers.AbstractTreeViewer;

import org.eclipse.jface.viewers.ISelection;

import org.eclipse.jface.viewers.ISelectionChangedListener;

import org.eclipse.jface.viewers.SelectionChangedEvent;

import org.eclipse.jface.viewers.StructuredSelection;

import org.eclipse.jface.viewers.TreeNode;

import org.eclipse.jface.viewers.TreeSelection;

import org.eclipse.jface.viewers.TreeViewer;

import org.eclipse.swt.SWT;

import org.eclipse.swt.dnd.DND;

import org.eclipse.swt.dnd.Transfer;

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.IEditorInput;

import org.eclipse.ui.IEditorPart;

import org.eclipse.ui.IEditorSite;

import org.eclipse.ui.PartInitException;

import org.eclipse.ui.part.EditorPart;

import eu.etaxonomy.cdm.api.conversation.ConversationHolder;

import eu.etaxonomy.cdm.api.conversation.IConversationEnabled;

import eu.etaxonomy.cdm.api.service.molecular.ISequenceService;

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

import eu.etaxonomy.cdm.model.molecular.Sequence;

import eu.etaxonomy.cdm.model.molecular.SingleRead;

import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;

import eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeMap;

import eu.etaxonomy.taxeditor.editor.Messages;

import eu.etaxonomy.taxeditor.editor.view.derivate.searchFilter.DerivateSearchCompositeController;

import eu.etaxonomy.taxeditor.model.IDirtyMarkable;

import eu.etaxonomy.taxeditor.model.IPartContentHasDetails;

import eu.etaxonomy.taxeditor.model.IPartContentHasFactualData;

import eu.etaxonomy.taxeditor.model.IPartContentHasMedia;

import eu.etaxonomy.taxeditor.model.IPartContentHasSupplementalData;

import eu.etaxonomy.taxeditor.operation.IPostOperationEnabled;

import eu.etaxonomy.taxeditor.store.CdmStore;

import eu.etaxonomy.taxeditor.ui.element.LayoutConstants;

import eu.etaxonomy.taxeditor.view.derivateSearch.DerivateContentProvider;

import eu.etaxonomy.taxeditor.view.derivateSearch.DerivateLabelProvider;

/**

 * Displays the derivate hierarchy of the specimen specified in the editor input.

 *

 */

public class DerivateView extends EditorPart implements IPartContentHasFactualData, IDirtyMarkable,

        IConversationEnabled, IPartContentHasDetails, IPartContentHasSupplementalData, IPartContentHasMedia,

        ISelectionChangedListener, IPostOperationEnabled{

    public static final String ID = "eu.etaxonomy.taxeditor.editor.view.derivate.DerivateView"; //$NON-NLS-1$

    public static final String YOU_NEED_TO_SAVE_BEFORE_PERFORMING_THIS_ACTION = Messages.DerivateView_YOU_NEED_TO_SAVE;

    public static final String VIEW_HAS_UNSAVED_CHANGES = Messages.DerivateView_UNSAVED_CHANGES;

	private ConversationHolder conversation;

	private TreeViewer viewer;

    private boolean isDirty;

    private final int dndOperations = DND.DROP_MOVE;

    private DerivateLabelProvider labelProvider;

    private Set<SingleRead> multiLinkSingleReads;

    private ISelection selection = null;

    private DerivateContentProvider contentProvider;

    private DerivateSearchCompositeController derivateSearchCompositeController;

    /**

     * Default constructor

     */

    public DerivateView() {

    }

    @Override

    public void createPartControl(Composite parent) {

        parent.setLayout(LayoutConstants.LAYOUT());

        parent.setLayoutData(LayoutConstants.FILL());

        //---search and filter---

        derivateSearchCompositeController = new DerivateSearchCompositeController(parent, this);

        derivateSearchCompositeController.setEnabled(CdmStore.isActive());

        derivateSearchCompositeController.setLayoutData(LayoutConstants.FILL_HORIZONTALLY());

        //---tree viewer---

        viewer = new TreeViewer(new Tree(parent, SWT.MULTI | SWT.H_SCROLL | SWT.V_SCROLL | SWT.FULL_SELECTION));

        viewer.getTree().setLayoutData(LayoutConstants.FILL());

        contentProvider = new DerivateContentProvider();

        viewer.setContentProvider(contentProvider);

        labelProvider = new DerivateLabelProvider();

        labelProvider.setConversation(conversation);

        viewer.setLabelProvider(labelProvider);

        viewer.setAutoExpandLevel(AbstractTreeViewer.ALL_LEVELS);

        // Propagate selection from viewer

        getSite().setSelectionProvider(viewer);

        //listen to selection changes

        viewer.addSelectionChangedListener(this);

        //create context menu

        MenuManager menuManager = new MenuManager();

        getSite().registerContextMenu(menuManager, viewer);

        Control control = viewer.getControl();

        Menu menu = menuManager.createContextMenu(control);

        control.setMenu(menu);

        generateMultiLinkSingleReads();

        labelProvider.setMultiLinkSingleReads(multiLinkSingleReads);

        IEditorInput editorInput = getEditorInput();

        viewer.setInput(((DerivateViewEditorInput) editorInput).getRootEntities());

        //set selection to selected derivate if only one was selected

        if(editorInput instanceof DerivateViewEditorInput){

            Set<SpecimenOrObservationBase<?>> derivateEntities = ((DerivateViewEditorInput) editorInput).getDerivateEntities();

            if(derivateEntities.size()==1){

                SpecimenOrObservationBase<?> specimen = derivateEntities.iterator().next();

                if(specimen != null){

                    viewer.setSelection(new StructuredSelection(new TreeNode(specimen)));

                }

            }

        }

        //add drag'n'drop support

        Transfer[] transfers = new Transfer[] {LocalSelectionTransfer.getTransfer(),};

        viewer.addDragSupport(dndOperations, transfers, new DerivateDragListener(this));

        viewer.addDropSupport(dndOperations, transfers, new DerivateDropListener(this));

    }

    @Override

    public void doSave(IProgressMonitor monitor) {

        String taskName = Messages.DerivateView_SAVING_HIERARCHY;

        if(getEditorInput() instanceof DerivateViewEditorInput){

            DerivateViewEditorInput derivateViewEditorInput = (DerivateViewEditorInput) getEditorInput();

            taskName += " "+derivateViewEditorInput.getName(); //$NON-NLS-1$

        }

        monitor.beginTask(taskName, 3);

        if (!conversation.isBound()) {

            conversation.bind();

        }

        monitor.worked(1);

        // commit the conversation and start a new transaction immediately

        conversation.commit(true);

        ((DerivateViewEditorInput) getEditorInput()).merge();

        monitor.worked(1);

        this.setDirty(false);

        monitor.worked(1);

        monitor.done();

        firePropertyChange(PROP_DIRTY);

        refreshTree();

    }

    @Override

    public void doSaveAs() {

    }

    @Override

    public String getTitleToolTip() {

        if(getEditorInput() instanceof DerivateViewEditorInput){

            DerivateViewEditorInput derivateViewEditorInput = (DerivateViewEditorInput) getEditorInput();

            return derivateViewEditorInput.getName();

        }

        return Messages.DerivateView_DERIVATIVE_EDITOR;

    }

    @Override

    public void init(IEditorSite site, IEditorInput input) throws PartInitException {

        setSite(site);

        setInput(input);

        if(input instanceof DerivateViewEditorInput){

            DerivateViewEditorInput derivateViewEditorInput = (DerivateViewEditorInput) input;

            conversation = derivateViewEditorInput.getConversationHolder();

            setPartName(derivateViewEditorInput.getName());

        }

    }

    @Override

    public boolean isDirty() {

        return isDirty;

    }

    /**

     * @param isDirty the isDirty to set

     */

    public void setDirty(boolean isDirty) {

        this.isDirty = isDirty;

    }

    @Override

    public boolean isSaveAsAllowed() {

        return false;

    }

    @Override

    public void setFocus() {

        viewer.getControl().setFocus();

        //make sure to bind again if maybe in another view the conversation was unbound

        if(!conversation.isBound()){

            conversation.bind();

        }

        ((DerivateViewEditorInput) getEditorInput()).bind();

    }

    @Override

    public void update(CdmDataChangeMap changeEvents) {

    }

    @Override

    public ConversationHolder getConversationHolder() {

        return conversation;

    }

    /**

     * @return the viewer

     */

    public TreeViewer getViewer() {

        return viewer;

    }

    @Override

    public void changed(Object element) {

        setDirty(true);

        firePropertyChange(IEditorPart.PROP_DIRTY);

        viewer.refresh();

    }

    @Override

    public void forceDirty() {

        changed(null);

    }

    /**

     * Refreshes the derivate hierarchy tree and expands the tree

     * to show and select the given object.

     *

     * @param expandTo the object to which the tree should be expanded

     */

    public void refreshTree(Object expandTo){

        refreshTree();

        TreeSelection selection = (TreeSelection) viewer.getSelection();

        viewer.expandToLevel(selection.getFirstElement(), 1);

        viewer.setSelection(new StructuredSelection(new TreeNode(expandTo)));

    }

    /**

     * Refreshes the derivate hierarchy tree

     */

    public void refreshTree(){

        generateMultiLinkSingleReads();

        labelProvider.setMultiLinkSingleReads(multiLinkSingleReads);

        viewer.refresh();

    }

    //FIXME:Remoting hack to make this work for remoting

    //This should actually be resolved using remoting post operations

    public void remove(Object obj) {

        Set<SpecimenOrObservationBase<?>> rootEntities = ((DerivateViewEditorInput) getEditorInput()).getRootEntities();

        rootEntities.remove(obj);

        viewer.setInput(rootEntities);

    }

    private void generateMultiLinkSingleReads() {

        Set<SingleRead> multiLinkSingleReads = new HashSet<SingleRead>();

        for(Entry<SingleRead, Collection<Sequence>> entry:CdmStore.getService(ISequenceService.class).getSingleReadSequencesMap().entrySet()){

            if(entry.getValue().size()>1){

                multiLinkSingleReads.add(entry.getKey());

            }

        }

        this.multiLinkSingleReads = multiLinkSingleReads;

    }

    /**

     * @return a set of {@link SingleRead}s that have multiple parents

     */

    public Set<SingleRead> getMultiLinkSingleReads() {

        return this.multiLinkSingleReads;

    }

    @Override

    public void selectionChanged(SelectionChangedEvent event) {

        this.selection  = event.getSelection();

    }

    public ISelection getSelection() {

        return selection;

    }

    public DerivateLabelProvider getLabelProvider() {

        return labelProvider;

    }

    @Override

    public boolean postOperation(CdmBase objectAffectedByOperation) {

        refreshTree();

        if(objectAffectedByOperation!=null){

            changed(objectAffectedByOperation);

        }

        return true;

    }

    @Override

    public boolean onComplete() {

        return true;

    }

    @Override

    public void dispose() {

        ((DerivateViewEditorInput) getEditorInput()).dispose();

        super.dispose();

    }

    @Override

    public boolean canAttachMedia() {

        return true;

    }

}