eu.etaxonomy.taxeditor.store/src/main/java/eu/etaxonomy/taxeditor/featuretree/AvailableFeaturesWizardPage.java

   1 // $Id$
   2 /**
   3 * Copyright (C) 2007 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
  11 package eu.etaxonomy.taxeditor.featuretree;
  12
  13 import java.util.HashSet;
  14 import java.util.List;
  15 import java.util.Set;
  16
  17 import org.eclipse.jface.viewers.CheckboxTableViewer;
  18 import org.eclipse.jface.wizard.WizardPage;
  19 import org.eclipse.swt.SWT;
  20 import org.eclipse.swt.layout.GridData;
  21 import org.eclipse.swt.layout.GridLayout;
  22 import org.eclipse.swt.widgets.Composite;
  23
  24 import eu.etaxonomy.cdm.model.description.Feature;
  25 import eu.etaxonomy.cdm.model.description.FeatureTree;
  26 import eu.etaxonomy.taxeditor.preference.menu.DefinedTermBaseContentProvider;
  27 import eu.etaxonomy.taxeditor.preference.menu.DefinedTermBaseLabelProvider;
  28 import eu.etaxonomy.taxeditor.store.CdmStore;
  29
  30 /**
  31  * <p>AvailableFeaturesWizardPage class.</p>
  32  *
  33  * @author n.hoffmann
  34  * @created Aug 5, 2010
  35  * @version 1.0
  36  */
  37 public class AvailableFeaturesWizardPage extends WizardPage {
  38
  39         private CheckboxTableViewer viewer;
  40         private FeatureTree featureTree;
  41
  42         /**
  43          * <p>Constructor for AvailableFeaturesWizardPage.</p>
  44          *
  45          * @param featureTree a {@link eu.etaxonomy.cdm.model.description.FeatureTree} object.
  46          */
  47         protected AvailableFeaturesWizardPage(FeatureTree featureTree) {
  48                 super("AvailableFeaturesWizardPage");
  49                 this.featureTree = featureTree;
  50                 setTitle("Available Features");
  51                 setDescription("Please select features you want to add to the tree. Features already in the tree do not show.");
  52         }
  53
  54         /* (non-Javadoc)
  55          * @see org.eclipse.jface.dialogs.IDialogPage#createControl(org.eclipse.swt.widgets.Composite)
  56          */
  57         /** {@inheritDoc} */
  58         @Override
  59         public void createControl(Composite parent) {
  60                 Composite composite = new Composite(parent, SWT.NULL);
  61                 composite.setLayout(new GridLayout());
  62
  63                 viewer = CheckboxTableViewer.newCheckList(composite, SWT.NULL);
  64
  65                 viewer.getControl().setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));
  66
  67                 viewer.setContentProvider(new DefinedTermBaseContentProvider());
  68                 viewer.setLabelProvider(new DefinedTermBaseLabelProvider());
  69
  70                 viewer.setInput(getAvailableFeatures());
  71
  72                 setControl(composite);
  73         }
  74
  75         /**
  76          * <p>getSelectedFeatures</p>
  77          *
  78          * @return a {@link java.util.Set} object.
  79          */
  80         public Set<Feature> getSelectedFeatures() {
  81
  82                 Object[] checkedElements = viewer.getCheckedElements();
  83
  84                 Set<Feature> features = new HashSet<Feature>(checkedElements.length);
  85
  86                 for(Object element : checkedElements){
  87                         features.add((Feature) element);
  88                 }
  89
  90                 return features;
  91         }
  92
  93         /**
  94          * We want to show only features that are not already in the featureTree
  95          *
  96          * @return
  97          */
  98         private List<Feature> getAvailableFeatures(){
  99
 100                 List<Feature> preferredFeatures = CdmStore.getTermManager().getPreferredTerms(Feature.class);
 101
 102                 for(Feature featureAlreadyInTree : featureTree.getDistinctFeatures()){
 103                         preferredFeatures.remove(featureAlreadyInTree);
 104                 }
 105
 106                 return preferredFeatures;
 107
 108         }
 109 }