summaryrefslogtreecommitdiff
path: root/JPratt/src/main/java/com/ashardalon/pratt/commands/impls/PreTernaryCommand.java
blob: cd0133327698fad8692a6ebe8a359f361600f866 (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
77
78
79
80
81
82
83
84
85
package com.ashardalon.pratt.commands.impls;

import bjc.data.Tree;

import com.ashardalon.pratt.ParserContext;
import com.ashardalon.pratt.blocks.ParseBlock;
import com.ashardalon.pratt.commands.AbstractInitialCommand;
import com.ashardalon.pratt.commands.CommandResult;
import com.ashardalon.pratt.commands.CommandResult.Status;
import com.ashardalon.pratt.tokens.Token;

import bjc.data.SimpleTree;
import bjc.utils.parserutils.ParserException;

/**
 * A prefix ternary operator, like an if/then/else group.
 *
 * @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 PreTernaryCommand<K, V, C> extends AbstractInitialCommand<K, V, C> {
	private final Token<K, V> trm;

	private final ParseBlock<K, V, C> condBlock;

	private final ParseBlock<K, V, C> opblock1;
	private final ParseBlock<K, V, C> opblock2;

	/**
	 * Create a new ternary statement.
	 *
	 * @param cond
	 *        The block for handling the condition.
	 *
	 * @param op1
	 *        The block for handling the first operator.
	 *
	 * @param op2
	 *        The block for handling the second operator.
	 *
	 * @param term
	 *        The token to use as the node for the AST.
	 */
	public PreTernaryCommand(final ParseBlock<K, V, C> cond, final ParseBlock<K, V, C> op1,
			final ParseBlock<K, V, C> op2, final Token<K, V> term) {
		super();

		if(cond == null)
			throw new NullPointerException("Cond block must not be null");
		else if(op1 == null)
			throw new NullPointerException("Op block #1 must not be null");
		else if(op2 == null) throw new NullPointerException("Op block #2 must not be null");

		condBlock = cond;
		opblock1 = op1;
		opblock2 = op2;

		trm = term;
	}

	@Override
	protected CommandResult<K, V> intNullDenotation(final Token<K, V> operator, final ParserContext<K, V, C> ctx)
			throws ParserException {
		final CommandResult<K,V> condRes = condBlock.parse(ctx);
		if (condRes.status != Status.SUCCESS) return condRes;
		Tree<Token<K, V>> cond = condRes.success();
		
		final CommandResult<K,V> op1Res = opblock1.parse(ctx);
		if (op1Res.status != Status.SUCCESS) return op1Res;
		Tree<Token<K, V>> op1 = op1Res.success();

		final CommandResult<K,V> op2Res = opblock2.parse(ctx);
		if (op2Res.status != Status.SUCCESS) return op2Res;
		Tree<Token<K, V>> op2 = op2Res.success();
		return CommandResult.success(new SimpleTree<>(trm, cond, op1, op2));
	}
}