Project

General

Profile

Download (2.27 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.editor.key.polytomous;
11

    
12
import java.util.ArrayList;
13
import java.util.List;
14

    
15
import org.eclipse.jface.viewers.Viewer;
16
import org.eclipse.zest.core.viewers.IGraphContentProvider;
17

    
18
import eu.etaxonomy.cdm.model.description.PolytomousKey;
19
import eu.etaxonomy.cdm.model.description.PolytomousKeyNode;
20
import eu.etaxonomy.taxeditor.editor.Messages;
21
import eu.etaxonomy.taxeditor.model.PolytomousKeyRelationship;
22

    
23
/**
24
 *
25
 * @author n.hoffmann
26
 * @created Mar 30, 2011
27
 * @version 1.0
28
 */
29
class PolytomousKeyGraphContentProvider implements IGraphContentProvider {
30

    
31
	private List<PolytomousKeyRelationship> relations;
32

    
33
	@Override
34
	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
35

    
36
	}
37

    
38
	@Override
39
	public void dispose() {
40

    
41
	}
42

    
43
	@Override
44
	public Object[] getElements(Object inputElement) {
45
		relations = new ArrayList<PolytomousKeyRelationship>();
46

    
47
		if (inputElement instanceof PolytomousKey) {
48
			getTreeRecursively(relations, inputElement);
49
		}
50
		return relations.toArray();
51
	}
52

    
53
	private void getTreeRecursively(List<PolytomousKeyRelationship> relations,
54
			Object parent) {
55
		List<PolytomousKeyNode> children;
56

    
57
		if (parent instanceof PolytomousKeyNode) {
58
			((PolytomousKeyNode) parent).removeNullValueFromChildren();
59
			children = ((PolytomousKeyNode) parent).getChildren();
60
		} else if (parent instanceof PolytomousKey) {
61
			children = new ArrayList<PolytomousKeyNode>();
62
			children.add(((PolytomousKey) parent).getRoot());
63
		} else {
64
			throw new RuntimeException(
65
					Messages.PolytomousKeyGraphContentProvider_WRONG_PARENT
66
							+ parent.getClass());
67
		}
68

    
69
		for (PolytomousKeyNode childNode : children) {
70
			relations.add(new PolytomousKeyRelationship(parent, childNode));
71
			getTreeRecursively(relations, childNode);
72
		}
73
	}
74

    
75
	@Override
76
	public Object getSource(Object relationship) {
77
		return ((PolytomousKeyRelationship) relationship).getSource();
78
	}
79

    
80
	@Override
81
	public Object getDestination(Object relationship) {
82
		return ((PolytomousKeyRelationship) relationship).getDestination();
83
	}
84

    
85
}
(4-4/11)