Mac/Modules/dlg/dlgsupport.py - platform/external/python - Git at Google

 # This script generates the Dialogs interface for Python.
 # It uses the "bgen" package to generate C code.
 # It execs the file dlggen.py which contain the function definitions
 # (dlggen.py was generated by dlgscan.py, scanning the <Dialogs.h> header file).

 from macsupport import *

 # Create the type objects

 DialogPtr = OpaqueByValueType("DialogPtr", "DlgObj")
 DialogRef = DialogPtr

 # An OptHandle is either a handle or None (in case NULL is passed in).
 # This is needed for GetDialogItem().
 OptHandle = OpaqueByValueType("Handle", "OptResObj")

 ModalFilterProcPtr = InputOnlyType("PyObject*", "O")
 ModalFilterProcPtr.passInput = lambda name: "Dlg_PassFilterProc(%s)" % name
 ModalFilterUPP = ModalFilterProcPtr

 RgnHandle = OpaqueByValueType("RgnHandle", "ResObj")
 TEHandle = OpaqueByValueType("TEHandle", "ResObj")
 CGrafPtr = OpaqueByValueType("CGrafPtr", "GrafObj")

 DITLMethod = Type("DITLMethod", "h")
 DialogItemIndex = Type("DialogItemIndex", "h")
 DialogItemType = Type("DialogItemType", "h")
 DialogItemIndexZeroBased = Type("DialogItemIndexZeroBased", "h")
 AlertType = Type("AlertType", "h")
 StringPtr = Str255
 EventMask = Type("EventMask", "H")

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

 #ifdef USE_TOOLBOX_OBJECT_GLUE
 extern PyObject *_DlgObj_New(DialogRef);
 extern PyObject *_DlgObj_WhichDialog(DialogRef);
 extern int _DlgObj_Convert(PyObject *, DialogRef *);

 #define DlgObj_New _DlgObj_New
 #define DlgObj_WhichDialog _DlgObj_WhichDialog
 #define DlgObj_Convert _DlgObj_Convert
 #endif

 /* XXX Shouldn't this be a stack? */
 static PyObject *Dlg_FilterProc_callback = NULL;

 static pascal Boolean Dlg_UnivFilterProc(DialogPtr dialog,
                                          EventRecord *event,
                                          short *itemHit)
 {
         Boolean rv;
         PyObject *args, *res;
         PyObject *callback = Dlg_FilterProc_callback;
         if (callback == NULL)
                 return 0; /* Default behavior */
         Dlg_FilterProc_callback = NULL; /* We'll restore it when call successful */
         args = Py_BuildValue("O&O&", DlgObj_WhichDialog, dialog, PyMac_BuildEventRecord, event);
         if (args == NULL)
                 res = NULL;
         else {
                 res = PyEval_CallObject(callback, args);
                 Py_DECREF(args);
         }
         if (res == NULL) {
                 PySys_WriteStderr("Exception in Dialog Filter\\n");
                 PyErr_Print();
                 *itemHit = -1; /* Fake return item */
                 return 1; /* We handled it */
         }
         else {
                 Dlg_FilterProc_callback = callback;
                 if (PyInt_Check(res)) {
                         *itemHit = PyInt_AsLong(res);
                         rv = 1;
                 }
                 else
                         rv = PyObject_IsTrue(res);
         }
         Py_DECREF(res);
         return rv;
 }

 static ModalFilterUPP
 Dlg_PassFilterProc(PyObject *callback)
 {
         PyObject *tmp = Dlg_FilterProc_callback;
         static ModalFilterUPP UnivFilterUpp = NULL;

         Dlg_FilterProc_callback = NULL;
         if (callback == Py_None) {
                 Py_XDECREF(tmp);
                 return NULL;
         }
         Py_INCREF(callback);
         Dlg_FilterProc_callback = callback;
         Py_XDECREF(tmp);
         if ( UnivFilterUpp == NULL )
                 UnivFilterUpp = NewModalFilterUPP(&Dlg_UnivFilterProc);
         return UnivFilterUpp;
 }

 static PyObject *Dlg_UserItemProc_callback = NULL;

 static pascal void Dlg_UnivUserItemProc(DialogPtr dialog,
                                          short item)
 {
         PyObject *args, *res;

         if (Dlg_UserItemProc_callback == NULL)
                 return; /* Default behavior */
         Dlg_FilterProc_callback = NULL; /* We'll restore it when call successful */
         args = Py_BuildValue("O&h", DlgObj_WhichDialog, dialog, item);
         if (args == NULL)
                 res = NULL;
         else {
                 res = PyEval_CallObject(Dlg_UserItemProc_callback, args);
                 Py_DECREF(args);
         }
         if (res == NULL) {
                 PySys_WriteStderr("Exception in Dialog UserItem proc\\n");
                 PyErr_Print();
         }
         Py_XDECREF(res);
         return;
 }

 #if 0
 /*
 ** Treating DialogObjects as WindowObjects is (I think) illegal under Carbon.
 ** However, as they are still identical under MacOS9 Carbon this is a problem, even
 ** if we neatly call GetDialogWindow() at the right places: there's one refcon field
 ** and it points to the DialogObject, so WinObj_WhichWindow will smartly return the
 ** dialog object, and therefore we still don't have a WindowObject.
 ** I'll leave the chaining code in place for now, with this comment to warn the
 ** unsuspecting victims (i.e. me, probably, in a few weeks:-)
 */
 extern PyMethodChain WinObj_chain;
 #endif
 """

 finalstuff = finalstuff + """
 /* Return the WindowPtr corresponding to a DialogObject */
 #if 0
 WindowPtr
 DlgObj_ConvertToWindow(PyObject *self)
 {
         if ( DlgObj_Check(self) )
                 return GetDialogWindow(((DialogObject *)self)->ob_itself);
         return NULL;
 }
 #endif
 /* Return the object corresponding to the dialog, or None */

 PyObject *
 DlgObj_WhichDialog(DialogPtr d)
 {
         PyObject *it;

         if (d == NULL) {
                 it = Py_None;
                 Py_INCREF(it);
         } else {
                 WindowPtr w = GetDialogWindow(d);

                 it = (PyObject *) GetWRefCon(w);
                 if (it == NULL || ((DialogObject *)it)->ob_itself != d || !DlgObj_Check(it)) {
 #if 0
                         /* Should do this, but we don't have an ob_freeit for dialogs yet. */
                         it = WinObj_New(w);
                         ((WindowObject *)it)->ob_freeit = NULL;
 #else
                         it = Py_None;
                         Py_INCREF(it);
 #endif
                 } else {
                         Py_INCREF(it);
                 }
         }
         return it;
 }
 """

 initstuff = initstuff + """
         PyMac_INIT_TOOLBOX_OBJECT_NEW(DialogPtr, DlgObj_New);
         PyMac_INIT_TOOLBOX_OBJECT_NEW(DialogPtr, DlgObj_WhichDialog);
         PyMac_INIT_TOOLBOX_OBJECT_CONVERT(DialogPtr, DlgObj_Convert);
 """


 # Define a class which specializes our object definition
 class MyObjectDefinition(PEP253Mixin, GlobalObjectDefinition):
     def __init__(self, name, prefix = None, itselftype = None):
         GlobalObjectDefinition.__init__(self, name, prefix, itselftype)
 ## This won't work in Carbon, so we disable it for all MacPythons:-(
 ## But see the comment above:-((
 ##              self.basechain = "&WinObj_chain"

     def outputInitStructMembers(self):
         GlobalObjectDefinition.outputInitStructMembers(self)
         Output("SetWRefCon(GetDialogWindow(itself), (long)it);")

     def outputCheckNewArg(self):
         Output("if (itself == NULL) { Py_INCREF(Py_None); return Py_None; }")

     def outputCheckConvertArg(self):
         Output("if (v == Py_None) { *p_itself = NULL; return 1; }")
         Output("if (PyInt_Check(v)) { *p_itself = (DialogPtr)PyInt_AsLong(v);")
         Output("                      return 1; }")

     def outputCompare(self):
         Output()
         Output("static int %s_compare(%s *self, %s *other)", self.prefix, self.objecttype, self.objecttype)
         OutLbrace()
         Output("if ( self->ob_itself > other->ob_itself ) return 1;")
         Output("if ( self->ob_itself < other->ob_itself ) return -1;")
         Output("return 0;")
         OutRbrace()

     def outputHash(self):
         Output()
         Output("static int %s_hash(%s *self)", self.prefix, self.objecttype)
         OutLbrace()
         Output("return (int)self->ob_itself;")
         OutRbrace()

     def outputFreeIt(self, itselfname):
         Output("DisposeDialog(%s);", itselfname)

 # Create the generator groups and link them
 module = MacModule('_Dlg', 'Dlg', includestuff, finalstuff, initstuff)
 object = MyObjectDefinition('Dialog', 'DlgObj', 'DialogPtr')
 module.addobject(object)

 # Create the generator classes used to populate the lists
 Function = OSErrWeakLinkFunctionGenerator
 Method = OSErrWeakLinkMethodGenerator

 # Create and populate the lists
 functions = []
 methods = []
 execfile("dlggen.py")

 # add the populated lists to the generator groups
 for f in functions: module.add(f)
 for f in methods: object.add(f)

 setuseritembody = """
         PyObject *new = NULL;


         if (!PyArg_ParseTuple(_args, "|O", &new))
                 return NULL;

         if (Dlg_UserItemProc_callback && new && new != Py_None) {
                 PyErr_SetString(Dlg_Error, "Another UserItemProc is already installed");
                 return NULL;
         }

         if (new == NULL || new == Py_None) {
                 new = NULL;
                 _res = Py_None;
                 Py_INCREF(Py_None);
         } else {
                 Py_INCREF(new);
                 _res = Py_BuildValue("O&", ResObj_New, (Handle)NewUserItemUPP(Dlg_UnivUserItemProc));
         }

         Dlg_UserItemProc_callback = new;
         return _res;
 """
 f = ManualGenerator("SetUserItemHandler", setuseritembody)
 module.add(f)

 # generate output
 SetOutputFileName('_Dlgmodule.c')
 module.generate()
	# This script generates the Dialogs interface for Python.
	# It uses the "bgen" package to generate C code.
	# It execs the file dlggen.py which contain the function definitions
	# (dlggen.py was generated by dlgscan.py, scanning the <Dialogs.h> header file).

	from macsupport import *

	# Create the type objects

	DialogPtr = OpaqueByValueType("DialogPtr", "DlgObj")
	DialogRef = DialogPtr

	# An OptHandle is either a handle or None (in case NULL is passed in).
	# This is needed for GetDialogItem().
	OptHandle = OpaqueByValueType("Handle", "OptResObj")

	ModalFilterProcPtr = InputOnlyType("PyObject*", "O")
	ModalFilterProcPtr.passInput = lambda name: "Dlg_PassFilterProc(%s)" % name
	ModalFilterUPP = ModalFilterProcPtr

	RgnHandle = OpaqueByValueType("RgnHandle", "ResObj")
	TEHandle = OpaqueByValueType("TEHandle", "ResObj")
	CGrafPtr = OpaqueByValueType("CGrafPtr", "GrafObj")

	DITLMethod = Type("DITLMethod", "h")
	DialogItemIndex = Type("DialogItemIndex", "h")
	DialogItemType = Type("DialogItemType", "h")
	DialogItemIndexZeroBased = Type("DialogItemIndexZeroBased", "h")
	AlertType = Type("AlertType", "h")
	StringPtr = Str255
	EventMask = Type("EventMask", "H")

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

	#ifdef USE_TOOLBOX_OBJECT_GLUE
	extern PyObject *_DlgObj_New(DialogRef);
	extern PyObject *_DlgObj_WhichDialog(DialogRef);
	extern int _DlgObj_Convert(PyObject , DialogRef );

	#define DlgObj_New _DlgObj_New
	#define DlgObj_WhichDialog _DlgObj_WhichDialog
	#define DlgObj_Convert _DlgObj_Convert
	#endif

	/* XXX Shouldn't this be a stack? */
	static PyObject *Dlg_FilterProc_callback = NULL;

	static pascal Boolean Dlg_UnivFilterProc(DialogPtr dialog,
	EventRecord *event,
	short *itemHit)
	{
	Boolean rv;
	PyObject args, res;
	PyObject *callback = Dlg_FilterProc_callback;
	if (callback == NULL)
	return 0; /* Default behavior */
	Dlg_FilterProc_callback = NULL; /* We'll restore it when call successful */
	args = Py_BuildValue("O&O&", DlgObj_WhichDialog, dialog, PyMac_BuildEventRecord, event);
	if (args == NULL)
	res = NULL;
	else {
	res = PyEval_CallObject(callback, args);
	Py_DECREF(args);
	}
	if (res == NULL) {
	PySys_WriteStderr("Exception in Dialog Filter\\n");
	PyErr_Print();
	itemHit = -1; / Fake return item */
	return 1; /* We handled it */
	}
	else {
	Dlg_FilterProc_callback = callback;
	if (PyInt_Check(res)) {
	*itemHit = PyInt_AsLong(res);
	rv = 1;
	}
	else
	rv = PyObject_IsTrue(res);
	}
	Py_DECREF(res);
	return rv;
	}

	static ModalFilterUPP
	Dlg_PassFilterProc(PyObject *callback)
	{
	PyObject *tmp = Dlg_FilterProc_callback;
	static ModalFilterUPP UnivFilterUpp = NULL;

	Dlg_FilterProc_callback = NULL;
	if (callback == Py_None) {
	Py_XDECREF(tmp);
	return NULL;
	}
	Py_INCREF(callback);
	Dlg_FilterProc_callback = callback;
	Py_XDECREF(tmp);
	if ( UnivFilterUpp == NULL )
	UnivFilterUpp = NewModalFilterUPP(&Dlg_UnivFilterProc);
	return UnivFilterUpp;
	}

	static PyObject *Dlg_UserItemProc_callback = NULL;

	static pascal void Dlg_UnivUserItemProc(DialogPtr dialog,
	short item)
	{
	PyObject args, res;

	if (Dlg_UserItemProc_callback == NULL)
	return; /* Default behavior */
	Dlg_FilterProc_callback = NULL; /* We'll restore it when call successful */
	args = Py_BuildValue("O&h", DlgObj_WhichDialog, dialog, item);
	if (args == NULL)
	res = NULL;
	else {
	res = PyEval_CallObject(Dlg_UserItemProc_callback, args);
	Py_DECREF(args);
	}
	if (res == NULL) {
	PySys_WriteStderr("Exception in Dialog UserItem proc\\n");
	PyErr_Print();
	}
	Py_XDECREF(res);
	return;
	}

	#if 0
	/*
	** Treating DialogObjects as WindowObjects is (I think) illegal under Carbon.
	** However, as they are still identical under MacOS9 Carbon this is a problem, even
	** if we neatly call GetDialogWindow() at the right places: there's one refcon field
	** and it points to the DialogObject, so WinObj_WhichWindow will smartly return the
	** dialog object, and therefore we still don't have a WindowObject.
	** I'll leave the chaining code in place for now, with this comment to warn the
	** unsuspecting victims (i.e. me, probably, in a few weeks:-)
	*/
	extern PyMethodChain WinObj_chain;
	#endif
	"""

	finalstuff = finalstuff + """
	/* Return the WindowPtr corresponding to a DialogObject */
	#if 0
	WindowPtr
	DlgObj_ConvertToWindow(PyObject *self)
	{
	if ( DlgObj_Check(self) )
	return GetDialogWindow(((DialogObject *)self)->ob_itself);
	return NULL;
	}
	#endif
	/* Return the object corresponding to the dialog, or None */

	PyObject *
	DlgObj_WhichDialog(DialogPtr d)
	{
	PyObject *it;

	if (d == NULL) {
	it = Py_None;
	Py_INCREF(it);
	} else {
	WindowPtr w = GetDialogWindow(d);

	it = (PyObject *) GetWRefCon(w);
	if (it == NULL \|\| ((DialogObject *)it)->ob_itself != d \|\| !DlgObj_Check(it)) {
	#if 0
	/* Should do this, but we don't have an ob_freeit for dialogs yet. */
	it = WinObj_New(w);
	((WindowObject *)it)->ob_freeit = NULL;
	#else
	it = Py_None;
	Py_INCREF(it);
	#endif
	} else {
	Py_INCREF(it);
	}
	}
	return it;
	}
	"""

	initstuff = initstuff + """
	PyMac_INIT_TOOLBOX_OBJECT_NEW(DialogPtr, DlgObj_New);
	PyMac_INIT_TOOLBOX_OBJECT_NEW(DialogPtr, DlgObj_WhichDialog);
	PyMac_INIT_TOOLBOX_OBJECT_CONVERT(DialogPtr, DlgObj_Convert);
	"""


	# Define a class which specializes our object definition
	class MyObjectDefinition(PEP253Mixin, GlobalObjectDefinition):
	def __init__(self, name, prefix = None, itselftype = None):
	GlobalObjectDefinition.__init__(self, name, prefix, itselftype)
	## This won't work in Carbon, so we disable it for all MacPythons:-(
	## But see the comment above:-((
	## self.basechain = "&WinObj_chain"

	def outputInitStructMembers(self):
	GlobalObjectDefinition.outputInitStructMembers(self)
	Output("SetWRefCon(GetDialogWindow(itself), (long)it);")

	def outputCheckNewArg(self):
	Output("if (itself == NULL) { Py_INCREF(Py_None); return Py_None; }")

	def outputCheckConvertArg(self):
	Output("if (v == Py_None) { *p_itself = NULL; return 1; }")
	Output("if (PyInt_Check(v)) { *p_itself = (DialogPtr)PyInt_AsLong(v);")
	Output(" return 1; }")

	def outputCompare(self):
	Output()
	Output("static int %s_compare(%s self, %s other)", self.prefix, self.objecttype, self.objecttype)
	OutLbrace()
	Output("if ( self->ob_itself > other->ob_itself ) return 1;")
	Output("if ( self->ob_itself < other->ob_itself ) return -1;")
	Output("return 0;")
	OutRbrace()

	def outputHash(self):
	Output()
	Output("static int %s_hash(%s *self)", self.prefix, self.objecttype)
	OutLbrace()
	Output("return (int)self->ob_itself;")
	OutRbrace()

	def outputFreeIt(self, itselfname):
	Output("DisposeDialog(%s);", itselfname)

	# Create the generator groups and link them
	module = MacModule('_Dlg', 'Dlg', includestuff, finalstuff, initstuff)
	object = MyObjectDefinition('Dialog', 'DlgObj', 'DialogPtr')
	module.addobject(object)

	# Create the generator classes used to populate the lists
	Function = OSErrWeakLinkFunctionGenerator
	Method = OSErrWeakLinkMethodGenerator

	# Create and populate the lists
	functions = []
	methods = []
	execfile("dlggen.py")

	# add the populated lists to the generator groups
	for f in functions: module.add(f)
	for f in methods: object.add(f)

	setuseritembody = """
	PyObject *new = NULL;


	if (!PyArg_ParseTuple(_args, "\|O", &new))
	return NULL;

	if (Dlg_UserItemProc_callback && new && new != Py_None) {
	PyErr_SetString(Dlg_Error, "Another UserItemProc is already installed");
	return NULL;
	}

	if (new == NULL \|\| new == Py_None) {
	new = NULL;
	_res = Py_None;
	Py_INCREF(Py_None);
	} else {
	Py_INCREF(new);
	_res = Py_BuildValue("O&", ResObj_New, (Handle)NewUserItemUPP(Dlg_UnivUserItemProc));
	}

	Dlg_UserItemProc_callback = new;
	return _res;
	"""
	f = ManualGenerator("SetUserItemHandler", setuseritembody)
	module.add(f)

	# generate output
	SetOutputFileName('_Dlgmodule.c')
	module.generate()