python/src/Mac/Modules/list/listsupport.py - toolchain/benchmark - Git at Google

 # This script generates a Python interface for an Apple Macintosh Manager.
 # It uses the "bgen" package to generate C code.
 # The function specifications are generated by scanning the mamager's header file,
 # using the "scantools" package (customized for this particular manager).

 import string

 # Declarations that change for each manager
 MACHEADERFILE = 'Lists.h'               # The Apple header file
 MODNAME = '_List'                               # The name of the module
 OBJECTNAME = 'List'                     # The basic name of the objects used here
 KIND = 'Handle'                         # Usually 'Ptr' or 'Handle'

 # The following is *usually* unchanged but may still require tuning
 MODPREFIX = 'List'                      # The prefix for module-wide routines
 OBJECTTYPE = "ListHandle"               # The C type used to represent them
 OBJECTPREFIX = MODPREFIX + 'Obj'        # The prefix for object methods
 INPUTFILE = string.lower(MODPREFIX) + 'gen.py' # The file generated by the scanner
 OUTPUTFILE = MODNAME + "module.c"       # The file generated by this program

 from macsupport import *

 # Create the type objects
 ListHandle = OpaqueByValueType("ListHandle", "ListObj")
 ListRef = ListHandle # Obsolete, but used in Lists.h
 Cell = Point
 ListBounds = Rect
 ListBounds_ptr = Rect_ptr

 ListDefSpec = ListDefSpec_ptr = OpaqueType("ListDefSpec", "PyMac_BuildListDefSpec", "PyMac_GetListDefSpec")

 VarOutBufferShortsize = VarHeapOutputBufferType('char', 'short', 's')   # (buf, &len)
 InBufferShortsize = VarInputBufferType('char', 'short', 's')            # (buf, len)

 RgnHandle = OpaqueByValueType("RgnHandle", "ResObj")
 DataHandle = OpaqueByValueType("DataHandle", "ResObj")
 Handle = OpaqueByValueType("Handle", "ResObj")
 CGrafPtr = OpaqueByValueType("CGrafPtr", "GrafObj")
 EventModifiers = Type("EventModifiers", "H")

 includestuff = includestuff + """
 #include <Carbon/Carbon.h>

 #ifdef USE_TOOLBOX_OBJECT_GLUE
 extern PyObject *_ListObj_New(ListHandle);
 extern int _ListObj_Convert(PyObject *, ListHandle *);

 #define ListObj_New _ListObj_New
 #define ListObj_Convert _ListObj_Convert
 #endif

 #define as_List(x) ((ListHandle)x)
 #define as_Resource(lh) ((Handle)lh)

 static ListDefUPP myListDefFunctionUPP;

 """

 initstuff = initstuff + """
 myListDefFunctionUPP = NewListDefUPP((ListDefProcPtr)myListDefFunction);

 PyMac_INIT_TOOLBOX_OBJECT_NEW(ListHandle, ListObj_New);
 PyMac_INIT_TOOLBOX_OBJECT_CONVERT(ListHandle, ListObj_Convert);
 """

 class ListMethodGenerator(MethodGenerator):
     """Similar to MethodGenerator, but has self as last argument"""

     def parseArgumentList(self, args):
         args, a0 = args[:-1], args[-1]
         t0, n0, m0 = a0
         if m0 != InMode:
             raise ValueError, "method's 'self' must be 'InMode'"
         self.itself = Variable(t0, "_self->ob_itself", SelfMode)
         FunctionGenerator.parseArgumentList(self, args)
         self.argumentList.append(self.itself)

 class MyObjectDefinition(PEP253Mixin, GlobalObjectDefinition):
     # XXXX Should inherit from Resource
     getsetlist = [(
             'listFlags',
             'return Py_BuildValue("l", (long)GetListFlags(self->ob_itself) & 0xff);',
             'if (!PyArg_Parse(v, "B", &(*self->ob_itself)->listFlags)) return -1;',
             None,
             ), (
             'selFlags',
             'return Py_BuildValue("l", (long)GetListSelectionFlags(self->ob_itself) & 0xff);',
             'if (!PyArg_Parse(v, "B", &(*self->ob_itself)->selFlags)) return -1;',
             None,
             ), (
             'cellSize',
             'return Py_BuildValue("O&", PyMac_BuildPoint, (*self->ob_itself)->cellSize);',
             'if (!PyArg_Parse(v, "O&", PyMac_GetPoint, &(*self->ob_itself)->cellSize)) return -1;',
             None
             )]

     def outputStructMembers(self):
         ObjectDefinition.outputStructMembers(self)
         Output("PyObject *ob_ldef_func;")
         Output("int ob_must_be_disposed;")

     def outputCheckNewArg(self):
         Output("""if (itself == NULL) {
                                 PyErr_SetString(List_Error,"Cannot create null List");
                                 return NULL;
                         }""")

     def outputInitStructMembers(self):
         ObjectDefinition.outputInitStructMembers(self)
         Output("it->ob_ldef_func = NULL;")
         Output("it->ob_must_be_disposed = 1;")
         Output("SetListRefCon(itself, (long)it);")

     def outputFreeIt(self, itselfname):
         Output("Py_XDECREF(self->ob_ldef_func);")
         Output("self->ob_ldef_func = NULL;")
         Output("SetListRefCon(self->ob_itself, (long)0);")
         Output("if (self->ob_must_be_disposed && %s) LDispose(%s);", itselfname, itselfname)

 # From here on it's basically all boiler plate...

 finalstuff = finalstuff + """
 static void myListDefFunction(SInt16 message,
                        Boolean selected,
                        Rect *cellRect,
                        Cell theCell,
                        SInt16 dataOffset,
                        SInt16 dataLen,
                        ListHandle theList)
 {
         PyObject *listDefFunc, *args, *rv=NULL;
         ListObject *self;

         self = (ListObject*)GetListRefCon(theList);
         if (self == NULL || self->ob_itself != theList)
                 return;  /* nothing we can do */
         listDefFunc = self->ob_ldef_func;
         if (listDefFunc == NULL)
                 return;  /* nothing we can do */
         args = Py_BuildValue("hbO&O&hhO", message,
                                           selected,
                                           PyMac_BuildRect, cellRect,
                                           PyMac_BuildPoint, theCell,
                                           dataOffset,
                                           dataLen,
                                           self);
         if (args != NULL) {
                 rv = PyEval_CallObject(listDefFunc, args);
                 Py_DECREF(args);
         }
         if (rv == NULL) {
                 PySys_WriteStderr("error in list definition callback:\\n");
                 PyErr_Print();
         } else {
                 Py_DECREF(rv);
         }
 }
 """

 # Create the generator groups and link them
 module = MacModule(MODNAME, MODPREFIX, includestuff, finalstuff, initstuff)
 object = MyObjectDefinition(OBJECTNAME, OBJECTPREFIX, OBJECTTYPE)
 module.addobject(object)

 # Create the generator classes used to populate the lists
 Function = FunctionGenerator
 Method = ListMethodGenerator

 # Create and populate the lists
 functions = []
 methods = []
 execfile(INPUTFILE)

 # Function to convert any handle to a list and vv.
 ##f = Function(ListHandle, 'as_List', (Handle, 'h', InMode))
 as_List_body = """
 Handle h;
 ListObject *l;
 if (!PyArg_ParseTuple(_args, "O&", ResObj_Convert, &h))
         return NULL;
 l = (ListObject *)ListObj_New(as_List(h));
 l->ob_must_be_disposed = 0;
 _res = Py_BuildValue("O", l);
 return _res;
 """
 f = ManualGenerator("as_List", as_List_body)
 f.docstring = lambda: "(Resource)->List.\nReturns List object (which is not auto-freed!)"
 functions.append(f)

 f = Method(Handle, 'as_Resource', (ListHandle, 'lh', InMode))
 methods.append(f)

 # Manual generator for CreateCustomList, due to callback ideosyncracies
 CreateCustomList_body = """\
 Rect rView;
 Rect dataBounds;
 Point cellSize;

 PyObject *listDefFunc;
 ListDefSpec theSpec;
 WindowPtr theWindow;
 Boolean drawIt;
 Boolean hasGrow;
 Boolean scrollHoriz;
 Boolean scrollVert;
 ListHandle outList;

 if (!PyArg_ParseTuple(_args, "O&O&O&(iO)O&bbbb",
                       PyMac_GetRect, &rView,
                       PyMac_GetRect, &dataBounds,
                       PyMac_GetPoint, &cellSize,
                       &theSpec.defType, &listDefFunc,
                       WinObj_Convert, &theWindow,
                       &drawIt,
                       &hasGrow,
                       &scrollHoriz,
                       &scrollVert))
         return NULL;


 /* Carbon applications use the CreateCustomList API */
 theSpec.u.userProc = myListDefFunctionUPP;
 CreateCustomList(&rView,
                  &dataBounds,
                  cellSize,
                  &theSpec,
                  theWindow,
                  drawIt,
                  hasGrow,
                  scrollHoriz,
                  scrollVert,
                  &outList);


 _res = ListObj_New(outList);
 if (_res == NULL)
         return NULL;
 Py_INCREF(listDefFunc);
 ((ListObject*)_res)->ob_ldef_func = listDefFunc;
 return _res;\
 """

 f = ManualGenerator("CreateCustomList", CreateCustomList_body);
 f.docstring = lambda: "(Rect rView, Rect dataBounds, Point cellSize, ListDefSpec theSpec, WindowPtr theWindow, Boolean drawIt, Boolean hasGrow, Boolean scrollHoriz, Boolean scrollVert) -> (ListHandle outList)"
 module.add(f)

 # add the populated lists to the generator groups
 # (in a different wordl the scan program would generate this)
 for f in functions: module.add(f)
 for f in methods: object.add(f)


 # generate output (open the output file as late as possible)
 SetOutputFileName(OUTPUTFILE)
 module.generate()
	# This script generates a Python interface for an Apple Macintosh Manager.
	# It uses the "bgen" package to generate C code.
	# The function specifications are generated by scanning the mamager's header file,
	# using the "scantools" package (customized for this particular manager).

	import string

	# Declarations that change for each manager
	MACHEADERFILE = 'Lists.h' # The Apple header file
	MODNAME = '_List' # The name of the module
	OBJECTNAME = 'List' # The basic name of the objects used here
	KIND = 'Handle' # Usually 'Ptr' or 'Handle'

	# The following is usually unchanged but may still require tuning
	MODPREFIX = 'List' # The prefix for module-wide routines
	OBJECTTYPE = "ListHandle" # The C type used to represent them
	OBJECTPREFIX = MODPREFIX + 'Obj' # The prefix for object methods
	INPUTFILE = string.lower(MODPREFIX) + 'gen.py' # The file generated by the scanner
	OUTPUTFILE = MODNAME + "module.c" # The file generated by this program

	from macsupport import *

	# Create the type objects
	ListHandle = OpaqueByValueType("ListHandle", "ListObj")
	ListRef = ListHandle # Obsolete, but used in Lists.h
	Cell = Point
	ListBounds = Rect
	ListBounds_ptr = Rect_ptr

	ListDefSpec = ListDefSpec_ptr = OpaqueType("ListDefSpec", "PyMac_BuildListDefSpec", "PyMac_GetListDefSpec")

	VarOutBufferShortsize = VarHeapOutputBufferType('char', 'short', 's') # (buf, &len)
	InBufferShortsize = VarInputBufferType('char', 'short', 's') # (buf, len)

	RgnHandle = OpaqueByValueType("RgnHandle", "ResObj")
	DataHandle = OpaqueByValueType("DataHandle", "ResObj")
	Handle = OpaqueByValueType("Handle", "ResObj")
	CGrafPtr = OpaqueByValueType("CGrafPtr", "GrafObj")
	EventModifiers = Type("EventModifiers", "H")

	includestuff = includestuff + """
	#include <Carbon/Carbon.h>

	#ifdef USE_TOOLBOX_OBJECT_GLUE
	extern PyObject *_ListObj_New(ListHandle);
	extern int _ListObj_Convert(PyObject , ListHandle );

	#define ListObj_New _ListObj_New
	#define ListObj_Convert _ListObj_Convert
	#endif

	#define as_List(x) ((ListHandle)x)
	#define as_Resource(lh) ((Handle)lh)

	static ListDefUPP myListDefFunctionUPP;

	"""

	initstuff = initstuff + """
	myListDefFunctionUPP = NewListDefUPP((ListDefProcPtr)myListDefFunction);

	PyMac_INIT_TOOLBOX_OBJECT_NEW(ListHandle, ListObj_New);
	PyMac_INIT_TOOLBOX_OBJECT_CONVERT(ListHandle, ListObj_Convert);
	"""

	class ListMethodGenerator(MethodGenerator):
	"""Similar to MethodGenerator, but has self as last argument"""

	def parseArgumentList(self, args):
	args, a0 = args[:-1], args[-1]
	t0, n0, m0 = a0
	if m0 != InMode:
	raise ValueError, "method's 'self' must be 'InMode'"
	self.itself = Variable(t0, "_self->ob_itself", SelfMode)
	FunctionGenerator.parseArgumentList(self, args)
	self.argumentList.append(self.itself)

	class MyObjectDefinition(PEP253Mixin, GlobalObjectDefinition):
	# XXXX Should inherit from Resource
	getsetlist = [(
	'listFlags',
	'return Py_BuildValue("l", (long)GetListFlags(self->ob_itself) & 0xff);',
	'if (!PyArg_Parse(v, "B", &(*self->ob_itself)->listFlags)) return -1;',
	None,
	), (
	'selFlags',
	'return Py_BuildValue("l", (long)GetListSelectionFlags(self->ob_itself) & 0xff);',
	'if (!PyArg_Parse(v, "B", &(*self->ob_itself)->selFlags)) return -1;',
	None,
	), (
	'cellSize',
	'return Py_BuildValue("O&", PyMac_BuildPoint, (*self->ob_itself)->cellSize);',
	'if (!PyArg_Parse(v, "O&", PyMac_GetPoint, &(*self->ob_itself)->cellSize)) return -1;',
	None
	)]

	def outputStructMembers(self):
	ObjectDefinition.outputStructMembers(self)
	Output("PyObject *ob_ldef_func;")
	Output("int ob_must_be_disposed;")

	def outputCheckNewArg(self):
	Output("""if (itself == NULL) {
	PyErr_SetString(List_Error,"Cannot create null List");
	return NULL;
	}""")

	def outputInitStructMembers(self):
	ObjectDefinition.outputInitStructMembers(self)
	Output("it->ob_ldef_func = NULL;")
	Output("it->ob_must_be_disposed = 1;")
	Output("SetListRefCon(itself, (long)it);")

	def outputFreeIt(self, itselfname):
	Output("Py_XDECREF(self->ob_ldef_func);")
	Output("self->ob_ldef_func = NULL;")
	Output("SetListRefCon(self->ob_itself, (long)0);")
	Output("if (self->ob_must_be_disposed && %s) LDispose(%s);", itselfname, itselfname)

	# From here on it's basically all boiler plate...

	finalstuff = finalstuff + """
	static void myListDefFunction(SInt16 message,
	Boolean selected,
	Rect *cellRect,
	Cell theCell,
	SInt16 dataOffset,
	SInt16 dataLen,
	ListHandle theList)
	{
	PyObject listDefFunc, args, *rv=NULL;
	ListObject *self;

	self = (ListObject*)GetListRefCon(theList);
	if (self == NULL \|\| self->ob_itself != theList)
	return; /* nothing we can do */
	listDefFunc = self->ob_ldef_func;
	if (listDefFunc == NULL)
	return; /* nothing we can do */
	args = Py_BuildValue("hbO&O&hhO", message,
	selected,
	PyMac_BuildRect, cellRect,
	PyMac_BuildPoint, theCell,
	dataOffset,
	dataLen,
	self);
	if (args != NULL) {
	rv = PyEval_CallObject(listDefFunc, args);
	Py_DECREF(args);
	}
	if (rv == NULL) {
	PySys_WriteStderr("error in list definition callback:\\n");
	PyErr_Print();
	} else {
	Py_DECREF(rv);
	}
	}
	"""

	# Create the generator groups and link them
	module = MacModule(MODNAME, MODPREFIX, includestuff, finalstuff, initstuff)
	object = MyObjectDefinition(OBJECTNAME, OBJECTPREFIX, OBJECTTYPE)
	module.addobject(object)

	# Create the generator classes used to populate the lists
	Function = FunctionGenerator
	Method = ListMethodGenerator

	# Create and populate the lists
	functions = []
	methods = []
	execfile(INPUTFILE)

	# Function to convert any handle to a list and vv.
	##f = Function(ListHandle, 'as_List', (Handle, 'h', InMode))
	as_List_body = """
	Handle h;
	ListObject *l;
	if (!PyArg_ParseTuple(_args, "O&", ResObj_Convert, &h))
	return NULL;
	l = (ListObject *)ListObj_New(as_List(h));
	l->ob_must_be_disposed = 0;
	_res = Py_BuildValue("O", l);
	return _res;
	"""
	f = ManualGenerator("as_List", as_List_body)
	f.docstring = lambda: "(Resource)->List.\nReturns List object (which is not auto-freed!)"
	functions.append(f)

	f = Method(Handle, 'as_Resource', (ListHandle, 'lh', InMode))
	methods.append(f)

	# Manual generator for CreateCustomList, due to callback ideosyncracies
	CreateCustomList_body = """\
	Rect rView;
	Rect dataBounds;
	Point cellSize;

	PyObject *listDefFunc;
	ListDefSpec theSpec;
	WindowPtr theWindow;
	Boolean drawIt;
	Boolean hasGrow;
	Boolean scrollHoriz;
	Boolean scrollVert;
	ListHandle outList;

	if (!PyArg_ParseTuple(_args, "O&O&O&(iO)O&bbbb",
	PyMac_GetRect, &rView,
	PyMac_GetRect, &dataBounds,
	PyMac_GetPoint, &cellSize,
	&theSpec.defType, &listDefFunc,
	WinObj_Convert, &theWindow,
	&drawIt,
	&hasGrow,
	&scrollHoriz,
	&scrollVert))
	return NULL;


	/* Carbon applications use the CreateCustomList API */
	theSpec.u.userProc = myListDefFunctionUPP;
	CreateCustomList(&rView,
	&dataBounds,
	cellSize,
	&theSpec,
	theWindow,
	drawIt,
	hasGrow,
	scrollHoriz,
	scrollVert,
	&outList);


	_res = ListObj_New(outList);
	if (_res == NULL)
	return NULL;
	Py_INCREF(listDefFunc);
	((ListObject*)_res)->ob_ldef_func = listDefFunc;
	return _res;\
	"""

	f = ManualGenerator("CreateCustomList", CreateCustomList_body);
	f.docstring = lambda: "(Rect rView, Rect dataBounds, Point cellSize, ListDefSpec theSpec, WindowPtr theWindow, Boolean drawIt, Boolean hasGrow, Boolean scrollHoriz, Boolean scrollVert) -> (ListHandle outList)"
	module.add(f)

	# add the populated lists to the generator groups
	# (in a different wordl the scan program would generate this)
	for f in functions: module.add(f)
	for f in methods: object.add(f)


	# generate output (open the output file as late as possible)
	SetOutputFileName(OUTPUTFILE)
	module.generate()