2015-05-18 17:07:40 +03:00
|
|
|
package com.annimon.ownlang;
|
|
|
|
|
2016-01-24 19:38:19 +02:00
|
|
|
import com.annimon.ownlang.lib.CallStack;
|
2015-05-18 17:07:40 +03:00
|
|
|
import com.annimon.ownlang.parser.Lexer;
|
|
|
|
import com.annimon.ownlang.parser.Parser;
|
|
|
|
import com.annimon.ownlang.parser.Token;
|
2015-05-20 14:59:16 +03:00
|
|
|
import com.annimon.ownlang.parser.ast.Statement;
|
2015-06-29 20:50:59 +03:00
|
|
|
import com.annimon.ownlang.parser.visitors.AssignValidator;
|
|
|
|
import com.annimon.ownlang.parser.visitors.FunctionAdder;
|
|
|
|
import com.annimon.ownlang.parser.visitors.VariablePrinter;
|
2015-05-25 23:20:10 +03:00
|
|
|
import java.io.IOException;
|
|
|
|
import java.nio.file.Files;
|
|
|
|
import java.nio.file.Paths;
|
2015-05-18 17:07:40 +03:00
|
|
|
import java.util.List;
|
2016-02-19 11:42:29 +02:00
|
|
|
import java.util.concurrent.TimeUnit;
|
2015-05-18 17:07:40 +03:00
|
|
|
|
|
|
|
/**
|
|
|
|
* @author aNNiMON
|
|
|
|
*/
|
|
|
|
public final class Main {
|
|
|
|
|
2015-05-25 23:20:10 +03:00
|
|
|
public static void main(String[] args) throws IOException {
|
2016-01-19 18:25:19 +02:00
|
|
|
if (args.length == 0) {
|
2016-02-19 11:42:29 +02:00
|
|
|
run(readFile("program.own"), true, true, true);
|
2016-01-19 18:25:19 +02:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2016-02-19 11:42:29 +02:00
|
|
|
boolean showTokens = false, showAst = false, showMeasurements = false;
|
2016-01-19 18:25:19 +02:00
|
|
|
String input = null;
|
|
|
|
for (int i = 0; i < args.length; i++) {
|
|
|
|
switch (args[i]) {
|
|
|
|
case "-a":
|
|
|
|
case "--showast":
|
|
|
|
showAst = true;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case "-t":
|
|
|
|
case "--showtokens":
|
|
|
|
showTokens = true;
|
|
|
|
break;
|
|
|
|
|
2016-02-19 11:42:29 +02:00
|
|
|
case "-m":
|
|
|
|
case "--showtime":
|
|
|
|
showMeasurements = true;
|
|
|
|
break;
|
|
|
|
|
2016-01-19 18:25:19 +02:00
|
|
|
case "-f":
|
|
|
|
case "--file":
|
|
|
|
if (i + 1 < args.length) {
|
|
|
|
input = readFile(args[i + 1]);
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
input = args[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (input == null) {
|
|
|
|
throw new IllegalArgumentException("Empty input");
|
|
|
|
}
|
2016-02-19 11:42:29 +02:00
|
|
|
run(input, showTokens, showAst, showMeasurements);
|
2016-01-19 18:25:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
private static String readFile(String file) throws IOException {
|
|
|
|
return new String( Files.readAllBytes(Paths.get(file)), "UTF-8");
|
|
|
|
}
|
|
|
|
|
2016-02-19 11:42:29 +02:00
|
|
|
private static void run(String input, boolean showTokens, boolean showAst, boolean showMeasurements) {
|
|
|
|
final TimeMeasurement measurement = new TimeMeasurement();
|
|
|
|
measurement.start("Tokenize time");
|
2015-05-25 23:20:10 +03:00
|
|
|
final List<Token> tokens = new Lexer(input).tokenize();
|
2016-02-19 11:42:29 +02:00
|
|
|
measurement.stop("Tokenize time");
|
2016-01-19 18:25:19 +02:00
|
|
|
if (showTokens) {
|
|
|
|
for (int i = 0; i < tokens.size(); i++) {
|
|
|
|
System.out.println(i + " " + tokens.get(i));
|
|
|
|
}
|
2015-05-18 17:07:40 +03:00
|
|
|
}
|
|
|
|
|
2016-02-19 11:42:29 +02:00
|
|
|
measurement.start("Parse time");
|
2016-02-14 21:03:28 +02:00
|
|
|
final Parser parser = new Parser(tokens);
|
|
|
|
final Statement program = parser.parse();
|
2016-02-19 11:42:29 +02:00
|
|
|
measurement.stop("Parse time");
|
2016-01-19 18:25:19 +02:00
|
|
|
if (showAst) {
|
|
|
|
System.out.println(program.toString());
|
|
|
|
}
|
2016-02-14 21:03:28 +02:00
|
|
|
if (parser.getParseErrors().hasErrors()) {
|
|
|
|
System.out.println(parser.getParseErrors());
|
|
|
|
return;
|
|
|
|
}
|
2015-06-29 20:50:59 +03:00
|
|
|
program.accept(new FunctionAdder());
|
2016-01-09 14:11:15 +02:00
|
|
|
// program.accept(new VariablePrinter());
|
2015-06-29 20:50:59 +03:00
|
|
|
program.accept(new AssignValidator());
|
2016-01-24 19:38:19 +02:00
|
|
|
try {
|
2016-02-19 11:42:29 +02:00
|
|
|
measurement.start("Execution time");
|
2016-01-24 19:38:19 +02:00
|
|
|
program.execute();
|
|
|
|
} catch (Exception ex) {
|
|
|
|
handleException(Thread.currentThread(), ex);
|
2016-02-19 11:42:29 +02:00
|
|
|
} finally {
|
|
|
|
if (showMeasurements) {
|
|
|
|
measurement.stop("Execution time");
|
|
|
|
System.out.println(measurement.summary(TimeUnit.MILLISECONDS, true));
|
|
|
|
}
|
2016-01-24 19:38:19 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
public static void handleException(Thread thread, Throwable throwable) {
|
|
|
|
System.err.printf("%s in %s\n", throwable.getMessage(), thread.getName());
|
|
|
|
for (CallStack.CallInfo call : CallStack.getCalls()) {
|
|
|
|
System.err.printf("\tat %s\n", call);
|
|
|
|
}
|
|
|
|
throwable.printStackTrace();
|
2015-05-18 17:07:40 +03:00
|
|
|
}
|
|
|
|
}
|