com.speedment.runtime.config.internal.util.Trees Maven / Gradle / Ivy
Show all versions of generator-deploy Show documentation
/*
*
* Copyright (c) 2006-2019, Speedment, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License"); You may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package com.speedment.runtime.config.internal.util;
import static com.speedment.common.invariant.NullUtil.requireNonNulls;
import java.util.ArrayDeque;
import java.util.Objects;
import java.util.Optional;
import java.util.Queue;
import java.util.function.Function;
import java.util.stream.Stream;
/**
*
* @author Per Minborg
*/
public final class Trees {
/**
* The TraversalType can control how a tree is walked.
*
* Wikipedia
* article
*/
public enum TraversalOrder {
DEPTH_FIRST_PRE,
/*DEPTH_FIRST_IN, Supported only for left/right trees*/
DEPTH_FIRST_POST,
BREADTH_FIRST
}
/**
* The walking order of the tree.
*/
public enum WalkingOrder {
FORWARD, BACKWARD
}
public static Stream walk(
T first,
Function traverser) {
requireNonNulls(first, traverser);
return walk(
first,
traverser,
WalkingOrder.FORWARD,
Stream.builder()
).build();
}
public static Stream walk(
T first,
Function traverser,
WalkingOrder order) {
requireNonNulls(first, traverser, order);
return walk(first, traverser, order, Stream.builder()).build();
}
public static Stream walkOptional(
T first,
Function> traverser) {
requireNonNulls(first, traverser);
return walkOptional(
first,
traverser,
WalkingOrder.FORWARD,
Stream.builder()
).build();
}
public static Stream walkOptional(
T first,
Function> traverser,
WalkingOrder order) {
requireNonNulls(first, traverser, order);
return walkOptional(first, traverser, order, Stream.builder()).build();
}
public static Stream traverse(
T first,
Function> traverser,
TraversalOrder traversalOrder) {
requireNonNulls(first, traverser, traversalOrder);
if (traversalOrder == TraversalOrder.BREADTH_FIRST) {
return traverseBredthFirst(
first,
traverser,
Stream.builder()
).build();
} else {
return traverse(
first,
traverser,
traversalOrder,
Stream.builder()
).build();
}
}
//
// Private support methods
//
private static Stream.Builder walkOptional(
T first,
Function> traverser,
WalkingOrder order,
Stream.Builder builder) {
requireNonNulls(first, traverser, order, builder);
if (order == WalkingOrder.FORWARD) {
builder.add(first);
}
traverser.apply(first)
.ifPresent(p -> walkOptional(p, traverser, order, builder));
if (order == WalkingOrder.BACKWARD) {
builder.add(first);
}
return builder;
}
private static Stream.Builder walk(
T first,
Function traverser,
WalkingOrder order,
Stream.Builder builder) {
requireNonNulls(first, traverser, order, builder);
if (order == WalkingOrder.FORWARD) {
builder.add(first);
}
final T next = traverser.apply(first);
if (next != null) {
walk(next, traverser, order, builder);
}
if (order == WalkingOrder.BACKWARD) {
builder.add(first);
}
return builder;
}
private static Stream.Builder traverse(
T first,
Function> traverser,
TraversalOrder traversalOrder,
Stream.Builder builder) {
requireNonNulls(first, traverser, traversalOrder, builder);
if (first == null) {
return builder;
}
if (traversalOrder == TraversalOrder.DEPTH_FIRST_PRE) {
builder.add(first);
}
final Stream next = traverser.apply(first);
if (next != null) {
next.filter(Objects::nonNull)
.forEach((T n) -> {
traverse(n, traverser, traversalOrder, builder);
});
}
if (traversalOrder == TraversalOrder.DEPTH_FIRST_POST) {
builder.add(first);
}
return builder;
}
private static Stream.Builder traverseBredthFirst(
T first,
Function> traverser,
Stream.Builder builder) {
requireNonNulls(first, traverser, builder);
if (first == null) {
return builder;
}
final Queue q = new ArrayDeque<>();
q.add(first);
while (!q.isEmpty()) {
final T node = q.poll();
builder.add(node);
traverser
.apply(node)
.filter(Objects::nonNull)
.forEach(q::add);
}
return builder;
}
/**
* Utility classes should not be instantiated.
*/
private Trees() {
throw new UnsupportedOperationException();
}
}