prompto.parser.ECleverParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Core Show documentation
The newest version!
package prompto.parser;

import java.io.IOException;
import java.io.InputStream;
import java.util.function.Supplier;

import org.antlr.v4.runtime.ANTLRErrorListener;
import org.antlr.v4.runtime.CharStream;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.TokenStream;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.ParseTreeWalker;

import prompto.declaration.DeclarationList;
import prompto.expression.IFetchExpression;
import prompto.problem.IProblemListener;
import prompto.type.IType;


public class ECleverParser extends EParser implements IParser {

	IProblemListener problemListener;
	String path = "";

	public ECleverParser(String input) {
		this(CharStreams.fromString(input));
	}
	
	public ECleverParser(InputStream input) throws IOException {
		this(CharStreams.fromStream(input));
	}
	
	public ECleverParser(String path, InputStream input) throws IOException {
		this(CharStreams.fromStream(input));
		setPath(path);
	}

	public ECleverParser(CharStream input) {
		this(new EIndentingLexer(input));
	}
	
	public ECleverParser(EIndentingLexer lexer) {
		this(new CommonTokenStream(lexer));
	}

	public ECleverParser(TokenStream input) {
		super(input);
		this.setErrorHandler(new ErrorStrategy());
	}

	@Override
	public void setProblemListener(IProblemListener problemListener) {
		this.removeErrorListeners();
		if(problemListener!=null)
			this.addErrorListener((ANTLRErrorListener)problemListener);
		getLexer().setProblemListener(problemListener);
		this.problemListener = problemListener;
	}
	
	@Override
	public EIndentingLexer getLexer() {
		return (EIndentingLexer)this.getInputStream().getTokenSource();
	}
	
	@Override
	public int equalToken() {
		return EParser.EQ;
	};

	public void setPath(String path) {
		this.path = path;
	}
	
	public String getPath() {
		return path;
	}

	@Override
	public DeclarationList parse(String path, InputStream data) throws Exception {
		setPath(path);
		getLexer().reset(data);
		setInputStream(new CommonTokenStream(getLexer()));
		return parse_declaration_list();
	}
	
	@Override
	public DeclarationList parse_declaration_list() throws Exception {
		return this.doParse(this::declaration_list, true);
	}
	
	public IType parse_standalone_type() throws Exception {
		return this.doParse(this::category_or_any_type, false);
	}

	public IFetchExpression parse_fetch_expression() throws Exception {
		return this.doParse(this::fetch_expression, false);
	}
	
	public  T doParse(Supplier method, boolean addLF) {
		getLexer().setAddLF(addLF);
		ParseTree tree = method.get();
		EPromptoBuilder builder = new EPromptoBuilder(this);
		ParseTreeWalker walker = new ParseTreeWalker();
		walker.walk(builder, tree);
		return builder.getNodeValue(tree);
	}

	
}