Project

General

Profile

Download (3.61 KB) Statistics
| Branch: | Tag: | Revision:
1
/**
2
* Copyright (C) 2009 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
package eu.etaxonomy.taxeditor.editor.definedterm;
10

    
11
import java.util.Collection;
12
import java.util.HashSet;
13
import java.util.Set;
14
import java.util.SortedSet;
15

    
16
import org.eclipse.jface.viewers.ITreeContentProvider;
17
import org.eclipse.jface.viewers.Viewer;
18

    
19
import eu.etaxonomy.cdm.model.common.DefinedTermBase;
20
import eu.etaxonomy.cdm.model.common.TermVocabulary;
21
import eu.etaxonomy.taxeditor.store.CdmStore;
22

    
23
/**
24
 * @author l.morris
25
 * @date 8 Dec 2011
26
 *
27
 */
28
public class TermContentProvider implements ITreeContentProvider {
29

    
30
	/* (non-Javadoc)
31
	 * @see org.eclipse.jface.viewers.IContentProvider#dispose()
32
	 */
33
	@Override
34
	public void dispose() {
35
		// TODO Auto-generated method stub
36

    
37
	}
38

    
39
	/* (non-Javadoc)
40
	 * @see org.eclipse.jface.viewers.IContentProvider#inputChanged(org.eclipse.jface.viewers.Viewer, java.lang.Object, java.lang.Object)
41
	 */
42
	@Override
43
	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
44
		// TODO Auto-generated method stub
45

    
46
	}
47

    
48
	/* (non-Javadoc)
49
	 * @see org.eclipse.jface.viewers.IStructuredContentProvider#getElements(java.lang.Object)
50
	 */
51
	@Override
52
	public Object[] getElements(Object inputElement) {
53

    
54
		Collection<TermVocabulary> inputElements = (Collection<TermVocabulary>) inputElement;
55
		return inputElements.toArray();
56

    
57
	}
58

    
59
	/* (non-Javadoc)
60
	 * @see org.eclipse.jface.viewers.ITreeContentProvider#getChildren(java.lang.Object)
61
	 */
62
	@Override
63
	public Object[] getChildren(Object parentElement) {
64

    
65
		if(parentElement instanceof TermVocabulary){
66
			return getTopLevelElements((TermVocabulary)parentElement);
67
		} else if (parentElement instanceof DefinedTermBase) {
68
			Set<Object> includes = ((DefinedTermBase) parentElement).getIncludes();
69
			Set<Object> generalizationOfs = ((DefinedTermBase) parentElement).getGeneralizationOf();
70
			generalizationOfs.addAll(includes);
71
            return generalizationOfs.toArray();
72
		}
73
		return null;
74
	}
75

    
76
	/**
77
	 *
78
	 * @param vocabulary
79
	 * @return An array of DefinedTermBase objects that do not have parents
80
	 *
81
	 * TODO: Needs to be implemented in cdmlib
82
	 */
83
	private Object[] getTopLevelElements(TermVocabulary vocabulary) {
84

    
85
		SortedSet<DefinedTermBase> terms = vocabulary.getTermsOrderedByLabels(CdmStore.getDefaultLanguage());
86
		Set<DefinedTermBase> topLevelTerms = new HashSet<DefinedTermBase>();
87

    
88
		for (DefinedTermBase term : terms){
89
			 if (term.getPartOf() == null && term.getKindOf() == null){
90
				 topLevelTerms.add(term);
91
			 }
92
		}
93
	 	return topLevelTerms.toArray();
94
	}
95

    
96
	/* (non-Javadoc)
97
	 * @see org.eclipse.jface.viewers.ITreeContentProvider#getParent(java.lang.Object)
98
	 */
99
	@Override
100
	public Object getParent(Object element) {
101

    
102
		if(element instanceof DefinedTermBase){
103
			DefinedTermBase definedTermBase = (DefinedTermBase)element;
104
			if (definedTermBase.getPartOf() != null) {
105
			    return definedTermBase.getPartOf();
106
			}else if (definedTermBase.getKindOf() != null) {
107
                return definedTermBase.getKindOf();
108
            } else {
109
			    return definedTermBase.getVocabulary();
110
			}
111
		}
112
		return null;
113

    
114
	}
115

    
116
	/* (non-Javadoc)
117
	 * @see org.eclipse.jface.viewers.ITreeContentProvider#hasChildren(java.lang.Object)
118
	 */
119
	@Override
120
	public boolean hasChildren(Object element) {
121
		if (getChildren(element) != null){
122
			return getChildren(element).length > 0;
123
		}
124
		return false;
125
	}
126

    
127
}
(6-6/8)