@SuppressWarnings("unchecked")
public <B extends E> boolean add(BoundedHandler<B> handler) {
Class<B> newOne = handler.getBoundClass();
if (newOne == getBoundClass()) { // same class as root
BoundedHandler<E> bh = (BoundedHandler<E>) handler;
boolean targetWasNull = target == null;
if (targetWasNull) {
target = bh;
}
return targetWasNull;
} else if (getBoundClass().isAssignableFrom(newOne)) { // child of root class
for (ClassTree<? extends E> subtree : subclasses.values()) {
Class<? extends E> subtreeClass = subtree.getBoundClass();
if (subtreeClass.isAssignableFrom(newOne)) {
// subtreeClass is a parent - simply add newOne to subtree
ClassTree<? super B> subtree2 = (ClassTree<? super B>) subtree;
return subtree2.add(handler);
} else if (newOne.isAssignableFrom(subtreeClass)) {
// newOne is a parent - remove subtree from this tree, add newOne, add subtreeClass to
// newOne as a child
ClassTree<? extends B> subtree2 = (ClassTree<? extends B>) subtree;
ClassTree<B> value = new ClassTree<B>(handler);
value.add(subtree2.target);
subclasses.remove(subtreeClass);
subclasses.put(newOne, value);
return true;
}
}
subclasses.put(newOne, new ClassTree<B>(handler));
return true;
} else {
return false;
}
}
@SuppressWarnings("unchecked")
public <B extends E> boolean remove(BoundedHandler<B> handler) {
Class<B> newOne = handler.getBoundClass();
if (newOne == getBoundClass()) {
target = null;
return true;
} else if (getBoundClass().isAssignableFrom(newOne)) {
for (ClassTree<? extends E> subtree : subclasses.values()) {
Class<? extends E> subtreeClass = subtree.getBoundClass();
if (subtreeClass.isAssignableFrom(newOne)) {
// subtreeClass is a parent - simply add subtreeClass to newOne subclasses
ClassTree<? super B> subtree2 = (ClassTree<? super B>) subtree;
return subtree2.remove(handler);
}
}
}
return false;
}
public ClassTree(BoundedHandler<E> target) {
super(target.getBoundClass());
this.target = target;
}