Mac/Modules/res/ressupport.py

   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
   9 class ResMixIn:
  10
  11         def checkit(self):
  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, FunctionGenerator): pass
  19 class ResMethod(ResMixIn, MethodGenerator): pass
  20
  21 # includestuff etc. are imported from macsupport
  22
  23 includestuff = includestuff + """
  24 #include <Resources.h>
  25 #include <string.h>
  26
  27 #define resNotFound -192 /* Can't include <Errors.h> because of Python's "errors.h" */
  28 """
  29
  30 finalstuff = finalstuff + """
  31
  32 /* Alternative version of ResObj_New, which returns None for null argument */
  33 PyObject *OptResObj_New(itself)
  34         Handle itself;
  35 {
  36         if (itself == NULL) {
  37                 Py_INCREF(Py_None);
  38                 return Py_None;
  39         }
  40         return ResObj_New(itself);
  41 }
  42
  43 OptResObj_Convert(v, p_itself)
  44         PyObject *v;
  45         Handle *p_itself;
  46 {
  47         if ( v == Py_None ) {
  48                 *p_itself = NULL;
  49                 return 1;
  50         }
  51         if (!ResObj_Check(v))
  52         {
  53                 PyErr_SetString(PyExc_TypeError, "Resource required");
  54                 return 0;
  55         }
  56         *p_itself = ((ResourceObject *)v)->ob_itself;
  57         return 1;
  58 }
  59
  60 """
  61
  62 initstuff = initstuff + """
  63 """
  64
  65 module = MacModule('Res', 'Res', includestuff, finalstuff, initstuff)
  66
  67 getattrHookCode = """
  68 if (strcmp(name, "size") == 0)
  69         return PyInt_FromLong(GetHandleSize(self->ob_itself));
  70 if (strcmp(name, "data") == 0) {
  71         PyObject *res;
  72         char state;
  73         state = HGetState(self->ob_itself);
  74         HLock(self->ob_itself);
  75         res = PyString_FromStringAndSize(
  76                 *self->ob_itself,
  77                 GetHandleSize(self->ob_itself));
  78         HUnlock(self->ob_itself);
  79         HSetState(self->ob_itself, state);
  80         return res;
  81 }
  82 if (strcmp(name, "__members__") == 0)
  83         return Py_BuildValue("[ss]", "data", "size");
  84 """
  85
  86 setattrCode = """
  87 static int
  88 ResObj_setattr(self, name, value)
  89         ResourceObject *self;
  90         char *name;
  91         PyObject *value;
  92 {
  93         char *data;
  94         long size;
  95
  96         if (strcmp(name, "data") != 0 || value == NULL )
  97                 return -1;
  98         if ( !PyString_Check(value) )
  99                 return -1;
 100         size = PyString_Size(value);
 101         data = PyString_AsString(value);
 102         /* XXXX Do I need the GetState/SetState calls? */
 103         SetHandleSize(self->ob_itself, size);
 104         if ( MemError())
 105                 return -1;
 106         HLock(self->ob_itself);
 107         memcpy((char *)*self->ob_itself, data, size);
 108         HUnlock(self->ob_itself);
 109         /* XXXX Should I do the Changed call immedeately? */
 110         return 0;
 111 }
 112 """
 113
 114 class ResDefiniton(GlobalObjectDefinition):
 115
 116         def outputCheckNewArg(self):
 117                 Output("if (itself == NULL) return PyMac_Error(resNotFound);")
 118
 119         def outputGetattrHook(self):
 120                 Output(getattrHookCode)
 121
 122         def outputSetattr(self):
 123                 Output(setattrCode)
 124
 125
 126 resobject = ResDefiniton('Resource', 'ResObj', 'Handle')
 127 module.addobject(resobject)
 128
 129 functions = []
 130 resmethods = []
 131
 132 execfile('resgen.py')
 133 execfile('resedit.py')
 134
 135 for f in functions: module.add(f)
 136 for f in resmethods: resobject.add(f)
 137
 138 SetOutputFileName('Resmodule.c')
 139 module.generate()