Method allocateStack

src/ast/ast_allocate_stack.cpp:929–1010

    // program

    void Program::allocateStack(TextWriter & logs, bool permanent, bool everything) {
        // string heap
        AllocateConstString vstr;
        for (auto & pm : library.modules) {
            for ( auto & var : pm->globals.each() ) {
                if (var->used && var->init) {
                    var->init->visit(vstr);
                }
            }
            for ( auto & func : pm->functions.each() ) {
                if (func->used) {
                    func->visit(vstr);
                }
            }
        }
        globalStringHeapSize = vstr.bytesTotal;
        // move some variables to CMRES
        VarCMRes vcm(this, everything);
        visit(vcm);
        // allocate stack for the rest of them
        AllocateStack context(this, permanent, everything, logs);
        visit(context);
        // adjust stack size for all the used variables
        for (auto & pm : library.modules) {
            for ( auto & var : pm->globals.each() ) {
                if ( var->used ) {
                    globalInitStackSize = das::max(globalInitStackSize, var->initStackSize);
                }
            }
        }
        // allocate used variables and functions indices
        totalVariables = 0;
        totalFunctions = 0;
        auto log = options.getBoolOption("log_stack");
        if ( log ) {
            logs << "INIT STACK SIZE:\t" << globalInitStackSize << "\n";
            logs << "FUNCTION TABLE:\n";
        }
        for (auto & pm : library.modules) {
            for ( auto & func : pm->functions.each() ) {
                if ( func->used && !func->builtIn && !func->isTemplate ) {
                    func->index = totalFunctions++;
                    if ( log ) {
                        logs << "\t" << func->index << "\t" << func->totalStackSize << "\t" << func->getMangledName();
                        if ( func->init ) {
                            logs << " [init";
                            if (func->lateInit ) logs << "(late)";
                            logs << "]";
                        }
                        if ( func->shutdown ) {
                            logs << " [finalize";
                            if (func->lateShutdown ) logs << "(late)";
                            logs << "]";
                        }
                        logs << " // " << HEX << func->getMangledNameHash() << DEC;
                        logs << "\n";
                    }
                }