95
93
for entry in entries:
97
95
"static PyObject *%s;" % entry.pystring_cname)
100
class ModuleNode(Node, BlockNode):
104
def analyse_declarations(self, env):
106
self.body.analyse_declarations(env)
108
def process_implementation(self, env, result):
109
self.analyse_declarations(env)
110
env.check_c_classes()
111
self.body.analyse_expressions(env)
112
env.return_type = PyrexTypes.c_void_type
113
self.generate_c_code(env, result)
114
self.generate_h_code(env, result)
116
def generate_h_code(self, env, result):
117
public_vars_and_funcs = []
118
public_extension_types = []
119
for entry in env.var_entries:
120
if entry.visibility == 'public':
121
public_vars_and_funcs.append(entry)
122
for entry in env.cfunc_entries:
123
if entry.visibility == 'public':
124
public_vars_and_funcs.append(entry)
125
for entry in env.c_class_entries:
126
if entry.visibility == 'public':
127
public_extension_types.append(entry)
128
if public_vars_and_funcs or public_extension_types:
129
result.h_file = replace_suffix(result.c_file, ".h")
130
result.i_file = replace_suffix(result.c_file, ".pxi")
131
h_code = Code.CCodeWriter(result.h_file)
132
i_code = Code.PyrexCodeWriter(result.i_file)
133
self.generate_extern_c_macro_definition(h_code)
134
for entry in public_vars_and_funcs:
135
h_code.putln("%s %s;" % (
136
Naming.extern_c_macro,
137
entry.type.declaration_code(
138
entry.cname, dll_linkage = "DL_IMPORT")))
139
i_code.putln("cdef extern %s" %
140
entry.type.declaration_code(entry.cname, pyrex = 1))
141
for entry in public_extension_types:
142
self.generate_cclass_header_code(entry.type, h_code)
143
self.generate_cclass_include_code(entry.type, i_code)
144
h_code.putln("PyMODINIT_FUNC init%s(void);" % env.module_name)
146
def generate_cclass_header_code(self, type, h_code):
147
#h_code.putln("extern DL_IMPORT(PyTypeObject) %s;" % type.typeobj_cname)
148
h_code.putln("%s DL_IMPORT(PyTypeObject) %s;" % (
149
Naming.extern_c_macro,
151
self.generate_obj_struct_definition(type, h_code)
153
def generate_cclass_include_code(self, type, i_code):
154
i_code.putln("cdef extern class %s.%s:" % (
155
type.module_name, type.name))
157
var_entries = type.scope.var_entries
159
for entry in var_entries:
160
i_code.putln("cdef %s" %
161
entry.type.declaration_code(entry.cname, pyrex = 1))
166
def generate_c_code(self, env, result):
168
self.find_referenced_modules(env, modules, {})
169
code = Code.CCodeWriter(result.c_file)
171
self.generate_module_preamble(env, modules, code)
172
for module in modules:
173
self.generate_declarations_for_module(module, code,
174
definition = module is env)
176
code.putln("/* Implementation of %s */" % env.qualified_name)
177
self.generate_const_definitions(env, code)
178
self.generate_interned_name_decls(env, code)
179
self.generate_py_string_decls(env, code)
180
self.body.generate_function_definitions(env, code)
181
self.generate_interned_name_table(env, code)
182
self.generate_py_string_table(env, code)
183
self.generate_typeobj_definitions(env, code)
184
self.generate_method_table(env, code)
185
self.generate_filename_init_prototype(code)
186
self.generate_module_init_func(modules[:-1], env, code)
187
self.generate_filename_table(code)
188
self.generate_utility_functions(env, code)
189
result.c_file_generated = 1
191
def find_referenced_modules(self, env, module_list, modules_seen):
192
if env not in modules_seen:
193
modules_seen[env] = 1
194
for imported_module in env.cimported_modules:
195
self.find_referenced_modules(imported_module, module_list, modules_seen)
196
module_list.append(env)
198
def generate_module_preamble(self, env, cimported_modules, code):
199
code.putln('/* Generated by Pyrex %s on %s */' % (
200
Version.version, time.asctime()))
202
code.putln('#define PY_SSIZE_T_CLEAN')
203
for filename in env.python_include_files:
204
code.putln('#include "%s"' % filename)
205
code.putln("#ifndef PY_LONG_LONG")
206
code.putln(" #define PY_LONG_LONG LONG_LONG")
208
code.putln("#if PY_VERSION_HEX < 0x02050000")
209
code.putln(" typedef int Py_ssize_t;")
210
code.putln(" #define PY_SSIZE_T_MAX INT_MAX")
211
code.putln(" #define PY_SSIZE_T_MIN INT_MIN")
212
code.putln(" #define PyInt_FromSsize_t(z) PyInt_FromLong(z)")
213
code.putln(" #define PyInt_AsSsize_t(o) PyInt_AsLong(o)")
215
self.generate_extern_c_macro_definition(code)
216
code.putln("%s double pow(double, double);" % Naming.extern_c_macro)
217
self.generate_includes(env, cimported_modules, code)
218
#for filename in env.include_files:
219
# code.putln('#include "%s"' % filename)
221
code.put(utility_function_predeclarations)
222
if Options.intern_names:
223
code.putln(get_name_interned_predeclaration)
225
code.putln(get_name_predeclaration)
227
code.putln('static PyObject *%s;' % env.module_cname)
228
code.putln('static PyObject *%s;' % Naming.builtins_cname)
229
code.putln('static int %s;' % Naming.lineno_cname)
230
code.putln('static char *%s;' % Naming.filename_cname)
231
code.putln('static char **%s;' % Naming.filetable_cname)
234
code.putln('static char %s[] = "%s";' % (env.doc_cname, env.doc))
236
def generate_extern_c_macro_definition(self, code):
237
name = Naming.extern_c_macro
238
code.putln("#ifdef __cplusplus")
239
code.putln('#define %s extern "C"' % name)
241
code.putln("#define %s extern" % name)
244
def generate_includes(self, env, cimported_modules, code):
245
includes = env.include_files[:]
246
for module in cimported_modules:
247
for filename in module.include_files:
248
if filename not in includes:
249
includes.append(filename)
250
for filename in includes:
251
code.putln('#include "%s"' % filename)
253
def generate_filename_table(self, code):
255
code.putln("static char *%s[] = {" % Naming.filenames_cname)
256
if code.filename_list:
257
for filename in code.filename_list:
258
filename = os.path.basename(filename)
259
escaped_filename = filename.replace("\\", "\\\\").replace('"', r'\"')
263
# Some C compilers don't like an empty array
267
def generate_declarations_for_module(self, env, code, definition):
269
code.putln("/* Declarations from %s */" % env.qualified_name)
270
self.generate_type_predeclarations(env, code)
271
self.generate_type_definitions(env, code)
272
self.generate_global_declarations(env, code, definition)
273
self.generate_cfunction_predeclarations(env, code)
275
def generate_type_predeclarations(self, env, code):
278
def generate_type_definitions(self, env, code):
279
# Generate definitions of structs/unions/enums.
280
for entry in env.sue_entries:
281
if not entry.in_cinclude:
283
if type.is_struct_or_union:
284
self.generate_struct_union_definition(entry, code)
286
self.generate_enum_definition(entry, code)
287
# Generate extension type object struct definitions.
288
for entry in env.c_class_entries:
289
if not entry.in_cinclude:
290
self.generate_typeobject_predeclaration(entry, code)
291
self.generate_obj_struct_definition(entry.type, code)
292
self.generate_exttype_vtable_struct(entry, code)
293
self.generate_exttype_vtabptr_declaration(entry, code)
295
def sue_header_footer(self, type, kind, name):
296
if type.typedef_flag:
297
header = "typedef %s {" % kind
298
footer = "} %s;" % name
300
header = "%s %s {" % (kind, name)
302
return header, footer
304
def generate_struct_union_definition(self, entry, code):
309
self.sue_header_footer(type, type.kind, type.cname)
312
var_entries = scope.var_entries
315
"Empty struct or union definition not allowed outside a"
316
" 'cdef extern from' block")
317
for attr in var_entries:
320
attr.type.declaration_code(attr.cname))
323
def generate_enum_definition(self, entry, code):
325
name = entry.cname or entry.name or ""
327
self.sue_header_footer(type, "enum", name)
330
enum_values = entry.enum_values
333
"Empty enum definition not allowed outside a"
334
" 'cdef extern from' block")
335
for value_entry in enum_values:
336
if value_entry.value == value_entry.name:
347
def generate_typeobject_predeclaration(self, entry, code):
349
name = entry.type.typeobj_cname
351
if entry.visibility == 'extern' and not entry.in_cinclude:
352
code.putln("%s DL_IMPORT(PyTypeObject) %s;" % (
353
Naming.extern_c_macro,
355
elif entry.visibility == 'public':
356
#code.putln("DL_EXPORT(PyTypeObject) %s;" % name)
357
code.putln("%s DL_EXPORT(PyTypeObject) %s;" % (
358
Naming.extern_c_macro,
360
# ??? Do we really need the rest of this? ???
362
# code.putln("staticforward PyTypeObject %s;" % name)
364
def generate_exttype_vtable_struct(self, entry, code):
365
# Generate struct declaration for an extension type's vtable.
368
if type.vtabstruct_cname:
372
type.vtabstruct_cname)
373
if type.base_type and type.base_type.vtabstruct_cname:
374
code.putln("struct %s %s;" % (
375
type.base_type.vtabstruct_cname,
376
Naming.obj_base_cname))
377
for method_entry in scope.cfunc_entries:
378
if not method_entry.is_inherited:
380
"%s;" % method_entry.type.declaration_code("(*%s)" % method_entry.name))
384
def generate_exttype_vtabptr_declaration(self, entry, code):
385
# Generate declaration of pointer to an extension type's vtable.
387
if type.vtabptr_cname:
388
code.putln("static struct %s *%s;" % (
389
type.vtabstruct_cname,
392
def generate_obj_struct_definition(self, type, code):
393
# Generate object struct definition for an
396
return # Forward declared but never defined
398
self.sue_header_footer(type, "struct", type.objstruct_cname)
401
base_type = type.base_type
405
("struct ", "")[base_type.typedef_flag],
406
base_type.objstruct_cname,
407
Naming.obj_base_cname))
411
if type.vtabslot_cname and not (type.base_type and type.base_type.vtabslot_cname):
414
type.vtabstruct_cname,
415
type.vtabslot_cname))
416
for attr in type.scope.var_entries:
419
attr.type.declaration_code(attr.cname))
422
def generate_global_declarations(self, env, code, definition):
424
for entry in env.c_class_entries:
425
code.putln("static PyTypeObject *%s = 0;" %
426
entry.type.typeptr_cname)
427
code.put_var_declarations(env.var_entries, static = 1,
428
dll_linkage = "DL_EXPORT", definition = definition)
429
code.put_var_declarations(env.default_entries, static = 1)
431
def generate_cfunction_predeclarations(self, env, code):
432
for entry in env.cfunc_entries:
433
if not entry.in_cinclude:
434
if entry.visibility == 'public':
435
dll_linkage = "DL_EXPORT"
438
header = entry.type.declaration_code(entry.cname,
439
dll_linkage = dll_linkage)
440
if entry.visibility <> 'private':
441
storage_class = "%s " % Naming.extern_c_macro
443
storage_class = "static "
444
code.putln("%s%s; /*proto*/" % (
448
def generate_typeobj_definitions(self, env, code):
449
full_module_name = env.qualified_name
450
for entry in env.c_class_entries:
451
#print "generate_typeobj_definitions:", entry.name
452
#print "...visibility =", entry.visibility
453
if entry.visibility <> 'extern':
456
if scope: # could be None if there was an error
457
self.generate_exttype_vtable(scope, code)
458
self.generate_new_function(scope, code)
459
self.generate_dealloc_function(scope, code)
460
self.generate_traverse_function(scope, code)
461
self.generate_clear_function(scope, code)
462
if scope.defines_any(["__getitem__"]):
463
self.generate_getitem_int_function(scope, code)
464
if scope.defines_any(["__setitem__", "__delitem__"]):
465
self.generate_ass_subscript_function(scope, code)
466
if scope.defines_any(["__setslice__", "__delslice__"]):
467
self.generate_ass_slice_function(scope, code)
468
if scope.defines_any(["__getattr__"]):
469
self.generate_getattro_function(scope, code)
470
if scope.defines_any(["__setattr__", "__delattr__"]):
471
self.generate_setattro_function(scope, code)
472
if scope.defines_any(["__get__"]):
473
self.generate_descr_get_function(scope, code)
474
if scope.defines_any(["__set__", "__delete__"]):
475
self.generate_descr_set_function(scope, code)
476
self.generate_property_accessors(scope, code)
477
self.generate_method_table(scope, code)
478
self.generate_member_table(scope, code)
479
self.generate_getset_table(scope, code)
480
self.generate_typeobj_definition(full_module_name, entry, code)
482
def generate_exttype_vtable(self, scope, code):
483
# Generate the definition of an extension type's vtable.
484
type = scope.parent_type
485
if type.vtable_cname:
486
code.putln("static struct %s %s;" % (
487
type.vtabstruct_cname,
490
def generate_self_cast(self, scope, code):
491
type = scope.parent_type
494
type.declaration_code("p"),
495
type.declaration_code("")))
497
def generate_new_function(self, scope, code):
498
base_type = scope.parent_type.base_type
501
"static PyObject *%s(PyTypeObject *t, PyObject *a, PyObject *k) {"
502
% scope.mangle_internal("tp_new"))
505
"PyObject *o = %s->tp_new(t, a, k);" %
506
base_type.typeptr_cname)
509
"PyObject *o = (*t->tp_alloc)(t, 0);")
510
self.generate_self_cast(scope, code)
511
type = scope.parent_type
512
if type.vtabslot_cname:
513
code.putln("*(struct %s **)&p->%s = %s;" % (
514
type.vtabstruct_cname,
517
for entry in scope.var_entries:
518
if entry.type.is_pyobject:
519
code.put_init_var_to_py_none(entry, "p->%s")
520
entry = scope.lookup_here("__new__")
523
"if (%s(o, a, k) < 0) {" %
525
code.put_decref_clear("o", py_object_type);
533
def generate_dealloc_function(self, scope, code):
534
base_type = scope.parent_type.base_type
537
"static void %s(PyObject *o) {"
538
% scope.mangle_internal("tp_dealloc"))
539
self.generate_self_cast(scope, code)
540
self.generate_usr_dealloc_call(scope, code)
541
for entry in scope.var_entries:
542
if entry.type.is_pyobject:
543
code.put_xdecref("p->%s" % entry.cname, entry.type)
546
"%s->tp_dealloc(o);" %
547
base_type.typeptr_cname)
550
"(*o->ob_type->tp_free)(o);")
554
def generate_usr_dealloc_call(self, scope, code):
555
entry = scope.lookup_here("__dealloc__")
560
"PyObject *etype, *eval, *etb;")
562
"PyErr_Fetch(&etype, &eval, &etb);")
569
"if (PyErr_Occurred()) PyErr_WriteUnraisable(o);")
573
"PyErr_Restore(etype, eval, etb);")
577
def generate_traverse_function(self, scope, code):
578
base_type = scope.parent_type.base_type
581
"static int %s(PyObject *o, visitproc v, void *a) {"
582
% scope.mangle_internal("tp_traverse"))
585
self.generate_self_cast(scope, code)
588
"e = %s->tp_traverse(o, v, a); if (e) return e;" %
589
base_type.typeptr_cname)
590
for entry in scope.var_entries:
591
if entry.type.is_pyobject:
592
var_code = "p->%s" % entry.cname
596
if entry.type.is_extension_type:
597
var_code = "((PyObject*)%s)" % var_code
599
"e = (*v)(%s, a); if (e) return e;"
608
def generate_clear_function(self, scope, code):
609
base_type = scope.parent_type.base_type
612
"static int %s(PyObject *o) {"
613
% scope.mangle_internal("tp_clear"))
614
self.generate_self_cast(scope, code)
618
base_type.typeptr_cname)
619
for entry in scope.var_entries:
620
if entry.type.is_pyobject:
621
name = "p->%s" % entry.cname
622
code.put_xdecref(name, entry.type)
623
#code.put_init_to_py_none(name)
624
code.put_init_var_to_py_none(entry, "p->%s")
630
def generate_getitem_int_function(self, scope, code):
631
# This function is put into the sq_item slot when
632
# a __getitem__ method is present. It converts its
633
# argument to a Python integer and calls mp_subscript.
635
"static PyObject *%s(PyObject *o, Py_ssize_t i) {" %
636
scope.mangle_internal("sq_item"))
640
"PyObject *x = PyInt_FromSsize_t(i); if(!x) return 0;")
642
"r = o->ob_type->tp_as_mapping->mp_subscript(o, x);")
650
def generate_ass_subscript_function(self, scope, code):
651
# Setting and deleting an item are both done through
652
# the ass_subscript method, so we dispatch to user's __setitem__
653
# or __delitem__, or raise an exception.
654
base_type = scope.parent_type.base_type
655
set_entry = scope.lookup_here("__setitem__")
656
del_entry = scope.lookup_here("__delitem__")
659
"static int %s(PyObject *o, PyObject *i, PyObject *v) {" %
660
scope.mangle_internal("mp_ass_subscript"))
665
"return %s(o, i, v);" %
666
set_entry.func_cname)
668
self.generate_guarded_basetype_call(
669
base_type, "tp_as_mapping", "mp_ass_subscript", "o, i, v", code)
671
"PyErr_Format(PyExc_NotImplementedError,")
673
' "Subscript assignment not supported by %s", o->ob_type->tp_name);')
683
del_entry.func_cname)
685
self.generate_guarded_basetype_call(
686
base_type, "tp_as_mapping", "mp_ass_subscript", "o, i, v", code)
688
"PyErr_Format(PyExc_NotImplementedError,")
690
' "Subscript deletion not supported by %s", o->ob_type->tp_name);')
698
def generate_guarded_basetype_call(
699
self, base_type, substructure, slot, args, code):
701
base_tpname = base_type.typeptr_cname
704
"if (%s->%s && %s->%s->%s)" % (
705
base_tpname, substructure, base_tpname, substructure, slot))
707
" return %s->%s->%s(%s);" % (
708
base_tpname, substructure, slot, args))
714
" return %s->%s(%s);" % (
715
base_tpname, slot, args))
717
def generate_ass_slice_function(self, scope, code):
718
# Setting and deleting a slice are both done through
719
# the ass_slice method, so we dispatch to user's __setslice__
720
# or __delslice__, or raise an exception.
721
base_type = scope.parent_type.base_type
722
set_entry = scope.lookup_here("__setslice__")
723
del_entry = scope.lookup_here("__delslice__")
726
"static int %s(PyObject *o, Py_ssize_t i, Py_ssize_t j, PyObject *v) {" %
727
scope.mangle_internal("sq_ass_slice"))
732
"return %s(o, i, j, v);" %
733
set_entry.func_cname)
735
self.generate_guarded_basetype_call(
736
base_type, "tp_as_sequence", "sq_ass_slice", "o, i, j, v", code)
738
"PyErr_Format(PyExc_NotImplementedError,")
740
' "2-element slice assignment not supported by %s", o->ob_type->tp_name);')
749
"return %s(o, i, j);" %
750
del_entry.func_cname)
752
self.generate_guarded_basetype_call(
753
base_type, "tp_as_sequence", "sq_ass_slice", "o, i, j, v", code)
755
"PyErr_Format(PyExc_NotImplementedError,")
757
' "2-element slice deletion not supported by %s", o->ob_type->tp_name);')
765
def generate_getattro_function(self, scope, code):
766
# First try to get the attribute using PyObject_GenericGetAttr.
767
# If that raises an AttributeError, call the user's __getattr__
769
entry = scope.lookup_here("__getattr__")
772
"static PyObject *%s(PyObject *o, PyObject *n) {"
773
% scope.mangle_internal("tp_getattro"))
775
"PyObject *v = PyObject_GenericGetAttr(o, n);")
777
"if (!v && PyErr_ExceptionMatches(PyExc_AttributeError)) {")
790
def generate_setattro_function(self, scope, code):
791
# Setting and deleting an attribute are both done through
792
# the setattro method, so we dispatch to user's __setattr__
793
# or __delattr__ or fall back on PyObject_GenericSetAttr.
794
base_type = scope.parent_type.base_type
795
set_entry = scope.lookup_here("__setattr__")
796
del_entry = scope.lookup_here("__delattr__")
799
"static int %s(PyObject *o, PyObject *n, PyObject *v) {" %
800
scope.mangle_internal("tp_setattro"))
805
"return %s(o, n, v);" %
806
set_entry.func_cname)
808
self.generate_guarded_basetype_call(
809
base_type, None, "tp_setattro", "o, n, v", code)
811
"return PyObject_GenericSetAttr(o, n, v);")
819
del_entry.func_cname)
821
self.generate_guarded_basetype_call(
822
base_type, None, "tp_setattro", "o, n, v", code)
824
"return PyObject_GenericSetAttr(o, n, 0);")
830
def generate_descr_get_function(self, scope, code):
831
# The __get__ function of a descriptor object can be
832
# called with NULL for the second or third arguments
833
# under some circumstances, so we replace them with
835
user_get_entry = scope.lookup_here("__get__")
838
"static PyObject *%s(PyObject *o, PyObject *i, PyObject *c) {" %
839
scope.mangle_internal("tp_descr_get"))
843
"if (!i) i = Py_None;")
845
"if (!c) c = Py_None;")
846
#code.put_incref("i", py_object_type)
847
#code.put_incref("c", py_object_type)
850
user_get_entry.func_cname)
851
#code.put_decref("i", py_object_type)
852
#code.put_decref("c", py_object_type)
858
def generate_descr_set_function(self, scope, code):
859
# Setting and deleting are both done through the __set__
860
# method of a descriptor, so we dispatch to user's __set__
861
# or __delete__ or raise an exception.
862
base_type = scope.parent_type.base_type
863
user_set_entry = scope.lookup_here("__set__")
864
user_del_entry = scope.lookup_here("__delete__")
867
"static int %s(PyObject *o, PyObject *i, PyObject *v) {" %
868
scope.mangle_internal("tp_descr_set"))
873
"return %s(o, i, v);" %
874
user_set_entry.func_cname)
876
self.generate_guarded_basetype_call(
877
base_type, None, "tp_descr_set", "o, i, v", code)
879
'PyErr_SetString(PyExc_NotImplementedError, "__set__");')
889
user_del_entry.func_cname)
891
self.generate_guarded_basetype_call(
892
base_type, None, "tp_descr_set", "o, i, v", code)
894
'PyErr_SetString(PyExc_NotImplementedError, "__delete__");')
902
def generate_property_accessors(self, cclass_scope, code):
903
for entry in cclass_scope.property_entries:
904
property_scope = entry.scope
905
if property_scope.defines_any(["__get__"]):
906
self.generate_property_get_function(entry, code)
907
if property_scope.defines_any(["__set__", "__del__"]):
908
self.generate_property_set_function(entry, code)
910
def generate_property_get_function(self, property_entry, code):
911
property_scope = property_entry.scope
912
property_entry.getter_cname = property_scope.parent_scope.mangle(
913
Naming.prop_get_prefix, property_entry.name)
914
get_entry = property_scope.lookup_here("__get__")
917
"static PyObject *%s(PyObject *o, void *x) {" %
918
property_entry.getter_cname)
921
get_entry.func_cname)
925
def generate_property_set_function(self, property_entry, code):
926
property_scope = property_entry.scope
927
property_entry.setter_cname = property_scope.parent_scope.mangle(
928
Naming.prop_set_prefix, property_entry.name)
929
set_entry = property_scope.lookup_here("__set__")
930
del_entry = property_scope.lookup_here("__del__")
933
"static int %s(PyObject *o, PyObject *v, void *x) {" %
934
property_entry.setter_cname)
940
set_entry.func_cname)
943
'PyErr_SetString(PyExc_NotImplementedError, "__set__");')
953
del_entry.func_cname)
956
'PyErr_SetString(PyExc_NotImplementedError, "__del__");')
964
def generate_typeobj_definition(self, modname, entry, code):
967
for suite in TypeSlots.substructures:
968
suite.generate_substructure(scope, code)
970
if entry.visibility == 'public':
971
header = "DL_EXPORT(PyTypeObject) %s = {"
973
#header = "statichere PyTypeObject %s = {"
974
header = "PyTypeObject %s = {"
975
#code.putln(header % scope.parent_type.typeobj_cname)
976
code.putln(header % type.typeobj_cname)
978
"PyObject_HEAD_INIT(0)")
982
'"%s.%s", /*tp_name*/' % (
983
modname, scope.class_name))
984
if type.typedef_flag:
985
objstruct = type.objstruct_cname
987
#objstruct = "struct %s" % scope.parent_type.objstruct_cname
988
objstruct = "struct %s" % type.objstruct_cname
990
"sizeof(%s), /*tp_basicsize*/" %
993
"0, /*tp_itemsize*/")
994
for slot in TypeSlots.slot_table:
995
slot.generate(scope, code)
999
def generate_method_table(self, env, code):
1002
"static struct PyMethodDef %s[] = {" %
1003
env.method_table_cname)
1004
for entry in env.pyfunc_entries:
1005
code.put_pymethoddef(entry, ",")
1011
def generate_member_table(self, env, code):
1012
#print "ModuleNode.generate_member_table: scope =", env ###
1013
if env.public_attr_entries:
1016
"static struct PyMemberDef %s[] = {" %
1017
env.member_table_cname)
1018
type = env.parent_type
1019
if type.typedef_flag:
1020
objstruct = type.objstruct_cname
1022
objstruct = "struct %s" % type.objstruct_cname
1023
for entry in env.public_attr_entries:
1024
type_code = entry.type.pymemberdef_typecode
1025
if entry.visibility == 'readonly':
1029
code.putln('{"%s", %s, %s, %s, 0},' % (
1032
"offsetof(%s, %s)" % (objstruct, entry.name),
1039
def generate_getset_table(self, env, code):
1040
if env.property_entries:
1043
"static struct PyGetSetDef %s[] = {" %
1044
env.getset_table_cname)
1045
for entry in env.property_entries:
1047
'{"%s", %s, %s, %s, 0},' % (
1049
entry.getter_cname or "0",
1050
entry.setter_cname or "0",
1051
entry.doc_cname or "0"))
1057
def generate_interned_name_table(self, env, code):
1058
items = env.intern_map.items()
1063
"static __Pyx_InternTabEntry %s[] = {" %
1064
Naming.intern_tab_cname)
1065
for (name, cname) in items:
1075
def generate_py_string_table(self, env, code):
1076
entries = env.all_pystring_entries
1080
"static __Pyx_StringTabEntry %s[] = {" %
1081
Naming.stringtab_cname)
1082
for entry in entries:
1084
"{&%s, %s, sizeof(%s)}," % (
1085
entry.pystring_cname,
1093
def generate_filename_init_prototype(self, code):
1095
code.putln("static void %s(void); /*proto*/" % Naming.fileinit_cname)
1097
def generate_module_init_func(self, imported_modules, env, code):
1099
header = "PyMODINIT_FUNC init%s(void)" % env.module_name
1100
code.putln("%s; /*proto*/" % header)
1101
code.putln("%s {" % header)
1102
code.put_var_declarations(env.temp_entries)
1103
#code.putln("/*--- Libary function declarations ---*/")
1104
env.generate_library_function_declarations(code)
1105
self.generate_filename_init_call(code)
1106
#code.putln("/*--- Module creation code ---*/")
1107
self.generate_module_creation_code(env, code)
1108
#code.putln("/*--- Intern code ---*/")
1109
self.generate_intern_code(env, code)
1110
#code.putln("/*--- String init code ---*/")
1111
self.generate_string_init_code(env, code)
1112
#code.putln("/*--- Global init code ---*/")
1113
self.generate_global_init_code(env, code)
1114
#code.putln("/*--- Type import code ---*/")
1115
for module in imported_modules:
1116
self.generate_type_import_code_for_module(module, env, code)
1117
#code.putln("/*--- Type init code ---*/")
1118
self.generate_type_init_code(env, code)
1119
#code.putln("/*--- Execution code ---*/")
1120
self.body.generate_execution_code(code)
1121
code.putln("return;")
1122
code.put_label(code.error_label)
1123
code.put_var_xdecrefs(env.temp_entries)
1124
code.putln('__Pyx_AddTraceback("%s");' % (env.qualified_name))
1125
env.use_utility_code(traceback_utility_code)
1128
def generate_filename_init_call(self, code):
1129
code.putln("%s();" % Naming.fileinit_cname)
1131
def generate_module_creation_code(self, env, code):
1132
# Generate code to create the module object and
1133
# install the builtins.
1139
'%s = Py_InitModule4("%s", %s, %s, 0, PYTHON_API_VERSION);' % (
1142
env.method_table_cname,
1147
code.error_goto(self.pos)));
1149
'%s = PyImport_AddModule("__builtin__");' %
1150
Naming.builtins_cname)
1153
Naming.builtins_cname,
1154
code.error_goto(self.pos)));
1156
'if (PyObject_SetAttrString(%s, "__builtins__", %s) < 0) %s;' % (
1158
Naming.builtins_cname,
1159
code.error_goto(self.pos)))
1161
def generate_intern_code(self, env, code):
1163
env.use_utility_code(init_intern_tab_utility_code);
1165
"if (__Pyx_InternStrings(%s) < 0) %s;" % (
1166
Naming.intern_tab_cname,
1167
code.error_goto(self.pos)))
1169
def generate_string_init_code(self, env, code):
1170
if env.all_pystring_entries:
1171
env.use_utility_code(init_string_tab_utility_code)
1173
"if (__Pyx_InitStrings(%s) < 0) %s;" % (
1174
Naming.stringtab_cname,
1175
code.error_goto(self.pos)))
1177
def generate_global_init_code(self, env, code):
1178
# Generate code to initialise global PyObject *
1179
# variables to None.
1180
for entry in env.var_entries:
1181
if entry.visibility <> 'extern':
1182
if entry.type.is_pyobject:
1183
code.put_init_var_to_py_none(entry)
1185
def generate_type_import_code_for_module(self, module, env, code):
1186
# Generate type import code for all extension types in
1187
# an imported module.
1188
if module.c_class_entries:
1189
for entry in module.c_class_entries:
1190
self.generate_type_import_code(env, entry, code)
1192
def generate_type_init_code(self, env, code):
1193
# Generate type import code for extern extension types
1194
# and type ready code for non-extern ones.
1195
for entry in env.c_class_entries:
1196
if entry.visibility == 'extern':
1197
self.generate_type_import_code(env, entry, code)
1199
self.generate_exttype_vtable_init_code(entry, code)
1200
self.generate_type_ready_code(env, entry, code)
1201
self.generate_typeptr_assignment_code(entry, code)
1203
def use_type_import_utility_code(self, env):
1205
env.use_utility_code(type_import_utility_code)
1206
env.use_utility_code(ExprNodes.import_utility_code)
1208
def generate_type_import_code(self, env, entry, code):
1209
# Generate code to import the typeobject of an
1210
# extension type defined in another module, and
1211
# extract its C method table pointer if any.
1213
if type.typedef_flag:
1214
objstruct = type.objstruct_cname
1216
objstruct = "struct %s" % type.objstruct_cname
1217
code.putln('%s = __Pyx_ImportType("%s", "%s", sizeof(%s)); if (!%s) %s' % (
1223
code.error_goto(entry.pos)))
1224
self.use_type_import_utility_code(env)
1225
if type.vtabptr_cname:
1227
"if (__Pyx_GetVtable(%s->tp_dict, &%s) < 0) %s" % (
1230
code.error_goto(entry.pos)))
1231
env.use_utility_code(get_vtable_utility_code)
1233
def generate_type_ready_code(self, env, entry, code):
1234
# Generate a call to PyType_Ready for an extension
1235
# type defined in this module.
1237
typeobj_cname = type.typeobj_cname
1239
if scope: # could be None if there was an error
1240
if entry.visibility <> 'extern':
1241
for slot in TypeSlots.slot_table:
1242
slot.generate_dynamic_init_code(scope, code)
1244
"if (PyType_Ready(&%s) < 0) %s" % (
1246
code.error_goto(entry.pos)))
1247
if type.vtable_cname:
1249
"if (__Pyx_SetVtable(%s.tp_dict, %s) < 0) %s" % (
1252
code.error_goto(entry.pos)))
1253
env.use_utility_code(set_vtable_utility_code)
1255
'if (PyObject_SetAttrString(%s, "%s", (PyObject *)&%s) < 0) %s' % (
1256
Naming.module_cname,
1259
code.error_goto(entry.pos)))
1260
weakref_entry = scope.lookup_here("__weakref__")
1262
if weakref_entry.type is py_object_type:
1263
tp_weaklistoffset = "%s.tp_weaklistoffset" % typeobj_cname
1264
code.putln("if (%s == 0) %s = offsetof(struct %s, %s);" % (
1267
type.objstruct_cname,
1268
weakref_entry.cname))
1270
error(weakref_entry.pos, "__weakref__ slot must be of type 'object'")
1272
def generate_exttype_vtable_init_code(self, entry, code):
1273
# Generate code to initialise the C method table of an
1276
if type.vtable_cname:
1281
if type.base_type and type.base_type.vtabptr_cname:
1285
Naming.obj_base_cname,
1286
type.base_type.vtabptr_cname))
1287
for meth_entry in type.scope.cfunc_entries:
1288
if meth_entry.func_cname:
1290
"*(void **)&%s.%s = (void *)%s;" % (
1293
meth_entry.func_cname))
1295
def generate_typeptr_assignment_code(self, entry, code):
1296
# Generate code to initialise the typeptr of an extension
1297
# type defined in this module to point to its type object.
1299
if type.typeobj_cname:
1302
type.typeptr_cname, type.typeobj_cname))
1304
def generate_utility_functions(self, env, code):
1306
code.putln("/* Runtime support code */")
1308
code.putln("static void %s(void) {" % Naming.fileinit_cname)
1309
code.putln("%s = %s;" %
1310
(Naming.filetable_cname, Naming.filenames_cname))
1312
for utility_code in env.utility_code_used:
1313
code.put(utility_code)
1316
98
class StatListNode(Node):