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.DefinedTermSorter;
  31 import eu.etaxonomy.taxeditor.editor.definedterm.TermContentProvider;
  32 import eu.etaxonomy.taxeditor.editor.definedterm.TermLabelProvider;
  33 import eu.etaxonomy.taxeditor.store.CdmStore;
  34 import eu.etaxonomy.taxeditor.ui.element.LayoutConstants;
  35
  36 /**
  37  * @author n.hoffmann
  38  * @created Aug 5, 2010
  39  * @version 1.0
  40  */
  41 public class AvailableFeaturesWizardPage extends WizardPage {
  42
  43         private CheckboxTreeViewer viewer;
  44
  45         protected AvailableFeaturesWizardPage() {
  46                 super("AvailableFeaturesWizardPage");
  47                 setTitle("Available Features");
  48                 setDescription("Please select features you want to add to the tree.");
  49         }
  50
  51         /** {@inheritDoc} */
  52         @Override
  53         public void createControl(Composite parent) {
  54                 Composite composite = new Composite(parent, SWT.NULL);
  55                 composite.setLayout(new GridLayout());
  56
  57                 viewer = new CheckboxTreeViewer(composite, SWT.NULL);
  58
  59                 viewer.getControl().setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));
  60
  61               viewer.getControl().setLayoutData(LayoutConstants.FILL());
  62                 viewer.setContentProvider(new TermContentProvider());
  63                 viewer.setLabelProvider(new TermLabelProvider());
  64                 viewer.setSorter(new DefinedTermSorter());
  65
  66         viewer.setContentProvider(new TermContentProvider());
  67         viewer.setLabelProvider(new TermLabelProvider());
  68         // When user checks a checkbox in the tree, check all its children
  69         viewer.addCheckStateListener(new ICheckStateListener() {
  70             @Override
  71             public void checkStateChanged(CheckStateChangedEvent event) {
  72                 viewer.setSubtreeChecked(event.getElement(), event.getChecked());
  73             }
  74         });
  75
  76         List<TermVocabulary<DefinedTermBase>> vocs = CdmStore.getService(IVocabularyService.class).findByTermType(TermType.Feature, null);
  77                 viewer.setInput(vocs);
  78
  79                 setControl(composite);
  80         }
  81
  82         public Set<Feature> getSelectedFeatures() {
  83
  84                 Object[] checkedElements = viewer.getCheckedElements();
  85
  86                 Set<Feature> features = new HashSet<Feature>(checkedElements.length);
  87
  88                 for(Object element : checkedElements){
  89                     if(element instanceof Feature){
  90                         features.add((Feature) element);
  91                     }
  92                 }
  93                 return features;
  94         }
  95
  96 }