/Mac/Modules/qd/qdsupport.py

http://unladen-swallow.googlecode.com/ · Python · 377 lines · 304 code · 29 blank · 44 comment · 3 complexity · 7ddf21b08870768f8051e60f1e9dbc36 MD5 · raw file

  1. # This script generates a Python interface for an Apple Macintosh Manager.
  2. # It uses the "bgen" package to generate C code.
  3. # The function specifications are generated by scanning the mamager's header file,
  4. # using the "scantools" package (customized for this particular manager).
  5. import string
  6. # Declarations that change for each manager
  7. MACHEADERFILE = 'QuickDraw.h' # The Apple header file
  8. MODNAME = '_Qd' # The name of the module
  9. OBJECTNAME = 'Graf' # The basic name of the objects used here
  10. # The following is *usually* unchanged but may still require tuning
  11. MODPREFIX = 'Qd' # The prefix for module-wide routines
  12. OBJECTTYPE = OBJECTNAME + 'Ptr' # The C type used to represent them
  13. OBJECTPREFIX = MODPREFIX + 'Obj' # The prefix for object methods
  14. INPUTFILE = string.lower(MODPREFIX) + 'gen.py' # The file generated by the scanner
  15. EXTRAFILE = string.lower(MODPREFIX) + 'edit.py' # A similar file but hand-made
  16. OUTPUTFILE = MODNAME + "module.c" # The file generated by this program
  17. from macsupport import *
  18. # Create the type objects
  19. class TextThingieClass(FixedInputBufferType):
  20. def getargsCheck(self, name):
  21. Output("/* Fool compiler warnings */")
  22. Output("%s__in_len__ = %s__in_len__;", name, name)
  23. def declareSize(self, name):
  24. Output("int %s__in_len__;", name)
  25. TextThingie = TextThingieClass(None)
  26. # These are temporary!
  27. RgnHandle = OpaqueByValueType("RgnHandle", "ResObj")
  28. OptRgnHandle = OpaqueByValueType("RgnHandle", "OptResObj")
  29. PicHandle = OpaqueByValueType("PicHandle", "ResObj")
  30. PolyHandle = OpaqueByValueType("PolyHandle", "ResObj")
  31. PixMapHandle = OpaqueByValueType("PixMapHandle", "ResObj")
  32. PixPatHandle = OpaqueByValueType("PixPatHandle", "ResObj")
  33. PatHandle = OpaqueByValueType("PatHandle", "ResObj")
  34. CursHandle = OpaqueByValueType("CursHandle", "ResObj")
  35. CCrsrHandle = OpaqueByValueType("CCrsrHandle", "ResObj")
  36. CIconHandle = OpaqueByValueType("CIconHandle", "ResObj")
  37. CTabHandle = OpaqueByValueType("CTabHandle", "ResObj")
  38. ITabHandle = OpaqueByValueType("ITabHandle", "ResObj")
  39. GDHandle = OpaqueByValueType("GDHandle", "ResObj")
  40. CGrafPtr = OpaqueByValueType("CGrafPtr", "GrafObj")
  41. GrafPtr = OpaqueByValueType("GrafPtr", "GrafObj")
  42. BitMap_ptr = OpaqueByValueType("BitMapPtr", "BMObj")
  43. const_BitMap_ptr = OpaqueByValueType("const BitMap *", "BMObj")
  44. BitMap = OpaqueType("BitMap", "BMObj_NewCopied", "BUG")
  45. RGBColor = OpaqueType('RGBColor', 'QdRGB')
  46. RGBColor_ptr = RGBColor
  47. FontInfo = OpaqueType('FontInfo', 'QdFI')
  48. Component = OpaqueByValueType('Component', 'CmpObj')
  49. ComponentInstance = OpaqueByValueType('ComponentInstance', 'CmpInstObj')
  50. Cursor = StructOutputBufferType('Cursor')
  51. Cursor_ptr = StructInputBufferType('Cursor')
  52. Pattern = StructOutputBufferType('Pattern')
  53. Pattern_ptr = StructInputBufferType('Pattern')
  54. PenState = StructOutputBufferType('PenState')
  55. PenState_ptr = StructInputBufferType('PenState')
  56. TruncCode = Type("TruncCode", "h")
  57. includestuff = includestuff + """
  58. #include <Carbon/Carbon.h>
  59. #ifdef USE_TOOLBOX_OBJECT_GLUE
  60. extern PyObject *_GrafObj_New(GrafPtr);
  61. extern int _GrafObj_Convert(PyObject *, GrafPtr *);
  62. extern PyObject *_BMObj_New(BitMapPtr);
  63. extern int _BMObj_Convert(PyObject *, BitMapPtr *);
  64. extern PyObject *_QdRGB_New(RGBColorPtr);
  65. extern int _QdRGB_Convert(PyObject *, RGBColorPtr);
  66. #define GrafObj_New _GrafObj_New
  67. #define GrafObj_Convert _GrafObj_Convert
  68. #define BMObj_New _BMObj_New
  69. #define BMObj_Convert _BMObj_Convert
  70. #define QdRGB_New _QdRGB_New
  71. #define QdRGB_Convert _QdRGB_Convert
  72. #endif
  73. static PyObject *BMObj_NewCopied(BitMapPtr);
  74. /*
  75. ** Parse/generate RGB records
  76. */
  77. PyObject *QdRGB_New(RGBColorPtr itself)
  78. {
  79. return Py_BuildValue("lll", (long)itself->red, (long)itself->green, (long)itself->blue);
  80. }
  81. int QdRGB_Convert(PyObject *v, RGBColorPtr p_itself)
  82. {
  83. long red, green, blue;
  84. if( !PyArg_ParseTuple(v, "lll", &red, &green, &blue) )
  85. return 0;
  86. p_itself->red = (unsigned short)red;
  87. p_itself->green = (unsigned short)green;
  88. p_itself->blue = (unsigned short)blue;
  89. return 1;
  90. }
  91. /*
  92. ** Generate FontInfo records
  93. */
  94. static
  95. PyObject *QdFI_New(FontInfo *itself)
  96. {
  97. return Py_BuildValue("hhhh", itself->ascent, itself->descent,
  98. itself->widMax, itself->leading);
  99. }
  100. """
  101. finalstuff = finalstuff + """
  102. /* Like BMObj_New, but the original bitmap data structure is copied (and
  103. ** released when the object is released)
  104. */
  105. PyObject *BMObj_NewCopied(BitMapPtr itself)
  106. {
  107. BitMapObject *it;
  108. BitMapPtr itself_copy;
  109. if ((itself_copy=(BitMapPtr)malloc(sizeof(BitMap))) == NULL)
  110. return PyErr_NoMemory();
  111. *itself_copy = *itself;
  112. it = (BitMapObject *)BMObj_New(itself_copy);
  113. it->referred_bitmap = itself_copy;
  114. return (PyObject *)it;
  115. }
  116. """
  117. variablestuff = ""
  118. initstuff = initstuff + """
  119. PyMac_INIT_TOOLBOX_OBJECT_NEW(BitMapPtr, BMObj_New);
  120. PyMac_INIT_TOOLBOX_OBJECT_CONVERT(BitMapPtr, BMObj_Convert);
  121. PyMac_INIT_TOOLBOX_OBJECT_NEW(GrafPtr, GrafObj_New);
  122. PyMac_INIT_TOOLBOX_OBJECT_CONVERT(GrafPtr, GrafObj_Convert);
  123. PyMac_INIT_TOOLBOX_OBJECT_NEW(RGBColorPtr, QdRGB_New);
  124. PyMac_INIT_TOOLBOX_OBJECT_CONVERT(RGBColor, QdRGB_Convert);
  125. """
  126. ## not yet...
  127. ##
  128. ##class Region_ObjectDefinition(GlobalObjectDefinition):
  129. ## def outputCheckNewArg(self):
  130. ## Output("if (itself == NULL) return PyMac_Error(resNotFound);")
  131. ## def outputFreeIt(self, itselfname):
  132. ## Output("DisposeRegion(%s);", itselfname)
  133. ##
  134. ##class Polygon_ObjectDefinition(GlobalObjectDefinition):
  135. ## def outputCheckNewArg(self):
  136. ## Output("if (itself == NULL) return PyMac_Error(resNotFound);")
  137. ## def outputFreeIt(self, itselfname):
  138. ## Output("KillPoly(%s);", itselfname)
  139. class MyGRObjectDefinition(PEP253Mixin, GlobalObjectDefinition):
  140. getsetlist = [
  141. ('visRgn',
  142. """RgnHandle h=NewRgn(); /* XXXX wrong dispose routine */
  143. return Py_BuildValue("O&", ResObj_New, (Handle)GetPortVisibleRegion(self->ob_itself, h));
  144. """,
  145. None,
  146. "Convenience attribute: return a copy of the visible region"
  147. ), (
  148. 'clipRgn',
  149. """RgnHandle h=NewRgn(); /* XXXX wrong dispose routine */
  150. return Py_BuildValue("O&", ResObj_New, (Handle)GetPortClipRegion(self->ob_itself, h));
  151. """,
  152. None,
  153. "Convenience attribute: return a copy of the clipping region"
  154. )]
  155. def outputCheckNewArg(self):
  156. Output("if (itself == NULL) return PyMac_Error(resNotFound);")
  157. def outputCheckConvertArg(self):
  158. Output("#if 1")
  159. OutLbrace()
  160. Output("WindowRef win;")
  161. OutLbrace("if (WinObj_Convert(v, &win) && v)")
  162. Output("*p_itself = (GrafPtr)GetWindowPort(win);")
  163. Output("return 1;")
  164. OutRbrace()
  165. Output("PyErr_Clear();")
  166. OutRbrace()
  167. Output("#else")
  168. OutLbrace("if (DlgObj_Check(v))")
  169. Output("DialogRef dlg = (DialogRef)((GrafPortObject *)v)->ob_itself;")
  170. Output("*p_itself = (GrafPtr)GetWindowPort(GetDialogWindow(dlg));")
  171. Output("return 1;")
  172. OutRbrace()
  173. OutLbrace("if (WinObj_Check(v))")
  174. Output("WindowRef win = (WindowRef)((GrafPortObject *)v)->ob_itself;")
  175. Output("*p_itself = (GrafPtr)GetWindowPort(win);")
  176. Output("return 1;")
  177. OutRbrace()
  178. Output("#endif")
  179. class MyBMObjectDefinition(PEP253Mixin, GlobalObjectDefinition):
  180. getsetlist = [
  181. (
  182. 'baseAddr',
  183. 'return PyInt_FromLong((long)self->ob_itself->baseAddr);',
  184. None,
  185. None
  186. ), (
  187. 'rowBytes',
  188. 'return PyInt_FromLong((long)self->ob_itself->rowBytes);',
  189. None,
  190. None
  191. ), (
  192. 'bounds',
  193. 'return Py_BuildValue("O&", PyMac_BuildRect, &self->ob_itself->bounds);',
  194. None,
  195. None
  196. ), (
  197. 'bitmap_data',
  198. 'return PyString_FromStringAndSize((char *)self->ob_itself, sizeof(BitMap));',
  199. None,
  200. None
  201. ), (
  202. 'pixmap_data',
  203. 'return PyString_FromStringAndSize((char *)self->ob_itself, sizeof(PixMap));',
  204. None,
  205. None
  206. )]
  207. def outputCheckNewArg(self):
  208. Output("if (itself == NULL) return PyMac_Error(resNotFound);")
  209. def outputStructMembers(self):
  210. # We need to more items: a pointer to privately allocated data
  211. # and a python object we're referring to.
  212. Output("%s ob_itself;", self.itselftype)
  213. Output("PyObject *referred_object;")
  214. Output("BitMap *referred_bitmap;")
  215. def outputInitStructMembers(self):
  216. Output("it->ob_itself = %sitself;", self.argref)
  217. Output("it->referred_object = NULL;")
  218. Output("it->referred_bitmap = NULL;")
  219. def outputCleanupStructMembers(self):
  220. Output("Py_XDECREF(self->referred_object);")
  221. Output("if (self->referred_bitmap) free(self->referred_bitmap);")
  222. # Create the generator groups and link them
  223. module = MacModule(MODNAME, MODPREFIX, includestuff, finalstuff, initstuff, variablestuff)
  224. ##r_object = Region_ObjectDefinition('Region', 'QdRgn', 'RgnHandle')
  225. ##po_object = Polygon_ObjectDefinition('Polygon', 'QdPgn', 'PolyHandle')
  226. ##module.addobject(r_object)
  227. ##module.addobject(po_object)
  228. gr_object = MyGRObjectDefinition("GrafPort", "GrafObj", "GrafPtr")
  229. module.addobject(gr_object)
  230. bm_object = MyBMObjectDefinition("BitMap", "BMObj", "BitMapPtr")
  231. module.addobject(bm_object)
  232. # Create the generator classes used to populate the lists
  233. Function = OSErrWeakLinkFunctionGenerator
  234. Method = OSErrWeakLinkMethodGenerator
  235. # Create and populate the lists
  236. functions = []
  237. gr_methods = []
  238. bm_methods = []
  239. #methods = []
  240. execfile(INPUTFILE)
  241. execfile(EXTRAFILE)
  242. # add the populated lists to the generator groups
  243. # (in a different wordl the scan program would generate this)
  244. for f in functions: module.add(f)
  245. for f in gr_methods: gr_object.add(f)
  246. for f in bm_methods: bm_object.add(f)
  247. # Manual generator: get data out of a bitmap
  248. getdata_body = """
  249. int from, length;
  250. char *cp;
  251. if ( !PyArg_ParseTuple(_args, "ii", &from, &length) )
  252. return NULL;
  253. cp = _self->ob_itself->baseAddr+from;
  254. _res = PyString_FromStringAndSize(cp, length);
  255. return _res;
  256. """
  257. f = ManualGenerator("getdata", getdata_body)
  258. f.docstring = lambda: """(int start, int size) -> string. Return bytes from the bitmap"""
  259. bm_object.add(f)
  260. # Manual generator: store data in a bitmap
  261. putdata_body = """
  262. int from, length;
  263. char *cp, *icp;
  264. if ( !PyArg_ParseTuple(_args, "is#", &from, &icp, &length) )
  265. return NULL;
  266. cp = _self->ob_itself->baseAddr+from;
  267. memcpy(cp, icp, length);
  268. Py_INCREF(Py_None);
  269. _res = Py_None;
  270. return _res;
  271. """
  272. f = ManualGenerator("putdata", putdata_body)
  273. f.docstring = lambda: """(int start, string data). Store bytes into the bitmap"""
  274. bm_object.add(f)
  275. #
  276. # We manually generate a routine to create a BitMap from python data.
  277. #
  278. BitMap_body = """
  279. BitMap *ptr;
  280. PyObject *source;
  281. Rect bounds;
  282. int rowbytes;
  283. char *data;
  284. if ( !PyArg_ParseTuple(_args, "O!iO&", &PyString_Type, &source, &rowbytes, PyMac_GetRect,
  285. &bounds) )
  286. return NULL;
  287. data = PyString_AsString(source);
  288. if ((ptr=(BitMap *)malloc(sizeof(BitMap))) == NULL )
  289. return PyErr_NoMemory();
  290. ptr->baseAddr = (Ptr)data;
  291. ptr->rowBytes = rowbytes;
  292. ptr->bounds = bounds;
  293. if ( (_res = BMObj_New(ptr)) == NULL ) {
  294. free(ptr);
  295. return NULL;
  296. }
  297. ((BitMapObject *)_res)->referred_object = source;
  298. Py_INCREF(source);
  299. ((BitMapObject *)_res)->referred_bitmap = ptr;
  300. return _res;
  301. """
  302. f = ManualGenerator("BitMap", BitMap_body)
  303. f.docstring = lambda: """Take (string, int, Rect) argument and create BitMap"""
  304. module.add(f)
  305. #
  306. # And again, for turning a correctly-formatted structure into the object
  307. #
  308. RawBitMap_body = """
  309. BitMap *ptr;
  310. PyObject *source;
  311. if ( !PyArg_ParseTuple(_args, "O!", &PyString_Type, &source) )
  312. return NULL;
  313. if ( PyString_Size(source) != sizeof(BitMap) && PyString_Size(source) != sizeof(PixMap) ) {
  314. PyErr_Format(PyExc_TypeError,
  315. "Argument size was %d, should be %d (sizeof BitMap) or %d (sizeof PixMap)",
  316. PyString_Size(source), sizeof(BitMap), sizeof(PixMap));
  317. return NULL;
  318. }
  319. ptr = (BitMapPtr)PyString_AsString(source);
  320. if ( (_res = BMObj_New(ptr)) == NULL ) {
  321. return NULL;
  322. }
  323. ((BitMapObject *)_res)->referred_object = source;
  324. Py_INCREF(source);
  325. return _res;
  326. """
  327. f = ManualGenerator("RawBitMap", RawBitMap_body)
  328. f.docstring = lambda: """Take string BitMap and turn into BitMap object"""
  329. module.add(f)
  330. # generate output (open the output file as late as possible)
  331. SetOutputFileName(OUTPUTFILE)
  332. module.generate()
  333. SetOutputFile() # Close it