Tools/freeze/makefreeze.py

   1 import marshal
   2 import string
   3
   4
   5 # Write a file containing frozen code for the modules in the dictionary.
   6
   7 header = """
   8 #include "Python.h"
   9
  10 static struct _frozen _PyImport_FrozenModules[] = {
  11 """
  12 trailer = """\
  13     {0, 0, 0} /* sentinel */
  14 };
  15 """
  16
  17 # if __debug__ == 0 (i.e. -O option given), set Py_OptimizeFlag in frozen app.
  18 default_entry_point = """
  19 int
  20 main(argc, argv)
  21     int argc;
  22     char **argv;
  23 {
  24 """ + ((not __debug__ and """
  25         Py_OptimizeFlag++;
  26 """) or "")  + """
  27         PyImport_FrozenModules = _PyImport_FrozenModules;
  28         return Py_FrozenMain(argc, argv);
  29 }
  30
  31 """
  32
  33 def makefreeze(outfp, dict, debug=0, entry_point = None):
  34     if entry_point is None: entry_point = default_entry_point
  35     done = []
  36     mods = dict.keys()
  37     mods.sort()
  38     for mod in mods:
  39         m = dict[mod]
  40         mangled = string.join(string.split(mod, "."), "__")
  41         if m.__code__:
  42             if debug:
  43                 print "freezing", mod, "..."
  44             str = marshal.dumps(m.__code__)
  45             size = len(str)
  46             if m.__path__:
  47                 # Indicate package by negative size
  48                 size = -size
  49             done.append((mod, mangled, size))
  50             writecode(outfp, mangled, str)
  51     if debug:
  52         print "generating table of frozen modules"
  53     outfp.write(header)
  54     for mod, mangled, size in done:
  55         outfp.write('\t{"%s", M_%s, %d},\n' % (mod, mangled, size))
  56     outfp.write(trailer)
  57     outfp.write(entry_point)
  58
  59
  60
  61 # Write a C initializer for a module containing the frozen python code.
  62 # The array is called M_<mod>.
  63
  64 def writecode(outfp, mod, str):
  65     outfp.write('static unsigned char M_%s[] = {' % mod)
  66     for i in range(0, len(str), 16):
  67         outfp.write('\n\t')
  68         for c in str[i:i+16]:
  69             outfp.write('%d,' % ord(c))
  70     outfp.write('\n};\n')