- added derivate hierarchy section
[taxeditor.git] / eu.etaxonomy.taxeditor.editor / src / main / java / eu / etaxonomy / taxeditor / editor / view / derivate / DerivateView.java
index 566fc6f9670cb2a6ba0258aeccc7cd1ba79ac5d6..8dc0e35a4d194e0638e7354ff64d19437d6ae50b 100644 (file)
@@ -1,11 +1,25 @@
 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 java.util.UUID;
+
 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;
@@ -18,30 +32,53 @@ import org.eclipse.ui.part.EditorPart;
 
 import eu.etaxonomy.cdm.api.conversation.ConversationHolder;
 import eu.etaxonomy.cdm.api.conversation.IConversationEnabled;
-import eu.etaxonomy.cdm.model.occurrence.FieldUnit;
+import eu.etaxonomy.cdm.api.service.IOccurrenceService;
+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.model.IDirtyMarkableSelectionProvider;
+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.ui.campanula.derivatesearch.DerivateLabelProvider;
+import eu.etaxonomy.taxeditor.operation.IPostOperationEnabled;
+import eu.etaxonomy.taxeditor.store.CdmStore;
+import eu.etaxonomy.taxeditor.view.derivateSearch.DerivateContentProvider;
+import eu.etaxonomy.taxeditor.view.derivateSearch.DerivateLabelProvider;
 
 /**
- * Displays the derivate hierarchy of all derivates belonging to a taxon.
+ * Displays the derivate hierarchy of the specimen specified in the editor input.
  *
  */
-public class DerivateView extends EditorPart implements IPartContentHasFactualData, IDirtyMarkableSelectionProvider, IConversationEnabled, IPartContentHasDetails, IPartContentHasSupplementalData {
+public class DerivateView extends EditorPart implements IPartContentHasFactualData, IDirtyMarkable,
+        IConversationEnabled, IPartContentHasDetails, IPartContentHasSupplementalData, IPartContentHasMedia,
+        ISelectionChangedListener, IPostOperationEnabled {
 
-       /** Constant <code>ID="eu.etaxonomy.taxeditor.editor.view.desc"{trunked}</code> */
-       public static final String ID = "eu.etaxonomy.taxeditor.editor.view.derivate.DerivateView";
+    public static final String ID = "eu.etaxonomy.taxeditor.editor.view.derivate.DerivateView";
 
        private ConversationHolder conversation;
 
-       protected TreeViewer viewer;
+       private TreeViewer viewer;
 
     private boolean isDirty;
 
+    private final int dndOperations = DND.DROP_MOVE;
+
+    private DerivateLabelProvider labelProvider;
+
+    private Set<SingleRead> multiLinkSingleReads;
+
+    private ISelection selection = null;
+
+    /**
+     * Default constructor
+     */
+    public DerivateView() {
+    }
+
     /* (non-Javadoc)
      * @see org.eclipse.ui.part.WorkbenchPart#createPartControl(org.eclipse.swt.widgets.Composite)
      */
@@ -49,29 +86,41 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
     public void createPartControl(Composite parent) {
         viewer = new TreeViewer(new Tree(parent, SWT.MULTI | SWT.H_SCROLL | SWT.V_SCROLL | SWT.FULL_SELECTION));
         viewer.setContentProvider(new DerivateContentProvider());
-        viewer.setLabelProvider(new DerivateLabelProvider());
+        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(editorInput);
+        viewer.setInput(((DerivateViewEditorInput) editorInput).getRootUUIDs());
+        //set selection to selected derivate if only one was selected
         if(editorInput instanceof DerivateViewEditorInput){
-            SpecimenOrObservationBase<?> specimen = ((DerivateViewEditorInput) editorInput).getDerivate();
-            FieldUnit fieldUnit = ((DerivateViewEditorInput) editorInput).getFieldUnit();
-            setPartName(fieldUnit.toString());
-            viewer.setSelection(new StructuredSelection(specimen));
+            Set<UUID> derivateUUIDs = ((DerivateViewEditorInput) editorInput).getDerivateUUIDs();
+            if(derivateUUIDs.size()==1){
+                SpecimenOrObservationBase<?> specimen = CdmStore.getService(IOccurrenceService.class).load(derivateUUIDs.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));
     }
 
     /* (non-Javadoc)
@@ -79,7 +128,12 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
      */
     @Override
     public void doSave(IProgressMonitor monitor) {
-        monitor.beginTask("Saving Derivates", 3);
+        String taskName = "Saving hierarchy";
+        if(getEditorInput() instanceof DerivateViewEditorInput){
+            DerivateViewEditorInput derivateViewEditorInput = (DerivateViewEditorInput) getEditorInput();
+            taskName += " "+derivateViewEditorInput.getName();
+        }
+        monitor.beginTask(taskName, 3);
         if (!conversation.isBound()) {
             conversation.bind();
         }
@@ -93,6 +147,7 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
         monitor.worked(1);
         monitor.done();
         firePropertyChange(PROP_DIRTY);
+        refreshTree();
     }
 
     /* (non-Javadoc)
@@ -100,8 +155,6 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
      */
     @Override
     public void doSaveAs() {
-        // TODO Auto-generated method stub
-
     }
 
     /* (non-Javadoc)
@@ -109,6 +162,10 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
      */
     @Override
     public String getTitleToolTip() {
+        if(getEditorInput() instanceof DerivateViewEditorInput){
+            DerivateViewEditorInput derivateViewEditorInput = (DerivateViewEditorInput) getEditorInput();
+            return derivateViewEditorInput.getName();
+        }
         return "Derivate View";
     }
 
@@ -120,7 +177,9 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
         setSite(site);
         setInput(input);
         if(input instanceof DerivateViewEditorInput){
-            conversation = ((DerivateViewEditorInput) input).getConversationHolder();
+            DerivateViewEditorInput derivateViewEditorInput = (DerivateViewEditorInput) input;
+            conversation = derivateViewEditorInput.getConversationHolder();
+            setPartName(derivateViewEditorInput.getName());
         }
     }
 
@@ -144,7 +203,6 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
      */
     @Override
     public boolean isSaveAsAllowed() {
-        // TODO Auto-generated method stub
         return false;
     }
 
@@ -154,6 +212,7 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
      */
     @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();
@@ -165,8 +224,6 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
      */
     @Override
     public void update(CdmDataChangeMap changeEvents) {
-        // TODO Auto-generated method stub
-
     }
 
     /* (non-Javadoc)
@@ -177,14 +234,105 @@ public class DerivateView extends EditorPart implements IPartContentHasFactualDa
         return conversation;
     }
 
+    /**
+     * @return the viewer
+     */
+    public TreeViewer getViewer() {
+        return viewer;
+    }
+
     /* (non-Javadoc)
      * @see eu.etaxonomy.taxeditor.model.IDirtyMarkableSelectionProvider#changed(java.lang.Object)
      */
     @Override
     public void changed(Object element) {
-        viewer.setInput(getEditorInput());
         setDirty(true);
         firePropertyChange(IEditorPart.PROP_DIRTY);
+        viewer.refresh();
+    }
+
+    /* (non-Javadoc)
+     * @see eu.etaxonomy.taxeditor.model.IDirtyMarkableSelectionProvider#forceDirty()
+     */
+    @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();
+    }
+
+
+    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;
+    }
+
+    /* (non-Javadoc)
+     * @see org.eclipse.jface.viewers.ISelectionChangedListener#selectionChanged(org.eclipse.jface.viewers.SelectionChangedEvent)
+     */
+    @Override
+    public void selectionChanged(SelectionChangedEvent event) {
+        this.selection  = event.getSelection();
+    }
+
+    public ISelection getSelection() {
+        return selection;
+    }
+
+    public DerivateLabelProvider getLabelProvider() {
+        return labelProvider;
+    }
+
+    /* (non-Javadoc)
+     * @see eu.etaxonomy.taxeditor.operation.IPostOperationEnabled#postOperation(eu.etaxonomy.cdm.model.common.CdmBase)
+     */
+    @Override
+    public boolean postOperation(CdmBase objectAffectedByOperation) {
+        refreshTree();
+        if(objectAffectedByOperation!=null){
+            changed(objectAffectedByOperation);
+        }
+        return true;
+    }
+
+    /* (non-Javadoc)
+     * @see eu.etaxonomy.taxeditor.operation.IPostOperationEnabled#onComplete()
+     */
+    @Override
+    public boolean onComplete() {
+        return true;
     }
 
 }