DanilaFe/Abacus
1
0
Fork 0
mirror of https://github.com/DanilaFe/abacus synced 2026-01-10 17:25:19 +00:00
Code Issues Releases Wiki Activity

Add sanitization to TreeBuilder.

Browse Source
This commit is contained in:
Danila Fedorin
2017-07-27 15:26:02 -07:00
parent d2b27da528
commit 96b09fdf2e
2 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/org/nwapw/abacus/tree/TreeBuilder.java
View File

@@ -3,6 +3,7 @@ package org.nwapw.abacus.tree;
import org.nwapw.abacus.function.OperatorAssociativity;
import org.nwapw.abacus.lexing.Lexer;
import org.nwapw.abacus.lexing.pattern.Match;
import org.nwapw.abacus.lexing.pattern.Pattern;
import java.util.*;
@@ -48,7 +49,7 @@ public class TreeBuilder {
* @param function the function to register.
*/
public void registerFunction(String function){
lexer.register(function, TokenType.FUNCTION);
lexer.register(Pattern.sanitize(function), TokenType.FUNCTION);
}
/**
@@ -58,7 +59,7 @@ public class TreeBuilder {
* @param associativity the associativity of the operator.
*/
public void registerOperator(String operator, int precedence, OperatorAssociativity associativity){
lexer.register(operator, TokenType.OP);
lexer.register(Pattern.sanitize(operator), TokenType.OP);
precedenceMap.put(operator, precedence);
associativityMap.put(operator, associativity);
}