DEADSOFTWARE

Добавлены встроенные процедуры
[dsw-obn.git] / generator.c
index 21506b0626fe5db7cfbd117a444d2295de3448a2..709e5a65cbfb513cd502e2bf7d70383b89468449 100644 (file)
@@ -8,21 +8,6 @@
 #include "oberon.h"
 #include "generator.h"
 
-#include <libgccjit.h>
-
-/*
- * oberon_var_t -> gvar == gcc_jit_lvalue;
- * oberon_type_t -> gtype == gcc_jit_type;
- * oberon_context_t -> gctx == gen_context_t;
- */ 
-
-typedef struct
-{
-       gcc_jit_context * gcc_context;
-       gcc_jit_block * gcc_block;
-       gcc_jit_result * gcc_result;
-} gen_context_t;
-
 static void printcontext(oberon_context_t * ctx, char * s)
 {
 /*
@@ -42,6 +27,24 @@ static void printcontext(oberon_context_t * ctx, char * s)
 //   ALLOC
 // ======================================================================= 
 
+static void
+oberon_generator_open_block(gen_context_t * gen_context, gcc_jit_block * gcc_block)
+{
+       gen_block_t * block = malloc(sizeof *block);
+       memset(block, 0, sizeof *block);
+
+       block -> gcc_block = gcc_block;
+       block -> up = gen_context -> block;
+
+       gen_context -> block = block;
+}
+
+static void
+oberon_generator_close_block(gen_context_t * gen_context)
+{
+       gen_context -> block = gen_context -> block -> up;
+}
+
 void
 oberon_generator_init_context(oberon_context_t * ctx)
 {
@@ -71,48 +74,206 @@ oberon_generator_destroy_context(oberon_context_t * ctx)
 void
 oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
 {
-       printcontext(ctx, "oberon_generator_init_type");
+       gen_type_t * gen_type = malloc(sizeof *gen_type);
+       memset(gen_type, 0, sizeof *gen_type);
+       type -> gen_type = gen_type;
 
        gen_context_t * gen_context = ctx -> gen_context;
        gcc_jit_context * gcc_context = gen_context -> gcc_context;
 
-       gcc_jit_type * gen_type;
-       if(type -> class == OBERON_TYPE_INTEGER)
+       gcc_jit_type * gcc_type = NULL;
+       gcc_jit_struct * gcc_struct = NULL;
+       if(type -> class == OBERON_TYPE_VOID)
        {
-               gen_type = gcc_jit_context_get_int_type(gcc_context, type -> size, 1);
+               gcc_type = gcc_jit_context_get_type(gcc_context, GCC_JIT_TYPE_VOID);
+       }
+       else if(type -> class == OBERON_TYPE_INTEGER)
+       {
+               gcc_type = gcc_jit_context_get_int_type(gcc_context, type -> size, 1);
        }
        else if(type -> class == OBERON_TYPE_BOOLEAN)
        {
-               gen_type = gcc_jit_context_get_type(gcc_context, GCC_JIT_TYPE_BOOL);
+               gcc_type = gcc_jit_context_get_type(gcc_context, GCC_JIT_TYPE_BOOL);
+       }
+       else if(type -> class == OBERON_TYPE_ARRAY)
+       {
+               gen_type_t * gen_base = type -> base -> gen_type;
+               gcc_jit_type * gcc_base = gen_base -> gcc_type;
+               gcc_type = gcc_jit_context_new_array_type(gcc_context, NULL, gcc_base, type -> size);
+       }
+       else if(type -> class == OBERON_TYPE_RECORD)
+       {
+               char name[32];
+               snprintf(name, 32, "RECORD%u", gen_context -> record_count);
+               gen_context -> record_count += 1;
+
+               gcc_struct = gcc_jit_context_new_opaque_struct(gcc_context, NULL, name);
+               gcc_type = gcc_jit_struct_as_type(gcc_struct);
+       }
+       else if(type -> class == OBERON_TYPE_POINTER)
+       {
+               gen_type_t * gen_base = type -> base -> gen_type;
+               gcc_jit_type * gcc_base = gen_base -> gcc_type;
+               gcc_type = gcc_jit_type_get_pointer(gcc_base);
+       }
+       else if(type -> class == OBERON_TYPE_PROCEDURE)
+       {
+               int num_params = type -> num_decl;
+               gcc_jit_type * params[num_params];
+               oberon_object_t * o = type -> decl;
+               for(int i = 0; i < num_params; i++)
+               {
+                       gen_type_t * gen_type = o -> type -> gen_type;
+                       params[i] = gen_type -> gcc_type;
+                       o = o -> next;
+               }
+
+               gen_type_t * base = type -> base -> gen_type;
+               gcc_jit_type * result_type = base -> gcc_type;
+
+               gcc_type = gcc_jit_context_new_function_ptr_type(
+                       gcc_context, NULL, result_type, num_params, params, 0
+               );
        }
        else
        {
                oberon_error(ctx, "oberon_generator_init_type: invalid type class %i", type -> class);
        }
 
-       type -> gen_type = gen_type;
+       assert(gcc_type);
+       gen_type -> gcc_type = gcc_type;
+       gen_type -> gcc_struct = gcc_struct;
 }
 
 void
-oberon_generator_init_var(oberon_context_t * ctx, oberon_var_t * var)
+oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * type)
 {
-       printcontext(ctx, "oberon_generator_init_var");
+       assert(type -> class == OBERON_TYPE_RECORD);
+
+       gen_type_t * gen_type = type -> gen_type;
+       gcc_jit_struct * gcc_struct = gen_type -> gcc_struct;
 
+       // TODO type exstension
+
+       int num_fields = type -> num_decl;
+       gcc_jit_field * fields[num_fields];
+       oberon_object_t * o = type -> decl;
+       for(int i = 0; i < num_fields; i++)
+       {
+               assert(o -> class == OBERON_CLASS_FIELD);
+               gen_var_t * var = o -> gen_var;
+               fields[i] = var -> gcc_field;
+               o = o -> next;
+       }
+
+       gcc_jit_struct_set_fields (gcc_struct, NULL, num_fields, fields);
+
+       //gcc_struct = gcc_jit_context_new_struct_type(gcc_context, NULL, "", num_fields, fields);
+}
+
+void
+oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
+{
        gen_context_t * gen_context = ctx -> gen_context;
+       gen_type_t * gen_type = var -> type -> gen_type;
+
+       gen_var_t * gen_var = malloc(sizeof *gen_var);
+       memset(gen_var, 0, sizeof *gen_var);
+       var -> gen_var = gen_var;
+
        gcc_jit_context * gcc_context = gen_context -> gcc_context;
-       gcc_jit_type * gen_type = var -> type -> gen_type;
+       gcc_jit_type * gcc_type = gen_type -> gcc_type;
        const char * name = var -> name;
 
-       gcc_jit_lvalue * gen_var;
-       gen_var = gcc_jit_context_new_global(gcc_context, NULL, GCC_JIT_GLOBAL_INTERNAL, gen_type, name);
+       gcc_jit_lvalue * gcc_lvalue = NULL;
+       gcc_jit_param * gcc_param = NULL;
+       gcc_jit_field * gcc_field = NULL;
+       if(var -> class == OBERON_CLASS_VAR)
+       {
+               if(var -> local)
+               {
+                       gen_proc_t * gen_func = var -> parent -> gen_proc;
+                       gcc_jit_function * func = gen_func -> gcc_func;
+                       
+                       gcc_lvalue = gcc_jit_function_new_local(func, NULL, gcc_type, name);
+               }
+               else
+               {
+                       gcc_lvalue = gcc_jit_context_new_global(
+                               gcc_context, NULL, GCC_JIT_GLOBAL_INTERNAL, gcc_type, name
+                       );
+               }
+       }
+       else if(var -> class == OBERON_CLASS_PARAM)
+       {
+               gcc_param = gcc_jit_context_new_param(gcc_context, NULL, gcc_type, name);
+               gcc_lvalue = gcc_jit_param_as_lvalue(gcc_param);
+       }
+       else if(var -> class == OBERON_CLASS_VAR_PARAM)
+       {
+               gcc_type = gcc_jit_type_get_pointer(gcc_type);
+               gcc_param = gcc_jit_context_new_param(gcc_context, NULL, gcc_type, name);
+               gcc_lvalue = gcc_jit_param_as_lvalue(gcc_param);
+       }
+       else if(var -> class == OBERON_CLASS_FIELD)
+       {
+               gcc_field = gcc_jit_context_new_field(gcc_context, NULL, gcc_type, name);
+       }
+       else
+       {
+               oberon_error(ctx, "oberon_generator_init_var: invalid class %i", var -> class);
+       }
 
-       var -> gen_var = gen_var;
+       gen_var -> gcc_lvalue = gcc_lvalue;
+       gen_var -> gcc_param = gcc_param;
+       gen_var -> gcc_field = gcc_field;
+}
+
+void
+oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
+{
+       if(proc -> local)
+       {
+               oberon_error(ctx, "generator: local procedures not supported");
+       }
+
+       gen_context_t * gen_context = ctx -> gen_context;
+       gcc_jit_context * gcc_context = gen_context -> gcc_context;
+
+       gen_proc_t * gen_proc = malloc(sizeof *gen_proc);
+       memset(gen_proc, 0, sizeof *gen_proc);
+       proc -> gen_proc = gen_proc;
+
+       const char * name = proc -> name;
+       gen_type_t * gen_result_type = proc -> type -> base -> gen_type;
+       gcc_jit_type * result_type = gen_result_type -> gcc_type;
+
+       /* Строим список параметров */
+       int num_param = proc -> type -> num_decl;
+       oberon_object_t * o = proc -> type -> decl;
+       gcc_jit_param * params[num_param];
+       for(int i = 0; i < num_param; i++)
+       {
+               gen_var_t * param_var = o -> gen_var;
+               params[i] = param_var -> gcc_param;
+               o = o -> next;
+       }
+
+       gcc_jit_function * gcc_func;
+       gcc_func = gcc_jit_context_new_function(
+               gcc_context, NULL, GCC_JIT_FUNCTION_EXPORTED, result_type, name, num_param, params, 0
+       );
+
+       gen_proc -> gcc_func = gcc_func;
 }
 
 // =======================================================================
 //   GENERATOR
 // ======================================================================= 
 
+static gcc_jit_rvalue * rvalue_from_item(oberon_context_t * ctx, oberon_item_t * item);
+static gcc_jit_rvalue * rvalue_from_expr(oberon_context_t * ctx, oberon_expr_t * expr);
+
 void
 oberon_generate_begin_module(oberon_context_t * ctx)
 {
@@ -127,7 +288,7 @@ oberon_generate_begin_module(oberon_context_t * ctx)
        );
        gcc_jit_block * gcc_block = gcc_jit_function_new_block(func, NULL);
 
-       gen_context -> gcc_block = gcc_block;
+       oberon_generator_open_block(gen_context, gcc_block);
 }
 
 void
@@ -136,11 +297,107 @@ oberon_generate_end_module(oberon_context_t * ctx)
        printcontext(ctx, "oberon_generate_end_module");
 
        gen_context_t * gen_context = ctx -> gen_context;
-       gcc_jit_block * gcc_block = gen_context -> gcc_block;
+       gcc_jit_block * gcc_block = gen_context -> block -> gcc_block;
 
        gcc_jit_block_end_with_void_return(gcc_block, NULL);
 
-       gen_context -> gcc_block = NULL;
+       oberon_generator_close_block(gen_context);
+}
+
+void
+oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
+{
+       gen_context_t * gen_context = ctx -> gen_context;
+       gen_proc_t * gen_proc = proc -> gen_proc;
+
+       gcc_jit_function * func = gen_proc -> gcc_func;
+       gcc_jit_block * gcc_block = gcc_jit_function_new_block(func, NULL);
+
+       oberon_generator_open_block(gen_context, gcc_block);
+}
+
+void
+oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
+{
+       gen_context_t * gen_context = ctx -> gen_context;
+       gen_block_t * gen_block = gen_context -> block;
+       gcc_jit_block * block = gen_block -> gcc_block;
+
+       gcc_jit_rvalue * return_value;
+       return_value = rvalue_from_expr(ctx, desig);
+       gcc_jit_block_add_eval(block, NULL, return_value);
+}
+
+void
+oberon_generate_end_proc(oberon_context_t * ctx)
+{
+       gen_context_t * gen_context = ctx -> gen_context;
+       oberon_generator_close_block(gen_context);
+}
+
+void
+oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
+{
+       gen_context_t * gen_context = ctx -> gen_context;
+       gen_block_t * gen_block = gen_context -> block;
+       gcc_jit_block * gcc_block = gen_block -> gcc_block;
+
+       if(expr == NULL)
+       {
+               gcc_jit_block_end_with_void_return(gcc_block, NULL);
+       }
+       else
+       {
+               gcc_jit_rvalue * r = rvalue_from_expr(ctx, expr);
+               gcc_jit_block_end_with_return(gcc_block, NULL, r);
+       }
+}
+
+static gcc_jit_lvalue *
+lvalue_from_item(oberon_context_t * ctx, oberon_item_t * item)
+{
+       gen_context_t * gen_context = ctx -> gen_context;
+       gcc_jit_context * gcc_context = gen_context -> gcc_context;
+
+       gcc_jit_lvalue * left;
+
+       if(item -> mode == MODE_VAR)
+       {
+               gen_var_t * gen_var = item -> var -> gen_var;
+               left = gen_var -> gcc_lvalue;
+               if(item -> var -> class == OBERON_CLASS_VAR_PARAM)
+               {
+                       gcc_jit_rvalue * r = gcc_jit_lvalue_as_rvalue(left);
+                       left = gcc_jit_rvalue_dereference(r, NULL);
+               }
+       }
+       else if(item -> mode == MODE_INDEX)
+       {
+               assert(item -> num_args == 1);
+               gcc_jit_rvalue * parent = rvalue_from_item(ctx, item -> parent);
+               gcc_jit_rvalue * index = rvalue_from_expr(ctx, item -> args);
+               left = gcc_jit_context_new_array_access(gcc_context, NULL, parent, index);
+       }
+       else if(item -> mode == MODE_FIELD)
+       {
+               printf("lvalue_from_item: %s\n", item -> var -> name);
+               gen_var_t * gen_var = item -> var -> gen_var;
+               gcc_jit_field * gcc_field = gen_var -> gcc_field;
+
+               gcc_jit_lvalue * parent = lvalue_from_item(ctx, item -> parent);
+               left = gcc_jit_lvalue_access_field(parent, NULL, gcc_field);
+       }
+       else if(item -> mode == MODE_DEREF)
+       {
+               gcc_jit_rvalue * parent = rvalue_from_item(ctx, item -> parent);
+               left = gcc_jit_rvalue_dereference(parent, NULL);
+       }
+       else
+       {
+               oberon_error(ctx, "invalid lvalue expression");
+       }
+
+       return left;
 }
 
 static gcc_jit_lvalue *
@@ -152,11 +409,7 @@ lvalue_from_expr(oberon_context_t *ctx, oberon_expr_t * expr)
        if(expr -> is_item)
        {
                item = (oberon_item_t *) expr;
-               if(item -> mode != MODE_VAR)
-               {
-                       oberon_error(ctx, "invalid lvalue expression");
-               }
-               left = item -> var -> gen_var;
+               left = lvalue_from_item(ctx, item);
        }
        else
        {
@@ -166,8 +419,6 @@ lvalue_from_expr(oberon_context_t *ctx, oberon_expr_t * expr)
        return left;
 }
 
-static gcc_jit_rvalue * rvalue_from_expr(oberon_context_t * ctx, oberon_expr_t * expr);
-
 static gcc_jit_rvalue *
 rvalue_from_item(oberon_context_t * ctx, oberon_item_t * item)
 {
@@ -177,8 +428,8 @@ rvalue_from_item(oberon_context_t * ctx, oberon_item_t * item)
        gcc_jit_rvalue * right;
        if(item -> mode == MODE_VAR)
        {
-               gcc_jit_lvalue * gen_var = item -> var -> gen_var;
-               right = gcc_jit_lvalue_as_rvalue(gen_var);
+               gcc_jit_lvalue * left = lvalue_from_item(ctx, item);
+               right = gcc_jit_lvalue_as_rvalue(left);
        }
        else if(item -> mode == MODE_INTEGER)
        {
@@ -197,9 +448,64 @@ rvalue_from_item(oberon_context_t * ctx, oberon_item_t * item)
                        right = gcc_jit_context_zero(gcc_context, bool_type);                   
                }
        }
+       else if(item -> mode == MODE_CALL)
+       {
+               assert(item -> var -> class == OBERON_CLASS_PROC);
+
+               oberon_type_t * signature = item -> var -> type;
+               gen_proc_t * gen_proc = item -> var -> gen_proc;
+
+               int num_args = item -> num_args;
+               gcc_jit_rvalue *args[num_args];
+
+               oberon_expr_t * expr = item -> args;
+               oberon_object_t * arg_param = signature -> decl;
+               for(int i = 0; i < num_args; i++)
+               {
+                       if(arg_param -> class == OBERON_CLASS_VAR_PARAM)
+                       {
+                               gcc_jit_lvalue * left = lvalue_from_expr(ctx, expr);
+                               args[i] = gcc_jit_lvalue_get_address(left, NULL);
+                       }
+                       else
+                       {
+                               args[i] = rvalue_from_expr(ctx, expr);
+                       }
+                       expr = expr -> next;
+                       arg_param = arg_param -> next;
+               }
+
+               gcc_jit_function * func = gen_proc -> gcc_func;
+               right = gcc_jit_context_new_call(
+                       gcc_context, NULL, func, num_args, args
+               );
+       }
+       else if(item -> mode == MODE_INDEX)
+       {
+               gcc_jit_lvalue * left = lvalue_from_item(ctx, item);
+               right = gcc_jit_lvalue_as_rvalue(left);
+       }
+       else if(item -> mode == MODE_FIELD)
+       {
+               gen_var_t * gen_var = item -> var -> gen_var;
+               gcc_jit_field * gcc_field = gen_var -> gcc_field;
+
+               gcc_jit_rvalue * parent = rvalue_from_item(ctx, item -> parent);
+               right = gcc_jit_rvalue_access_field(parent, NULL, gcc_field);
+       }
+       else if(item -> mode == MODE_DEREF)
+       {
+               gcc_jit_lvalue * left = lvalue_from_item(ctx, item);
+               right = gcc_jit_lvalue_as_rvalue(left);
+       }
+       else if(item -> mode == MODE_NIL)
+       {
+               gcc_jit_type * type = gcc_jit_context_get_type(gcc_context, GCC_JIT_TYPE_VOID_PTR);
+               right = gcc_jit_context_null(gcc_context, type);
+       }
        else
        {
-               oberon_error(ctx, "oberon_generate_push: invalid mode %i", item -> mode);
+               oberon_error(ctx, "rvalue_from_item: invalid mode %i", item -> mode);
        }
 
        return right;
@@ -213,13 +519,19 @@ struct {
                enum gcc_jit_comparison comp_op;
        };
 } op_table[] = {
-       { 0, .unary_op = GCC_JIT_UNARY_OP_LOGICAL_NEGATE },
        { 0, .unary_op = GCC_JIT_UNARY_OP_MINUS },
+       { 0, .unary_op = GCC_JIT_UNARY_OP_BITWISE_NEGATE },
+       { 0, .unary_op = GCC_JIT_UNARY_OP_LOGICAL_NEGATE },
+       { 0, .unary_op = GCC_JIT_UNARY_OP_ABS },
 
        { 1, .binary_op = GCC_JIT_BINARY_OP_PLUS },
        { 1, .binary_op = GCC_JIT_BINARY_OP_MINUS },
+       { 1, .binary_op = GCC_JIT_BINARY_OP_MULT },
        { 1, .binary_op = GCC_JIT_BINARY_OP_DIVIDE },
        { 1, .binary_op = GCC_JIT_BINARY_OP_MODULO },
+       { 1, .binary_op = GCC_JIT_BINARY_OP_BITWISE_AND },
+       { 1, .binary_op = GCC_JIT_BINARY_OP_BITWISE_XOR },
+       { 1, .binary_op = GCC_JIT_BINARY_OP_BITWISE_OR },
        { 1, .binary_op = GCC_JIT_BINARY_OP_LOGICAL_AND },
        { 1, .binary_op = GCC_JIT_BINARY_OP_LOGICAL_OR },
 
@@ -237,8 +549,9 @@ rvalue_from_operator(oberon_context_t * ctx, oberon_oper_t * operator)
        gcc_jit_rvalue * right;
 
        gen_context_t * gen_context = ctx -> gen_context;
+       gen_type_t * gen_type = operator -> result -> gen_type;
        gcc_jit_context * gcc_context = gen_context -> gcc_context;
-       gcc_jit_type * result_type = operator -> result -> gen_type;
+       gcc_jit_type * result_type = gen_type -> gcc_type;
 
        int expr_type = op_table[operator -> op].type;
        if(expr_type == 0)
@@ -297,8 +610,21 @@ oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_
        gcc_jit_rvalue * right;
        right = rvalue_from_expr(ctx, src);
 
+       if(src -> is_item)
+       {
+               if(src -> item.mode == MODE_NIL)
+               {
+                       gen_context_t * gen_context = ctx -> gen_context;
+                       gcc_jit_context * gcc_context = gen_context -> gcc_context;
+                       gen_type_t * gen_type = dst -> result -> gen_type;
+                       gcc_jit_type * cast_to_type = gen_type -> gcc_type;
+                       right = gcc_jit_context_new_cast(gcc_context, NULL, right, cast_to_type);
+               }
+       }
+
        gen_context_t * gen_context = ctx -> gen_context;
-       gcc_jit_block * gcc_block = gen_context -> gcc_block;
+       gen_block_t * gen_block = gen_context -> block;
+       gcc_jit_block * gcc_block = gen_block -> gcc_block;
        gcc_jit_block_add_assignment(gcc_block, NULL, left, right);
 }
 
@@ -314,3 +640,11 @@ oberon_generate_code(oberon_context_t * ctx)
        gen_context -> gcc_result = gcc_result;
        ctx -> mod -> begin = gcc_jit_result_get_code(gcc_result, "BEGIN");
 }
+
+void
+oberon_generator_dump(oberon_context_t * ctx, char * path)
+{
+       gen_context_t * gen_context = ctx -> gen_context;
+       gcc_jit_context * gcc_context = gen_context -> gcc_context;
+       gcc_jit_context_dump_to_file(gcc_context, path, 0);
+}