summaryrefslogtreecommitdiff
path: root/JPratt/src/main/java/bjc/pratt/commands/impls/TernaryCommand.java
blob: 786dfec7ded4964485f59f96f2c33e4af8e89a7a (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
package bjc.pratt.commands.impls;

import bjc.pratt.ParserContext;
import bjc.pratt.blocks.ParseBlock;
import bjc.pratt.commands.BinaryPostCommand;
import bjc.pratt.commands.CommandResult;
import bjc.pratt.commands.CommandResult.Status;
import bjc.pratt.tokens.Token;
import bjc.data.Tree;
import bjc.data.SimpleTree;
import bjc.utils.parserutils.ParserException;

/**
 * A ternary command, like C's ?:
 *
 * @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 TernaryCommand<K, V, C> extends BinaryPostCommand<K, V, C> {
	private final ParseBlock<K, V, C> innerBlck;

	private final Token<K, V> mark;

	private final boolean nonassoc;

	/**
	 * Create a new ternary command.
	 *
	 * @param precedence The precedence of this operator.
	 *
	 * @param innerBlock The representation of the inner block of the expression.
	 *
	 * @param marker     The token to use as the root of the AST node.
	 *
	 * @param isNonassoc Whether or not the conditional is associative.
	 */
	public TernaryCommand(final int precedence, final ParseBlock<K, V, C> innerBlock, final Token<K, V> marker,
			final boolean isNonassoc) {
		super(precedence);

		if (innerBlock == null)
			throw new NullPointerException("Inner block must not be null");
		else if (marker == null)
			throw new NullPointerException("Marker must not be null");

		innerBlck = innerBlock;
		mark = marker;
		nonassoc = isNonassoc;
	}

	@Override
	public CommandResult<K, V> denote(final Tree<Token<K, V>> operand, final Token<K, V> operator,
			final ParserContext<K, V, C> ctx) throws ParserException {
		final CommandResult<K, V> innerRes = innerBlck.parse(ctx);
		if (innerRes.status != Status.SUCCESS) return innerRes;
		Tree<Token<K, V>> inner = innerRes.success();
		
		final CommandResult<K,V> outerRes = ctx.parse.parseExpression(1 + leftBinding(), ctx.tokens, ctx.state, false);
		if (outerRes.status != Status.SUCCESS) return outerRes;
		Tree<Token<K, V>> outer = outerRes.success();
		return CommandResult.success(new SimpleTree<>(mark, inner, operand, outer));
	}

	@Override
	public int nextBinding() {
		if (nonassoc)
			return leftBinding() - 1;

		return leftBinding();
	}
}