eu.etaxonomy.taxeditor.editor/src/main/java/eu/etaxonomy/taxeditor/editor/view/derivate/searchFilter/DerivateSearchCompositeController.java

   1 // $Id$
   2 /**
   3 * Copyright (C) 2013 EDIT
   4 * European Distributed Institute of Taxonomy
   5 * http://www.e-taxonomy.eu
   6 *
   7 * The contents of this file are subject to the Mozilla Public License Version 1.1
   8 * See LICENSE.TXT at the top of this package for the full license terms.
   9 */
  10 package eu.etaxonomy.taxeditor.editor.view.derivate.searchFilter;
  11
  12 import java.util.ArrayList;
  13 import java.util.List;
  14 import java.util.UUID;
  15
  16 import org.eclipse.swt.SWT;
  17 import org.eclipse.swt.widgets.Composite;
  18 import org.eclipse.swt.widgets.Event;
  19 import org.eclipse.swt.widgets.Listener;
  20 import org.eclipse.swt.widgets.Widget;
  21
  22 import eu.etaxonomy.cdm.api.conversation.ConversationHolder;
  23 import eu.etaxonomy.cdm.api.service.IOccurrenceService;
  24 import eu.etaxonomy.cdm.api.service.config.FindOccurrencesConfigurator;
  25 import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;
  26 import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationType;
  27 import eu.etaxonomy.cdm.model.taxon.Taxon;
  28 import eu.etaxonomy.taxeditor.editor.view.derivate.DerivateView;
  29 import eu.etaxonomy.taxeditor.editor.view.derivate.DerivateViewEditorInput;
  30 import eu.etaxonomy.taxeditor.store.CdmStore;
  31 import eu.etaxonomy.taxeditor.store.SearchManager;
  32 import eu.etaxonomy.taxeditor.ui.dialog.selection.SelectionDialogFactory;
  33 import eu.etaxonomy.taxeditor.ui.mvc.combo.EnumTermComboController;
  34 import eu.etaxonomy.taxeditor.view.derivateSearch.DerivateLabelProvider;
  35
  36 /**
  37  * Controller class for the {@link DerivateSearchComposite}.<br>
  38  * This controller contains the logic to search the data base for {@link SpecimenOrObservationBase}s
  39  * and handles GUI events.
  40  *
  41  */
  42 public class DerivateSearchCompositeController implements Listener{
  43
  44     private final DerivateSearchComposite derivateSearchComposite;
  45     private EnumTermComboController<SpecimenOrObservationType> comboSpecimenTypeController;
  46         private DerivateLabelProvider labelProvider;
  47         private Taxon selectedTaxon;
  48     private DerivateView derivativeEditor;
  49
  50     public DerivateSearchCompositeController(Composite parent, DerivateView derivativeEditor) {
  51         this.derivateSearchComposite = new DerivateSearchComposite(parent, SWT.NONE);
  52         this.derivativeEditor = derivativeEditor;
  53         init();
  54     }
  55
  56     private void init(){
  57         //derivate type combo
  58         comboSpecimenTypeController = new EnumTermComboController<SpecimenOrObservationType>(derivateSearchComposite.getComboDerivateType(), null, null, SpecimenOrObservationType.class, 0);
  59         comboSpecimenTypeController.setSelection(SpecimenOrObservationType.Unknown);
  60
  61         //add listeners
  62         derivateSearchComposite.getSearchField().addListener(SWT.KeyDown, this);
  63         derivateSearchComposite.getButtonSearch().addListener(SWT.Selection, this);
  64         derivateSearchComposite.getBtnBrowseTaxa().addListener(SWT.Selection, this);
  65         derivateSearchComposite.getBtnClearTaxon().addListener(SWT.Selection, this);
  66
  67     }
  68
  69     private void searchDerivates(){
  70 //        if(!conversation.isBound()){
  71 //            conversation.bind();
  72 //        }
  73
  74         String queryString = derivateSearchComposite.getSearchField().getText();
  75         SpecimenOrObservationType specimenType = comboSpecimenTypeController.getSelection();
  76
  77         FindOccurrencesConfigurator config = new FindOccurrencesConfigurator();
  78         config.setAssociatedTaxonUuid(selectedTaxon!=null?selectedTaxon.getUuid():null);
  79         config.setSpecimenType(specimenType.equals(SpecimenOrObservationType.Unknown)?null:specimenType);//unknown := all types
  80         config.setTitleSearchString(queryString);
  81         if(selectedTaxon!=null){
  82             config.setRetrieveIndirectlyAssociatedSpecimens(true);
  83             config.setSpecimenType(null);
  84         }
  85
  86         SearchManager searchManager = new SearchManager();
  87         List<SpecimenOrObservationBase> occurrences = searchManager.findOccurrences(config, true);
  88
  89         //filter out specimens of wrong type (only if an associated taxon was selected)
  90         if(selectedTaxon!=null && !specimenType.equals(SpecimenOrObservationType.Unknown)){
  91             List<SpecimenOrObservationBase> specimensOfCorrectType = new ArrayList<SpecimenOrObservationBase>();
  92             for(SpecimenOrObservationBase<?> occurrence:occurrences){
  93                 if(occurrence.getRecordBasis().equals(specimenType)
  94                         || occurrence.getRecordBasis().isKindOf(specimenType)){
  95                     specimensOfCorrectType.add(occurrence);
  96                 }
  97             }
  98             occurrences = specimensOfCorrectType;
  99         }
 100
 101         //filter out assigned specimens
 102         if(derivateSearchComposite.getBtnFilterUndeterminedSpecimen().getSelection()){
 103             List<SpecimenOrObservationBase> specimensWithNoDetermination = new ArrayList<SpecimenOrObservationBase>();
 104             for(SpecimenOrObservationBase<?> result:occurrences){
 105                 if(CdmStore.getService(IOccurrenceService.class).countDeterminations(result, null)==0){
 106                     specimensWithNoDetermination.add(result);
 107                 }
 108             }
 109             occurrences = specimensWithNoDetermination;
 110         }
 111         List<UUID> derivateUuids = new ArrayList<UUID>();
 112         for (SpecimenOrObservationBase specimenOrObservationBase : occurrences) {
 113             derivateUuids.add(specimenOrObservationBase.getUuid());
 114         }
 115         //update tree
 116         DerivateViewEditorInput editorInput = (DerivateViewEditorInput)derivativeEditor.getEditorInput();
 117         editorInput.updateRootEntities(derivateUuids);
 118         derivativeEditor.getViewer().setInput(editorInput.getRootEntities());
 119         derivativeEditor.refreshTree();
 120     }
 121
 122     @Override
 123     public void handleEvent(Event event) {
 124         Widget eventSource = event.widget;
 125         if(eventSource == derivateSearchComposite.getButtonSearch()
 126                 || (eventSource == derivateSearchComposite.getSearchField() && event.keyCode == SWT.CR)){
 127             searchDerivates();
 128         }
 129         else if(eventSource==derivateSearchComposite.getBtnBrowseTaxa()){
 130             selectedTaxon = SelectionDialogFactory.getSelectionFromDialog(Taxon.class, this.derivateSearchComposite.getShell(), derivativeEditor.getConversationHolder(), null);
 131             if(selectedTaxon!=null){
 132                 derivateSearchComposite.getTextTaxonName().setText(selectedTaxon.getTitleCache());
 133             }
 134         }
 135         else if(eventSource==derivateSearchComposite.getBtnClearTaxon()){
 136             selectedTaxon = null;
 137             derivateSearchComposite.getTextTaxonName().setText("");
 138         }
 139     }
 140
 141     /**
 142      * Sets enable status of all input fields
 143      * @param enabled
 144      */
 145     public void setEnabled(boolean enabled) {
 146         if(!derivateSearchComposite.isDisposed()){
 147             derivateSearchComposite.setEnabled(enabled);
 148         }
 149     }
 150
 151     public void setLayoutData(Object layoutData){
 152         derivateSearchComposite.setLayoutData(layoutData);
 153     }
 154
 155     /**
 156      * Resets all input fields
 157      */
 158     public void reset(){
 159         if(!derivateSearchComposite.isDisposed()){
 160             derivateSearchComposite.getSearchField().setText("");
 161             derivateSearchComposite.getComboDerivateType().deselectAll();
 162             comboSpecimenTypeController.setSelection(SpecimenOrObservationType.Unknown);
 163         }
 164         selectedTaxon = null;
 165     }
 166
 167     public void setConversation(ConversationHolder conversationHolder) {
 168         labelProvider.setConversation(conversationHolder);
 169     }
 170
 171 }