Project

General

Profile

Download (2.96 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

    
14
import org.eclipse.jface.viewers.TreeNodeContentProvider;
15

    
16
import eu.etaxonomy.cdm.api.service.ITermService;
17
import eu.etaxonomy.cdm.api.service.IVocabularyService;
18
import eu.etaxonomy.cdm.model.common.DefinedTermBase;
19
import eu.etaxonomy.cdm.model.common.TermVocabulary;
20
import eu.etaxonomy.cdm.persistence.dto.UuidAndTitleCache;
21
import eu.etaxonomy.taxeditor.store.CdmStore;
22

    
23
/**
24
 *
25
 * @author pplitzner
26
 * @since Oct 29, 2018
27
 *
28
 */
29
public class TermUuidAndTitleCacheContentProvider extends TreeNodeContentProvider {
30

    
31
	@Override
32
	public Object[] getElements(Object inputElement) {
33
		Collection<TermVocabulary> inputElements = (Collection<TermVocabulary>) inputElement;
34
		return inputElements.toArray();
35
	}
36

    
37
	@Override
38
	public Object[] getChildren(Object parentElement) {
39
	    Collection<Object> children = new HashSet<>();
40
	    if(parentElement instanceof TermVocabulary){
41
	        children.addAll(CdmStore.getService(IVocabularyService.class).getTopLevelTerms(((TermVocabulary)parentElement).getId()));
42
	    } else {
43
	        Collection<Object> includes = new HashSet<>();
44
	        Collection<Object> generalizationOfs = new HashSet<>();
45
	        if (parentElement instanceof DefinedTermBase) {
46
	            includes = ((DefinedTermBase) parentElement).getIncludes();
47
	            generalizationOfs = ((DefinedTermBase) parentElement).getGeneralizationOf();
48
	        }
49
	        else if(parentElement instanceof UuidAndTitleCache){
50
	            includes.addAll(CdmStore.getService(ITermService.class).getIncludesAsUuidAndTitleCache((UuidAndTitleCache<DefinedTermBase>) parentElement));
51
	            generalizationOfs.addAll(CdmStore.getService(ITermService.class).getKindOfsAsUuidAndTitleCache((UuidAndTitleCache<DefinedTermBase>) parentElement));
52
	        }
53
	        children.addAll(includes);
54
	        children.addAll(generalizationOfs);
55
	    }
56
	    return children.toArray();
57
	}
58

    
59
	@Override
60
	public Object getParent(Object element) {
61
		if(element instanceof DefinedTermBase){
62
			DefinedTermBase definedTermBase = (DefinedTermBase)element;
63
			if (definedTermBase.getPartOf() != null) {
64
			    return definedTermBase.getPartOf();
65
			}else if (definedTermBase.getKindOf() != null) {
66
                return definedTermBase.getKindOf();
67
            } else {
68
			    return definedTermBase.getVocabulary();
69
			}
70
		}
71
		else if(element instanceof UuidAndTitleCache){
72
		    return CdmStore.getService(ITermService.class).getParentUuidAndTitleCache((UuidAndTitleCache)element);
73
		}
74
		return null;
75

    
76
	}
77

    
78
	@Override
79
	public boolean hasChildren(Object element) {
80
		if (getChildren(element) != null){
81
			return getChildren(element).length > 0;
82
		}
83
		return false;
84
	}
85

    
86
}
(6-6/9)