Project

General

Profile

Download (2.79 KB) Statistics
| Branch: | Tag: | Revision:
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.Arrays;
13
import java.util.HashSet;
14
import java.util.List;
15
import java.util.Set;
16
import java.util.stream.Collectors;
17

    
18
import org.eclipse.jface.layout.GridLayoutFactory;
19
import org.eclipse.jface.wizard.WizardPage;
20
import org.eclipse.swt.SWT;
21
import org.eclipse.swt.widgets.Composite;
22

    
23
import eu.etaxonomy.cdm.api.service.ITermService;
24
import eu.etaxonomy.cdm.api.service.IVocabularyService;
25
import eu.etaxonomy.cdm.model.common.DefinedTermBase;
26
import eu.etaxonomy.cdm.model.common.TermType;
27
import eu.etaxonomy.cdm.model.description.Feature;
28
import eu.etaxonomy.cdm.persistence.dto.TermDto;
29
import eu.etaxonomy.cdm.persistence.dto.TermVocabularyDto;
30
import eu.etaxonomy.taxeditor.editor.definedterm.DefinedTermSorter;
31
import eu.etaxonomy.taxeditor.editor.definedterm.TermDtoContentProvider;
32
import eu.etaxonomy.taxeditor.editor.definedterm.TermDtoLabelProvider;
33
import eu.etaxonomy.taxeditor.preference.wizard.CheckBoxTreeComposite;
34
import eu.etaxonomy.taxeditor.store.CdmStore;
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 CheckBoxTreeComposite treeComposite;
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
        treeComposite = new CheckBoxTreeComposite(parent, new TermDtoContentProvider(), new TermDtoLabelProvider(), SWT.NONE);
55
        List<TermVocabularyDto> vocabularyDtos = CdmStore.getService(IVocabularyService.class).findVocabularyDtoByTermType(TermType.Feature);
56
        treeComposite.getViewer().setComparator(new DefinedTermSorter());
57
        treeComposite.getViewer().setInput(vocabularyDtos);
58
        GridLayoutFactory.fillDefaults().applyTo(treeComposite);
59

    
60
		setControl(treeComposite);
61
	}
62

    
63
	public Set<Feature> getSelectedFeatures() {
64

    
65
        List<Object> checkedElements = Arrays.asList(treeComposite.getViewer().getCheckedElements());
66
        checkedElements = checkedElements.stream().filter(element->element instanceof TermDto).collect(Collectors.toList());
67

    
68
		Set<Feature> features = new HashSet<Feature>();
69

    
70
		for(Object element : checkedElements){
71
		    if(element instanceof TermDto){
72
		        DefinedTermBase feature = CdmStore.getService(ITermService.class).load(((TermDto) element).getUuid());
73
		        features.add((Feature) feature);
74
		    }
75
		}
76
		return features;
77
	}
78

    
79
}
(2-2/5)