ir.c

   1 #include <stdlib.h>
   2 #include <string.h>
   3 #include "ir.h"
   4
   5 ir_builder* ir_builder_new(const char *modulename)
   6 {
   7         ir_builder* self;
   8
   9         self = (ir_builder*)mem_a(sizeof(*self));
  10         MEM_VECTOR_INIT(self, functions);
  11         MEM_VECTOR_INIT(self, globals);
  12         self->name = NULL;
  13         ir_builder_set_name(self, modulename);
  14
  15         /* globals which always exist */
  16
  17         /* for now we give it a vector size */
  18         ir_builder_create_global(self, "OFS_RETURN", qc_variant);
  19
  20         return self;
  21 }
  22
  23 MEM_VEC_FUNCTIONS(ir_builder, ir_value*, globals)
  24 MEM_VECTOR_FUNCTIONS(ir_builder, ir_function*, functions)
  25
  26 void ir_builder_delete(ir_builder* self)
  27 {
  28         size_t i;
  29         mem_d((void*)self->name);
  30         for (i = 0; i != self->functions_count; ++i) {
  31                 ir_function_delete(self->functions[i]);
  32         }
  33         VEC_CLEAR(self, functions);
  34         for (i = 0; i != self->globals_count; ++i) {
  35                 ir_value_delete(self->globals[i]);
  36         }
  37         VEC_CLEAR(self, globals);
  38         mem_d(self);
  39 }
  40
  41 void ir_builder_set_name(ir_builder *self, const char *name)
  42 {
  43         if (self->name)
  44                 mem_d((void*)self->name);
  45         self->name = util_strdup(name);
  46 }
  47
  48 ir_function* ir_builder_get_function(ir_builder *self, const char *name)
  49 {
  50         size_t i;
  51         for (i = 0; i < self->functions_count; ++i) {
  52                 if (!strcmp(name, self->functions[i]->name))
  53                         return self->functions[i];
  54         }
  55         return NULL;
  56 }
  57
  58 ir_function* ir_builder_create_function(ir_builder *self, const char *name)
  59 {
  60         ir_function *fn = ir_builder_get_function(self, name);
  61         if (fn) {
  62                 return NULL;
  63         }
  64
  65         fn = ir_function_new(self);
  66         ir_function_set_name(fn, name);
  67         ir_builder_functions_add(self, fn);
  68         return fn;
  69 }
  70
  71 ir_value* ir_builder_get_global(ir_builder *self, const char *name)
  72 {
  73         size_t i;
  74         for (i = 0; i < self->globals_count; ++i) {
  75                 if (!strcmp(self->globals[i]->name, name))
  76                         return self->globals[i];
  77         }
  78         return NULL;
  79 }
  80
  81 ir_value* ir_builder_create_global(ir_builder *self, const char *name, ir_type_t vtype)
  82 {
  83         ir_value *ve = ir_builder_get_global(self, name);
  84         if (ve) {
  85                 return NULL;
  86         }
  87
  88         ve = ir_value_var(name, qc_global, vtype);
  89         ir_builder_globals_add(self, ve);
  90         return ve;
  91 }