make %constant directive to work with structs/classes

This commit is contained in:
Paweł Tomulik 2014-10-25 22:56:41 +02:00
commit c21e2423a0
2 changed files with 53 additions and 1 deletions

View file

@ -0,0 +1,21 @@
import constant_directive
if type(constant_directive.TYPE1_CONSTANT1) != constant_directive.Type1:
print("TYPE1_CONSTANT1 type: {}".format(type(constant_directive.TYPE1_CONSTANT1)))
raise RuntimeError("fail");
if type(constant_directive.getType1Instance()) != constant_directive.Type1:
print("getType1Instance() type: {}".format(type(constant_directive.getType1Instance())))
raise RuntimeError("fail");
if constant_directive.TYPE1_CONSTANT1.val != 1:
print "constant_directive.TYPE1_CONSTANT1.val != 1"
print "constant_directive.TYPE1_CONSTANT1.val is %r" % constant_directive.TYPE1_CONSTANT1.val
raise RuntimeError("fail")
if constant_directive.TYPE1_CONSTANT2.val != 2:
print "constant_directive.TYPE1_CONSTANT2.val != 2"
print "constant_directive.TYPE1_CONSTANT2.val is %r" % constant_directive.TYPE1_CONSTANT2.val
raise RuntimeError("fail")
if constant_directive.TYPE1_CONSTANT3.val != 3:
print "constant_directive.TYPE1_CONSTANT3.val != 3"
print "constant_directive.TYPE1_CONSTANT3.val is %r" % constant_directive.TYPE1_CONSTANT3.val
raise RuntimeError("fail")

View file

@ -3169,7 +3169,32 @@ public:
Replaceall(tm, "$source", value);
Replaceall(tm, "$target", name);
Replaceall(tm, "$value", value);
Printf(f_init, "%s\n", tm);
if (!builtin && (shadow) && (!(shadow & PYSHADOW_MEMBER)) && (!in_class || !Getattr(n, "feature:python:callback"))) {
// Generate method which registers the new constant
Printf(f_wrappers, "SWIGINTERN PyObject* %s_swigregister(PyObject* SWIGUNUSEDPARM(self), PyObject* args) {\n", iname);
Printf(f_wrappers, tab2 "PyObject *m;\n", tm);
if (modernargs) {
if (fastunpack) {
Printf(f_wrappers, tab2 "if (!SWIG_Python_UnpackTuple(args,(char*)\"swigregister\", 1, 1,&m)) return NULL;\n");
} else {
Printf(f_wrappers, tab2 "if (!PyArg_UnpackTuple(args,(char*)\"swigregister\", 1, 1,&m)) return NULL;\n");
}
} else {
Printf(f_wrappers, tab2 "if (!PyArg_ParseTuple(args,(char*)\"O:swigregister\", &m)) return NULL;\n");
}
Printf(f_wrappers, tab2 "PyObject* d = PyModule_GetDict(m);\n");
Printf(f_wrappers, tab2 "if(!d) return NULL;\n");
Printf(f_wrappers, tab2 "%s\n", tm);
Printf(f_wrappers, tab2 "return SWIG_Py_Void();\n");
Printf(f_wrappers, "}\n\n\n");
// Register the method in SwigMethods array
String *cname = NewStringf("%s_swigregister", iname);
add_method(cname, cname, 0);
Delete(cname);
} else {
Printf(f_init, "%s\n", tm);
}
Delete(tm);
have_tm = 1;
}
@ -3184,9 +3209,15 @@ public:
if (!builtin && (shadow) && (!(shadow & PYSHADOW_MEMBER))) {
if (!in_class) {
Printv(f_shadow, "\n",NIL);
Printv(f_shadow, module, ".", iname, "_swigregister(",module,")\n", NIL);
Printv(f_shadow, iname, "_swigregister = ", module, ".", iname, "_swigregister\n", NIL);
Printv(f_shadow, iname, " = ", module, ".", iname, "\n", NIL);
} else {
if (!(Getattr(n, "feature:python:callback"))) {
Printv(f_shadow_stubs, "\n",NIL);
Printv(f_shadow_stubs, module, ".", iname, "_swigregister(", module, ")\n", NIL);
Printv(f_shadow_stubs, iname, "_swigregister = ", module, ".", iname, "_swigregister\n", NIL);
Printv(f_shadow_stubs, iname, " = ", module, ".", iname, "\n", NIL);
}
}