Merge branch 'master' of https://github.com/bculkin2442/JPratt.git

1 files changed, 53 insertions, 0 deletions

diff --git a/JPratt/src/main/java/bjc/pratt/commands/impls/TransformingInitialCommand.java b/JPratt/src/main/java/bjc/pratt/commands/impls/TransformingInitialCommand.java
new file mode 100644
index 0000000..e6c509c
--- /dev/null
+++ b/JPratt/src/main/java/bjc/pratt/commands/impls/TransformingInitialCommand.java
@@ -0,0 +1,53 @@
+package bjc.pratt.commands.impls;
+
+import java.util.function.UnaryOperator;
+
+import bjc.pratt.ParserContext;
+import bjc.pratt.commands.AbstractInitialCommand;
+import bjc.pratt.commands.InitialCommand;
+import bjc.pratt.tokens.Token;
+import bjc.utils.data.ITree;
+import bjc.utils.parserutils.ParserException;
+
+/**
+ * An initial command that transforms the result of another command.
+ *
+ * @author bjculkin
+ *
+ * @param <K>
+ *                The key type of the tokens.
+ *
+ * @param <V>
+ *                The value type of the tokens.
+ *
+ * @param <C>
+ *                The state type of the parser.
+ */
+public class TransformingInitialCommand<K, V, C> extends AbstractInitialCommand<K, V, C> {
+	private final InitialCommand<K, V, C> internl;
+
+	private final UnaryOperator<ITree<Token<K, V>>> transfrm;
+
+	/**
+	 * Create a new transforming initial command.
+	 *
+	 * @param internal
+	 *                The initial command to delegate to.
+	 *
+	 * @param transform
+	 *                The transform to apply to the returned tree.
+	 */
+	public TransformingInitialCommand(final InitialCommand<K, V, C> internal,
+			final UnaryOperator<ITree<Token<K, V>>> transform) {
+		super();
+		internl = internal;
+		transfrm = transform;
+	}
+
+	@Override
+	protected ITree<Token<K, V>> intNullDenotation(final Token<K, V> operator, final ParserContext<K, V, C> ctx)
+			throws ParserException {
+		return transfrm.apply(internl.denote(operator, ctx));
+	}
+
+}