summaryrefslogtreecommitdiff
path: root/BJC-Utils2/src/main/java/bjc/utils/parserutils/pratt/commands/ChainCommand.java
blob: aa85f75d9c8f84d586fd8ef90b491f1403af1c78 (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
package bjc.utils.parserutils.pratt.commands;

import bjc.utils.data.ITree;
import bjc.utils.data.Tree;
import bjc.utils.parserutils.ParserException;
import bjc.utils.parserutils.pratt.ParserContext;
import bjc.utils.parserutils.pratt.Token;

import java.util.Set;

public class ChainCommand<K, V, C> extends BinaryPostCommand<K, V, C> {
	private Set<K> chainWith;

	private Token<K, V> chain;

	public ChainCommand(int leftPower, Set<K> chainSet, Token<K, V> chainMarker) {
		super(leftPower);

		chainWith = chainSet;
		chain = chainMarker;
	}

	@Override
	public ITree<Token<K, V>> denote(ITree<Token<K, V>> operand, Token<K, V> operator,
			ParserContext<K, V, C> ctx) throws ParserException {
		ITree<Token<K, V>> tree = ctx.parse.parseExpression(1 + leftBinding(), ctx.tokens, ctx.state, false);

		ITree<Token<K, V>> res = new Tree<>(operator, operand, tree);

		if (chainWith.contains(ctx.tokens.current().getKey())) {
			Token<K, V> tok = ctx.tokens.current();
			ctx.tokens.next();

			ITree<Token<K, V>> other = denote(tree, tok,
					new ParserContext<>(ctx.tokens, ctx.parse, ctx.state));

			return new Tree<>(chain, res, other);
		} else {
			return res;
		}
	}

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