write-lua-bytecode.js

define(['./parse', './bcompile', './bytecode-table', './top-level', './str-escape', './tests', './stdlib', './extensions'], function(parse, bcompile, bytecode_table, top_level, str_escape, tests, stdlib) {
    var fake_require =
        "var __modules__ = {};\n"+
        "define = function _define(name, deps, init_func) {\n"+
        "  var d = deps.map(function(m) { return __modules__[m]; });\n"+
        "  __modules__[name] = init_func.apply(this, d);\n"+
        "};\n";
    var make_compile_from_source = function(parse, bcompile, TOP_LEVEL) {
       var compile_from_source = function (source, as_object) {
           source = source || '{ return 1+2; }';
           var tree = parse(source, TOP_LEVEL);
           var bc = bcompile(tree, true/*don't desugar frame get*/);
           var result = as_object ? bc : bc.encode();
           return result;
       };
       compile_from_source.make_repl = function() {
           var state = null;
           return function(source) {
               var rv = parse.repl(state, source, TOP_LEVEL);
               state = rv.state;
               return bcompile(rv.tree).encode();
           };
       };
       return compile_from_source;
    };
    var cfs_source = make_compile_from_source.toSource ?
        make_compile_from_source.toSource() :
        make_compile_from_source.toString();
    cfs_source = 'define("compile_from_source", ["parse","bcompile","top-level"], '+
        cfs_source + ');';
    var top_level_source = 'define("top-level", [], function() { return ' +
        str_escape(top_level) + '; });';
    var source = '{\n'+
        stdlib.source()+'\n'+
        fake_require +
        tests.lookup("tokenize")+"\n"+
        tests.lookup("parse")+"\n"+
        tests.lookup("bytecode-table")+"\n"+
        tests.lookup("bcompile")+"\n"+
        top_level_source+"\n"+
        cfs_source + '\n' +
        /*
        "var test_nan = function() { return NaN; };\n" +
        "var test_inf = function() { return Infinity; };\n" +
        "var test_neg_inf = function() { return -Infinity; };\n" +
        */
        "return __modules__['compile_from_source']; }\n";

    /* XXX Hacks for initial tests:
    source = "{ console.log('Hello,', 'world!'); }";
    source = "{ var fib=function(n){return (n<2)?1:fib(n-1)+fib(n-2);}; return fib(10); }";
    source = '{ return 1+2; }';
    source = "{ return 0+'x'; }";
    */

    var compile_from_source = make_compile_from_source(parse, bcompile, top_level);
    var bc = compile_from_source(source, true/*as object*/);

    var lua_esc = function(str) {

        var re = /[\uD800-\uDBFF][\uDC00-\uDFFF]|[^A-Za-z0-9_ !#-\/:-@\[\]^`{|}~]/g;
        return '"' + str.replace(re, function(c) {
            var code = c.charCodeAt(0);
            if (c.length===2) {
                var next = c.charCodeAt(1);
                code = ((code - 0xD800) * 0x400) +
                    (next - 0xDC00) + 0x10000;
            }

            var esc = function(d) {
                var s = d.toString(10);
                while (s.length < 3) { s = '0' + s; }
                return "\\" + s;
            };
            if (code < 0x80) {
                return esc(code);
            }
            var prefix, s='', repeat;
            if (code < 0x800) {
                prefix = esc(0xC0 | (code >>> 6));
                repeat = 1;
            } else if (code < 0x10000) {
                prefix = esc(0xE0 | (code >>> 12));
                repeat = 2;
            } else {
                prefix = esc(0xF0 | (code >>> 18));
                repeat = 3;
            }
            while (repeat > 0) {
                s = esc(0x80 | (code & 0x3F)) + s;
                code = code >>> 6;
                repeat--;
            }
            return prefix + s;
        }) + '"';
    };

    var pad = function(s, width) {
        if (width===undefined) { width = 10; }
        while (s.length < width) { s += ' '; }
        return s;
    };

    console.log('-- generated by TurtleScript write-lua-bytecode.js');
    console.log('local jsval = require("luaturtle.jsval")');
    console.log('local ifunc = require("luaturtle.ifunc")');
    console.log('');
    console.log('local startup = {}');
    console.log('');
    console.log('-- Populate the function and literal arrays with the precompiled');
    console.log('-- startup code, including the compiler and standard library.');
    console.log('');
    console.log('startup.functions = {');
    var mkComma = function(len) {
        len--;
        return function(i) { return (i < len) ? ',' : ''; };
    };
    var comma = mkComma(bc.functions.length);
    bc.functions.forEach(function(f, i) {
        var name = f.name ? (' -- '+JSON.stringify(f.name)) : '';
        console.log('  ifunc.Function:new{' + name);
        name = f.name ? ('jsval.newString(' + lua_esc(f.name) + ')') : 'jsval.Undefined';
        console.log('    name = ' + name + ',');
        console.log('    id = ' + f.id + ',');
        console.log('    nargs = ' + f.nargs + ',');
        console.log('    max_stack = ' + f.max_stack + ',');
        console.log('    bytecode = {');
        var j = 0;
        while (j < f.bytecode.length) {
            var pc = j;
            var bc = bytecode_table.for_num(f.bytecode[j]);
            var a = f.bytecode.slice(j, j+=bc.args+1);
            a = a.map(function(b) {
                if (typeof(b) !== 'number') { b = b.label; }
                return ''+b;
            });
            var b = a.slice(1);
            a = a.join(', ') + ((j < f.bytecode.length) ? ',' : '');
            b = bc.name + ( b.length ? ( '(' + b.join(',') + ')' ) : '' );
            console.log('      ' + pad(a) + '-- ' + pc + ': ' + b);
        }
        console.log('    }');
        console.log('  }' + comma(i));
    });
    console.log('}');

    console.log('-- literals');
    console.log('startup.literals = {');
    comma = mkComma(bc.literals.length);
    bc.literals.forEach(function(lv, i) {
        var str;
        if (typeof(lv) === "number" ) {
            str = lv.toString();
            if (isNaN(lv)) { str = '0/0'; }
            else if (!isFinite(lv)) { str = lv > 0 ? '1/0' : '-1/0'; }

            else if (lv === 0 && (1/lv)===-Infinity) { str = '-1/(1/0)'; }
            str = 'jsval.newNumber(' + str + ')';
        } else if (typeof(lv) === "string") {
            str = lua_esc(lv); // Note: UTF8!
            str = 'jsval.newString(' + str + ')'; // convert to UTF16
        } else if (typeof(lv) === "boolean") {
            str = 'jsval.' + (lv ? "True" : "False");
        } else if (lv === null) {
            str = "jsval.Null";
        } else if (lv === undefined) {
            str = "jsval.Undefined";
        } else {
            console.assert(false);
        }
        console.log('  ' + pad(str + comma(i)) + ' -- ' + i);
    });
    console.log('}');
    console.log('');

    console.log('return startup');
});