From 9ecc2d9f80b565e30a516e25944119a5c0e51120 Mon Sep 17 00:00:00 2001
From: bjculkin <bjculkin@WIT-136XG42.wvu-ad.wvu.edu>
Date: Fri, 24 Feb 2017 10:46:05 -0500
Subject: Add new top-down transform type

---
 .../src/main/java/bjc/utils/data/TopDownTransformIterator.java     | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'BJC-Utils2/src/main/java/bjc/utils/data/TopDownTransformIterator.java')

diff --git a/BJC-Utils2/src/main/java/bjc/utils/data/TopDownTransformIterator.java b/BJC-Utils2/src/main/java/bjc/utils/data/TopDownTransformIterator.java
index 3886264..4c5ec8a 100644
--- a/BJC-Utils2/src/main/java/bjc/utils/data/TopDownTransformIterator.java
+++ b/BJC-Utils2/src/main/java/bjc/utils/data/TopDownTransformIterator.java
@@ -8,6 +8,8 @@ import java.util.function.BiFunction;
 import java.util.function.Consumer;
 import java.util.function.Function;
 
+import static bjc.utils.data.TopDownTransformResult.*;
+
 public class TopDownTransformIterator<ContainedType> implements Iterator<ITree<ContainedType>> {
 	private Function<ContainedType, TopDownTransformResult> picker;
 	private BiFunction<ITree<ContainedType>, 
@@ -83,6 +85,9 @@ public class TopDownTransformIterator<ContainedType> implements Iterator<ITree<C
 				case TRANSFORM:
 					done = true;
 					return transform.apply(preParent, this::addYield);
+				case RTRANSFORM:
+					preParent = transform.apply(preParent, this::addYield);
+					break;
 				case PUSHDOWN:
 					if(preParent.getChildrenCount() != 0) {
 						for(int i = 0; i < preParent.getChildrenCount(); i++) {
@@ -115,7 +120,7 @@ public class TopDownTransformIterator<ContainedType> implements Iterator<ITree<C
 					throw new IllegalArgumentException("Unknown result type " + res);
 			}
 
-			initial = false;
+			if(res != RTRANSFORM) initial = false;
 		}
 
 		if(curYield != null) {
-- 
cgit v1.2.3