import marshal

import bkfile

# Write a file containing frozen code for the modules in the dictionary.

header = """

#include "Python.h"

static struct _frozen _PyImport_FrozenModules[] = {

"""

trailer = """\

    {0, 0, 0} /* sentinel */

};

"""

# if __debug__ == 0 (i.e. -O option given), set Py_OptimizeFlag in frozen app.

default_entry_point = """

int

main(int argc, char **argv)

{

        extern int Py_FrozenMain(int, char **);

""" + ((not __debug__ and """

        Py_OptimizeFlag++;

""") or "")  + """

        PyImport_FrozenModules = _PyImport_FrozenModules;

        return Py_FrozenMain(argc, argv);

}

"""

def makefreeze(base, dict, debug=0, entry_point=None, fail_import=()):

    if entry_point is None: entry_point = default_entry_point

    done = []

    files = []

    mods = dict.keys()

    mods.sort()

    for mod in mods:

        m = dict[mod]

        mangled = "__".join(mod.split("."))

        if m.__code__:

            file = 'M_' + mangled + '.c'

            outfp = bkfile.open(base + file, 'w')

            files.append(file)

            if debug:

                print "freezing", mod, "..."

            str = marshal.dumps(m.__code__)

            size = len(str)

            if m.__path__:

                # Indicate package by negative size

                size = -size

            done.append((mod, mangled, size))

            writecode(outfp, mangled, str)

            outfp.close()

    if debug:

        print "generating table of frozen modules"

    outfp = bkfile.open(base + 'frozen.c', 'w')

    for mod, mangled, size in done:

        outfp.write('extern unsigned char M_%s[];\n' % mangled)

    outfp.write(header)

    for mod, mangled, size in done:

        outfp.write('\t{"%s", M_%s, %d},\n' % (mod, mangled, size))

    outfp.write('\n')

    # The following modules have a NULL code pointer, indicating

    # that the prozen program should not search for them on the host

    # system. Importing them will *always* raise an ImportError.

    # The zero value size is never used.

    for mod in fail_import:

        outfp.write('\t{"%s", NULL, 0},\n' % (mod,))

    outfp.write(trailer)

    outfp.write(entry_point)

    outfp.close()

    return files

# Write a C initializer for a module containing the frozen python code.

# The array is called M_<mod>.

def writecode(outfp, mod, str):

    outfp.write('unsigned char M_%s[] = {' % mod)

    for i in range(0, len(str), 16):

        outfp.write('\n\t')

        for c in str[i:i+16]:

            outfp.write('%d,' % ord(c))

    outfp.write('\n};\n')

## def writecode(outfp, mod, str):

##     outfp.write('unsigned char M_%s[%d] = "%s";\n' % (mod, len(str),

##     '\\"'.join(map(lambda s: repr(s)[1:-1], str.split('"')))))