OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 """ |
| 3 Create a C file for embedding one or more Cython source files. |
| 4 Requires Cython 0.11.2 (or perhaps newer). |
| 5 |
| 6 See Demos/freeze/README.txt for more details. |
| 7 """ |
| 8 |
| 9 import optparse |
| 10 from os.path import splitext, basename |
| 11 |
| 12 usage= '%prog [-o outfile] [-p] module [module ...]' |
| 13 description = 'Create a C file for embedding Cython modules.' |
| 14 p = optparse.OptionParser(usage=usage, description=description) |
| 15 p.add_option('-o', '--output', metavar='FILE', |
| 16 help='write output to FILE instead of standard output') |
| 17 p.add_option('-p', '--pymain', action='store_true', default=False, |
| 18 help='do not automatically run the first module as __main__') |
| 19 |
| 20 options, args = p.parse_args() |
| 21 |
| 22 if len(args) < 1: |
| 23 p.print_help() |
| 24 p.exit(1) |
| 25 |
| 26 if options.output: |
| 27 import sys |
| 28 old_stdout = sys.stdout |
| 29 sys.stdout = open(options.output, 'w') |
| 30 |
| 31 modules = [basename(splitext(x)[0]).replace('.', '_') for x in args] |
| 32 |
| 33 print """\ |
| 34 #include <Python.h> |
| 35 #include <locale.h> |
| 36 #include <stdio.h> |
| 37 #include <stdlib.h> |
| 38 |
| 39 #ifdef __FreeBSD__ |
| 40 #include <floatingpoint.h> |
| 41 #endif |
| 42 |
| 43 #if PY_MAJOR_VERSION < 3 |
| 44 # define MODINIT(name) init ## name |
| 45 #else |
| 46 # define MODINIT(name) PyInit_ ## name |
| 47 #endif |
| 48 """ |
| 49 |
| 50 for name in modules: |
| 51 print "PyMODINIT_FUNC MODINIT(%s) (void);" % name |
| 52 |
| 53 print """ |
| 54 static struct _inittab inittab[] = {""" |
| 55 |
| 56 for name in modules: |
| 57 print ' {"%(name)s", MODINIT(%(name)s)},' % {'name' : name} |
| 58 |
| 59 print """ {NULL, NULL} |
| 60 }; |
| 61 """, |
| 62 |
| 63 if not options.pymain: |
| 64 print "\nextern int __pyx_module_is_main_%s;" % modules[0] |
| 65 |
| 66 print """ |
| 67 #if PY_MAJOR_VERSION < 3 |
| 68 int main(int argc, char** argv) { |
| 69 #elif defined(WIN32) || defined(MS_WINDOWS) |
| 70 int wmain(int argc, wchar_t **argv) { |
| 71 #else |
| 72 static int python_main(int argc, wchar_t **argv) { |
| 73 #endif |
| 74 """, |
| 75 if not options.pymain: |
| 76 print """\ |
| 77 PyObject *m = NULL; |
| 78 int r = 0; |
| 79 """, |
| 80 print """\ |
| 81 /* 754 requires that FP exceptions run in "no stop" mode by default, |
| 82 * and until C vendors implement C99's ways to control FP exceptions, |
| 83 * Python requires non-stop mode. Alas, some platforms enable FP |
| 84 * exceptions by default. Here we disable them. |
| 85 */ |
| 86 #ifdef __FreeBSD__ |
| 87 fp_except_t m; |
| 88 |
| 89 m = fpgetmask(); |
| 90 fpsetmask(m & ~FP_X_OFL); |
| 91 #endif |
| 92 if (PyImport_ExtendInittab(inittab)) { |
| 93 fprintf(stderr, "No memory\\n"); |
| 94 exit(1); |
| 95 } |
| 96 """, |
| 97 if options.pymain: |
| 98 print """\ |
| 99 return Py_Main(argc, argv); |
| 100 } |
| 101 """ |
| 102 else: |
| 103 print """\ |
| 104 Py_SetProgramName(argv[0]); |
| 105 Py_Initialize(); |
| 106 PySys_SetArgv(argc, argv); |
| 107 __pyx_module_is_main_%(main)s = 1; |
| 108 m = PyImport_ImportModule(inittab[0].name); |
| 109 if (!m) { |
| 110 r = 1; |
| 111 PyErr_Print(); /* This exits with the right code if SystemExit. */ |
| 112 #if PY_MAJOR_VERSION < 3 |
| 113 if (Py_FlushLine()) |
| 114 PyErr_Clear(); |
| 115 #endif |
| 116 } |
| 117 Py_XDECREF(m); |
| 118 Py_Finalize(); |
| 119 return r; |
| 120 } |
| 121 """ % {'main' : modules[0]}, |
| 122 |
| 123 print r""" |
| 124 #if PY_MAJOR_VERSION >= 3 && !defined(WIN32) && !defined(MS_WINDOWS) |
| 125 static wchar_t* |
| 126 char2wchar(char* arg) |
| 127 { |
| 128 wchar_t *res; |
| 129 #ifdef HAVE_BROKEN_MBSTOWCS |
| 130 /* Some platforms have a broken implementation of |
| 131 * mbstowcs which does not count the characters that |
| 132 * would result from conversion. Use an upper bound. |
| 133 */ |
| 134 size_t argsize = strlen(arg); |
| 135 #else |
| 136 size_t argsize = mbstowcs(NULL, arg, 0); |
| 137 #endif |
| 138 size_t count; |
| 139 unsigned char *in; |
| 140 wchar_t *out; |
| 141 #ifdef HAVE_MBRTOWC |
| 142 mbstate_t mbs; |
| 143 #endif |
| 144 if (argsize != (size_t)-1) { |
| 145 res = (wchar_t *)malloc((argsize+1)*sizeof(wchar_t)); |
| 146 if (!res) |
| 147 goto oom; |
| 148 count = mbstowcs(res, arg, argsize+1); |
| 149 if (count != (size_t)-1) { |
| 150 wchar_t *tmp; |
| 151 /* Only use the result if it contains no |
| 152 surrogate characters. */ |
| 153 for (tmp = res; *tmp != 0 && |
| 154 (*tmp < 0xd800 || *tmp > 0xdfff); tmp++) |
| 155 ; |
| 156 if (*tmp == 0) |
| 157 return res; |
| 158 } |
| 159 free(res); |
| 160 } |
| 161 /* Conversion failed. Fall back to escaping with surrogateescape. */ |
| 162 #ifdef HAVE_MBRTOWC |
| 163 /* Try conversion with mbrtwoc (C99), and escape non-decodable bytes. */ |
| 164 |
| 165 /* Overallocate; as multi-byte characters are in the argument, the |
| 166 actual output could use less memory. */ |
| 167 argsize = strlen(arg) + 1; |
| 168 res = malloc(argsize*sizeof(wchar_t)); |
| 169 if (!res) goto oom; |
| 170 in = (unsigned char*)arg; |
| 171 out = res; |
| 172 memset(&mbs, 0, sizeof mbs); |
| 173 while (argsize) { |
| 174 size_t converted = mbrtowc(out, (char*)in, argsize, &mbs); |
| 175 if (converted == 0) |
| 176 /* Reached end of string; null char stored. */ |
| 177 break; |
| 178 if (converted == (size_t)-2) { |
| 179 /* Incomplete character. This should never happen, |
| 180 since we provide everything that we have - |
| 181 unless there is a bug in the C library, or I |
| 182 misunderstood how mbrtowc works. */ |
| 183 fprintf(stderr, "unexpected mbrtowc result -2\n"); |
| 184 return NULL; |
| 185 } |
| 186 if (converted == (size_t)-1) { |
| 187 /* Conversion error. Escape as UTF-8b, and start over |
| 188 in the initial shift state. */ |
| 189 *out++ = 0xdc00 + *in++; |
| 190 argsize--; |
| 191 memset(&mbs, 0, sizeof mbs); |
| 192 continue; |
| 193 } |
| 194 if (*out >= 0xd800 && *out <= 0xdfff) { |
| 195 /* Surrogate character. Escape the original |
| 196 byte sequence with surrogateescape. */ |
| 197 argsize -= converted; |
| 198 while (converted--) |
| 199 *out++ = 0xdc00 + *in++; |
| 200 continue; |
| 201 } |
| 202 /* successfully converted some bytes */ |
| 203 in += converted; |
| 204 argsize -= converted; |
| 205 out++; |
| 206 } |
| 207 #else |
| 208 /* Cannot use C locale for escaping; manually escape as if charset |
| 209 is ASCII (i.e. escape all bytes > 128. This will still roundtrip |
| 210 correctly in the locale's charset, which must be an ASCII superset. *
/ |
| 211 res = malloc((strlen(arg)+1)*sizeof(wchar_t)); |
| 212 if (!res) goto oom; |
| 213 in = (unsigned char*)arg; |
| 214 out = res; |
| 215 while(*in) |
| 216 if(*in < 128) |
| 217 *out++ = *in++; |
| 218 else |
| 219 *out++ = 0xdc00 + *in++; |
| 220 *out = 0; |
| 221 #endif |
| 222 return res; |
| 223 oom: |
| 224 fprintf(stderr, "out of memory\n"); |
| 225 return NULL; |
| 226 } |
| 227 |
| 228 int |
| 229 main(int argc, char **argv) |
| 230 { |
| 231 wchar_t **argv_copy = (wchar_t **)malloc(sizeof(wchar_t*)*argc); |
| 232 /* We need a second copies, as Python might modify the first one. */ |
| 233 wchar_t **argv_copy2 = (wchar_t **)malloc(sizeof(wchar_t*)*argc); |
| 234 int i, res; |
| 235 char *oldloc; |
| 236 if (!argv_copy || !argv_copy2) { |
| 237 fprintf(stderr, "out of memory\n"); |
| 238 return 1; |
| 239 } |
| 240 oldloc = strdup(setlocale(LC_ALL, NULL)); |
| 241 setlocale(LC_ALL, ""); |
| 242 for (i = 0; i < argc; i++) { |
| 243 argv_copy2[i] = argv_copy[i] = char2wchar(argv[i]); |
| 244 if (!argv_copy[i]) |
| 245 return 1; |
| 246 } |
| 247 setlocale(LC_ALL, oldloc); |
| 248 free(oldloc); |
| 249 res = python_main(argc, argv_copy); |
| 250 for (i = 0; i < argc; i++) { |
| 251 free(argv_copy2[i]); |
| 252 } |
| 253 free(argv_copy); |
| 254 free(argv_copy2); |
| 255 return res; |
| 256 } |
| 257 #endif""" |
OLD | NEW |