From acf7fa23e7a254fc690f036aab61fe4d9633fc30 Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Tue, 21 Jun 2016 22:41:57 +0300 Subject: Add builtin function 'as_int' --- .../runtime/builtins/AsIntBuiltinFunction.java | 55 ++++++++++++++++++++++ main/core/runtime/builtins/BuiltinManager.java | 1 + test/resources/program.bork | 14 ++---- 3 files changed, 61 insertions(+), 9 deletions(-) create mode 100644 main/core/runtime/builtins/AsIntBuiltinFunction.java diff --git a/main/core/runtime/builtins/AsIntBuiltinFunction.java b/main/core/runtime/builtins/AsIntBuiltinFunction.java new file mode 100644 index 0000000..3e1c71c --- /dev/null +++ b/main/core/runtime/builtins/AsIntBuiltinFunction.java @@ -0,0 +1,55 @@ +package com.jantuomi.interpreter.main.core.runtime.builtins; + +import com.jantuomi.interpreter.main.core.parser.datatype.BooleanDataContainer; +import com.jantuomi.interpreter.main.core.parser.datatype.DataContainer; +import com.jantuomi.interpreter.main.core.parser.datatype.IntegerDataContainer; +import com.jantuomi.interpreter.main.core.parser.datatype.StringDataContainer; +import com.jantuomi.interpreter.main.core.runtime.Function; +import com.jantuomi.interpreter.main.exception.ExceptionManager; +import com.jantuomi.interpreter.main.exception.InterpreterException; + +import java.util.Arrays; +import java.util.List; + +/** + * Created by jan on 21.6.2016. + */ +public class AsIntBuiltinFunction extends Function { + public AsIntBuiltinFunction() { + super( + Arrays.asList("expression"), + null + ); + } + + @Override + public String getName() { + return "as_int"; + } + + @Override + public DataContainer evaluate(List params) throws InterpreterException { + if (params.size() != 1) { + ExceptionManager.raise(InterpreterException.ExceptionType.SyntaxError, -1, "amount of parameters to as_int"); + } + DataContainer param = params.get(0); + int i = 0; + if (param instanceof StringDataContainer) { + try { + i = Integer.parseInt(((StringDataContainer) param).getData()); + } + catch (NumberFormatException ex) { + ExceptionManager.raise(InterpreterException.ExceptionType.SyntaxError, -1, + String.format("parameter %s to as_int", param.toString())); + return null; + } + } + else if (param instanceof IntegerDataContainer) { + i = ((IntegerDataContainer) param).getData(); + } + else if (param instanceof BooleanDataContainer) { + i = ((BooleanDataContainer) param).getData() ? 1 : 0; + } + return new IntegerDataContainer(i); + } +} diff --git a/main/core/runtime/builtins/BuiltinManager.java b/main/core/runtime/builtins/BuiltinManager.java index a16d8e9..c6c9e26 100644 --- a/main/core/runtime/builtins/BuiltinManager.java +++ b/main/core/runtime/builtins/BuiltinManager.java @@ -21,6 +21,7 @@ public class BuiltinManager { builtins.add(new OutBuiltinFunction()); builtins.add(new InBuiltinFunction()); builtins.add(new ConcatBuiltinFunction()); + builtins.add(new AsIntBuiltinFunction()); } public List getBuiltins() { diff --git a/test/resources/program.bork b/test/resources/program.bork index 0901bd7..1b7717d 100644 --- a/test/resources/program.bork +++ b/test/resources/program.bork @@ -13,18 +13,14 @@ sum(x y) */ decl user_input -set user_input in("give input: ") +set user_input in("give integer: ") out( concat("your input was: " user_input "!") ) -decl var -set var true +decl operand set operand 10 -if var then - out("var on tosi!") -end +decl sum +set sum + as_int(user_input) operand -if not var then - out("var on epatosi!") -end +out( concat(user_input " + " operand " = " sum) ) "done" \ No newline at end of file -- cgit v1.3