1 # This script will generate the Resources interface for Python.

     2 # It uses the "bgen" package to generate C code.

     3 # It execs the file resgen.py which contain the function definitions

     4 # (resgen.py was generated by resscan.py, scanning the <Resources.h> header file).

     5 

     6 from macsupport import *

     7 

     8 class ResMixIn:

     9 

    10     def checkit(self):

    11         if self.returntype.__class__ != OSErrType:

    12             OutLbrace()

    13             Output("OSErr _err = ResError();")

    14             Output("if (_err != noErr) return PyMac_Error(_err);")

    15             OutRbrace()

    16         FunctionGenerator.checkit(self) # XXX

    17 

    18 class ResFunction(ResMixIn, OSErrWeakLinkFunctionGenerator): pass

    19 class ResMethod(ResMixIn, OSErrWeakLinkMethodGenerator): pass

    20 

    21 RsrcChainLocation = Type("RsrcChainLocation", "h")

    22 FSCatalogInfoBitmap = FakeType("0") # Type("FSCatalogInfoBitmap", "l")

    23 FSCatalogInfo_ptr = FakeType("(FSCatalogInfo *)0")

    24 

    25 # includestuff etc. are imported from macsupport

    26 

    27 includestuff = includestuff + """

    28 #include <Carbon/Carbon.h>

    29 

    30 #ifdef USE_TOOLBOX_OBJECT_GLUE

    31 extern PyObject *_ResObj_New(Handle);

    32 extern int _ResObj_Convert(PyObject *, Handle *);

    33 extern PyObject *_OptResObj_New(Handle);

    34 extern int _OptResObj_Convert(PyObject *, Handle *);

    35 #define ResObj_New _ResObj_New

    36 #define ResObj_Convert _ResObj_Convert

    37 #define OptResObj_New _OptResObj_New

    38 #define OptResObj_Convert _OptResObj_Convert

    39 #endif

    40 

    41 /* Function to dispose a resource, with a "normal" calling sequence */

    42 static void

    43 PyMac_AutoDisposeHandle(Handle h)

    44 {

    45         DisposeHandle(h);

    46 }

    47 """

    48 

    49 finalstuff = finalstuff + """

    50 

    51 /* Alternative version of ResObj_New, which returns None for null argument */

    52 PyObject *OptResObj_New(Handle itself)

    53 {

    54         if (itself == NULL) {

    55                 Py_INCREF(Py_None);

    56                 return Py_None;

    57         }

    58         return ResObj_New(itself);

    59 }

    60 

    61 int OptResObj_Convert(PyObject *v, Handle *p_itself)

    62 {

    63         PyObject *tmp;

    64 

    65         if ( v == Py_None ) {

    66                 *p_itself = NULL;

    67                 return 1;

    68         }

    69         if (ResObj_Check(v))

    70         {

    71                 *p_itself = ((ResourceObject *)v)->ob_itself;

    72                 return 1;

    73         }

    74         /* If it isn't a resource yet see whether it is convertible */

    75         if ( (tmp=PyObject_CallMethod(v, "as_Resource", "")) ) {

    76                 *p_itself = ((ResourceObject *)tmp)->ob_itself;

    77                 Py_DECREF(tmp);

    78                 return 1;

    79         }

    80         PyErr_Clear();

    81         PyErr_SetString(PyExc_TypeError, "Resource required");

    82         return 0;

    83 }

    84 """

    85 

    86 initstuff = initstuff + """

    87         PyMac_INIT_TOOLBOX_OBJECT_NEW(Handle, ResObj_New);

    88         PyMac_INIT_TOOLBOX_OBJECT_CONVERT(Handle, ResObj_Convert);

    89         PyMac_INIT_TOOLBOX_OBJECT_NEW(Handle, OptResObj_New);

    90         PyMac_INIT_TOOLBOX_OBJECT_CONVERT(Handle, OptResObj_Convert);

    91 """

    92 

    93 module = MacModule('_Res', 'Res', includestuff, finalstuff, initstuff)

    94 

    95 class ResDefinition(PEP253Mixin, GlobalObjectDefinition):

    96     getsetlist = [

    97             ('data',

    98             """

    99             PyObject *res;

   100             char state;

   101 

   102             state = HGetState(self->ob_itself);

   103             HLock(self->ob_itself);

   104             res = PyString_FromStringAndSize(

   105                     *self->ob_itself,

   106                     GetHandleSize(self->ob_itself));

   107             HUnlock(self->ob_itself);

   108             HSetState(self->ob_itself, state);

   109             return res;

   110             """,

   111             """

   112             char *data;

   113             long size;

   114 

   115             if ( v == NULL )

   116                     return -1;

   117             if ( !PyString_Check(v) )

   118                     return -1;

   119             size = PyString_Size(v);

   120             data = PyString_AsString(v);

   121             /* XXXX Do I need the GetState/SetState calls? */

   122             SetHandleSize(self->ob_itself, size);

   123             if ( MemError())

   124                     return -1;

   125             HLock(self->ob_itself);

   126             memcpy((char *)*self->ob_itself, data, size);

   127             HUnlock(self->ob_itself);

   128             /* XXXX Should I do the Changed call immedeately? */

   129             return 0;

   130             """,

   131             'The resource data'

   132             ), (

   133             'size',

   134             'return PyInt_FromLong(GetHandleSize(self->ob_itself));',

   135             None,

   136             'The length of the resource data'

   137             )]

   138 

   139     def outputCheckNewArg(self):

   140         Output("if (itself == NULL) return PyMac_Error(resNotFound);")

   141 

   142     def outputCheckConvertArg(self):

   143         # if it isn't a resource we may be able to coerce it

   144         Output("if (!%s_Check(v))", self.prefix)

   145         OutLbrace()

   146         Output("PyObject *tmp;")

   147         Output('if ( (tmp=PyObject_CallMethod(v, "as_Resource", "")) )')

   148         OutLbrace()

   149         Output("*p_itself = ((ResourceObject *)tmp)->ob_itself;")

   150         Output("Py_DECREF(tmp);")

   151         Output("return 1;")

   152         OutRbrace()

   153         Output("PyErr_Clear();")

   154         OutRbrace()

   155 

   156     def outputStructMembers(self):

   157         GlobalObjectDefinition.outputStructMembers(self)

   158         Output("void (*ob_freeit)(%s ptr);", self.itselftype)

   159 

   160     def outputInitStructMembers(self):

   161         GlobalObjectDefinition.outputInitStructMembers(self)

   162         Output("it->ob_freeit = NULL;")

   163 

   164     def outputCleanupStructMembers(self):

   165         Output("if (self->ob_freeit && self->ob_itself)")

   166         OutLbrace()

   167         Output("self->ob_freeit(self->ob_itself);")

   168         OutRbrace()

   169         Output("self->ob_itself = NULL;")

   170 

   171     def output_tp_newBody(self):

   172         Output("PyObject *self;")

   173         Output

   174         Output("if ((self = type->tp_alloc(type, 0)) == NULL) return NULL;")

   175         Output("((%s *)self)->ob_itself = NULL;", self.objecttype)

   176         Output("((%s *)self)->ob_freeit = NULL;", self.objecttype)

   177         Output("return self;")

   178 

   179     def output_tp_initBody(self):

   180         Output("char *srcdata = NULL;")

   181         Output("int srclen = 0;")

   182         Output("%s itself;", self.itselftype);

   183         Output("char *kw[] = {\"itself\", 0};")

   184         Output()

   185         Output("if (PyArg_ParseTupleAndKeywords(_args, _kwds, \"O&\", kw, %s_Convert, &itself))",

   186                 self.prefix);

   187         OutLbrace()

   188         Output("((%s *)_self)->ob_itself = itself;", self.objecttype)

   189         Output("return 0;")

   190         OutRbrace()

   191         Output("PyErr_Clear();")

   192         Output("if (!PyArg_ParseTupleAndKeywords(_args, _kwds, \"|s#\", kw, &srcdata, &srclen)) return -1;")

   193         Output("if ((itself = NewHandle(srclen)) == NULL)")

   194         OutLbrace()

   195         Output("PyErr_NoMemory();")

   196         Output("return 0;")

   197         OutRbrace()

   198         Output("((%s *)_self)->ob_itself = itself;", self.objecttype)

   199 # XXXX          Output("((%s *)self)->ob_freeit = PyMac_AutoDisposeHandle;")

   200         Output("if (srclen && srcdata)")

   201         OutLbrace()

   202         Output("HLock(itself);")

   203         Output("memcpy(*itself, srcdata, srclen);")

   204         Output("HUnlock(itself);")

   205         OutRbrace()

   206         Output("return 0;")

   207 

   208 resobject = ResDefinition('Resource', 'ResObj', 'Handle')

   209 module.addobject(resobject)

   210 

   211 functions = []

   212 resmethods = []

   213 

   214 execfile('resgen.py')

   215 execfile('resedit.py')

   216 

   217 for f in functions: module.add(f)

   218 for f in resmethods: resobject.add(f)

   219 

   220 SetOutputFileName('_Resmodule.c')

   221 module.generate()