prompto.parser.MCleverParser 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.problem.IProblemListener;

public class MCleverParser extends MParser implements IParser {

	IProblemListener problemListener;
	String path = "";

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

	public MCleverParser(CharStream input) {
		this(new MIndentingLexer(input));
	}
	
	public MCleverParser(MIndentingLexer lexer) {
		this(new CommonTokenStream(lexer));
	}

	public MCleverParser(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 MIndentingLexer getLexer() {
		return (MIndentingLexer)this.getInputStream().getTokenSource();
	}

	@Override
	public int equalToken() {
		return MParser.EQ;
	};

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

	@Override
	public DeclarationList parse(String path, InputStream input) throws Exception {
		this.path = path;
		getLexer().reset(input);
		getInputStream().seek(0);
		return parse_declaration_list();
	}
	
	@Override
	public DeclarationList parse_declaration_list() throws Exception {
		return this.doParse(this::declaration_list, true);
	}
	
	public  T doParse(Supplier method, boolean addLF) {
		getLexer().setAddLF(addLF);
		ParseTree tree = method.get();
		MPromptoBuilder builder = new MPromptoBuilder(this);
		ParseTreeWalker walker = new ParseTreeWalker();
		walker.walk(builder, tree);
		return builder.getNodeValue(tree);
	}

}