nsz Git - cparser/blob - write_fluffy.c

   1 #include <config.h>
   2
   3 #include <errno.h>
   4 #include <string.h>
   5
   6 #include "write_fluffy.h"
   7 #include "ast_t.h"
   8 #include "type_t.h"
   9 #include "type.h"
  10 #include "adt/error.h"
  11
  12 static const context_t *global_context;
  13 static FILE            *out;
  14
  15 static void write_type(const type_t *type);
  16
  17 static const char *get_atomic_type_string(const atomic_type_type_t type)
  18 {
  19         switch(type) {
  20         case ATOMIC_TYPE_VOID:       return "void";
  21         case ATOMIC_TYPE_CHAR:       return "byte";
  22         case ATOMIC_TYPE_SCHAR:      return "byte";
  23         case ATOMIC_TYPE_UCHAR:      return "unsigned byte";
  24         case ATOMIC_TYPE_SHORT:      return "short";
  25         case ATOMIC_TYPE_USHORT:     return "unsigned short";
  26         case ATOMIC_TYPE_INT:        return "int";
  27         case ATOMIC_TYPE_UINT:       return "unsigned int";
  28         case ATOMIC_TYPE_LONG:       return "int";
  29         case ATOMIC_TYPE_ULONG:      return "unsigned int";
  30         case ATOMIC_TYPE_LONGLONG:   return "long";
  31         case ATOMIC_TYPE_ULONGLONG:  return "unsigned long";
  32         case ATOMIC_TYPE_FLOAT:      return "float";
  33         case ATOMIC_TYPE_DOUBLE:     return "double";
  34         case ATOMIC_TYPE_LONG_DOUBLE: return "double";
  35         case ATOMIC_TYPE_BOOL:       return "bool";
  36         default:                     panic("unsupported atomic type");
  37         }
  38 }
  39
  40 static void write_atomic_type(const atomic_type_t *type)
  41 {
  42         fprintf(out, "%s", get_atomic_type_string(type->atype));
  43 }
  44
  45 static void write_pointer_type(const pointer_type_t *type)
  46 {
  47         write_type(type->points_to);
  48         fputc('*', out);
  49 }
  50
  51 static declaration_t *find_typedef(const type_t *type)
  52 {
  53         /* first: search for a matching typedef in the global type... */
  54         declaration_t *declaration = global_context->declarations;
  55         while(declaration != NULL) {
  56                 if(! (declaration->storage_class == STORAGE_CLASS_TYPEDEF)) {
  57                         declaration = declaration->next;
  58                         continue;
  59                 }
  60                 if(declaration->type == type)
  61                         break;
  62                 declaration = declaration->next;
  63         }
  64
  65         return declaration;
  66 }
  67
  68 static void write_compound_type(const compound_type_t *type)
  69 {
  70         declaration_t *declaration = find_typedef((const type_t*) type);
  71         if(declaration != NULL) {
  72                 fprintf(out, "%s", declaration->symbol->string);
  73                 return;
  74         }
  75
  76         /* does the struct have a name? */
  77         symbol_t *symbol = type->declaration->symbol;
  78         if(symbol != NULL) {
  79                 /* TODO: make sure we create a struct for it... */
  80                 fprintf(out, "%s", symbol->string);
  81                 return;
  82         }
  83         /* TODO: create a struct and use its name here... */
  84         fprintf(out, "/* TODO anonymous struct */byte");
  85 }
  86
  87 static void write_enum_type(const enum_type_t *type)
  88 {
  89         declaration_t *declaration = find_typedef((const type_t*) type);
  90         if(declaration != NULL) {
  91                 fprintf(out, "%s", declaration->symbol->string);
  92                 return;
  93         }
  94
  95         /* does the enum have a name? */
  96         symbol_t *symbol = type->declaration->symbol;
  97         if(symbol != NULL) {
  98                 /* TODO: make sure we create an enum for it... */
  99                 fprintf(out, "%s", symbol->string);
 100                 return;
 101         }
 102         /* TODO: create a struct and use its name here... */
 103         fprintf(out, "/* TODO anonymous enum */byte");
 104 }
 105
 106 static void write_function_type(const function_type_t *type)
 107 {
 108         fprintf(out, "(func(");
 109
 110         function_parameter_t *parameter = type->parameters;
 111         int                   first     = 1;
 112         while(parameter != NULL) {
 113                 if(!first) {
 114                         fprintf(out, ", ");
 115                 } else {
 116                         first = 0;
 117                 }
 118
 119 #if 0
 120                 if(parameter->symbol != NULL) {
 121                         fprintf(out, "%s : ", parameter->symbol->string);
 122                 } else {
 123                         /* TODO make up some unused names (or allow _ in fluffy?) */
 124                         fprintf(out, "_ : ");
 125                 }
 126 #endif
 127                 fputs("_ : ", out);
 128                 write_type(parameter->type);
 129
 130                 parameter = parameter->next;
 131         }
 132
 133         fprintf(out, ") : ");
 134         write_type(type->result_type);
 135         fprintf(out, ")");
 136 }
 137
 138 static void write_type(const type_t *type)
 139 {
 140         switch(type->type) {
 141         case TYPE_ATOMIC:
 142                 write_atomic_type((const atomic_type_t*) type);
 143                 return;
 144         case TYPE_POINTER:
 145                 write_pointer_type((const pointer_type_t*) type);
 146                 return;
 147         case TYPE_COMPOUND_UNION:
 148         case TYPE_COMPOUND_STRUCT:
 149                 write_compound_type((const compound_type_t*) type);
 150                 return;
 151         case TYPE_ENUM:
 152                 write_enum_type((const enum_type_t*) type);
 153                 return;
 154         case TYPE_FUNCTION:
 155                 write_function_type((const function_type_t*) type);
 156                 return;
 157         case TYPE_INVALID:
 158                 panic("invalid type found");
 159                 break;
 160         default:
 161                 fprintf(out, "/* TODO type */");
 162                 break;
 163         }
 164 }
 165
 166 static void write_struct_entry(const declaration_t *declaration)
 167 {
 168         fprintf(out, "\t%s : ", declaration->symbol->string);
 169         write_type(declaration->type);
 170         fprintf(out, "\n");
 171 }
 172
 173 static void write_struct(const symbol_t *symbol, const compound_type_t *type)
 174 {
 175         fprintf(out, "struct %s:\n", symbol->string);
 176
 177         const declaration_t *declaration = type->declaration->context.declarations;
 178         while(declaration != NULL) {
 179                 write_struct_entry(declaration);
 180                 declaration = declaration->next;
 181         }
 182
 183         fprintf(out, "\n");
 184 }
 185
 186 static void write_union(const symbol_t *symbol, const compound_type_t *type)
 187 {
 188         fprintf(out, "union %s:\n", symbol->string);
 189
 190         const declaration_t *declaration = type->declaration->context.declarations;
 191         while(declaration != NULL) {
 192                 write_struct_entry(declaration);
 193                 declaration = declaration->next;
 194         }
 195
 196         fprintf(out, "\n");
 197 }
 198
 199 static void write_expression(const expression_t *expression);
 200
 201 static void write_unary_expression(const unary_expression_t *expression)
 202 {
 203         switch(expression->type) {
 204         case UNEXPR_NEGATE:
 205                 fputc('-', out);
 206                 break;
 207         case UNEXPR_NOT:
 208                 fputc('!', out);
 209                 break;
 210         default:
 211                 panic("unimeplemented unary expression found");
 212         }
 213         write_expression(expression->value);
 214 }
 215
 216 static void write_expression(const expression_t *expression)
 217 {
 218         const const_expression_t *constant;
 219         /* TODO */
 220         switch(expression->type) {
 221         case EXPR_CONST:
 222                 constant = &expression->conste;
 223                 if(is_type_integer(expression->base.datatype)) {
 224                         fprintf(out, "%lld", constant->v.int_value);
 225                 } else {
 226                         fprintf(out, "%Lf", constant->v.float_value);
 227                 }
 228                 break;
 229         case EXPR_UNARY:
 230                 write_unary_expression((const unary_expression_t*) expression);
 231                 break;
 232         default:
 233                 panic("not implemented expression");
 234         }
 235 }
 236
 237 static void write_enum(const symbol_t *symbol, const enum_type_t *type)
 238 {
 239         fprintf(out, "enum %s:\n", symbol->string);
 240
 241         declaration_t *entry = type->declaration->next;
 242         for ( ; entry != NULL && entry->storage_class == STORAGE_CLASS_ENUM_ENTRY;
 243                         entry = entry->next) {
 244                 fprintf(out, "\t%s", entry->symbol->string);
 245                 if(entry->init.initializer != NULL) {
 246                         fprintf(out, " <- ");
 247                         write_expression(entry->init.enum_value);
 248                 }
 249                 fputc('\n', out);
 250         }
 251         fprintf(out, "typealias %s <- int\n", symbol->string);
 252         fprintf(out, "\n");
 253 }
 254
 255 static void write_variable(const declaration_t *declaration)
 256 {
 257         fprintf(out, "var %s : ", declaration->symbol->string);
 258         write_type(declaration->type);
 259         /* TODO: initializers */
 260         fprintf(out, "\n");
 261 }
 262
 263 static void write_function(const declaration_t *declaration)
 264 {
 265         if(declaration->init.statement != NULL) {
 266                 fprintf(stderr, "Warning: can't convert function bodies (at %s)\n",
 267                         declaration->symbol->string);
 268         }
 269
 270         fprintf(out, "func extern %s(",
 271                 declaration->symbol->string);
 272
 273         const function_type_t *function_type
 274                 = (const function_type_t*) declaration->type;
 275
 276         declaration_t *parameter = declaration->context.declarations;
 277         int            first     = 1;
 278         for( ; parameter != NULL; parameter = parameter->next) {
 279                 if(!first) {
 280                         fprintf(out, ", ");
 281                 } else {
 282                         first = 0;
 283                 }
 284                 if(parameter->symbol != NULL) {
 285                         fprintf(out, "%s : ", parameter->symbol->string);
 286                 } else {
 287                         fputs("_ : ", out);
 288                 }
 289                 write_type(parameter->type);
 290         }
 291         if(function_type->variadic) {
 292                 if(!first) {
 293                         fprintf(out, ", ");
 294                 } else {
 295                         first = 0;
 296                 }
 297                 fputs("...", out);
 298         }
 299         fprintf(out, ")");
 300
 301         const type_t *result_type = function_type->result_type;
 302         if(result_type->type != TYPE_ATOMIC ||
 303                         ((const atomic_type_t*) result_type)->atype != ATOMIC_TYPE_VOID) {
 304                 fprintf(out, " : ");
 305                 write_type(result_type);
 306         }
 307         fputc('\n', out);
 308 }
 309
 310 void write_fluffy_decls(const translation_unit_t *unit)
 311 {
 312 #if 0
 313         out = fopen("out.fluffy", "w");
 314         if(out == NULL) {
 315                 fprintf(stderr, "Couldn't open out.fluffy: %s\n", strerror(errno));
 316                 exit(1);
 317         }
 318 #endif
 319         out            = stdout;
 320         global_context = &unit->context;
 321
 322         fprintf(out, "/* WARNING: Automatically generated file */\n");
 323
 324         /* write structs,unions + enums */
 325         declaration_t *declaration = unit->context.declarations;
 326         for( ; declaration != NULL; declaration = declaration->next) {
 327                 //fprintf(out, "// Decl: %s\n", declaration->symbol->string);
 328                 if(! (declaration->storage_class == STORAGE_CLASS_TYPEDEF)) {
 329                         continue;
 330                 }
 331                 type_t *type = declaration->type;
 332                 if(type->type == TYPE_COMPOUND_STRUCT) {
 333                         write_struct(declaration->symbol, (compound_type_t*) type);
 334                 } else if(type->type == TYPE_COMPOUND_UNION) {
 335                         write_union(declaration->symbol, (compound_type_t*) type);
 336                 } else if(type->type == TYPE_ENUM) {
 337                         write_enum(declaration->symbol, (enum_type_t*) type);
 338                 }
 339         }
 340
 341         /* write global variables */
 342         declaration = unit->context.declarations;
 343         for( ; declaration != NULL; declaration = declaration->next) {
 344                 if(declaration->namespc != NAMESPACE_NORMAL)
 345                         continue;
 346                 if(declaration->storage_class == STORAGE_CLASS_TYPEDEF
 347                                 || declaration->storage_class == STORAGE_CLASS_ENUM_ENTRY)
 348                         continue;
 349
 350                 type_t *type = declaration->type;
 351                 if(type->type == TYPE_FUNCTION)
 352                         continue;
 353
 354                 write_variable(declaration);
 355         }
 356
 357         /* write functions */
 358         declaration = unit->context.declarations;
 359         for( ; declaration != NULL; declaration = declaration->next) {
 360                 if(declaration->namespc != NAMESPACE_NORMAL)
 361                         continue;
 362                 if(declaration->storage_class == STORAGE_CLASS_TYPEDEF
 363                                 || declaration->storage_class == STORAGE_CLASS_ENUM_ENTRY)
 364                         continue;
 365
 366                 type_t *type = declaration->type;
 367                 if(type->type != TYPE_FUNCTION)
 368                         continue;
 369
 370                 write_function(declaration);
 371         }
 372
 373         //fclose(out);
 374 }