typegen.lua

   1 #! /usr/bin/env lua
   2 require "luax"
   3
   4 local h   = io.open("types.h",   "w")
   5 local c   = io.open("types.c",   "w")
   6 local def = io.open("types.def", "r")
   7
   8 local function emit_h(str)
   9         h:write(str)
  10 end
  11
  12 local function emit_c(str)
  13         c:write(str)
  14 end
  15
  16 local function emit(fun, code)
  17         emit_h(fun .. ";\n")
  18         emit_c(fun ..  "\n" .. code)
  19 end
  20
  21 -- fn prefixes
  22 local export_prefix = ""
  23 local local_prefix = "__attribute__((unused)) static inline "
  24
  25 -- head
  26
  27 local disclaimer = [[
  28 /*
  29         This file was automatically generated by Dragontype.
  30         DO NOT EDIT it manually. Instead, edit types.def and re-run dragontype typegen.
  31 */
  32
  33 ]]
  34
  35 emit_h(disclaimer)
  36 emit_h([[
  37 #ifndef _DRAGONTYPE_TYPES_H_
  38 #define _DRAGONTYPE_TYPES_H_
  39
  40 #ifdef USE_DRAGONNET
  41 #include <dragonnet/peer.h>
  42 #endif
  43
  44 #include <stdbool.h>
  45 #include <stddef.h>
  46 #include <stdint.h>
  47
  48 typedef char *String;
  49
  50 typedef struct {
  51         size_t siz;
  52         unsigned char *data;
  53 } Blob;
  54
  55 ]]
  56 )
  57
  58 emit_c(disclaimer)
  59 emit_c([[
  60 #ifdef USE_DRAGONNET
  61 #include <dragonnet/send.h>
  62 #include <dragonnet/recv.h>
  63 #endif
  64
  65 #include <endian.h/endian.h>
  66 #include <stdio.h>
  67 #include <stdlib.h>
  68 #include <stdlib.h>
  69 #include <string.h>
  70 #include <zlib.h>
  71
  72 #define htobe8(x) (x)
  73 #define be8toh(x) (x)
  74
  75 #include "types.h"
  76
  77 ]] .. local_prefix ..  [[void raw_write(Blob *buffer, const void *data, size_t len)
  78 {
  79         buffer->data = realloc(buffer->data, len + buffer->siz);
  80         memcpy(&buffer->data[buffer->siz], data, len);
  81         buffer->siz += len;
  82 }
  83
  84 ]] .. local_prefix ..  [[bool raw_read(Blob *buffer, void *data, size_t len)
  85 {
  86         if (len == 0)
  87                 return true;
  88
  89         if (buffer->siz < len) {
  90                 fprintf(stderr, "[warning] buffer exhausted (requested bytes: %lu, remaining bytes: %lu)\n", len, buffer->siz);
  91                 return false;
  92         }
  93
  94         memcpy(data, buffer->data, len);
  95         buffer->data += len;
  96         buffer->siz -= len;
  97         return true;
  98 }
  99
 100 ]]
 101 )
 102
 103 -- existing types
 104 local existing_types = {}
 105 local has_deallocator = {}
 106
 107 -- vector components
 108
 109 local base_vector_components = {"x", "y", "z", "w"}
 110 local vector_components = {}
 111
 112 for i = 2, 4 do
 113         local components = {}
 114
 115         for j = 1, i do
 116                 table.insert(components, base_vector_components[j])
 117         end
 118
 119         vector_components[i] = components
 120 end
 121
 122 -- numeric types
 123
 124 local function emit_vector(type, l)
 125         local name = "v" .. l .. type
 126         local box = "aabb" .. l .. type
 127
 128         existing_types[name] = true
 129         has_deallocator[name] = false
 130
 131         existing_types[box] = true
 132         has_deallocator[box] = false
 133
 134         local typedef, equals, add, clamp, cmp, mix, write, read, send, recv =
 135                    "",     "",  "",    "",  "",  "",    "",   "",   "",   ""
 136
 137         typedef = typedef .. "\t" .. type .. " "
 138         equals  = equals  .. "\treturn "
 139         add     = add     .. "\treturn (" .. name .. ") {"
 140         clamp   = clamp   .. "\treturn (" .. name .. ") {"
 141         cmp     = cmp     .. "\tint i;\n"
 142         mix     = mix     .. "\treturn (" .. name .. ") {"
 143
 144         for i, c in ipairs(vector_components[l]) do
 145                 local last = i == l
 146
 147                 typedef = typedef
 148                         .. c ..
 149                         (last
 150                                 and ";\n"
 151                                 or ", "
 152                         )
 153
 154                 equals = equals
 155                         .. "a." .. c .. " == "
 156                         .. "b." .. c ..
 157                         (last
 158                                 and ";\n"
 159                                 or " && "
 160                         )
 161
 162                 add = add
 163                         .. "a." .. c .. " + "
 164                         .. "b." .. c ..
 165                         (last
 166                                 and "};\n"
 167                                 or ", "
 168                         )
 169
 170                 clamp = clamp
 171                         .. type .. "_clamp("
 172                         .. "val." .. c .. ", "
 173                         .. "min." .. c .. ", "
 174                         .. "max." .. c .. ")" ..
 175                         (last
 176                                 and "};\n"
 177                                 or ", "
 178                         )
 179
 180                 cmp = cmp
 181                         .. "\tif ((i = " .. type .. "_cmp("
 182                         .. "&((const " .. name .. " *) a)->" .. c .. ", "
 183                         .. "&((const " .. name .. " *) b)->" .. c .. ")) != 0)"
 184                         .. "\n\t\treturn i;\n"
 185
 186                 mix = mix
 187                         .. type .. "_mix("
 188                         .. "a." .. c .. ", "
 189                         .. "b." .. c .. ", "
 190                         .. "f" .. ")" ..
 191                         (last
 192                                 and "};\n"
 193                                 or ", "
 194                         )
 195
 196                 write = write
 197                         .. "\t" .. type .. "_write(buffer, &val->" .. c .. ");\n"
 198
 199                 read = read
 200                         .. "\tif (! " .. type .. "_read(buffer, &val->" .. c .. "))\n\t\treturn false;\n"
 201
 202                 send = send
 203                         .. "\tif (! " .. type .. "_send(peer, " .. (last and "submit" or "false") .. ", &val->" .. c .. "))\n\t\treturn false;\n"
 204
 205                 recv = recv
 206                         .. "\tif (! " .. type .. "_recv(peer, &val->" .. c .. "))\n\t\treturn false;\n"
 207         end
 208
 209         emit_h("typedef struct {\n" .. typedef ..  "} " .. name .. ";\n")
 210
 211         emit(export_prefix .. "bool " .. name .. "_equals(" .. name .. " a, " .. name .. " b)", "{\n" .. equals .. "}\n\n")
 212         emit(export_prefix .. name .. " " .. name .. "_add(" .. name .. " a, " .. name .. " b)", "{\n" .. add .. "}\n\n")
 213         emit(export_prefix .. name .. " " .. name .. "_clamp(" .. name .. " val, " .. name .. " min, " .. name .. " max)", "{\n" .. clamp .. "}\n\n")
 214         emit(export_prefix .. "int " .. name .. "_cmp(const void *a, const void *b)", "{\n" .. cmp .. "\treturn 0;\n}\n\n")
 215
 216         if type:sub(1, 1) == "f" then
 217                 emit(export_prefix .. name .. " " .. name .. "_mix(" .. name .. " a, " .. name .. " b, " .. type .. " f)", "{\n" .. mix .. "}\n\n")
 218         end
 219
 220         emit(export_prefix .. "void " .. name .. "_write(Blob *buffer, " .. name .. " *val)", "{\n" .. write .. "}\n\n")
 221         emit(export_prefix .. "bool " .. name .. "_read(Blob *buffer, " .. name .. " *val)", "{\n" .. read .. "\treturn true;\n}\n\n")
 222
 223         emit_c("#ifdef USE_DRAGONNET\n")
 224         emit_c(local_prefix .. "bool " .. name .. "_send(DragonnetPeer *peer, bool submit, " .. name .. " *val)\n{\n" .. send .. "\treturn true;\n}\n\n")
 225         emit_c(local_prefix .. "bool " .. name .. "_recv(DragonnetPeer *peer, " .. name .. " *val)\n{\n" .. recv .. "\treturn true;\n}\n")
 226         emit_c("#endif\n\n")
 227
 228         emit_h("\n")
 229
 230         emit_h("typedef struct {\n\t" .. name .. " min, max;\n} " .. box .. ";\n")
 231
 232         emit(export_prefix .. "void " .. box .. "_write(Blob *buffer, " .. box .. " *val)", "{\n\t" .. name .. "_write(buffer, &val->min);\n\t" .. name .. "_write(buffer, &val->max);\n}\n\n")
 233         emit(export_prefix .. "bool " .. box .. "_read(Blob *buffer, " .. box .. " *val)", "{\n\tif (! " .. name .. "_read(buffer, &val->min))\n\t\treturn false;\n\tif (! " .. name .. "_read(buffer, &val->max))\n\t\treturn false;\n\treturn true;\n}\n\n")
 234
 235         emit_c("#ifdef USE_DRAGONNET\n")
 236         emit_c(local_prefix .. "bool " .. box .. "_send(DragonnetPeer *peer, bool submit, " .. box .. " *val)\n{\n\tif (! " .. name .. "_send(peer, false, &val->min))\n\t\treturn false;\n\tif (! " .. name .. "_send(peer, submit, &val->max))\n\t\treturn false;\n\treturn true;\n}\n\n")
 237         emit_c(local_prefix .. "bool " .. box .. "_recv(DragonnetPeer *peer, " .. box .. " *val)\n{\n\tif (! " .. name .. "_recv(peer, &val->min))\n\t\treturn false;\n\t if (! " .. name .. "_recv(peer, &val->max))\n\t\treturn false;\n\treturn true;\n}\n")
 238         emit_c("#endif\n\n")
 239
 240         emit_h("\n")
 241 end
 242
 243 local function emit_numeric(class, bits, alias)
 244         local name = class .. bits
 245
 246         existing_types[name] = true
 247         has_deallocator[name] = false
 248
 249         emit_h("typedef " .. alias .. " " .. name .. ";\n")
 250
 251         emit(export_prefix .. name .. " " .. name .. "_min(" .. name .. " a, " .. name .. " b)", "{\n\treturn a < b ? a : b;\n}\n\n")
 252         emit(export_prefix .. name .. " " .. name .. "_max(" .. name .. " a, " .. name .. " b)", "{\n\treturn a > b ? a : b;\n}\n\n")
 253         emit(export_prefix .. name .. " " .. name .. "_clamp(" .. name .. " val, " .. name .. " min, " .. name .. " max)", "{\n\treturn val < min ? min : val > max ? max : val;\n}\n\n")
 254         emit(export_prefix .. "int " .. name .. "_cmp(const void *a, const void *b)", "{\n\treturn\n\t\t*(const " .. name .. " *) a < *(const " .. name .. " *) b ? -1 :\n\t\t*(const " .. name .. " *) a > *(const " .. name .. " *) b ? +1 :\n\t\t0;\n}\n\n")
 255
 256         if class == "f" then
 257                 emit(export_prefix .. name .. " " .. name .. "_mix(" .. name .. " a, " .. name .. " b, " .. name .. " f)", "{\n\treturn (1.0 - f) * a + b * f;\n}\n\n")
 258         end
 259
 260         emit(export_prefix .. "void " .. name .. "_write(Blob *buffer, " .. name .. " *val)", "{\n" .. (class == "u"
 261                 and "\t" .. name .. " be = htobe" .. bits .. "(*val);\n\traw_write(buffer, &be, sizeof be);\n"
 262                 or  "\tu" .. bits .. "_write(buffer, (u" .. bits .. " *) val);\n"
 263         ) .. "}\n\n")
 264         emit(export_prefix .. "bool " .. name .. "_read(Blob *buffer, " .. name .. " *val)", "{\n" .. (class == "u"
 265                 and "\t" .. name .. " be;\n\tif (! raw_read(buffer, &be, sizeof be))\n\t\treturn false;\n\t*val = be" .. bits .. "toh(be);\n\treturn true;\n"
 266                 or  "\treturn u" .. bits .. "_read(buffer, (u" .. bits .. " *) val);\n"
 267         ) .. "}\n\n")
 268
 269         emit_c("#ifdef USE_DRAGONNET\n")
 270         emit_c(local_prefix .. "bool " .. name .. "_send(DragonnetPeer *peer, bool submit, " .. name .. " *val)\n{\n" .. (class == "u"
 271                 and "\t" .. name .. " be = htobe" .. bits .. "(*val);\n\treturn dragonnet_send_raw(peer, submit, &be, sizeof be);\n"
 272                 or  "\treturn u" .. bits .. "_send(peer, submit, (u" .. bits .. " *) val);\n"
 273         ) .. "}\n\n")
 274         emit_c(local_prefix .. "bool " .. name .. "_recv(DragonnetPeer *peer, " .. name .. " *val)\n{\n" .. (class == "u"
 275                 and "\t" .. name .. " be;\n\tif (! dragonnet_recv_raw(peer, &be, sizeof be))\n\t\treturn false;\n\t*val = be" .. bits .. "toh(be);\n\treturn true;\n"
 276                 or  "\treturn u" .. bits .. "_recv(peer, (u" .. bits .. " *) val);\n"
 277         ) .. "}\n")
 278         emit_c("#endif\n\n")
 279
 280         emit_h("\n")
 281
 282         for i = 2, 4 do
 283                 emit_vector(name, i)
 284         end
 285 end
 286
 287 for i = 0, 3 do
 288         local bytes = math.floor(2 ^ i)
 289         local bits = 8 * bytes
 290
 291         emit_numeric("u", bits, "uint" .. bits .. "_t")
 292         emit_numeric("s", bits,  "int" .. bits .. "_t")
 293
 294         if i >= 2 then
 295                 emit_numeric("f", bits, ({"float", "double"})[i - 1])
 296         end
 297 end
 298
 299 -- string
 300
 301 existing_types.String = true
 302 has_deallocator.String = true
 303
 304 emit(
 305 export_prefix .. "void String_free(String *val)", [[
 306 {
 307         if (*val)
 308                 free(*val);
 309 }
 310
 311 ]]
 312 )
 313
 314 emit(
 315 export_prefix .. "void String_write(Blob *buffer, String *val)", [[
 316 {
 317         *val ? raw_write(buffer, *val, strlen(*val) + 1) : raw_write(buffer, "", 1);
 318 }
 319
 320 ]]
 321 )
 322
 323 emit(
 324 export_prefix .. "bool String_read(Blob *buffer, String *val)", [[
 325 {
 326         String v = malloc(1 + (1 << 16));
 327
 328         char c;
 329         for (u16 i = 0;; i++) {
 330                 if (! raw_read(buffer, &c, 1)) {
 331                         free(v);
 332                         return false;
 333                 }
 334
 335                 v[i] = c;
 336                 if (c == '\0')
 337                         break;
 338         }
 339
 340         *val = realloc(v, strlen(v) + 1);
 341
 342         return true;
 343 }
 344
 345 ]]
 346 )
 347
 348 emit_c(
 349 [[
 350 #ifdef USE_DRAGONNET
 351 ]] .. local_prefix .. [[bool String_send(DragonnetPeer *peer, bool submit, String *val)
 352 {
 353         return *val ? dragonnet_send_raw(peer, submit, *val, strlen(*val) + 1) : dragonnet_send_raw(peer, submit, "", 1);
 354 }
 355
 356 ]] .. local_prefix .. [[bool String_recv(DragonnetPeer *peer, String *val)
 357 {
 358         String v = malloc(1 + (1 << 16));
 359
 360         char c;
 361         for (u16 i = 0;; i++) {
 362                 if (! dragonnet_recv_raw(peer, &c, 1)) {
 363                         free(v);
 364                         return false;
 365                 }
 366
 367                 v[i] = c;
 368                 if (c == '\0')
 369                         break;
 370         }
 371
 372         *val = realloc(v, strlen(v) + 1);
 373
 374         return true;
 375 }
 376 #endif
 377
 378 ]]
 379 )
 380
 381 emit_h("\n")
 382
 383 -- blob
 384
 385 existing_types.Blob = true
 386 has_deallocator.Blob = true
 387
 388 emit(
 389 export_prefix .. "void Blob_compress(Blob *buffer, Blob *val)", [[
 390 {
 391         buffer->siz = 8 + 2 + val->siz;
 392         buffer->data = malloc(buffer->siz);
 393
 394         *(u64 *) buffer->data = val->siz;
 395
 396         z_stream s;
 397         s.zalloc = Z_NULL;
 398         s.zfree = Z_NULL;
 399         s.opaque = Z_NULL;
 400
 401         s.avail_in = val->siz;
 402         s.next_in = (Bytef *) val->data;
 403         s.avail_out = buffer->siz - 8 + 1;
 404         s.next_out = (Bytef *) buffer->data + 8;
 405
 406         deflateInit(&s, Z_BEST_COMPRESSION);
 407         deflate(&s, Z_FINISH);
 408         deflateEnd(&s);
 409
 410         buffer->siz = s.total_out + 8;
 411 }
 412
 413 ]]
 414 )
 415
 416 emit(
 417 export_prefix .. "void Blob_decompress(Blob *buffer, Blob *val)", [[
 418 {
 419         buffer->siz = *(u64 *) val->data;
 420         buffer->data = malloc(buffer->siz);
 421
 422         z_stream s;
 423         s.zalloc = Z_NULL;
 424         s.zfree = Z_NULL;
 425         s.opaque = Z_NULL;
 426
 427         s.avail_in = val->siz - 8;
 428         s.next_in = val->data + 8;
 429         s.avail_out = buffer->siz;
 430         s.next_out = (Bytef *) buffer->data;
 431
 432         inflateInit(&s);
 433         inflate(&s, Z_NO_FLUSH);
 434         inflateEnd(&s);
 435
 436         buffer->siz = s.total_out;
 437 }
 438
 439 ]]
 440 )
 441
 442 emit(
 443 export_prefix .. "void Blob_shrink(Blob *val)", [[
 444 {
 445         val->data = realloc(val->data, val->siz);
 446 }
 447
 448 ]]
 449 )
 450
 451 emit(
 452 export_prefix .. "void Blob_free(Blob *val)", [[
 453 {
 454         if (val->data)
 455                 free(val->data);
 456 }
 457
 458 ]]
 459 )
 460
 461 emit(
 462 export_prefix .. "void Blob_write(Blob *buffer, Blob *val)", [[
 463 {
 464         u64_write(buffer, &val->siz);
 465         raw_write(buffer, val->data, val->siz);
 466 }
 467
 468 ]]
 469 )
 470
 471 emit(
 472 export_prefix .. "bool Blob_read(Blob *buffer, Blob *val)", [[
 473 {
 474         if (! u64_read(buffer, &val->siz))
 475                 return false;
 476
 477         if (! raw_read(buffer, val->data = malloc(val->siz), val->siz)) {
 478                 free(val->data);
 479                 val->data = NULL;
 480                 return false;
 481         }
 482
 483         return true;
 484 }
 485
 486 ]]
 487 )
 488
 489 emit_c(
 490 [[
 491 #ifdef USE_DRAGONNET
 492 ]] .. local_prefix .. [[bool Blob_send(DragonnetPeer *peer, bool submit, Blob *val)
 493 {
 494         if (! u64_send(peer, false, &val->siz))
 495                 return false;
 496         return dragonnet_send_raw(peer, submit, val->data, val->siz);
 497 }
 498
 499 ]] .. local_prefix .. [[bool Blob_recv(DragonnetPeer *peer, Blob *val)
 500 {
 501         if (! u64_recv(peer, &val->siz))
 502                 return false;
 503
 504         if (! dragonnet_recv_raw(peer, val->data = malloc(val->siz), val->siz)) {
 505                 free(val->data);
 506                 val->data = NULL;
 507                 return false;
 508         }
 509
 510         return true;
 511 }
 512 #endif
 513
 514 ]]
 515 )
 516
 517 emit_h("\n")
 518
 519
 520 emit_c(
 521 [[
 522 ]] .. local_prefix .. [[void raw_write_compressed(Blob *buffer, void *val, void (*val_write)(Blob *, void *))
 523 {
 524         Blob compressed, raw = {0};
 525         val_write(&raw, val);
 526         Blob_compress(&compressed, &raw);
 527         Blob_write(buffer, &compressed);
 528
 529         Blob_free(&compressed);
 530         Blob_free(&raw);
 531 }
 532
 533 ]] .. local_prefix .. [[bool raw_read_compressed(Blob *buffer, void *val, bool (*val_read)(Blob *, void *))
 534 {
 535         Blob compressed, raw = {0};
 536         bool success = Blob_read(buffer, &compressed);
 537
 538         if (success) {
 539                 Blob_decompress(&raw, &compressed);
 540                 Blob raw_buffer = raw;
 541                 success = success && val_read(&raw_buffer, val);
 542         }
 543
 544         Blob_free(&compressed);
 545         Blob_free(&raw);
 546
 547         return success;
 548 }
 549
 550 #ifdef USE_DRAGONNET
 551 ]] .. local_prefix .. [[bool raw_send_compressed(DragonnetPeer *peer, bool submit, void *val, void (*val_write)(Blob *, void *))
 552 {
 553         Blob compressed, raw = {0};
 554         val_write(&raw, val);
 555         Blob_compress(&compressed, &raw);
 556         bool success = Blob_send(peer, submit, &compressed);
 557
 558         Blob_free(&compressed);
 559         Blob_free(&raw);
 560
 561         return success;
 562 }
 563
 564 ]] .. local_prefix .. [[bool raw_recv_compressed(DragonnetPeer *peer, void *val, bool (*val_read)(Blob *, void *))
 565 {
 566         Blob compressed, raw = {0};
 567         bool success = Blob_recv(peer, &compressed);
 568
 569         if (success) {
 570                 Blob_decompress(&raw, &compressed);
 571                 Blob raw_buffer = raw;
 572                 success = success && val_read(&raw_buffer, val);
 573         }
 574
 575         Blob_free(&compressed);
 576         Blob_free(&raw);
 577
 578         return success;
 579 }
 580 #endif
 581
 582 ]]
 583 )
 584
 585 local custom_types = {}
 586 local current_type
 587
 588 local function consume_name(name)
 589         if current_type then
 590                 table.insert(custom_types, current_type)
 591         end
 592
 593         if name == "" then
 594                 current_type = nil
 595         else
 596                 current_type = {components = {}, component_names = {}}
 597
 598                 current_type.flags = name:split(" ")
 599                 current_type.name = table.remove(current_type.flags, #current_type.flags)
 600
 601                 if existing_types[current_type.name] then
 602                         error("redeclaration of type " .. current_type.name)
 603                 end
 604
 605                 existing_types[current_type.name] = true
 606                 has_deallocator[current_type.name] = false
 607         end
 608 end
 609
 610 local function consume_comp(comp)
 611         if not current_type then
 612                 error("component without a type: " .. comp)
 613         end
 614
 615         local component = {}
 616
 617         component.flags = comp:split(" ")
 618         component.name = table.remove(component.flags, #component.flags)
 619         component.type = table.remove(component.flags, #component.flags)
 620
 621         component.full_name = current_type.name .. "." .. component.name
 622
 623         local base_type = ""
 624         local met_brace = false
 625         local brace_level = 0
 626
 627         for i = 1, #component.type do
 628                 local c = component.type:sub(i, i)
 629                 local append = false
 630
 631                 if c == "[" then
 632                         if not met_brace then
 633                                 component.array = {}
 634                         end
 635
 636                         met_brace = true
 637                         brace_level = brace_level + 1
 638
 639                         if brace_level == 1 then
 640                                 table.insert(component.array, "")
 641                         else
 642                                 append = true
 643                         end
 644                 elseif c == "]" then
 645                         brace_level = brace_level - 1
 646
 647                         if brace_level < 0 then
 648                                 error("missing [ in " .. component.full_name)
 649                         elseif brace_level > 0 then
 650                                 append = true
 651                         end
 652                 elseif not met_brace then
 653                         base_type = base_type .. c
 654                 elseif brace_level == 1 then
 655                         append = true
 656                 elseif brace_level == 0 then
 657                         error("invalid character " .. c .. " outside of braces in " .. component.full_name)
 658                 end
 659
 660                 if append then
 661                         component.array[#component.array] = component.array[#component.array] .. c
 662                 end
 663         end
 664
 665         component.type = base_type
 666
 667         if brace_level > 0 then
 668                 error("missing ] in " .. component.full_name)
 669         end
 670
 671         if not existing_types[component.type] then
 672                 error("type " .. component.type .. " of " .. component.full_name .. " does not exist ")
 673         end
 674
 675         has_deallocator[current_type.name] = has_deallocator[current_type.name] or has_deallocator[component.type]
 676
 677         if current_type.component_names[component.name] then
 678                 error("component " .. component.full_name .. " redeclared")
 679         end
 680
 681         current_type.component_names[component.name] = true
 682
 683         table.insert(current_type.components, component)
 684 end
 685
 686 if def then
 687         for l in def:lines() do
 688                 local f = l:sub(1, 1)
 689
 690                 if f == "\t" then
 691                         consume_comp(l:sub(2, #l))
 692                 elseif f == "#" then
 693                         emit_h(l .. "\n\n")
 694                 elseif f ~= ";" then
 695                         consume_name(l)
 696                 end
 697         end
 698 end
 699
 700 consume_name("")
 701
 702 local dragonnet_types_h = ""
 703 local dragonnet_types_c = ""
 704
 705 for _, t in ipairs(custom_types) do
 706         local pkt
 707
 708         for _, f in pairs(t.flags) do
 709                 if f == "pkt" then
 710                         pkt = true
 711                 else
 712                         error("invalid flag " .. f .. " for " .. t.name)
 713                 end
 714         end
 715
 716         local typedef, free, write, read, send, recv =
 717                    "",   "",    "",   "",   "",   ""
 718
 719         for ic, c in ipairs(t.components) do
 720                 typedef = typedef
 721                         .. "\t" .. c.type .. " " .. c.name
 722
 723                 if c.array then
 724                         typedef = typedef
 725                                 .. "[" .. table.concat(c.array, "][") .. "]"
 726                 end
 727
 728                 typedef = typedef
 729                         .. ";\n"
 730
 731                 local compressed = false
 732
 733                 for _, f in pairs(c.flags) do
 734                         if f == "compressed" then
 735                                 compressed = true
 736                         else
 737                                 error("invalid flag " .. f .. " for " .. c.full_name)
 738                         end
 739                 end
 740
 741                 local indent = "\t"
 742                 local loop = ""
 743                 local index = ""
 744                 local array_submit = ""
 745
 746                 if c.array then
 747                         for ia, a in ipairs(c.array) do
 748                                 local it = "i" .. ia
 749
 750                                 loop = loop .. indent .. "for (int " .. it .. " = 0; " .. it .. " < " .. a .. "; " .. it .. "++)\n"
 751                                 indent = indent .. "\t"
 752
 753                                 index = index .. "[" .. it .. "]"
 754
 755                                 array_submit = array_submit .. " && " .. it .. " == " .. a
 756                         end
 757                 end
 758
 759                 local addr = "&val->" .. c.name .. index
 760
 761                 if has_deallocator[c.type] then
 762                         free = free
 763                                 .. loop .. indent .. c.type .. "_free(" .. addr .. ");\n"
 764                 end
 765
 766                 write = write
 767                         .. loop .. indent .. (compressed
 768                                 and "raw_write_compressed(buffer, " .. addr .. ", (void *) &" .. c.type .. "_write);\n"
 769                                 or      c.type .. "_write(buffer, " .. addr .. ");\n"
 770                         )
 771
 772                 read = read
 773                         .. loop .. indent .. "if (! " .. (compressed
 774                                 and "raw_read_compressed(buffer, " .. addr .. ", (void *) &" .. c.type .. "_read)"
 775                                 or      c.type .. "_read(buffer, " .. addr .. ")"
 776                         ) .. ")\n" .. indent .. "\treturn false;\n"
 777
 778                 local submit = ic == #t.components and "submit" .. array_submit or "false"
 779                 send = send
 780                         .. loop .. indent .. "if (! " .. (compressed
 781                                 and "raw_send_compressed(peer, " .. submit .. ", " .. addr .. ", (void *) &" .. c.type .. "_write)"
 782                                 or      c.type .. "_send(peer, " .. submit .. ", " .. addr .. ")"
 783                         ) .. ")\n" .. indent .. "\treturn false;\n"
 784
 785                 recv = recv
 786                         .. loop .. indent .. "if (! " .. (compressed
 787                                 and "raw_recv_compressed(peer, " .. addr .. ", (void *) &" .. c.type .. "_read)"
 788                                 or      c.type .. "_recv(peer, " .. addr .. ")"
 789                         ) .. ")\n" .. indent .. "\treturn false;\n"
 790         end
 791
 792         emit_h("typedef struct {\n" .. typedef .. "} " .. t.name .. ";\n")
 793
 794         if has_deallocator[t.name] then
 795                 emit(export_prefix .. "void " .. t.name .. "_free(" .. t.name .. " *val)", "{\n" .. free .. "}\n\n")
 796         end
 797
 798         emit(export_prefix .. "void " .. t.name .. "_write(Blob *buffer, " .. t.name .. " *val)", "{\n" .. write .. "}\n\n")
 799         emit(export_prefix .. "bool " .. t.name .. "_read(Blob *buffer, " .. t.name .. " *val)", "{\n" .. read .. "\treturn true;\n}\n\n")
 800
 801         if pkt then
 802                 emit_h("#ifdef USE_DRAGONNET\n")
 803         end
 804
 805         emit_c("#ifdef USE_DRAGONNET\n")
 806
 807         emit_c(local_prefix .. "bool " .. t.name .. "_send(DragonnetPeer *peer, bool submit, " .. t.name .. " *val)\n{\n" .. send .. "\treturn true;\n}\n\n")
 808         emit_c(local_prefix .. "bool " .. t.name .. "_recv(DragonnetPeer *peer, " .. t.name .. " *val)\n{\n" .. recv .. "\treturn true;\n}\n")
 809
 810         if pkt then
 811                 emit_c("\n")
 812                 emit_c("static DragonnetTypeId " .. t.name .. "_type_id = DRAGONNET_TYPE_" .. t.name .. ";\n")
 813
 814                 emit("bool dragonnet_peer_send_" .. t.name .. "(DragonnetPeer *peer, " .. t.name .. " *val)", "{\n\tpthread_mutex_lock(&peer->mtx);\n\tif (! u16_send(peer, false, &" .. t.name .. "_type_id))\n\t\treturn false;\n\tif (! " .. t.name .. "_send(peer, true, val))\n\t\treturn false;\n\treturn true;\n}\n")
 815                 emit_h("#endif\n")
 816
 817                 dragonnet_types_h = dragonnet_types_h
 818                         .. "\tDRAGONNET_TYPE_" .. t.name .. ",\n"
 819
 820                 dragonnet_types_c = dragonnet_types_c
 821                         .. "\t{\n\t\t.siz = sizeof(" .. t.name .. "),\n\t\t.deserialize = (void *) &" .. t.name .. "_recv,\n\t\t.free = " .. (has_deallocator[t.name] and "(void *) &" .. t.name .. "_free" or "NULL") .. ",\n\t},\n"
 822         end
 823
 824         emit_c("#endif\n\n")
 825
 826         emit_h("\n")
 827 end
 828
 829 emit_h("#ifdef USE_DRAGONNET\n")
 830 emit_h("typedef enum {\n" .. dragonnet_types_h .. "\tDRAGONNET_NUM_TYPES\n} DragonnetTypeNum;\n")
 831 emit_h("#endif\n\n")
 832
 833 emit_c("#ifdef USE_DRAGONNET\n")
 834 emit_c("DragonnetTypeId dragonnet_num_types = DRAGONNET_NUM_TYPES;\nDragonnetType dragonnet_types[] = {\n" .. dragonnet_types_c .. "};\n")
 835 emit_c("#endif\n")
 836
 837 emit_h("#endif\n")
 838
 839 h:close()
 840 c:close()
 841 def:close()