creating and generating builtin functions, ast-macros for builtins, todo: params

[xonotic/gmqcc.git] / test / ast-test.c

diff --git a/test/ast-test.c b/test/ast-test.c
index 00e8c052e3220dba789651fde674f7473099ef97..2217a396e5ccc1373e9800f86e54a512a929601e 100644 (file)
--- a/test/ast-test.c
+++ b/test/ast-test.c
@@ -12,98 +12,52 @@
 VECTOR_MAKE(ast_value*, globals);
 VECTOR_MAKE(ast_function*, functions);
 
+#include "ast-macros.h"
+
 int main()
 {
-    /* AST */
-    ast_expression *exp;
-    ast_value      *gfoo    = NULL;
-    ast_value      *gbar    = NULL;
-    ast_value      *const_3 = NULL;
-    ast_value      *const_4 = NULL;
-    ast_value      *vmain   = NULL;
-    ast_function   *fmain   = NULL;
-    ast_block      *ba      = NULL;
-    ast_value      *li      = NULL;
-
-    /* IR */
-    ir_builder     *ir = NULL;
-
-    /* common stuff */
-
     size_t i;
-    lex_ctx ctx;
-    ctx.file = NULL;
-    ctx.line = 1;
-
-    /* globals */
-    gfoo = ast_value_new(ctx, "foo", TYPE_FLOAT);
-        assert(gfoo);
-    assert(globals_add(gfoo) >= 0);
-
-    gbar = ast_value_new(ctx, "bar", TYPE_FLOAT);
-        assert(gbar);
-    assert(globals_add(gbar) >= 0);
-
-    const_3 = ast_value_new(ctx, "3", TYPE_FLOAT);
-        assert(const_3);
-    assert(globals_add(const_3) >= 0);
-    const_3->isconst = true;
-    const_3->constval.vfloat = 3.0;
-
-    const_4 = ast_value_new(ctx, "4", TYPE_FLOAT);
-        assert(const_4);
-    assert(globals_add(const_4) >= 0);
-    const_4->isconst = true;
-    const_4->constval.vfloat = 4.0;
-
-    /* defining a function */
-    vmain = ast_value_new(ctx, "main", TYPE_FUNCTION);
-        assert(vmain);
-    assert(globals_add(vmain) >= 0);
-    /* This happens in ast_function_new:
-        vmain->isconst = true;
-        vmain->constval.vfunc = fmain;
-       Creating a function node connects the global to the function.
-       You still have to delete *BOTH*, deleting one will NOT delete the other,
-       it will only unlink them.
-     */
 
-    /* creating a function body */
-    fmain = ast_function_new(ctx, "main", vmain);
-        assert(fmain);
-    assert(functions_add(fmain) >= 0);
-
-    /* { block */
-    ba = ast_block_new(ctx);
-        assert(ba);
-    assert(ast_function_blocks_add(fmain, ba));
-
-    /* local variable i */
-    li = ast_value_new(ctx, "i", TYPE_FLOAT);
-        assert(li);
-    assert(ast_block_locals_add(ba, li));
-
-    /* I realize having to provide the opcode isn't the best way to go, but,
-     * urgh... */
-    /* foo = 3; */
-    exp = (ast_expression*)ast_store_new(ctx, INSTR_STORE_F, gfoo, (ast_expression*)const_3);
-        assert(exp);
-    assert(ast_block_exprs_add(ba, exp));
-    /* bar = 4; */
-    exp = (ast_expression*)ast_store_new(ctx, INSTR_STORE_F, gbar, (ast_expression*)const_4);
-        assert(exp);
-    assert(ast_block_exprs_add(ba, exp));
-
-    /* i = foo * bar */
-    exp = (ast_expression*)ast_binary_new(ctx,  INSTR_MUL_F, (ast_expression*)gbar, (ast_expression*)gfoo);
-        assert(exp);
-    exp = (ast_expression*)ast_store_new(ctx, INSTR_STORE_F, li, exp);
-        assert(exp);
-    assert(ast_block_exprs_add(ba, exp));
-
-    /* } block */
-
-    /* Next up: Having the AST generate IR */
+    ir_builder     *ir;
+
+    TESTVARS();
+
+    DEFVAR(vi);
+    DEFVAR(vx);
+    DEFVAR(f0);
+    DEFVAR(f1);
+    DEFVAR(f5);
+    DEFVAR(print);
+
+#if 0
+    BUILTIN(print, TYPE_VOID, -1);
+    PARAM(TYPE_STRING, text);
+    ENDBUILTIN();
+#endif
+
+    TESTINIT();
+VAR(TYPE_FLOAT, f0);
+VAR(TYPE_FLOAT, f1);
+VAR(TYPE_FLOAT, f5);
+MKCONSTFLOAT(f0, 0.0);
+MKCONSTFLOAT(f1, 1.0);
+MKCONSTFLOAT(f5, 5.0);
+
+FUNCTION(main, TYPE_VOID);
+
+VAR(TYPE_FLOAT, vi);
+VAR(TYPE_FLOAT, vx);
+
+MKLOCAL(vi);
+MKLOCAL(vx);
+
+STATE(ASSIGN(STORE_F, vi, f0));
+WHILE(BIN(LT, vi, f5));
+STATE(ASSIGN(STORE_F, vx, BIN(MUL_F, vi, f5)));
+STATE(ASSIGN(STORE_F, vi, BIN(ADD_F, vi, f1)));
+ENDWHILE();
+
+ENDFUNCTION(main);
 
     ir = ir_builder_new("ast_test");
     assert(ir);
@@ -120,11 +74,18 @@ int main()
         if (!ast_function_codegen(functions_data[i], ir)) {
             assert(!"failed to generate function");
         }
+        if (!ir_function_finalize(functions_data[i]->ir_func))
+            assert(!"finalize on function failed...");
     }
 
+
     /* dump */
     ir_builder_dump(ir, printf);
 
+    /* Now create a file */
+    if (!ir_builder_generate(ir, "test_ast.dat"))
+        printf("*** failed to generate code\n");
+
     /* ir cleanup */
     ir_builder_delete(ir);