summaryrefslogtreecommitdiff
path: root/BJC-Utils2/src/main/java/bjc/utils/funcdata/bst/BinarySearchTreeLeaf.java
blob: 2696577bafc7d5a7fc89e204ed4f17fa4d58c3db (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package bjc.utils.funcdata.bst;

import java.util.Comparator;
import java.util.function.BiFunction;
import java.util.function.Function;
import java.util.function.Predicate;

/**
 * A leaf in a tree.
 *
 * @author ben
 *
 * @param <T>
 *                The data stored in the tree.
 */
public class BinarySearchTreeLeaf<T> implements ITreePart<T> {
	/**
	 * The data held in this tree leaf
	 */
	protected T data;

	/**
	 * Whether this node is soft-deleted or not
	 */
	protected boolean isDeleted;

	/**
	 * Create a new leaf holding the specified data.
	 *
	 * @param element
	 *                The data for the leaf to hold.
	 */
	public BinarySearchTreeLeaf(T element) {
		data = element;
	}

	@Override
	public void add(T element, Comparator<T> comparator) {
		throw new IllegalArgumentException("Can't add to a leaf.");
	}

	@Override
	public <E> E collapse(Function<T, E> leafTransformer, BiFunction<E, E, E> branchCollapser) {
		if(leafTransformer == null) throw new NullPointerException("Transformer must not be null");

		return leafTransformer.apply(data);
	}

	@Override
	public boolean contains(T element, Comparator<T> comparator) {
		return this.data.equals(element);
	}

	@Override
	public T data() {
		return data;
	}

	@Override
	public void delete(T element, Comparator<T> comparator) {
		if(data.equals(element)) {
			isDeleted = true;
		}
	}

	@Override
	public boolean directedWalk(DirectedWalkFunction<T> treeWalker) {
		if(treeWalker == null) throw new NullPointerException("Tree walker must not be null");

		switch(treeWalker.walk(data)) {
		case SUCCESS:
			return true;
		// We don't have any children to care about
		case FAILURE:
		case LEFT:
		case RIGHT:
		default:
			return false;
		}
	}

	@Override
	public boolean forEach(TreeLinearizationMethod linearizationMethod, Predicate<T> traversalPredicate) {
		if(traversalPredicate == null) throw new NullPointerException("Predicate must not be null");

		return traversalPredicate.test(data);
	}
}