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.Collection;
  14 import java.util.List;
  15
  16 import org.eclipse.swt.SWT;
  17 import org.eclipse.swt.layout.GridData;
  18 import org.eclipse.swt.widgets.Composite;
  19 import org.eclipse.swt.widgets.Event;
  20 import org.eclipse.swt.widgets.Listener;
  21 import org.eclipse.swt.widgets.Widget;
  22
  23 import eu.etaxonomy.cdm.api.conversation.ConversationHolder;
  24 import eu.etaxonomy.cdm.api.service.IOccurrenceService;
  25 import eu.etaxonomy.cdm.api.service.config.FindOccurrencesConfigurator;
  26 import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;
  27 import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationType;
  28 import eu.etaxonomy.cdm.model.taxon.Taxon;
  29 import eu.etaxonomy.cdm.model.taxon.TaxonBase;
  30 import eu.etaxonomy.taxeditor.editor.view.derivate.DerivateView;
  31 import eu.etaxonomy.taxeditor.store.CdmStore;
  32 import eu.etaxonomy.taxeditor.store.SearchManager;
  33 import eu.etaxonomy.taxeditor.ui.dialog.selection.SelectionDialogFactory;
  34 import eu.etaxonomy.taxeditor.ui.mvc.combo.EnumTermComboController;
  35 import eu.etaxonomy.taxeditor.view.derivateSearch.DerivateLabelProvider;
  36
  37 /**
  38  * Controller class for the {@link DerivateSearchComposite}.<br>
  39  * This controller contains the logic to search the data base for {@link SpecimenOrObservationBase}s
  40  * and handles GUI events.
  41  *
  42  */
  43 public class DerivateSearchCompositeController implements Listener{
  44
  45     private final DerivateSearchComposite derivateSearchComposite;
  46     private EnumTermComboController<SpecimenOrObservationType> comboSpecimenTypeController;
  47         private DerivateLabelProvider labelProvider;
  48         private Taxon selectedTaxon;
  49     private DerivateView derivativeEditor;
  50
  51     public DerivateSearchCompositeController(Composite parent, DerivateView derivativeEditor) {
  52         this.derivateSearchComposite = new DerivateSearchComposite(parent, SWT.NONE);
  53         this.derivativeEditor = derivativeEditor;
  54         init();
  55     }
  56
  57     private void init(){
  58         //derivate type combo
  59         comboSpecimenTypeController = new EnumTermComboController<SpecimenOrObservationType>(derivateSearchComposite.getComboDerivateType(), null, null, SpecimenOrObservationType.class, 0);
  60         comboSpecimenTypeController.setSelection(SpecimenOrObservationType.Unknown);
  61
  62         //add listeners
  63         derivateSearchComposite.getSearchField().addListener(SWT.KeyDown, this);
  64         derivateSearchComposite.getButtonSearch().addListener(SWT.Selection, this);
  65         derivateSearchComposite.getBtnBrowseTaxa().addListener(SWT.Selection, this);
  66         derivateSearchComposite.getBtnClearTaxon().addListener(SWT.Selection, this);
  67
  68     }
  69
  70     private void searchDerivates(){
  71         String queryString = derivateSearchComposite.getSearchField().getText();
  72         SpecimenOrObservationType specimenType = comboSpecimenTypeController.getSelection();
  73
  74         FindOccurrencesConfigurator config = new FindOccurrencesConfigurator();
  75         config.setAssociatedTaxonUuid(selectedTaxon!=null?selectedTaxon.getUuid():null);
  76         config.setSpecimenType(specimenType.equals(SpecimenOrObservationType.Unknown)?null:specimenType);//unknown := all types
  77         config.setTitleSearchString(queryString);
  78         if(selectedTaxon!=null){
  79             config.setSpecimenType(null);
  80         }
  81
  82         SearchManager searchManager = new SearchManager();
  83         List<SpecimenOrObservationBase> occurrences = searchManager.findOccurrences(config, true);
  84
  85         //filter out specimens of wrong type (only if an associated taxon was selected)
  86         if(selectedTaxon!=null && !specimenType.equals(SpecimenOrObservationType.Unknown)){
  87             List<SpecimenOrObservationBase> specimensOfCorrectType = new ArrayList<SpecimenOrObservationBase>();
  88             for(SpecimenOrObservationBase<?> occurrence:occurrences){
  89                 if(occurrence.getRecordBasis().equals(specimenType)
  90                         || occurrence.getRecordBasis().isKindOf(specimenType)){
  91                     specimensOfCorrectType.add(occurrence);
  92                 }
  93             }
  94             occurrences = specimensOfCorrectType;
  95         }
  96
  97         //filter out (un-)assigned specimens
  98         if(selectedTaxon==null){
  99             int selectionIndex = derivateSearchComposite.getComboTaxonAssignment().getSelectionIndex();
 100             List<SpecimenOrObservationBase<?>> specimenWithAssociations = new ArrayList<SpecimenOrObservationBase<?>>();
 101             if(selectionIndex!=DerivateSearchComposite.ALL_SPECIMENS){
 102                 for (SpecimenOrObservationBase specimenOrObservationBase : occurrences) {
 103                     Collection<TaxonBase<?>> associatedTaxa = CdmStore.getService(IOccurrenceService.class).listAssociatedTaxa(specimenOrObservationBase, null, null, null, null);
 104                     if(!associatedTaxa.isEmpty()){
 105                         specimenWithAssociations.add(specimenOrObservationBase);
 106                         specimenWithAssociations.addAll(CdmStore.getService(IOccurrenceService.class).getFieldUnits(specimenOrObservationBase.getUuid()));
 107                     }
 108                 }
 109             }
 110             if(selectionIndex==DerivateSearchComposite.UNASSIGNED_SPECIMENS){
 111                 occurrences.removeAll(specimenWithAssociations);
 112             }
 113             else if(selectionIndex==DerivateSearchComposite.ASSIGNED_SPECIMENS){
 114                 occurrences = new ArrayList<SpecimenOrObservationBase>(specimenWithAssociations);
 115             }
 116         }
 117         derivativeEditor.updateRootEntities(occurrences);
 118     }
 119
 120     @Override
 121     public void handleEvent(Event event) {
 122         Widget eventSource = event.widget;
 123         if(eventSource == derivateSearchComposite.getButtonSearch()
 124                 || (eventSource == derivateSearchComposite.getSearchField() && event.keyCode == SWT.CR)){
 125             searchDerivates();
 126         }
 127         else if(eventSource==derivateSearchComposite.getBtnBrowseTaxa()){
 128             selectedTaxon = SelectionDialogFactory.getSelectionFromDialog(Taxon.class, this.derivateSearchComposite.getShell(), derivativeEditor.getConversationHolder(), null);
 129             if(selectedTaxon!=null){
 130                 derivateSearchComposite.getTextTaxonName().setText(selectedTaxon.getTitleCache());
 131             }
 132             derivateSearchComposite.getComboTaxonAssignment().select(DerivateSearchComposite.ASSIGNED_SPECIMENS);
 133             derivateSearchComposite.getComboTaxonAssignment().setEnabled(false);
 134             derivateSearchComposite.getLbltaxonAssignment().setEnabled(false);
 135         }
 136         else if(eventSource==derivateSearchComposite.getBtnClearTaxon()){
 137             selectedTaxon = null;
 138             derivateSearchComposite.getTextTaxonName().setText("");
 139             derivateSearchComposite.getComboTaxonAssignment().select(DerivateSearchComposite.ALL_SPECIMENS);
 140             derivateSearchComposite.getComboTaxonAssignment().setEnabled(true);
 141             derivateSearchComposite.getLbltaxonAssignment().setEnabled(true);
 142         }
 143     }
 144
 145     /**
 146      * Sets enable status of all input fields
 147      * @param enabled
 148      */
 149     public void setEnabled(boolean enabled) {
 150         if(!derivateSearchComposite.isDisposed()){
 151             GridData layoutData = (GridData) derivateSearchComposite.getLayoutData();
 152             layoutData.exclude = !enabled;
 153             derivateSearchComposite.setVisible(enabled);
 154             derivateSearchComposite.getParent().layout(true);
 155         }
 156     }
 157
 158     public void setLayoutData(Object layoutData){
 159         derivateSearchComposite.setLayoutData(layoutData);
 160     }
 161
 162     /**
 163      * Resets all input fields
 164      */
 165     public void reset(){
 166         if(!derivateSearchComposite.isDisposed()){
 167             derivateSearchComposite.getSearchField().setText("");
 168             derivateSearchComposite.getComboDerivateType().deselectAll();
 169             comboSpecimenTypeController.setSelection(SpecimenOrObservationType.Unknown);
 170         }
 171         selectedTaxon = null;
 172     }
 173
 174     public void setConversation(ConversationHolder conversationHolder) {
 175         labelProvider.setConversation(conversationHolder);
 176     }
 177
 178 }