eu.etaxonomy.taxeditor.editor/src/main/java/eu/etaxonomy/taxeditor/editor/view/derivate/contextMenu/CreateDerivateContextMenu.java

   1 package eu.etaxonomy.taxeditor.editor.view.derivate.contextMenu;
   2
   3 import org.eclipse.jface.action.ContributionItem;
   4 import org.eclipse.jface.action.IContributionItem;
   5 import org.eclipse.jface.viewers.ISelection;
   6 import org.eclipse.jface.viewers.TreeNode;
   7 import org.eclipse.swt.SWT;
   8 import org.eclipse.swt.events.SelectionAdapter;
   9 import org.eclipse.swt.events.SelectionEvent;
  10 import org.eclipse.swt.widgets.Menu;
  11 import org.eclipse.swt.widgets.MenuItem;
  12 import org.eclipse.ui.IWorkbenchPart;
  13 import org.eclipse.ui.IWorkbenchWindow;
  14 import org.eclipse.ui.PlatformUI;
  15 import org.eclipse.ui.actions.CompoundContributionItem;
  16
  17 import eu.etaxonomy.cdm.api.facade.DerivedUnitFacadeCacheStrategy;
  18 import eu.etaxonomy.cdm.model.media.Media;
  19 import eu.etaxonomy.cdm.model.molecular.DnaSample;
  20 import eu.etaxonomy.cdm.model.molecular.Sequence;
  21 import eu.etaxonomy.cdm.model.molecular.SingleRead;
  22 import eu.etaxonomy.cdm.model.occurrence.DerivationEvent;
  23 import eu.etaxonomy.cdm.model.occurrence.DerivationEventType;
  24 import eu.etaxonomy.cdm.model.occurrence.DerivedUnit;
  25 import eu.etaxonomy.cdm.model.occurrence.FieldUnit;
  26 import eu.etaxonomy.cdm.model.occurrence.MediaSpecimen;
  27 import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;
  28 import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationType;
  29 import eu.etaxonomy.taxeditor.editor.EditorUtil;
  30 import eu.etaxonomy.taxeditor.editor.view.derivate.DerivateView;
  31 import eu.etaxonomy.taxeditor.model.AbstractUtility;
  32
  33 /**
  34  * Context menu for the creation of derivates in the derivate hierarchy.
  35  *
  36  */
  37 public class CreateDerivateContextMenu extends CompoundContributionItem {
  38
  39     /* (non-Javadoc)
  40      * @see org.eclipse.ui.actions.CompoundContributionItem#getContributionItems()
  41      */
  42     @Override
  43     protected IContributionItem[] getContributionItems() {
  44         IContributionItem[] contributionItems = new IContributionItem[] {
  45                 new ContributionItem() {
  46                     @Override
  47                     public void fill(Menu menu, int index) {
  48                         MenuItem addItem = new MenuItem(menu, SWT.CASCADE);
  49                         addItem.setText("Add...");
  50                         Menu addMenu = new Menu(menu);
  51                         addItem.setMenu(addMenu);
  52                         final IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
  53                         final ISelection selection = window.getActivePage().getSelection();
  54                         TreeNode selectedTreeNode = EditorUtil.getTreeNodeOfSelection(selection);
  55                         if(selectedTreeNode!=null){
  56                             //context menu for FieldUnit
  57                             if(selectedTreeNode.getValue() instanceof FieldUnit){
  58                                 createMenuItem(addMenu, "Specimen", selectedTreeNode, DerivedUnit.class, SpecimenOrObservationType.PreservedSpecimen);
  59                                 new MenuItem(addMenu, SWT.SEPARATOR);
  60                                 createMenuItem(addMenu, "Tissue Sample", selectedTreeNode, DerivedUnit.class, SpecimenOrObservationType.TissueSample);
  61                                 createMenuItem(addMenu, "DNA Sample", selectedTreeNode, DnaSample.class, SpecimenOrObservationType.DnaSample);
  62                                 createMenuItem(addMenu, "Media Specimen", selectedTreeNode, MediaSpecimen.class, SpecimenOrObservationType.Media);
  63                             }
  64                             //context menu for Specimen
  65                             else if(selectedTreeNode.getValue() instanceof DerivedUnit && ((DerivedUnit)selectedTreeNode.getValue()).getRecordBasis()==SpecimenOrObservationType.PreservedSpecimen){
  66
  67                                 createMenuItem(addMenu, "Tissue Sample", selectedTreeNode, DerivedUnit.class, SpecimenOrObservationType.TissueSample);
  68                                 createMenuItem(addMenu, "DNA Sample", selectedTreeNode, DnaSample.class, SpecimenOrObservationType.DnaSample);
  69                                 createMenuItem(addMenu, "Media Specimen", selectedTreeNode, MediaSpecimen.class, SpecimenOrObservationType.Media);
  70                             }
  71                             //context menu for TissueSample
  72                             else if(selectedTreeNode.getValue() instanceof DerivedUnit && ((DerivedUnit)selectedTreeNode.getValue()).getRecordBasis()==SpecimenOrObservationType.TissueSample){
  73                                 createMenuItem(addMenu, "DnaSample", selectedTreeNode, DnaSample.class, null);
  74                             }
  75                             //context menu for DnaSample
  76                             else if(selectedTreeNode.getValue() instanceof DnaSample){
  77                                 createMenuItem(addMenu, "Consensus Sequence", selectedTreeNode, Sequence.class, null);
  78                             }
  79                             //context menu for Sequence
  80                             else if(selectedTreeNode.getValue() instanceof Sequence){
  81                                 createMenuItem(addMenu, "Single Read", selectedTreeNode, SingleRead.class, null);
  82                             }
  83                             //default
  84                             else{
  85                                 MenuItem menuItem = new MenuItem(addMenu, SWT.NONE);
  86                                 menuItem.setText("No child derivates");
  87                                 menuItem.setEnabled(false);
  88                             }
  89                         }
  90                     }
  91
  92
  93                 }
  94         };
  95         return contributionItems;
  96     }
  97
  98     private void createMenuItem(Menu menu, String menuLabel, TreeNode selectedNode, Class<?> childClass, SpecimenOrObservationType specimenType) {
  99         MenuItem menuItem = new MenuItem(menu, SWT.NONE);
 100         menuItem.setText(menuLabel);
 101         menuItem.addSelectionListener(new WidgetSelectionListener(selectedNode, childClass, specimenType));
 102     }
 103
 104     private class WidgetSelectionListener extends SelectionAdapter{
 105
 106         private final TreeNode selectedNode;
 107         //TODO: we need a common interface for the generic like e.g. "IDerivateHierarchyNode"
 108         private final Class<?> childClass;
 109         private final SpecimenOrObservationType specimenType;
 110
 111         /**
 112          * @param selectedNode
 113          * @param specimenType
 114          */
 115         public WidgetSelectionListener(TreeNode selectedNode, Class<?> childClass, SpecimenOrObservationType specimenType) {
 116             this.selectedNode = selectedNode;
 117             this.childClass = childClass;
 118             this.specimenType = specimenType;
 119         }
 120
 121         /* (non-Javadoc)
 122          * @see org.eclipse.swt.events.SelectionListener#widgetSelected(org.eclipse.swt.events.SelectionEvent)
 123          */
 124         @Override
 125         public void widgetSelected(SelectionEvent e) {
 126             Object selectedDerivate = selectedNode.getValue();
 127             if(selectedDerivate instanceof DnaSample){
 128                 DnaSample dnaSample = (DnaSample)selectedDerivate;
 129                 if(childClass.equals(Sequence.class)){
 130                     dnaSample.addSequence(Sequence.NewInstance(""));//why does this not have DnaSample as constructor parameter?
 131                 }
 132             }
 133             else if(selectedDerivate instanceof Sequence){
 134                 ((Sequence) selectedDerivate).addSingleRead(SingleRead.NewInstance());
 135             }
 136             else if(selectedDerivate instanceof SpecimenOrObservationBase && DerivedUnit.class.isAssignableFrom(childClass)){
 137                 SpecimenOrObservationBase<?> specimenOrObservationBase = (SpecimenOrObservationBase<?>)selectedDerivate;
 138                 DerivedUnit derivedUnit;
 139                 if(childClass.equals(DnaSample.class)){
 140                     derivedUnit = DnaSample.NewInstance();
 141                 }
 142                 else if(childClass.equals(MediaSpecimen.class)){
 143                     derivedUnit = MediaSpecimen.NewInstance(specimenType);
 144                     //a MediaSpecimen with no media attached does not make sense. Hence we add one
 145                     ((MediaSpecimen)derivedUnit).setMediaSpecimen(Media.NewInstance());
 146                 }
 147                 else{
 148                     derivedUnit = DerivedUnit.NewInstance(specimenType);
 149                 }
 150                 specimenOrObservationBase.addDerivationEvent(DerivationEvent.NewSimpleInstance(specimenOrObservationBase, derivedUnit, DerivationEventType.GATHERING_IN_SITU()));
 151                 derivedUnit.setCacheStrategy(new DerivedUnitFacadeCacheStrategy());
 152                 derivedUnit.getTitleCache(); //update title cache
 153             }
 154
 155             //refresh view
 156             IWorkbenchPart activePart = AbstractUtility.getActivePart();
 157             if(activePart instanceof DerivateView){
 158                 DerivateView derivateView = (DerivateView)activePart;
 159                 derivateView.changed(null);
 160                 derivateView.refreshTree();
 161             }
 162         }
 163     }
 164
 165 }