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

   1 /**
   2 * Copyright (C) 2007 EDIT
   3 * European Distributed Institute of Taxonomy
   4 * http://www.e-taxonomy.eu
   5 *
   6 * The contents of this file are subject to the Mozilla Public License Version 1.1
   7 * See LICENSE.TXT at the top of this package for the full license terms.
   8 */
   9
  10 package eu.etaxonomy.taxeditor.featuretree;
  11
  12 import java.util.HashSet;
  13 import java.util.List;
  14 import java.util.Set;
  15
  16 import org.eclipse.jface.viewers.CheckStateChangedEvent;
  17 import org.eclipse.jface.viewers.CheckboxTreeViewer;
  18 import org.eclipse.jface.viewers.ICheckStateListener;
  19 import org.eclipse.jface.wizard.WizardPage;
  20 import org.eclipse.swt.SWT;
  21 import org.eclipse.swt.layout.GridData;
  22 import org.eclipse.swt.layout.GridLayout;
  23 import org.eclipse.swt.widgets.Composite;
  24
  25 import eu.etaxonomy.cdm.api.service.IVocabularyService;
  26 import eu.etaxonomy.cdm.model.common.DefinedTermBase;
  27 import eu.etaxonomy.cdm.model.common.TermType;
  28 import eu.etaxonomy.cdm.model.common.TermVocabulary;
  29 import eu.etaxonomy.cdm.model.description.Feature;
  30 import eu.etaxonomy.taxeditor.editor.definedterm.TermContentProvider;
  31 import eu.etaxonomy.taxeditor.editor.definedterm.TermLabelProvider;
  32 import eu.etaxonomy.taxeditor.store.CdmStore;
  33
  34 /**
  35  * @author n.hoffmann
  36  * @created Aug 5, 2010
  37  * @version 1.0
  38  */
  39 public class AvailableFeaturesWizardPage extends WizardPage {
  40
  41         private CheckboxTreeViewer viewer;
  42
  43         protected AvailableFeaturesWizardPage() {
  44                 super("AvailableFeaturesWizardPage");
  45                 setTitle("Available Features");
  46                 setDescription("Please select features you want to add to the tree.");
  47         }
  48
  49         /** {@inheritDoc} */
  50         @Override
  51         public void createControl(Composite parent) {
  52                 Composite composite = new Composite(parent, SWT.NULL);
  53                 composite.setLayout(new GridLayout());
  54
  55                 viewer = new CheckboxTreeViewer(composite, SWT.NULL);
  56
  57                 viewer.getControl().setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));
  58
  59         viewer.setContentProvider(new TermContentProvider());
  60         viewer.setLabelProvider(new TermLabelProvider());
  61         // When user checks a checkbox in the tree, check all its children
  62         viewer.addCheckStateListener(new ICheckStateListener() {
  63             @Override
  64             public void checkStateChanged(CheckStateChangedEvent event) {
  65                 viewer.setSubtreeChecked(event.getElement(), event.getChecked());
  66             }
  67         });
  68
  69         List<TermVocabulary<DefinedTermBase>> vocs = CdmStore.getService(IVocabularyService.class).findByTermType(TermType.Feature, null);
  70                 viewer.setInput(vocs);
  71
  72                 setControl(composite);
  73         }
  74
  75         public Set<Feature> getSelectedFeatures() {
  76
  77                 Object[] checkedElements = viewer.getCheckedElements();
  78
  79                 Set<Feature> features = new HashSet<Feature>(checkedElements.length);
  80
  81                 for(Object element : checkedElements){
  82                     if(element instanceof Feature){
  83                         features.add((Feature) element);
  84                     }
  85                 }
  86                 return features;
  87         }
  88
  89 }